♻️ refactor(utils): move stats modules to utils

Move wakatime_stats and music_stats to new utils dir
2024-08-07 08:20:30 -04:00
parent 25352b81a0
commit e1e59bfea5
2 changed files with 13 additions and 0 deletions
@@ -0,0 +1,82 @@
+--          ╭─────────────────────────────────────────────────────────╮
+--          │                     WakaTime Stats                      │
+--          ╰─────────────────────────────────────────────────────────╯
+local M = {}
+local cache_file = os.getenv("HOME") .. "/.cache/wakatime_cache.txt"
+
+local function read_cache()
+  local file = io.open(cache_file, "r")
+  if file then
+    local content = file:read("*a")
+    file:close()
+    return content
+  end
+  return "Error: Cache file not found"
+end
+
+local function get_wakatime_today()
+  local result = read_cache()
+  if not result or result == "" then
+    return ""
+  end
+
+  -- Handle common error messages
+  if
+    result:match("command not found")
+    or result:match("No such file or directory")
+  then
+    return "Error: wakatime-cli not found"
+  end
+
+  return result:match("^%s*(.-)%s*$") -- Trim any whitespace
+end
+
+function M.get_today()
+  return get_wakatime_today()
+end
+
+function M.get_icon()
+  local text = M.get_today()
+  if text and text:match("Error:") then
+    return "󰥕 " -- Icon for error or issue
+  else
+    return "󱦺 " -- Default icon
+  end
+end
+
+function M.get_icon_with_text()
+  return M.get_icon() .. " " .. (M.get_today() or "Loading...")
+end
+
+function M.get_total_minutes()
+  local wakatime_string = M.get_today()
+  if not wakatime_string or wakatime_string:match("Error:") then
+    return 0 -- Return 0 if there's an error in the wakatime string
+  end
+
+  local hours, mins = wakatime_string:match("(%d+)%s*hrs?%s*(%d+)%s*mins?")
+  if not hours then
+    mins = wakatime_string:match("(%d+)%s*mins?")
+    hours = 0
+  end
+
+  hours = tonumber(hours) or 0
+  mins = tonumber(mins) or 0
+
+  return (hours * 60) + mins
+end
+
+function M.get_color()
+  local total_minutes = M.get_total_minutes()
+  local rooticolor = require("utils.rooticolor") -- Adjust according to your actual setup
+
+  if total_minutes >= 60 then
+    return { fg = rooticolor.get_fg_color("GitSignsAdd") }
+  elseif total_minutes >= 30 then
+    return { fg = rooticolor.get_fg_color("GitSignsChange") }
+  else
+    return { fg = rooticolor.get_fg_color("GitSignsDelete") }
+  end
+end
+
+return M