add telescope extension ^& small refactoring (still not done)

lua/lazygit.lua

@@ -2,6 +2,19 @@ vim = vim
 local api = vim.api
 local fn = vim.fn
 
+-- store all git repositories visited in this session
+local lazygit_visited_git_repos = {}
+local function append_git_repo_path(repo_path)
+    -- TODO: could be done better :)
+    for _, path in ipairs(lazygit_visited_git_repos) do
+        if path == repo_path then
+            return
+        end
+    end
+
+    table.insert(lazygit_visited_git_repos, tostring(repo_path))
+end
+
 LAZYGIT_BUFFER = nil
 LAZYGIT_LOADED = false
 vim.g.lazygit_opened = 0
@@ -31,19 +44,25 @@ local function project_root_dir()
   end
 
   -- try symlinked file location instead
-  gitdir = fn.system(
+  local gitdir = fn.system(
                'cd "' .. fn.fnamemodify(fn.resolve(fn.expand('%:p')), ':h') .. '" && git rev-parse --show-toplevel')
-  isgitdir = fn.matchstr(gitdir, '^fatal:.*') == ''
+  local isgitdir = fn.matchstr(gitdir, '^fatal:.*') == ''
+
+  -- TODO: not sure the right way to do this
   if isgitdir then
     vim.o.shell = oldshell
+    append_git_repo_path(gitdir)
     return trim(gitdir)
   end
 
   -- revert to old shell
   vim.o.shell = oldshell
 
+  local repo_path = fn.getcwd(0, 0)
+  append_git_repo_path(repo_path)
+
   -- just return current working directory
-  return fn.getcwd(0, 0)
+  return repo_path
 end
 
 --- on_exit callback function to delete the open buffer when lazygit exits in a neovim terminal
@@ -158,13 +177,30 @@ local function lazygit(path)
     end
   end
   open_floating_window()
+
+  -- TODO: this should be configurable and not hardcoded like this
+  -- this is convinient if you want to carry your lazygit config in a custom location
+  local function get_nvim_root()
+      local nvim_root_path = vim.api.nvim_eval('$MYVIMRC')
+      return nvim_root_path:match("(.*".."\\"..")")
+  end
   local cmd = 'lazygit'
-  if path ~= nil and not vim.env.GIT_DIR then
+  _ = project_root_dir()
-    cmd = cmd .. ' -g "' .. path .. '/.git/"'
+
-  end
+  if path ~= nil then
-  if path ~= nil and not vim.env.GIT_WORK_TREE then
+      cmd = cmd .. ' -p ' .. path
-    cmd = cmd .. ' -w "' .. path .. '"'
   end
+
+  cmd = cmd .. ' -ucf=' .. get_nvim_root() .. '/config/config.yml'
+
+  -- if path ~= nil and not vim.env.GIT_DIR then
+  --   cmd = cmd .. ' -g "' .. path .. '/.git/"'
+  -- end
+
+  -- if path ~= nil and not vim.env.GIT_WORK_TREE then
+  --   cmd = cmd .. ' -w "' .. path .. '"'
+  -- end
+
   exec_lazygit_command(cmd)
 end
 
@@ -217,4 +253,5 @@ return {
   lazygitfilter = lazygitfilter,
   lazygitconfig = lazygitconfig,
   project_root_dir = project_root_dir,
+  lazygit_visited_git_repos = lazygit_visited_git_repos,
 }

lua/telescope/_extensions/lazygit_telescope.lua

@@ -0,0 +1,84 @@
+local Path = require("plenary.path")
+local Window = require("plenary.window.float")
+local strings = require("plenary.strings")
+local pickers = require("telescope.pickers")
+local finders = require("telescope.finders")
+local actions = require("telescope.actions")
+local utils = require("telescope.utils")
+local action_set = require("telescope.actions.set")
+local action_state = require("telescope.actions.state")
+local conf = require("telescope.config").values
+local lazygit = require("lazygit")
+
+
+local function lazygit_toggle(path)
+    local cmd = [[lua require"lazygit".lazygit('%s')]]
+    cmd= cmd:format(path:gsub("%s", ""))
+    vim.api.nvim_command(cmd)
+    vim.api.nvim_buf_set_keymap(0, 't', '<Esc>', '<Esc>', {noremap = true, silent = true})
+end
+
+local function open_lazygit(prompt_buf)
+    actions.close(prompt_buf)
+    local entry = action_state.get_selected_entry()
+    lazygit_toggle(entry.value)
+end
+
+
+local lazygit_repos = function(opts)
+    local displayer = require("telescope.pickers.entry_display").create {
+        separator = "",
+        items = {
+            {width = 4},
+            {width = 55},
+            {remaining = true},
+        },
+    }
+
+    local repos = {}
+    for _, v in pairs(lazygit.lazygit_visited_git_repos) do
+        local index = #repos + 1
+        -- retrieve the git repo name
+        local entry =
+        {
+            idx = index,
+            value = v:gsub("%s", ""),
+            repo_name= v:gsub("%s", ""):match("^.+/(.+)$"),
+        }
+
+        table.insert(repos, index, entry)
+    end
+
+    pickers.new(opts or {}, {
+        prompt_title = "lazygit repos",
+        finder = finders.new_table {
+            results = repos,
+            entry_maker = function(entry)
+                local make_display = function()
+                    return displayer
+                    {
+                        {entry.idx},
+                        {entry.repo_name},
+                    }
+                end
+
+                return {
+                    value = entry.value,
+                    ordinal = string.format("%s %s", entry.idx, entry.repo_name),
+                    display = make_display,
+                }
+            end,
+        },
+        sorter = conf.generic_sorter(opts),
+        attach_mappings = function(_, _)
+            action_set.select:replace(open_lazygit)
+            return true
+        end
+    }):find()
+end
+
+return require("telescope").register_extension({
+    exports = {
+        lazygit = lazygit_repos,
+    }
+})