From 6bf989871e78f1cf1e7acbde81b31f9dbb5fecf7 Mon Sep 17 00:00:00 2001
From: Sanchayan Maity <sanchayan@sanchayanmaity.net>
Date: Sun, 13 Feb 2022 12:33:17 +0530
Subject: [PATCH] nvim: statusline: Fix tree-sitter status

Fixing based on file type was not good enough. Check for error
return and act accordingly.
---
 nvim/.config/nvim/after/plugin/statusline.lua | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/nvim/.config/nvim/after/plugin/statusline.lua b/nvim/.config/nvim/after/plugin/statusline.lua
index c07ce57..a06946d 100644
--- a/nvim/.config/nvim/after/plugin/statusline.lua
+++ b/nvim/.config/nvim/after/plugin/statusline.lua
@@ -171,16 +171,15 @@ M.progress = function()
 end
 
 M.treesitter_status = function()
-  local filetype = vim.bo.filetype
-  local ignore_ft = { 'diff', 'fugitive', 'git', 'markdown', 'qf', 'text', '' }
-  for _, v in pairs(ignore_ft) do
-    if v == filetype then
-      return ""
-    end
-  end
-
   if gps.is_available then
-    return string.format("%s%s", ' %#TSStatus#', gps.get_location())
+    local location = gps.get_location()
+    if location == 'error' then
+      return ''
+    else
+      return string.format("%s%s", ' %#TSStatus#', location)
+    end
+  else
+    return ''
   end
 end