-- misc stuff implemented in lua

local M = {}

function M.do_under_cursor(obj, cb)
  return cb(vim.fn.expand(vim.fn.expand(obj)))
end

function M.word_under_cursor(cb)
  return M.do_under_cursor('<cword>', cb)
end

function M.expr_under_cursor(cb)
  return M.do_under_cursor('<cexpr>', cb)
end

function M.file_under_cursor(cb)
  return M.do_under_cursor('<cfile>', cb)
end

function M.open_under_cursor(cmd, detect_cmd)
  return M.file_under_cursor(function(txt)
    txt = vim.trim(txt)
    if not cmd and detect_cmd then
      if vim.regex([[^http.*$]]):match_str(txt) then
        cmd = 'url-launcher'
      end
      if vim.regex([[^(\.\./|[\w\d_/\-])*(\.[\w\d]+)?$]]):match_str(txt) then
        cmd = ':edit'
      end
      vim.notify("open_under_cursor: detected command " .. txt .. " => " .. (cmd or "<nil>"))
    end
    if not cmd then
      vim.fn.inputsave()
      vim.ui.input('open with: ', function(result) cmd = result end)
      vim.fn.inputrestore()
      if not cmd then return end
    end
    -- detect vim command
    if cmd:sub(1, 1) == ":" then
      vim.cmd(cmd:sub(2) .. ' ' .. txt)
    else
      vim.loop.spawn(cmd, {args = {txt}})
    end
  end)
end

local function get_visual_selection()
  local top = vim.fn.getpos("'<")
  top = { ln = top[2], col = top[3] }
  local bot = vim.fn.getpos("'>")
  bot = { ln = bot[2], col = bot[3] }
  local lines = vim.api.nvim_buf_get_lines(0, (top.ln - 1), bot.ln, false)
  if vim.opt.selection:get() == 'inclusive' then
    lines[#lines] = lines[#lines]:sub(1, bot.col)
  else
    lines[#lines] = lines[#lines]:sub(1, (bot.col -1))
  end
  lines[1] = lines[1]:sub(top.col)
  return top, bot, lines
end

function M.sort_lines(_, preview_ns, preview_bufnr)
  local bufnr = vim.api.nvim_get_current_buf()

  local top, bot, lines = get_visual_selection()
  for i=1, #lines do
    lines[i] = vim.fn.join(vim.fn.sort(vim.fn.split(lines[i], " ")))
  end

  if not preview_ns then
    vim.api.nvim_buf_set_lines(bufnr, top.ln-1, bot.ln, false, lines)
    vim.notify('no preview')
    return 0
  end

  -- inccommand preview
  if preview_ns ~= nil then
    for i, line in ipairs(lines) do
      vim.api.nvim_buf_set_extmark(bufnr, preview_ns, top.ln+i - 2, 0, {
        hl_mode = 'combine',
        virt_text_pos = 'overlay',
        virt_text = {{line, 'Substitute'}},
      })

      if preview_bufnr ~= nil then
        local prefix = string.format('|%d| ', top.ln + i - 1)
        vim.api.nvim_buf_set_lines(preview_bufnr, i-1, -1, false, { prefix .. line })
        vim.api.nvim_buf_add_highlight(preview_bufnr, preview_ns, 'Substitute', i, #prefix, #prefix+#line)
      end
    end
    return (#lines > 1 and 2 or 1)
  end
end

local function get_searchdirs(dirs)
  -- skip asking if we're looking at known filetypes
  local ft = vim.api.nvim_buf_get_option(0, 'filetype')
  if not dirs and (ft == 'dirbuf' or ft == 'alpha') then
    dirs = '%:p:h'
  end
  if not dirs then
    vim.ui.input('Search directories: ', function(result) dirs = result end)
  end
  -- default to something reasonable
  if not dirs then dirs = '%:p:h' end
  return vim.split(vim.fn.expand(dirs), ',')
end

function M.live_grep(dirs, opts)
  opts = opts or {}
  local search_dirs = get_searchdirs(dirs)
  opts.search_dirs = search_dirs
  opts.prompt_title = 'Grep: '.. vim.inspect(search_dirs)
  return require('telescope.builtin').live_grep(opts)
end

function M.find_files(dirs, opts)
  opts = opts or {}
  local search_dirs = get_searchdirs(dirs)
  opts.search_dirs = search_dirs
  opts.prompt_title = 'Files: '.. vim.inspect(search_dirs)
  return require('telescope.builtin').find_files(opts)
end


function M.fold_block() -- {{{
  local Comment = require('Comment.api')
  local m = vim.api.nvim_buf_get_mark
  local inital_cusor_pos = vim.api.nvim_win_get_cursor(0)
  local spos, epos = m(0, '<'), m(0, '>')

  -- do start of selection
  vim.api.nvim_win_set_cursor(0, spos)
  Comment.insert_linewise_eol()
  vim.api.nvim_feedkeys('\\<esc>', 'ni', true)
  local sln = vim.api.nvim_get_current_line()
  vim.api.nvim_set_current_line(sln .. '{{{')

  -- do end of selection
  vim.api.nvim_win_set_cursor(0, epos)
  Comment.insert_linewise_eol()
  vim.api.nvim_feedkeys('\\<esc>', 'ni', true)
  local eln = vim.api.nvim_get_current_line()
  vim.api.nvim_set_current_line(eln .. '}}}')

  -- restore initial cursor postition
  vim.api.nvim_win_set_cursor(0, inital_cusor_pos)
end -- }}}

local function render_link_preview(buf, ln, link)
  local j = require('plenary.job'):new({
    command = vim.env.SHELL,
    args = {'-c', [[ curl -sSfL ]] .. link .. [[ | grep -iPo '(?<=property="og:title" content=")([^\"]+)(?=")' ]]},
    enable_recording = true,
  })
  local results, code = j:sync()
  if not code == 0 or #results == 0 then return end

  local ns = vim.api.nvim_create_namespace('')
  vim.api.nvim_buf_set_extmark(buf, ns, ln, 0, {
    virt_text = {{results[1], 'Error'}},
    virt_text_pos = 'eol'
  })
end
function M.link_preview()
  local link = vim.fn.expand('<cfile>')
  if not link then return end

  local ln = (vim.api.nvim_win_get_cursor(0)[1]) - 1

  render_link_preview(0, ln, link)
end

return M