feat: repo map (#496)

* feat: repo map

* chore: remove breakline

* chore: remove spaces

* fix: golang public method

* feat: mentions for editing input

lua/avante/path.lua

@@ -28,17 +28,17 @@ end
 H.get_mode_file = function(mode) return string.format("custom.%s.avanterules", mode) end
 
 -- History path
-local M = {}
+local History = {}
 
 -- Returns the Path to the chat history file for the given buffer.
 ---@param bufnr integer
 ---@return Path
-M.get = function(bufnr) return Path:new(Config.history.storage_path):joinpath(H.filename(bufnr)) end
+History.get = function(bufnr) return Path:new(Config.history.storage_path):joinpath(H.filename(bufnr)) end
 
 -- Loads the chat history for the given buffer.
 ---@param bufnr integer
-M.load = function(bufnr)
-  local history_file = M.get(bufnr)
+History.load = function(bufnr)
+  local history_file = History.get(bufnr)
   if history_file:exists() then
     local content = history_file:read()
     return content ~= nil and vim.json.decode(content) or {}
@@ -49,29 +49,29 @@ end
 -- Saves the chat history for the given buffer.
 ---@param bufnr integer
 ---@param history table
-M.save = function(bufnr, history)
-  local history_file = M.get(bufnr)
+History.save = function(bufnr, history)
+  local history_file = History.get(bufnr)
   history_file:write(vim.json.encode(history), "w")
 end
 
-P.history = M
+P.history = History
 
 -- Prompt path
-local N = {}
+local Prompt = {}
 
 ---@class AvanteTemplates
 ---@field initialize fun(directory: string): nil
 ---@field render fun(template: string, context: TemplateOptions): string
 local templates = nil
 
-N.templates = { planning = nil, editing = nil, suggesting = nil }
+Prompt.templates = { planning = nil, editing = nil, suggesting = nil }
 
 -- Creates a directory in the cache path for the given buffer and copies the custom prompts to it.
 -- We need to do this beacuse the prompt template engine requires a given directory to load all required files.
 -- PERF: Hmm instead of copy to cache, we can also load in globals context, but it requires some work on bindings. (eh maybe?)
 ---@param bufnr number
 ---@return string the resulted cache_directory to be loaded with avante_templates
-N.get = function(bufnr)
+Prompt.get = function(bufnr)
   if not P.available() then error("Make sure to build avante (missing avante_templates)", 2) end
 
   -- get root directory of given bufnr
@@ -85,19 +85,19 @@ N.get = function(bufnr)
   local scanner = Scan.scan_dir(directory:absolute(), { depth = 1, add_dirs = true })
   for _, entry in ipairs(scanner) do
     local file = Path:new(entry)
-    if entry:find("planning") and N.templates.planning == nil then
-      N.templates.planning = file:read()
-    elseif entry:find("editing") and N.templates.editing == nil then
-      N.templates.editing = file:read()
-    elseif entry:find("suggesting") and N.templates.suggesting == nil then
-      N.templates.suggesting = file:read()
+    if entry:find("planning") and Prompt.templates.planning == nil then
+      Prompt.templates.planning = file:read()
+    elseif entry:find("editing") and Prompt.templates.editing == nil then
+      Prompt.templates.editing = file:read()
+    elseif entry:find("suggesting") and Prompt.templates.suggesting == nil then
+      Prompt.templates.suggesting = file:read()
     end
   end
 
   Path:new(debug.getinfo(1).source:match("@?(.*/)"):gsub("/lua/avante/path.lua$", "") .. "templates")
     :copy({ destination = cache_prompt_dir, recursive = true })
 
-  vim.iter(N.templates):filter(function(_, v) return v ~= nil end):each(function(k, v)
+  vim.iter(Prompt.templates):filter(function(_, v) return v ~= nil end):each(function(k, v)
     local f = cache_prompt_dir:joinpath(H.get_mode_file(k))
     f:write(v, "w")
   end)
@@ -106,22 +106,53 @@ N.get = function(bufnr)
 end
 
 ---@param mode LlmMode
-N.get_file = function(mode)
-  if N.templates[mode] ~= nil then return H.get_mode_file(mode) end
+Prompt.get_file = function(mode)
+  if Prompt.templates[mode] ~= nil then return H.get_mode_file(mode) end
   return string.format("%s.avanterules", mode)
 end
 
 ---@param path string
 ---@param opts TemplateOptions
-N.render_file = function(path, opts) return templates.render(path, opts) end
+Prompt.render_file = function(path, opts) return templates.render(path, opts) end
 
 ---@param mode LlmMode
 ---@param opts TemplateOptions
-N.render_mode = function(mode, opts) return templates.render(N.get_file(mode), opts) end
+Prompt.render_mode = function(mode, opts) return templates.render(Prompt.get_file(mode), opts) end
 
-N.initialize = function(directory) templates.initialize(directory) end
+Prompt.initialize = function(directory) templates.initialize(directory) end
 
-P.prompts = N
+P.prompts = Prompt
+
+local RepoMap = {}
+
+-- Get a chat history file name given a buffer
+---@param project_root string
+---@param ext string
+---@return string
+RepoMap.filename = function(project_root, ext)
+  -- Replace path separators with double underscores
+  local path_with_separators = fn.substitute(project_root, "/", "__", "g")
+  -- Replace other non-alphanumeric characters with single underscores
+  return fn.substitute(path_with_separators, "[^A-Za-z0-9._]", "_", "g") .. "." .. ext .. ".repo_map.json"
+end
+
+RepoMap.get = function(project_root, ext) return Path:new(P.data_path):joinpath(RepoMap.filename(project_root, ext)) end
+
+RepoMap.save = function(project_root, ext, data)
+  local file = RepoMap.get(project_root, ext)
+  file:write(vim.json.encode(data), "w")
+end
+
+RepoMap.load = function(project_root, ext)
+  local file = RepoMap.get(project_root, ext)
+  if file:exists() then
+    local content = file:read()
+    return content ~= nil and vim.json.decode(content) or {}
+  end
+  return nil
+end
+
+P.repo_map = RepoMap
 
 P.setup = function()
   local history_path = Path:new(Config.history.storage_path)
@@ -132,6 +163,10 @@ P.setup = function()
   if not cache_path:exists() then cache_path:mkdir({ parents = true }) end
   P.cache_path = cache_path
 
+  local data_path = Path:new(vim.fn.stdpath("data") .. "/avante")
+  if not data_path:exists() then data_path:mkdir({ parents = true }) end
+  P.data_path = data_path
+
   vim.defer_fn(function()
     local ok, module = pcall(require, "avante_templates")
     ---@cast module AvanteTemplates