telescope-frecency.nvim/lua/telescope/_extensions/frecency/db_client.lua

local sqlwrap = require("telescope._extensions.frecency.sql_wrapper")
local util    = require("telescope._extensions.frecency.util")

local MAX_TIMESTAMPS = 10

-- modifier used as a weight in the recency_score calculation:
local recency_modifier = {
  [1] = { age = 240   , value = 100 }, -- past 4 hours
  [2] = { age = 1440  , value = 80  }, -- past day
  [3] = { age = 4320  , value = 60  }, -- past 3 days
  [4] = { age = 10080 , value = 40  }, -- past week
  [5] = { age = 43200 , value = 20  }, -- past month
  [6] = { age = 129600, value = 10  }  -- past 90 days
}

local sql_wrapper = nil

local function init()
  if sql_wrapper then return end

  sql_wrapper = sqlwrap:new()
  sql_wrapper:bootstrap()

  -- setup autocommands
  vim.api.nvim_command("augroup TelescopeFrecency")
  vim.api.nvim_command("autocmd!")
  vim.api.nvim_command("autocmd BufWinEnter,BufWritePost * lua require'telescope._extensions.frecency.db_client'.autocmd_handler(vim.fn.expand('<amatch>'))")
  vim.api.nvim_command("augroup END")
end

local function calculate_file_score(frequency, timestamps)
  local recency_score = 0
  for _, ts in pairs(timestamps) do
    for _, rank in ipairs(recency_modifier) do
      if ts.age <= rank.age then
        recency_score = recency_score + rank.value
        goto continue
      end
    end
    ::continue::
  end

  return frequency * recency_score / MAX_TIMESTAMPS
end

local function filter_timestamps(timestamps, file_id)
  local res = {}
  for _, entry in pairs(timestamps) do
    if entry.file_id == file_id then
      table.insert(res, entry)
    end
  end
  return res
end

local function get_file_scores()
  if not sql_wrapper then return {} end

  local queries = sql_wrapper.queries
  local scores = {}
  local files           = sql_wrapper:do_transaction(queries.file_get_entries, {})
  local timestamp_ages  = sql_wrapper:do_transaction(queries.timestamp_get_all_entry_ages, {})

  -- print(vim.inspect(files))
  -- print(vim.inspect(timestamp_ages))
  if vim.tbl_isempty(files) then return scores end

  for _, file_entry in ipairs(files) do
    table.insert(scores, {
      filename = file_entry.path,
      score    = calculate_file_score(file_entry.count, filter_timestamps(timestamp_ages, file_entry.id))
    })
  end

  -- sort the table
  local function compare(a, b)
    return a.score > b.score
  end
  table.sort(scores, compare)

  return scores
end

local function autocmd_handler(filepath)
  if not sql_wrapper or util.string_isempty(filepath) then return end

  -- check if file is registered as loaded
  if not vim.b.frecency_registered then
    -- allow noname files to go unregistered until BufWritePost
    if not util.fs_stat(filepath).exists then return end

    -- TODO: only register buffer if update did something?
    -- TODO: apply filetype_ignore here?
    vim.b.frecency_registered = 1
    -- print("registered buffer")
    sql_wrapper:update(filepath)
  end
end

local function validate()
  if not sql_wrapper then return {} end

  local queries = sql_wrapper.queries
  local files = sql_wrapper:do_transaction(queries.file_get_entry, {})
  for _, entry in pairs(files) do
    if not util.fs_stat(entry.path).exists then
      -- remove entries from file and timestamp tables
      print("removing entry: " .. entry.path .. "[" .. entry.id .."]")
      sql_wrapper:do_eval(queries.file_delete_entry, { id = entry.id })
      sql_wrapper:do_eval(queries.timestamp_delete_with_file_id, { file_id = entry.id })
    end
  end
end

return {
  init            = init,
  get_file_scores = get_file_scores,
  autocmd_handler = autocmd_handler,
  validate        = validate,
}