gitsigns.nvim/lua/gitsigns/current_line_blame.lua

local a = require('gitsigns.async')
local wrap = a.wrap
local void = a.void
local scheduler = a.scheduler

local cache = require('gitsigns.cache').cache
local config = require('gitsigns.config').config
local BlameInfo = require('gitsigns.git').BlameInfo
local util = require('gitsigns.util')
local nvim = require('gitsigns.nvim')

local api = vim.api

local current_buf = api.nvim_get_current_buf

local namespace = api.nvim_create_namespace('gitsigns_blame')

local timer = vim.loop.new_timer()

local M = {}



local wait_timer = wrap(vim.loop.timer_start, 4)

local function set_extmark(bufnr, row, opts)
   opts = opts or {}
   opts.id = 1
   api.nvim_buf_set_extmark(bufnr, namespace, row - 1, 0, opts)
end

local function get_extmark(bufnr)
   local pos = api.nvim_buf_get_extmark_by_id(bufnr, namespace, 1, {})
   if pos[1] then
      return pos[1] + 1
   end
   return
end

local reset = function(bufnr)
   bufnr = bufnr or current_buf()
   api.nvim_buf_del_extmark(bufnr, namespace, 1)
   pcall(api.nvim_buf_del_var, bufnr, 'gitsigns_blame_line_dict')
end


local max_cache_size = 1000

local BlameCache = {Elem = {}, }








BlameCache.contents = {}

function BlameCache:add(bufnr, lnum, x)
   if not config._blame_cache then return end
   local scache = self.contents[bufnr]
   if scache.size <= max_cache_size then
      scache.cache[lnum] = x
      scache.size = scache.size + 1
   end
end

function BlameCache:get(bufnr, lnum)
   if not config._blame_cache then return end


   local tick = vim.b[bufnr].changedtick
   if not self.contents[bufnr] or self.contents[bufnr].tick ~= tick then
      self.contents[bufnr] = { tick = tick, cache = {}, size = 0 }
   end

   return self.contents[bufnr].cache[lnum]
end

local function expand_blame_format(fmt, name, info)
   local m
   if info.author == name then
      info.author = 'You'
   end

   for k, v in pairs({
         author_time = info.author_time,
         committer_time = info.committer_time,
      }) do
      for _ = 1, 10 do
         m = fmt:match('<' .. k .. ':([^>]+)>')
         if not m then
            break
         end
         if m:match('%%R') then
            m = m:gsub('%%R', util.get_relative_time(v))
         end
         m = os.date(m, v)
         fmt = fmt:gsub('<' .. k .. ':[^>]+>', m)
      end
   end

   for k, v in pairs(info) do
      for _ = 1, 10 do
         m = fmt:match('<' .. k .. '>')
         if not m then
            break
         end
         if vim.endswith(k, '_time') then
            if config.current_line_blame_formatter_opts.relative_time then
               v = util.get_relative_time(v)
            else
               v = os.date('%Y-%m-%d', v)
            end
         end
         fmt = fmt:gsub('<' .. k .. '>', v)
      end
   end
   return fmt
end


local update = void(function()
   local bufnr = current_buf()
   local lnum = api.nvim_win_get_cursor(0)[1]

   local old_lnum = get_extmark(bufnr)
   if old_lnum and lnum == old_lnum and BlameCache:get(bufnr, lnum) then

      return
   end

   if api.nvim_get_mode().mode == 'i' then
      reset(bufnr)
      return
   end





   if get_extmark(bufnr) then
      reset(bufnr)
      set_extmark(bufnr, lnum)
   end

   local opts = config.current_line_blame_opts


   wait_timer(timer, opts.delay, 0)
   scheduler()

   local bcache = cache[bufnr]
   if not bcache or not bcache.git_obj.object_name then
      return
   end

   local result = BlameCache:get(bufnr, lnum)
   if not result then
      local buftext = util.buf_lines(bufnr)
      result = bcache.git_obj:run_blame(buftext, lnum, opts.ignore_whitespace)
      BlameCache:add(bufnr, lnum, result)
      scheduler()
   end

   local lnum1 = api.nvim_win_get_cursor(0)[1]
   if bufnr == current_buf() and lnum ~= lnum1 then

      return
   end

   if not api.nvim_buf_is_loaded(bufnr) then

      return
   end

   api.nvim_buf_set_var(bufnr, 'gitsigns_blame_line_dict', result)
   if opts.virt_text and result then
      local virt_text
      local clb_formatter = result.author == 'Not Committed Yet' and
      config.current_line_blame_formatter_nc or
      config.current_line_blame_formatter
      if type(clb_formatter) == "string" then
         virt_text = { {
            expand_blame_format(clb_formatter, bcache.git_obj.repo.username, result),
            'GitSignsCurrentLineBlame',
         }, }
      else
         virt_text = clb_formatter(
         bcache.git_obj.repo.username,
         result,
         config.current_line_blame_formatter_opts)

      end

      set_extmark(bufnr, lnum, {
         virt_text = virt_text,
         virt_text_pos = opts.virt_text_pos,
         priority = opts.virt_text_priority,
         hl_mode = 'combine',
      })
   end
end)

M.setup = function()
   nvim.augroup('gitsigns_blame')

   for k, _ in pairs(cache) do
      reset(k)
   end

   if config.current_line_blame then
      nvim.autocmd(
      { 'FocusGained', 'BufEnter', 'CursorMoved', 'CursorMovedI' },
      { group = 'gitsigns_blame', callback = function() update() end })


      nvim.autocmd(
      { 'InsertEnter', 'FocusLost', 'BufLeave' },
      { group = 'gitsigns_blame', callback = function() reset() end })


      update()
   end
end

return M