local M = {}

local function set_telescope_map(key, picker)
  vim.keymap.set(
    'n',
    string.format('<c-;>%s', key),
    string.format('<cmd>Telescope %s<cr>', picker)
  )
  vim.keymap.set(
    'n',
    string.format('<c-;><c-%s>', key),
    string.format('<cmd>Telescope %s<cr>', picker)
  )
end

function M.setup()
  require('telescope').setup {
    defaults = {
      mappings = {
        i = {
          ['<c-s>'] = 'select_horizontal',
          ['<c-t>'] = 'select_tab',
          ['<c-v>'] = 'select_vertical',
          ['<esc>'] = 'close',
        },
      },
    },
    pickers = {
      buffers = { theme = 'ivy' },
      find_files = { theme = 'ivy' },
      live_grep = { theme = 'ivy' },
      lsp_definitions = { theme = 'cursor' },
      lsp_implementations = { theme = 'cursor' },
      lsp_type_definitions = { theme = 'cursor' },
      lsp_workspace_symbols = { theme = 'ivy' },
    },
  }

  -- General
  set_telescope_map(';', 'resume')
  set_telescope_map('b', 'buffers')
  set_telescope_map('f', 'find_files')
  set_telescope_map('g', 'live_grep')

  -- LSP
  set_telescope_map('?', 'diagnostics')
  set_telescope_map(']', 'lsp_definitions')
  set_telescope_map('i', 'lsp_implementations')
  set_telescope_map('s', 'lsp_workspace_symbols')
  set_telescope_map('t', 'lsp_type_definitions')

  vim.api.nvim_create_autocmd('User', {
    pattern = 'TelescopePreviewerLoaded',
    callback = function()
      vim.wo.wrap = true
    end,
  })
end

return M