summaryrefslogtreecommitdiff
path: root/lua/telescope/mappings.lua
blob: 6c8641c42c165b1ea5c60b45192a6f6cfc9d4a7b (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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
-- TODO: Customize keymap
local a = vim.api

local mappings = {}

local keymap_store = setmetatable({}, {
  __index = function(t, k)
    rawset(t, k, {})

    return rawget(t, k)
  end
})

local _mapping_key_id = 0
local get_next_id = function()
  _mapping_key_id = _mapping_key_id + 1
  return _mapping_key_id
end

local assign_function = function(prompt_bufnr, func)
  local func_id = get_next_id()

  keymap_store[prompt_bufnr][func_id] = func

  return func_id
end


--[[
Usage:

mappings.apply_keymap(42, <function>, {
  n = {
    ["<leader>x"] = "just do this string",

    ["<CR>"] = function(picker, prompt_bufnr)
      actions.close_prompt()

>     local filename = ...
      vim.cmd(string.format(":e %s", filename))
    end,
  },

  i = {
  }
})
--]]
local telescope_map = function(prompt_bufnr, mode, key_bind, key_func, opts)
  opts = opts or {
    silent = true
  }

  if type(key_func) == "string" then
    a.nvim_buf_set_keymap(
      prompt_bufnr,
      mode,
      key_bind,
      key_func,
      opts or {
        silent = true
      }
    )
  else
    local key_id = assign_function(prompt_bufnr, key_func)
    local prefix = ""

    local map_string
    if opts.expr then
      map_string = string.format(
        [[luaeval("require('telescope.mappings').execute_keymap(%s, %s)")]],
        prompt_bufnr,
        key_id
      )
    else
      if mode == "i" and not opts.expr then
        prefix = "<C-O>"
      end

      map_string = string.format(
        "%s:lua require('telescope.mappings').execute_keymap(%s, %s)<CR>",
        prefix,
        prompt_bufnr,
        key_id
      )
    end

    a.nvim_buf_set_keymap(
      prompt_bufnr,
      mode,
      key_bind,
      map_string,
      opts
    )
  end
end

mappings.apply_keymap = function(prompt_bufnr, attach_mappings, buffer_keymap)
  local applied_mappings = { n = {}, i = {} }

  local map = function(mode, key_bind, key_func, opts)
    applied_mappings[mode][key_bind] = true

    telescope_map(prompt_bufnr, mode, key_bind, key_func, opts)
  end

  if attach_mappings and not attach_mappings(map) then
    return
  end

  for mode, mode_map in pairs(buffer_keymap) do
    -- TODO: Probalby should not overwrite any keymaps
    -- local buffer_keymaps 

    for key_bind, key_func in pairs(mode_map) do
      if not applied_mappings[mode][key_bind] then
        telescope_map(prompt_bufnr, mode, key_bind, key_func)
      end
    end
  end

  vim.cmd(string.format(
    [[autocmd BufDelete %s :lua require('telescope.mappings').clear(%s)]],
    prompt_bufnr,
    prompt_bufnr
  ))
end

mappings.execute_keymap = function(prompt_bufnr, keymap_identifier)
  local key_func = keymap_store[prompt_bufnr][keymap_identifier]

  assert(
    key_func,
    string.format(
      "Unsure of how we got this failure: %s %s",
      prompt_bufnr,
      keymap_identifier
    )
  )

  key_func(prompt_bufnr)
end

mappings.clear = function(prompt_bufnr)
  keymap_store[prompt_bufnr] = nil
end

return mappings