small refactoring

lua/lazygit/utils.lua

@@ -0,0 +1,97 @@
+local fn = vim.fn
+
+-- store all git repositories visited in this session
+local lazygit_visited_git_repos = {}
+
+-- TODO:check if the repo isa git repo
+local function append_git_repo_path(repo_path)
+    if repo_path == nil or not fn.isdirectory(repo_path) then
+        return
+    end
+
+    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
+
+
+--- Strip leading and lagging whitespace
+local function trim(str)
+  return str:gsub('^%s+', ''):gsub('%s+$', '')
+end
+
+
+local function get_root()
+    local cwd = vim.loop.cwd()
+    local status, job = pcall(require, 'plenary.job')
+    if not status then
+        local gitdir = fn.system('git rev-parse --show-toplevel')
+    end
+
+    local gitroot_job = job:new({
+        'git',
+        'rev-parse',
+        '--show-toplevel',
+        cwd=cwd
+    })
+
+    local _, code = gitroot_job:sync()
+    if (code ~= 0) then
+        return nil
+    end
+
+    return fn.getcwd(0, 0)
+end
+
+--- Get project_root_dir for git repository
+local function project_root_dir()
+  -- always use bash on Unix based systems.
+  local oldshell = vim.o.shell
+  if vim.fn.has('win32') == 0 then
+      vim.o.shell = 'bash'
+  end
+
+  local root = get_root()
+  -- try symlinked file location instead
+  local gitdir = fn.system(
+               'cd "' .. fn.fnamemodify(fn.resolve(fn.expand('%:p')), ':h') .. '" && git rev-parse --show-toplevel')
+  local isgitdir = fn.matchstr(gitdir, '^fatal:.*') == ''
+
+  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 repo_path
+end
+
+--- Check if lazygit is available
+local function is_lazygit_available()
+  return fn.executable('lazygit') == 1
+end
+
+local function is_symlink()
+  local resolved = fn.resolve(fn.expand('%:p'))
+  return resolved ~= fn.expand('%:p')
+end
+
+
+return {
+    get_root = get_root,
+    project_root_dir = project_root_dir,
+    lazygit_visited_git_repos = lazygit_visited_git_repos,
+    is_lazygit_available = is_lazygit_available,
+    is_symlink = is_symlink,
+}