local core = require "core"
local command = require "core.command"
local common = require "core.common"
local config = require "core.config"
local translate = require "core.doc.translate"
local DocView = require "core.docview"


local function dv()
  return core.active_view
end


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


local function doc_multiline_selections(sort)
  local iter, state, idx, line1, col1, line2, col2 = doc():get_selections(sort)
  return function()
    idx, line1, col1, line2, col2 = iter(state, idx)
    if idx and line2 > line1 and col2 == 1 then
      line2 = line2 - 1
      col2 = #doc().lines[line2]
    end
    return idx, line1, col1, line2, col2
  end
end

local function append_line_if_last_line(line)
  if line >= #doc().lines then
    doc():insert(line, math.huge, "\n")
  end
end

local function save(filename)
  local abs_filename
  if filename then
    filename = core.normalize_to_project_dir(filename)
    abs_filename = core.project_absolute_path(filename)
  end
  doc():save(filename, abs_filename)
  local saved_filename = doc().filename
  core.log("Saved \"%s\"", saved_filename)
end

local function cut_or_copy(delete)
  local full_text = ""
  local text = ""
  core.cursor_clipboard = {}
  core.cursor_clipboard_whole_line = {}
  for idx, line1, col1, line2, col2 in doc():get_selections() do
    if line1 ~= line2 or col1 ~= col2 then
      text = doc():get_text(line1, col1, line2, col2)
      full_text = full_text == "" and text or (full_text .. " " .. text)
      core.cursor_clipboard_whole_line[idx] = false
      if delete then
        doc():delete_to_cursor(idx, 0)
      end
    else -- Cut/copy whole line
      text = doc().lines[line1]
      full_text = full_text == "" and text or (full_text .. text)
      core.cursor_clipboard_whole_line[idx] = true
      if delete then
        if line1 < #doc().lines then
          doc():remove(line1, 1, line1 + 1, 1)
        else
          doc():remove(line1 - 1, math.huge, line1, math.huge)
        end
      end
    end
    core.cursor_clipboard[idx] = text
  end
  core.cursor_clipboard["full"] = full_text
  system.set_clipboard(full_text)
end

local function split_cursor(direction)
  local new_cursors = {}
  for _, line1, col1 in doc():get_selections() do
    if line1 + direction >= 1 and line1 + direction <= #doc().lines then
      table.insert(new_cursors, { line1 + direction, col1 })
    end
  end
  for i,v in ipairs(new_cursors) do doc():add_selection(v[1], v[2]) end
  core.blink_reset()
end

local function set_cursor(x, y, snap_type)
  local line, col = dv():resolve_screen_position(x, y)
  doc():set_selection(line, col, line, col)
  if snap_type == "word" or snap_type == "lines" then
    command.perform("doc:select-" .. snap_type)
  end
  dv().mouse_selecting = { line, col, snap_type }
  core.blink_reset()
end

local selection_commands = {
  ["doc:select-none"] = function()
    local line, col = doc():get_selection()
    doc():set_selection(line, col)
  end
}

local commands = {
  ["doc:cut"] = function()
    cut_or_copy(true)
  end,

  ["doc:copy"] = function()
    cut_or_copy(false)
  end,

  ["doc:undo"] = function()
    doc():undo()
  end,

  ["doc:redo"] = function()
    doc():redo()
  end,

  ["doc:paste"] = function()
    local clipboard = system.get_clipboard()
    -- If the clipboard has changed since our last look, use that instead
    local external_paste = core.cursor_clipboard["full"] ~= clipboard
    if external_paste then
      core.cursor_clipboard = {}
      core.cursor_clipboard_whole_line = {}
    end
    local value, whole_line
    for idx, line1, col1, line2, col2 in doc():get_selections() do
      if #core.cursor_clipboard_whole_line == (#doc().selections/4) then
        value = core.cursor_clipboard[idx]
        whole_line = core.cursor_clipboard_whole_line[idx] == true
      else
        value = clipboard
        whole_line = not external_paste and clipboard:find("\n") ~= nil
      end
      if whole_line then
        doc():insert(line1, 1, value:gsub("\r", ""))
        if col1 == 1 then
          doc():move_to_cursor(idx, #value)
        end
      else
        doc():text_input(value:gsub("\r", ""), idx)
      end
    end
  end,

  ["doc:newline"] = function()
    for idx, line, col in doc():get_selections(false, true) do
      local indent = doc().lines[line]:match("^[\t ]*")
      if col <= #indent then
        indent = indent:sub(#indent + 2 - col)
      end
      doc():text_input("\n" .. indent, idx)
    end
  end,

  ["doc:newline-below"] = function()
    for idx, line in doc():get_selections(false, true) do
      local indent = doc().lines[line]:match("^[\t ]*")
      doc():insert(line, math.huge, "\n" .. indent)
      doc():set_selections(idx, line + 1, math.huge)
    end
  end,

  ["doc:newline-above"] = function()
    for idx, line in doc():get_selections(false, true) do
      local indent = doc().lines[line]:match("^[\t ]*")
      doc():insert(line, 1, indent .. "\n")
      doc():set_selections(idx, line, math.huge)
    end
  end,

  ["doc:delete"] = function()
    for idx, line1, col1, line2, col2 in doc():get_selections() do
      if line1 == line2 and col1 == col2 and doc().lines[line1]:find("^%s*$", col1) then
        doc():remove(line1, col1, line1, math.huge)
      end
      doc():delete_to_cursor(idx, translate.next_char)
    end
  end,

  ["doc:backspace"] = function()
    local _, indent_size = doc():get_indent_info()
    for idx, line1, col1, line2, col2 in doc():get_selections() do
      if line1 == line2 and col1 == col2 then
        local text = doc():get_text(line1, 1, line1, col1)
        if #text >= indent_size and text:find("^ *$") then
          doc():delete_to_cursor(idx, 0, -indent_size)
          return
        end
      end
      doc():delete_to_cursor(idx, translate.previous_char)
    end
  end,

  ["doc:select-all"] = function()
    doc():set_selection(1, 1, math.huge, math.huge)
  end,

  ["doc:select-lines"] = function()
    for idx, line1, _, line2 in doc():get_selections(true) do
      append_line_if_last_line(line2)
      doc():set_selections(idx, line1, 1, line2 + 1, 1)
    end
  end,

  ["doc:select-word"] = function()
    for idx, line1, col1 in doc():get_selections(true) do
      local line1, col1 = translate.start_of_word(doc(), line1, col1)
      local line2, col2 = translate.end_of_word(doc(), line1, col1)
      doc():set_selections(idx, line2, col2, line1, col1)
    end
  end,

  ["doc:join-lines"] = function()
    for idx, line1, col1, line2, col2 in doc():get_selections(true) do
      if line1 == line2 then line2 = line2 + 1 end
      local text = doc():get_text(line1, 1, line2, math.huge)
      text = text:gsub("(.-)\n[\t ]*", function(x)
        return x:find("^%s*$") and x or x .. " "
      end)
      doc():insert(line1, 1, text)
      doc():remove(line1, #text + 1, line2, math.huge)
      if line1 ~= line2 or col1 ~= col2 then
        doc():set_selections(idx, line1, math.huge)
      end
    end
  end,

  ["doc:indent"] = function()
    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      local l1, c1, l2, c2 = doc():indent_text(false, line1, col1, line2, col2)
      if l1 then
        doc():set_selections(idx, l1, c1, l2, c2)
      end
    end
  end,

  ["doc:unindent"] = function()
    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      local l1, c1, l2, c2 = doc():indent_text(true, line1, col1, line2, col2)
      if l1 then
        doc():set_selections(idx, l1, c1, l2, c2)
      end
    end
  end,

  ["doc:duplicate-lines"] = function()
    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      append_line_if_last_line(line2)
      local text = doc():get_text(line1, 1, line2 + 1, 1)
      doc():insert(line2 + 1, 1, text)
      local n = line2 - line1 + 1
      doc():set_selections(idx, line1 + n, col1, line2 + n, col2)
    end
  end,

  ["doc:delete-lines"] = function()
    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      append_line_if_last_line(line2)
      doc():remove(line1, 1, line2 + 1, 1)
      doc():set_selections(idx, line1, col1)
    end
  end,

  ["doc:move-lines-up"] = function()
    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      append_line_if_last_line(line2)
      if line1 > 1 then
        local text = doc().lines[line1 - 1]
        doc():insert(line2 + 1, 1, text)
        doc():remove(line1 - 1, 1, line1, 1)
        doc():set_selections(idx, line1 - 1, col1, line2 - 1, col2)
      end
    end
  end,

  ["doc:move-lines-down"] = function()
    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      append_line_if_last_line(line2 + 1)
      if line2 < #doc().lines then
        local text = doc().lines[line2 + 1]
        doc():remove(line2 + 1, 1, line2 + 2, 1)
        doc():insert(line1, 1, text)
        doc():set_selections(idx, line1 + 1, col1, line2 + 1, col2)
      end
    end
  end,

  ["doc:toggle-block-comments"] = function()
    local comment = doc().syntax.block_comment
    if not comment then
      if doc().syntax.comment then
        command.perform "doc:toggle-line-comments"
      end
      return
    end

    for idx, line1, col1, line2, col2 in doc_multiline_selections(true) do
      local text = doc():get_text(line1, col1, line2, col2)

      -- might need to deal with unicode later...
      local cs, cse = text:find(comment[1], 1, true)
      local ce = text:find(comment[2], #text - #comment[2] + 1, true)

      local new_col1, new_col2 = col1, col2
      -- uncomment
      if cs and ce then
        if line1 == line2 then
          new_col2 = new_col2 - #table.concat(comment)
        else
          new_col2 = new_col2 - #comment[2]
        end

        -- remove 1 whitespace if possible
        if text:sub(cse + 1, cse + 1) == " " then
          cse = cse + 1
          if line1 == line2 then
            new_col2 = new_col2 - 1
          end
        end
        if text:sub(ce - 1, ce - 1) == " " then
          ce = ce - 1
          new_col2 = new_col2 - 1
        end

        text = text:sub(cse + 1, ce - 1)
      -- comment
      else
        text = comment[1] .. " " .. text .. " " .. comment[2]
        if line1 == line2 then
          new_col2 = new_col2 + #table.concat(comment) + 2
        else
          new_col2 = new_col2 + #comment[2] + 1
        end
      end
      doc():remove(line1, col1, line2, col2)
      doc():insert(line1, col1, text)
      doc():set_selections(idx, line1, new_col1, line2, new_col2)
    end
  end,

  ["doc:toggle-line-comments"] = function()
    local comment = doc().syntax.comment
    if not comment then
      if doc().syntax.block_comment then
        command.perform "doc:toggle-block-comments"
      end
      return
    end
    local indentation = doc():get_indent_string()
    local comment_text = comment .. " "
    for idx, line1, _, line2 in doc_multiline_selections(true) do
      local uncomment = true
      local start_offset = math.huge
      for line = line1, line2 do
        local text = doc().lines[line]
        local s = text:find("%S")
        local cs, ce = text:find(comment_text, s, true)
        if s and cs ~= s then
          uncomment = false
          start_offset = math.min(start_offset, s)
        end
      end
      for line = line1, line2 do
        local text = doc().lines[line]
        local s = text:find("%S")
        if uncomment then
          local cs, ce = text:find(comment_text, s, true)
          if ce then
            doc():remove(line, cs, line, ce + 1)
          end
        elseif s then
          doc():insert(line, start_offset, comment_text)
        end
      end
    end
  end,

  ["doc:upper-case"] = function()
    doc():replace(string.upper)
  end,

  ["doc:lower-case"] = function()
    doc():replace(string.lower)
  end,

  ["doc:go-to-line"] = function()
    local dv = dv()

    local items
    local function init_items()
      if items then return end
      items = {}
      local mt = { __tostring = function(x) return x.text end }
      for i, line in ipairs(dv.doc.lines) do
        local item = { text = line:sub(1, -2), line = i, info = "line: " .. i }
        table.insert(items, setmetatable(item, mt))
      end
    end

    core.command_view:enter("Go To Line", function(text, item)
      local line = item and item.line or tonumber(text)
      if not line then
        core.error("Invalid line number or unmatched string")
        return
      end
      dv.doc:set_selection(line, 1  )
      dv:scroll_to_line(line, true)

    end, function(text)
      if not text:find("^%d*$") then
        init_items()
        return common.fuzzy_match(items, text)
      end
    end)
  end,

  ["doc:toggle-line-ending"] = function()
    doc().crlf = not doc().crlf
  end,

  ["doc:save-as"] = function()
    local last_doc = core.last_active_view and core.last_active_view.doc
    if doc().filename then
      core.command_view:set_text(doc().filename)
    elseif last_doc and last_doc.filename then
      local dirname, filename = core.last_active_view.doc.abs_filename:match("(.*)[/\\](.+)$")
      core.command_view:set_text(core.normalize_to_project_dir(dirname) .. PATHSEP)
    end
    core.command_view:enter("Save As", function(filename)
      save(common.home_expand(filename))
    end, function (text)
      return common.home_encode_list(common.path_suggest(common.home_expand(text)))
    end)
  end,

  ["doc:save"] = function()
    if doc().filename then
      save()
    else
      command.perform("doc:save-as")
    end
  end,

  ["file:rename"] = function()
    local old_filename = doc().filename
    if not old_filename then
      core.error("Cannot rename unsaved doc")
      return
    end
    core.command_view:set_text(old_filename)
    core.command_view:enter("Rename", function(filename)
      save(common.home_expand(filename))
      core.log("Renamed \"%s\" to \"%s\"", old_filename, filename)
      if filename ~= old_filename then
        os.remove(old_filename)
      end
    end, function (text)
      return common.home_encode_list(common.path_suggest(common.home_expand(text)))
    end)
  end,


  ["file:delete"] = function()
    local filename = doc().abs_filename
    if not filename then
      core.error("Cannot remove unsaved doc")
      return
    end
    for i,docview in ipairs(core.get_views_referencing_doc(doc())) do
      local node = core.root_view.root_node:get_node_for_view(docview)
      node:close_view(core.root_view, docview)
    end
    os.remove(filename)
    core.log("Removed \"%s\"", filename)
  end,

  ["doc:select-to-cursor"] = function(x, y, clicks)
    local line1, col1 = select(3, doc():get_selection())
    local line2, col2 = dv():resolve_screen_position(x, y)
    dv().mouse_selecting = { line1, col1, nil }
    doc():set_selection(line2, col2, line1, col1)
  end,

  ["doc:set-cursor"] = function(x, y)
    set_cursor(x, y, "set")
  end,

  ["doc:set-cursor-word"] = function(x, y)
    set_cursor(x, y, "word")
  end,

  ["doc:set-cursor-line"] = function(x, y, clicks)
    set_cursor(x, y, "lines")
  end,

  ["doc:split-cursor"] = function(x, y, clicks)
    local line, col = dv():resolve_screen_position(x, y)
    doc():add_selection(line, col, line, col)
  end,

  ["doc:create-cursor-previous-line"] = function()
    split_cursor(-1)
    doc():merge_cursors()
  end,

  ["doc:create-cursor-next-line"] = function()
    split_cursor(1)
    doc():merge_cursors()
  end

}


local translations = {
  ["previous-char"] = translate.previous_char,
  ["next-char"] = translate.next_char,
  ["previous-word-start"] = translate.previous_word_start,
  ["next-word-end"] = translate.next_word_end,
  ["previous-block-start"] = translate.previous_block_start,
  ["next-block-end"] = translate.next_block_end,
  ["start-of-doc"] = translate.start_of_doc,
  ["end-of-doc"] = translate.end_of_doc,
  ["start-of-line"] = translate.start_of_line,
  ["end-of-line"] = translate.end_of_line,
  ["start-of-word"] = translate.start_of_word,
  ["start-of-indentation"] = translate.start_of_indentation,
  ["end-of-word"] = translate.end_of_word,
  ["previous-line"] = DocView.translate.previous_line,
  ["next-line"] = DocView.translate.next_line,
  ["previous-page"] = DocView.translate.previous_page,
  ["next-page"] = DocView.translate.next_page,
}

for name, fn in pairs(translations) do
  commands["doc:move-to-" .. name] = function() doc():move_to(fn, dv()) end
  commands["doc:select-to-" .. name] = function() doc():select_to(fn, dv()) end
  commands["doc:delete-to-" .. name] = function() doc():delete_to(fn, dv()) end
end

commands["doc:move-to-previous-char"] = function()
  for idx, line1, col1, line2, col2 in doc():get_selections(true) do
    if line1 ~= line2 or col1 ~= col2 then
      doc():set_selections(idx, line1, col1)
    end
  end
  doc():move_to(translate.previous_char)
end

commands["doc:move-to-next-char"] = function()
  for idx, line1, col1, line2, col2 in doc():get_selections(true) do
    if line1 ~= line2 or col1 ~= col2 then
      doc():set_selections(idx, line2, col2)
    end
  end
  doc():move_to(translate.next_char)
end

command.add("core.docview", commands)
command.add(function()
  return core.active_view:is(DocView) and core.active_view.doc:has_any_selection()
end ,selection_commands)