com.lua

local core = require "core"
local command = require "core.command"
local common = require "core.common"
local config = require "core.config"
local style = require "core.style"
local translate = require "core.doc.translate"

local misc = require "plugins.lite-xl-vibe.misc"
local ResultsView = require "plugins.lite-xl-vibe.ResultsView"

local function dv()
  return core.active_view
end

local function doc()
  return core.active_view.doc
end

local com = {}

com.caret_width__orig = style.caret_width

command.add(nil, {
  ["vibe:switch-to-insert-mode"] = function()
    core.vibe.mode = "insert"
  end,
  ["vibe:switch-to-normal-mode"] = function()
    core.vibe.mode = "normal"
  end,
  ["vibe:escape"] = function()
    if misc.has_selection() then
      misc.drop_selection()
    else
      core.vibe.reset_seq()
    end
  end,
  ["vibe:run-strokes"] = function()
    core.command_view:enter("Strokes to run:", function(text)
      core.vibe.run_stroke_seq(text)
    end)
  end,

  ["vibe:repeat"] = function()
    -- I was deleting the last stroke seq here
    --  (the one activating this repeat command)
    -- but that's wrong
    --  (for one I could in theory run it via CommandView)
    --   so I moved the deletion to vibe.process_stroke
    core.log('vibe:repeat seq=|%s|', core.vibe.last_executed_seq)
    core.vibe.run_stroke_seq(core.vibe.last_executed_seq)
  end,

  ["vibe:open-scratch-buffer"] = function()
    core.root_view:open_doc(core.open_doc(misc.scratch_filepath()))
  end,

  ["vibe:switch-to-last-tab"] = function()
    misc.last_active_view.vibe_parent_node:set_active_view(misc.last_active_view)
  end,

})


command.add(misc.has_selection, {
  ["vibe:copy"] = function()
    core.log('vibe:copy')
    command.perform("doc:copy")
    if core.vibe.target_register then
      core.vibe.registers[core.vibe.target_register] = system.get_clipboard()
      core.vibe.debug_str = "copied to " .. core.vibe.target_register
      -- aand zero it back for further actions
      core.vibe.target_register = nil
    else
      core.vibe.debug_str = "copied"
    end
    misc.drop_selection()
  end,
  ["vibe:delete"] = function()
    core.log('vibe:delete')
    local text = doc():get_text(doc():get_selection())
    if core.vibe.target_register then
      core.vibe.registers[core.vibe.target_register] = text
      -- aand zero it back for further actions
      core.vibe.target_register = nil
    end
    system.set_clipboard(text)
    command.perform("doc:delete")
  end,
  ["vibe:change"] = function()
    command.perform("vibe:delete")
    command.perform("vibe:switch-to-insert-mode")
  end,
  ["vibe:indent"] = function()
    command.perform("doc:indent")
  end,
  ["vibe:unindent"] = function()
    command.perform("doc:unindent")
  end,
})

command.add(nil, {
  ["vibe:switch-to-tab-search"] = function()
    core.command_view:enter("Switch to tab:", function(text, item)
      if item then
        core.root_view:open_doc(item.doc)
      else
        local filename = system.absolute_path(common.home_expand(text))
        core.root_view:open_doc(core.open_doc(filename))
      end
    end, function(text)
      local items = {}
      for _, doc in ipairs(core.docs) do
        table.insert(items, {
          ["text"] = misc.doc_abs_filename(doc),
          ["doc"]  = doc,
        })
      end
      return misc.fuzzy_match_key(items, 'text', text)
    end)
  end,

  ["vibe:tabs-list"] = function()
    if core.vibe.tabs_list_view then
      core.vibe.tabs_list_view:refresh()
      local node = core.root_view:get_active_node()
      node:set_active_view(core.vibe.tabs_list_view)
    else
      core.vibe.tabs_list_view = ResultsView.new_and_add({
        title = "Opened Files",
        items_fun = misc.get_tabs_list,
        on_click_fun = function(res)
          core.root_view:open_doc(res.doc)
        end
      })
    end
  end,
})

command.add("core.docview", {
  ["vibe:rotate-clipboard-ring"] = function()
    misc.clipboard_ring_rotate()
  end,

  ["vibe:move-to-start-of-doc"] = function()
    if core.vibe.num_arg == '' then
      if misc.has_selection() then
        command.perform("doc:select-to-start-of-doc")
      else
        command.perform("doc:move-to-start-of-doc")
      end
    else
      misc.move_to_line(tonumber(core.vibe.num_arg))
    end
  end,

  ["vibe:move-to-end-of-doc"] = function()
    if core.vibe.num_arg == '' then
      if misc.has_selection() then
        command.perform("doc:select-to-end-of-doc")
      else
        command.perform("doc:move-to-end-of-doc")
      end
    else
      misc.move_to_line(tonumber(core.vibe.num_arg))
    end
  end,

  ["vibe:delete-symbol-under-cursor"] = function()
    local doc = core.active_view and core.active_view.doc
    if doc then
      local line, col, line2, col2 = doc:get_selection()
      if line2 ~= line or col2 ~= col then
        doc:delete_to(line2, col2)
      else
        doc:set_selection(line, col)
        doc:delete_to(translate.next_char)
        doc:set_selection(line, col, line2, col2)
      end
    end
  end,

  ["vibe:repeat-find-in-line"] = function()
    if core.vibe.last_line_find == nil then
      core.vibe.debug_str = 'no last line search..'
      return
    end
    doc():move_to(function(doc, line, col)
      return misc.find_in_line(
        core.vibe.last_line_find["symbol"],
        core.vibe.last_line_find["backwards"],
        core.vibe.last_line_find["include"],
        doc, line, col
      )
    end, dv())
  end,

  ["vibe:paste"] = function()
    core.log('vibe:paste')
    local text
    if core.vibe.target_register
        and core.vibe.registers[core.vibe.target_register] then
      system.set_clipboard(core.vibe.registers[core.vibe.target_register], true)
      -- aand zero it back for further actions
      core.vibe.target_register = nil
    end
    if doc():has_selection() then
      text = doc():get_text(doc():get_selection())
    end
    command.perform("doc:paste")
    if text then
      system.set_clipboard(text)
    end
  end,
})

-- can't put this into misc since ResultsView depends on misc
command.add(nil, {
  ["core:exec-history"] = function()
    ResultsView.new_and_add({
      title = "Execution History",
      items = misc.exec_history,
      on_click_fun = function(res)
        misc.exec_text = res.text
        command.perform('root:close')
        command.perform("core:exec-input")
      end
    })
  end,
})

com.inline_search = {}
com.inline_search.search_q = nil
-- com.inline_search.match_fun = function(s, q) return common.fuzzy_match(s, q) and 1 end
com.inline_search.match_fun = function(s, q) return s:lower():find(q, nil, true) end
com.inline_search.thread_name = nil




local function find_all_matches_in_file(t, abs_filename, fn)
  local fp = io.open(abs_filename)
  if not fp then return t end
  local n = 1
  for line in fp:lines() do
    local s = fn(line)
    if s then
      -- Insert maximum 256 characters. If we insert more, for compiled files, which can have very long lines
      -- things tend to get sluggish. If our line is longer than 80 characters, begin to truncate the thing.
      local start_index = math.max(s - 80, 1)
      table.insert(t, {
        abs_filename = abs_filename,
        text = core.normalize_to_project_dir(abs_filename) ..
            ' | ' .. (start_index > 1 and "..." or "") .. line:sub(start_index, 256 + start_index),
        line = n,
        col = s
      })
      core.redraw = true
      if #t >= config.vibe.inline_search_maxN then
        return
      end
    end
    if n % 100 == 0 then
      coroutine.yield()
    end
    n = n + 1
    core.redraw = true
  end
  fp:close()
end

local function project_search(_, list, _, fn)
  local i = 1
  for dir_name, file in core.get_project_files() do
    if com.inline_search.stop_flag then
      return
    end
    if file.type == "file" then
      local path = (dir_name == core.project_dir and "" or (dir_name .. PATHSEP))
      find_all_matches_in_file(list, path .. file.filename, fn)
      if #list >= config.vibe.inline_search_maxN then
        return
      end
    end
    i = i + 1
  end
  core.redraw = true
end

function com.inline_search:update_q(q)
  if self.search_q == q then
    -- pass
  else
    -- stop search
    if com.inline_search.thread_name and core.threads[com.inline_search.thread_name] then
      core.threads[com.inline_search.thread_name] = nil -- .stop = true
    end
    self.search_q = q
    if q == nil then
      return
    end
    local thread_name = 'inline_search' .. tostring(system.get_time)
    -- restart search
    core.add_thread(function()
      project_search(thread_name, core.command_view.suggestions, q,
        function(s) return com.inline_search.match_fun(s, q) end)
    end, thread_name)
    com.inline_search.thread_name = thread_name
  end
end

com.inline_search.general_command_fun = function(match_fun)
  return function()
    com.inline_search.match_fun = match_fun
    core.command_view:enter("Find in project", function(_, item)
        if item then
          misc.goto_mark(item)
        else

        end
      end,
      --suggest
      function(text)
        com.inline_search:update_q(text)
      end,
      -- cancel
      function()
        com.inline_search:update_q()
      end)
  end
end


command.add(nil, {
  ["vibe:inline-project-search"] = com.inline_search.general_command_fun(
    function(s, q) return s:lower():find(q, nil, true) end),
  ["vibe:inline-project-fuzzy-search"] = com.inline_search.general_command_fun(
    function(s, q) return common.fuzzy_match(s, q) and 1 end),
})

if rawget(_G, "regex") then
  command.add(nil, {
    ["vibe:inline-project-regex-search"] = com.inline_search.general_command_fun(
      function(s, q)
        local re = regex.compile(q, "i")
        return regex.cmatch(re, s)
      end),
  })
end

return com