Fixing contexts and adding auto repoMap

Author: W3ND31

lua/eca/api.lua

@@ -75,9 +75,19 @@ function M.add_file_context(file_path)
     content = content
   }
 
-  -- For now, store it for next message
-  -- TODO: Implement context management
-  Utils.debug("File context added: " .. file_path)
+  -- Get current sidebar and add context
+  local sidebar = eca.get()
+  if not sidebar then
+    Utils.info("Opening ECA sidebar to add context...")
+    M.chat()
+    sidebar = eca.get()
+  end
+  
+  if sidebar then
+    sidebar:add_context(context)
+  else
+    Utils.error("Failed to create ECA sidebar")
+  end
 end
 
 ---@param directory_path string
@@ -123,8 +133,112 @@ function M.add_selection_context()
   local lines = vim.api.nvim_buf_get_lines(0, start_pos[2] - 1, end_pos[2], false)
   if #lines > 0 then
     local selection_text = table.concat(lines, "\n")
-    Utils.info("Adding selection context (" .. #lines .. " lines)")
-    -- TODO: Send selection to ECA server as context
+    local current_file = vim.api.nvim_buf_get_name(0)
+    local context_path = current_file .. ":" .. start_pos[2] .. "-" .. end_pos[2]
+    
+    -- Create context object
+    local context = {
+      type = "selection",
+      path = context_path,
+      content = selection_text,
+      source_file = current_file,
+      start_line = start_pos[2],
+      end_line = end_pos[2]
+    }
+    
+    -- Get current sidebar and add context
+    local eca = require("eca")
+    local sidebar = eca.get()
+    if not sidebar then
+      Utils.info("Opening ECA sidebar to add context...")
+      M.chat()
+      sidebar = eca.get()
+    end
+    
+    if sidebar then
+      sidebar:add_context(context)
+      Utils.info("Added selection context (" .. #lines .. " lines)")
+    else
+      Utils.error("Failed to create ECA sidebar")
+    end
+  end
+end
+
+function M.list_contexts()
+  local eca = require("eca")
+  local sidebar = eca.get()
+  if not sidebar then
+    Utils.warn("No active ECA sidebar")
+    return
+  end
+  
+  local contexts = sidebar:get_contexts()
+  if #contexts == 0 then
+    Utils.info("No active contexts")
+    return
+  end
+  
+  Utils.info("Active contexts (" .. #contexts .. "):")
+  for i, context in ipairs(contexts) do
+    local size_info = ""
+    if context.content then
+      local lines = vim.split(context.content, "\n")
+      size_info = " (" .. #lines .. " lines)"
+    end
+    Utils.info(i .. ". " .. context.type .. ": " .. context.path .. size_info)
+  end
+end
+
+function M.clear_contexts()
+  local eca = require("eca")
+  local sidebar = eca.get()
+  if not sidebar then
+    Utils.warn("No active ECA sidebar")
+    return
+  end
+  
+  sidebar:clear_contexts()
+end
+
+function M.remove_context(path)
+  local eca = require("eca")
+  local sidebar = eca.get()
+  if not sidebar then
+    Utils.warn("No active ECA sidebar")
+    return
+  end
+  
+  sidebar:remove_context(path)
+end
+
+function M.add_repo_map_context()
+  local eca = require("eca")
+  local sidebar = eca.get()
+  if not sidebar then
+    Utils.info("Opening ECA sidebar to add repoMap context...")
+    M.chat()
+    sidebar = eca.get()
+  end
+  
+  if sidebar then
+    -- Check if repoMap already exists
+    local contexts = sidebar:get_contexts()
+    for _, context in ipairs(contexts) do
+      if context.type == "repoMap" then
+        Utils.info("RepoMap context already added")
+        return
+      end
+    end
+    
+    -- Add repoMap context
+    sidebar:add_context({
+      type = "repoMap",
+      path = "repoMap",
+      content = "Repository structure and code mapping for better project understanding"
+    })
+    Utils.info("Added repoMap context")
+  else
+    Utils.error("Failed to create ECA sidebar")
   end
 end
 

lua/eca/commands.lua

@@ -47,6 +47,36 @@ function M.setup()
     desc = "Add current selection as context to ECA",
   })
 
+  vim.api.nvim_create_user_command("EcaListContexts", function()
+    require("eca.api").list_contexts()
+  end, {
+    desc = "List active contexts in ECA",
+  })
+
+  vim.api.nvim_create_user_command("EcaClearContexts", function()
+    require("eca.api").clear_contexts()
+  end, {
+    desc = "Clear all contexts from ECA",
+  })
+
+  vim.api.nvim_create_user_command("EcaRemoveContext", function(opts)
+    if opts.args and opts.args ~= "" then
+      require("eca.api").remove_context(opts.args)
+    else
+      Utils.warn("Please provide a file path to remove")
+    end
+  end, {
+    desc = "Remove specific context from ECA",
+    nargs = "+",
+    complete = "file",
+  })
+
+  vim.api.nvim_create_user_command("EcaAddRepoMap", function()
+    require("eca.api").add_repo_map_context()
+  end, {
+    desc = "Add repository map context to ECA",
+  })
+
   vim.api.nvim_create_user_command("EcaServerStart", function()
     require("eca.api").start_server()
   end, {

lua/eca/config.lua

@@ -21,6 +21,9 @@ M._defaults = {
     enable = true, -- Enable markview.nvim integration
     filetypes = { "markdown", "Eca" }, -- Filetypes to enable markview
   },
+  context = {
+    auto_repo_map = true, -- Automatically add repoMap context when starting new chat
+  },
   mappings = {
     chat = "<leader>ec",
     focus = "<leader>ef",

lua/eca/path_finder.lua

@@ -137,7 +137,7 @@ function M:_download_latest_server(server_path, version)
 
   local download_path = self._cache_dir .. "/" .. artifact_name
 
-  Utils.info("Downloading ECA server from: " .. download_url)
+  Utils.info("Downloading latest ECA server version from: " .. download_url)
 
   -- Ensure cache directory exists
   vim.fn.mkdir(self._cache_dir, "p")