feat: optimise, lazy load

Author: tduyng

lua/codeme/backend.lua

@@ -0,0 +1,130 @@
+local M = {}
+
+-- Cache backend binary path
+local _binary_path = nil
+
+---Find codeme binary
+---@return string|nil
+local function find_binary()
+	if _binary_path then
+		return _binary_path
+	end
+
+	-- Check environment variable
+	local env_path = os.getenv("CODEME_BIN")
+	if env_path and vim.fn.executable(env_path) == 1 then
+		_binary_path = env_path
+		return _binary_path
+	end
+
+	-- Check system PATH
+	if vim.fn.executable("codeme") == 1 then
+		_binary_path = "codeme"
+		return _binary_path
+	end
+
+	-- Check local installation
+	local local_path = vim.fn.stdpath("data") .. "/codeme/codeme"
+	if vim.fn.executable(local_path) == 1 then
+		_binary_path = local_path
+		return _binary_path
+	end
+
+	return nil
+end
+
+---Execute codeme command asynchronously
+---@param args string[] Command arguments
+---@param callback fun(success: boolean, data: any, error: string?)
+local function exec_async(args, callback)
+	local binary = find_binary()
+	if not binary then
+		callback(false, nil, "codeme binary not found")
+		return
+	end
+
+	vim.system({ binary, unpack(args) }, { text = true }, function(obj)
+		vim.schedule(function()
+			if obj.code ~= 0 then
+				callback(false, nil, obj.stderr or "Command failed")
+				return
+			end
+
+			-- Parse JSON output
+			local stdout = obj.stdout or ""
+			local json_str = stdout:match("({.*})")
+			if not json_str then
+				callback(false, nil, "No JSON in output")
+				return
+			end
+
+			local ok, data = pcall(vim.json.decode, json_str)
+			if not ok then
+				callback(false, nil, "JSON parse error: " .. tostring(data))
+				return
+			end
+
+			callback(true, data, nil)
+		end)
+	end)
+end
+
+---Get stats from backend
+---@param today_only boolean? Only fetch today's stats
+---@param callback fun(stats: table)
+function M.get_stats(today_only, callback)
+	local args = { "api", "--compact" }
+	if today_only then
+		table.insert(args, "--today")
+	end
+
+	exec_async(args, function(success, data, err)
+		if not success then
+			vim.notify("CodeMe: " .. (err or "Unknown error"), vim.log.levels.ERROR)
+			callback({})
+			return
+		end
+		callback(data or {})
+	end)
+end
+
+---Send heartbeat to backend
+---@param opts table Heartbeat options
+---@param callback fun(success: boolean)?
+function M.send_heartbeat(opts, callback)
+	local binary = find_binary()
+	if not binary then
+		if callback then
+			callback(false)
+		end
+		return
+	end
+
+	local args = {
+		"track",
+		"--file",
+		opts.filepath,
+		"--lang",
+		opts.language or "",
+		"--editor",
+		"neovim",
+		"--lines",
+		tostring(opts.lines or 0),
+	}
+
+	vim.system({ binary, unpack(args) }, { detach = true }, function(obj)
+		if callback then
+			vim.schedule(function()
+				callback(obj.code == 0)
+			end)
+		end
+	end)
+end
+
+---Check if binary is installed
+---@return boolean
+function M.is_installed()
+	return find_binary() ~= nil
+end
+
+return M

lua/codeme/dashboard.lua

@@ -0,0 +1,173 @@
+local M = {}
+
+---Format seconds to human-readable duration
+---@param seconds number
+---@return string
+function M.format_duration(seconds)
+	if not seconds or seconds == 0 then
+		return "0s"
+	end
+
+	if seconds < 60 then
+		return string.format("%ds", seconds)
+	elseif seconds < 3600 then
+		return string.format("%dm", math.floor(seconds / 60))
+	else
+		local hours = math.floor(seconds / 3600)
+		local mins = math.floor((seconds % 3600) / 60)
+		if mins > 0 then
+			return string.format("%dh %dm", hours, mins)
+		end
+		return string.format("%dh", hours)
+	end
+end
+
+---Format number with commas
+---@param n number
+---@return string
+function M.format_number(n)
+	if not n then
+		return "0"
+	end
+	local s = tostring(math.floor(n))
+	return s:reverse():gsub("(%d%d%d)", "%1,"):reverse():gsub("^,", "")
+end
+
+---Calculate percentage
+---@param value number
+---@param total number
+---@return number
+function M.calculate_percentage(value, total)
+	if not total or total == 0 then
+		return 0
+	end
+	return math.floor((value / total) * 100)
+end
+
+---Get progress bar characters
+---@param percentage number 0-100
+---@param width number Bar width in characters
+---@return string filled, string empty
+function M.get_progress_bar(percentage, width)
+	percentage = math.max(0, math.min(100, percentage))
+	local filled = math.floor(percentage / 100 * width)
+	return string.rep("█", filled), string.rep("░", width - filled)
+end
+
+---Parse ISO date to timestamp
+---@param iso_string string
+---@return number|nil
+function M.parse_iso_date(iso_string)
+	if not iso_string or iso_string == "" then
+		return nil
+	end
+	local pattern = "(%d+)-(%d+)-(%d+)T(%d+):(%d+):(%d+)"
+	local year, month, day, hour, min, sec = iso_string:match(pattern)
+	if year then
+		return os.time({
+			year = tonumber(year),
+			month = tonumber(month),
+			day = tonumber(day),
+			hour = tonumber(hour),
+			min = tonumber(min),
+			sec = tonumber(sec),
+		})
+	end
+	return nil
+end
+
+---Format date string
+---@param date_str string YYYY-MM-DD
+---@return string
+function M.format_date(date_str)
+	if not date_str then
+		return "Unknown"
+	end
+	local year, month, day = date_str:match("(%d+)-(%d+)-(%d+)")
+	if not year then
+		return "Unknown"
+	end
+
+	local timestamp = os.time({ year = year, month = month, day = day, hour = 12 })
+	local today = os.date("%Y-%m-%d")
+
+	if date_str == today then
+		return os.date("%A, %b %d", timestamp)
+	end
+	return os.date("%a, %b %d", timestamp)
+end
+
+---Calculate streak display
+---@param streak_days number
+---@return string icon, string color
+function M.get_streak_display(streak_days)
+	if streak_days == 0 then
+		return "─", "commentfg"
+	elseif streak_days < 3 then
+		return string.rep("🔥", streak_days), "exred"
+	elseif streak_days < 7 then
+		return "🔥🔥🔥 +" .. (streak_days - 3), "exred"
+	elseif streak_days < 30 then
+		return "🔥🔥🔥 ⚡ +" .. (streak_days - 3), "exred"
+	else
+		return "🔥🔥🔥 ⚡⚡ +" .. (streak_days - 3), "exred"
+	end
+end
+
+---Get trend indicator
+---@param current number
+---@param previous number
+---@return string text, string color
+function M.get_trend(current, previous)
+	if not current or not previous or previous == 0 then
+		return "", "commentfg"
+	end
+
+	local diff = current - previous
+	local pct = math.floor(math.abs(diff) / previous * 100)
+
+	if diff > 0 then
+		return " ↑" .. pct .. "%", "exgreen"
+	elseif diff < 0 then
+		return " ↓" .. pct .. "%", "exred"
+	end
+	return " →", "commentfg"
+end
+
+---Safe array length (handles vim.NIL)
+---@param arr any
+---@return number
+function M.safe_length(arr)
+	if not arr or arr == vim.NIL then
+		return 0
+	end
+	if type(arr) == "table" then
+		return #arr
+	end
+	return 0
+end
+
+---Get top N items from list
+---@param list table
+---@param max number
+---@return string
+function M.top_items(list, max)
+	if not list or #list == 0 then
+		return ""
+	end
+
+	max = max or #list
+	local result = {}
+
+	for i = 1, math.min(max, #list) do
+		result[#result + 1] = tostring(list[i])
+	end
+
+	if #list > max then
+		result[#result + 1] = string.format("+%d more", #list - max)
+	end
+
+	return table.concat(result, ", ")
+end
+
+return M