1 files changed, 34 insertions, 8 deletions
diff --git a/lua/encrypt/helpers.lua b/lua/encrypt/helpers.lua
index 37a96b3..c88b448 100644
--- a/lua/encrypt/helpers.lua
+++ b/lua/encrypt/helpers.lua
@@ -1,13 +1,39 @@
----@return string
-local function getPassword()
-  local password = vim.b["password"]
-  if password == nil then
-    password = vim.fn.inputsecret("Enter password: ")
-    vim.b["password"] = password
+---@return function string
+local getPasswordFactory = function()
+  local bufferPasswordMap = {}
+
+  return function()
+    local key = string.format("%s", vim.fn.bufnr())
+    if bufferPasswordMap[key] == nil then
+      password = vim.fn.inputsecret("Enter password: ")
+      if password == "" then
+        vim.notify("Password is cancelled", vim.log.levels.WARN)
+      end
+      bufferPasswordMap[key] = password
+    end
+    return bufferPasswordMap[key]
   end
-  return password
 end
 
+local getPassword = getPasswordFactory()
 local ENCRYPTED_PREFIX = "# <<<encrypted>>>"
 
-return { getPassword = getPassword, ENCRYPTED_PREFIX = ENCRYPTED_PREFIX }
+---@enum buftype
+local BUFTYPE = {
+  encrypted = 1,
+  plaintext = 2,
+}
+
+local function bufferEncrypted()
+  local first_line = vim.api.nvim_buf_get_lines(0, 0, 1, false)[1]
+  if first_line == ENCRYPTED_PREFIX then
+    return true
+  end
+end
+
+return {
+  getPassword = getPassword,
+  ENCRYPTED_PREFIX = ENCRYPTED_PREFIX,
+  BUFTYPE = BUFTYPE,
+  bufferEncrypted = bufferEncrypted,
+}