summaryrefslogtreecommitdiff
path: root/lua/telescope/previewers/previewer.lua
blob: f986dac799cddb3cbd99849142bb22ec961df95e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
local utils = require "telescope.utils"

local Previewer = {}
Previewer.__index = Previewer

local force_function_wrap = function(value)
  if value ~= nil then
    if type(value) ~= "function" then
      return function()
        return tostring(value)
      end
    else
      return value
    end
  end
end

function Previewer:new(opts)
  opts = opts or {}

  return setmetatable({
    state = nil,
    _title_fn = force_function_wrap(opts.title),
    _dyn_title_fn = force_function_wrap(opts.dyn_title),
    _setup_func = opts.setup,
    _teardown_func = opts.teardown,
    _send_input = opts.send_input,
    _scroll_fn = opts.scroll_fn,
    preview_fn = opts.preview_fn,
    _empty_bufnr = nil,
  }, Previewer)
end

function Previewer:preview(entry, status)
  if not entry then
    if not self._empty_bufnr then
      self._empty_bufnr = vim.api.nvim_create_buf(false, true)
    end

    if vim.api.nvim_buf_is_valid(self._empty_bufnr) then
      vim.api.nvim_win_set_buf(status.preview_win, self._empty_bufnr)
    end
    return
  end

  if not self.state then
    if self._setup_func then
      self.state = self:_setup_func(status)
    else
      self.state = {}
    end
  end

  return self:preview_fn(entry, status)
end

function Previewer:title(entry, dynamic)
  if dynamic == true and self._dyn_title_fn ~= nil then
    if entry == nil then
      if self._title_fn ~= nil then
        return self:_title_fn()
      else
        return ""
      end
    end
    return self:_dyn_title_fn(entry)
  end
  if self._title_fn ~= nil then
    return self:_title_fn()
  end
end

function Previewer:teardown()
  if self._empty_bufnr then
    utils.buf_delete(self._empty_bufnr)
  end
  if self._teardown_func then
    self:_teardown_func()
  end
end

function Previewer:send_input(input)
  if self._send_input then
    self:_send_input(input)
  else
    vim.api.nvim_err_writeln "send_input is not defined for this previewer"
  end
end

function Previewer:scroll_fn(direction)
  if self._scroll_fn then
    self:_scroll_fn(direction)
  else
    vim.api.nvim_err_writeln "scroll_fn is not defined for this previewer"
  end
end

return Previewer