nvim lualine

home/user/.config/nvim/lua/base/plugins/lualine.lua

@@ -1,7 +1,41 @@
+
+function get_formatted_cwd()
+    return vim.fn.getcwd():gsub(os.getenv("HOME"), "~")
+end
+
+local function get_virtual_env()
+    return vim.fs.basename(os.getenv("VIRTUAL_ENV"))
+end
+
+function get_formatted_virtual_env()
+    if vim.bo.filetype == 'python' then
+        return "<" .. get_virtual_env() .. ">"
+    end
+    return ""
+end
+
+function detect_indent_type()
+    local shiftwidth = vim.bo.shiftwidth
+    if vim.bo.expandtab then
+        return shiftwidth .. " spaces"
+    else
+        return shiftwidth .. " tab"
+    end
+end
+
 return {
     'nvim-lualine/lualine.nvim',
     config = function()
-        require("lualine").setup()
+        require("lualine").setup({
+            sections = {
+                lualine_a = {'mode'},
+                lualine_b = {'branch', 'diff', 'diagnostics'},
+                lualine_c = {'get_formatted_cwd()'},
+                lualine_x = {'get_formatted_virtual_env()', 'detect_indent_type()', 'filetype'},
+                lualine_y = {'progress'},
+                lualine_z = {'location'},
+              },
+        })
     end,
     dependencies = { 'kyazdani42/nvim-web-devicons' }
 }