diff options
Diffstat (limited to 'lua/encrypt/init.lua')
| -rw-r--r-- | lua/encrypt/init.lua | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/lua/encrypt/init.lua b/lua/encrypt/init.lua index cfa1aa6..ea5368f 100644 --- a/lua/encrypt/init.lua +++ b/lua/encrypt/init.lua @@ -5,14 +5,18 @@ local helpers = require("encrypt.helpers") local function setup() vim.api.nvim_create_user_command("X", function() - setupBuffer(helpers.BUFTYPE.decrypted) - encrypt() + if helpers.bufferEncrypted() then + setupBuffer(helpers.BUFTYPE.encrypted) + decrypt() + else + setupBuffer(helpers.BUFTYPE.plaintext) + encrypt() + end end, {}) vim.api.nvim_create_autocmd({ "BufReadPost" }, { callback = function() - local first_line = vim.api.nvim_buf_get_lines(0, 0, 1, false)[1] - if first_line == helpers.ENCRYPTED_PREFIX then + if helpers.bufferEncrypted() then setupBuffer(helpers.BUFTYPE.encrypted) decrypt() end |