telescope-frecency.nvim/lua/frecency/timer.lua

local config = require "frecency.config"
local log = require "frecency.log"

---@class FrecencyTimer
---@field has_lazy boolean?
local M = {}

---@param event string
---@return nil
function M.track(event)
  if not config.debug then
    return
  elseif M.has_lazy == nil then
    M.has_lazy = (pcall(require, "lazy.stats"))
    if not M.has_lazy then
      log.debug "frecency.timer needs lazy.nvim"
    end
  end
  if M.has_lazy then
    local stats = require "lazy.stats"
    ---@param n integer
    ---@return string
    local function make_key(n)
      return ("[telescope-frecency] %s: %d"):format(event, n)
    end
    local key
    local num = 0
    while true do
      key = make_key(num)
      if not stats._stats.times[key] then
        break
      end
      num = num + 1
    end
    stats.track(key)
  end
end

---@return string
function M.pp()
  local times = require("lazy.stats")._stats.times
  local result = vim.tbl_map(function(k)
    return { event = k, t = times[k] }
  end, vim.tbl_keys(times))
  table.sort(result, function(a, b)
    return a.t < b.t
  end)
  return table.concat(
    vim.tbl_map(function(r)
      return ("%8.3f : %s"):format(r.t, r.event)
    end, result),
    "\n"
  )
end

return M