Show normal line numbers in insert mode

lua/valeth/plugins/statuscol.lua

@@ -16,16 +16,29 @@ spec.config = function()
         foldclose = "",
     })
 
+    local function line_numbers(args, fa)
+        if vim.api.nvim_get_mode().mode == "n" then
+            return builtin.lnumfunc(args, fa)
+        else
+            return "%l"
+        end
+    end
+
+    local signs = {
+        text = { "%s" },
+    }
+    local folds = {
+        text = { builtin.foldfunc, " " },
+        condition = { builtin.not_empty },
+    }
+    local numbers = {
+        text = { line_numbers, " " },
+        condition = { true, builtin.not_empty },
+    }
+
     statuscol.setup({
         relculright = true,
-        segments = {
+        segments = { signs, folds, numbers }
-            { text = { "%s" }},
-            {
-                text = { builtin.lnumfunc, " " },
-                condition = { true, builtin.not_empty },
-            },
-            { text = { builtin.foldfunc, " " }},
-        }
     })
 end