local finders = require "telescope.finders"
local make_entry = require "telescope.make_entry"
local pickers = require "telescope.pickers"
local sorters = require "telescope.sorters"
local conf = require("telescope.config").values
local flatten = vim.tbl_flatten

local M = {}

function M.list()
  require("telescope.builtin").find_files {
    prompt_title = "Notes",
    cwd = "~/.notes"
  }
end

function M.tags()
  local opts = {}

  local vimgrep_arguments = conf.vimgrep_arguments
  opts.cwd = vim.fn.expand("~/.notes")

  local additional_args = {}
  if opts.additional_args ~= nil and type(opts.additional_args) == "function" then
    additional_args = opts.additional_args(opts)
  end

  local live_grepper = finders.new_job(function(prompt)
    if not prompt or prompt == "" then
      prompt = "#[A-z]+"
    else
      prompt = "#" .. prompt
    end

    local command = flatten { vimgrep_arguments, additional_args, "--", prompt }
    return command
  end, make_entry.gen_from_vimgrep(opts), opts.max_results, opts.cwd)

  pickers.new(opts, {
    prompt_title = "Note Tags",
    finder = live_grepper,
    previewer = conf.grep_previewer(opts),
    sorter = sorters.highlighter_only(opts),
  }):find()
end

return M