local txtobj = require 'u.txtobj'
local Pos = require 'u.pos'
local Range = require 'u.range'
local Buffer = require 'u.buffer'

local M = {}

function M.setup()
  -- Select whole file:
  txtobj.define('ag', function() return Buffer.current():all() end)

  -- Select current line:
  txtobj.define('a.', function() return Buffer.current():line(Pos.from_pos('.').lnum) end)

  -- Select the nearest quote:
  txtobj.define('aq', function() return Range.find_nearest_quotes() end)
  txtobj.define('iq', function()
    local range = Range.find_nearest_quotes()
    if range == nil then return end
    return range:shrink(1)
  end)

  ---Selects the next quote object (searches forward)
  --- @param q string
  local function define_quote_obj(q)
    local function select_around() return Range.from_txtobj('a' .. q) end

    txtobj.define('a' .. q, function() return select_around() end)
    txtobj.define('i' .. q, function()
      local range = select_around()
      if range == nil or range:is_empty() then return range end

      local start_next = range.start:next(1) or range.start
      local stop_prev = range.stop:next(-1)
      if start_next > stop_prev then return Range.new(start_next) end
      return range:shrink(1) or range
    end)
  end
  define_quote_obj [["]]
  define_quote_obj [[']]
  define_quote_obj [[`]]

  ---Selects the "last" quote object (searches backward)
  --- @param q string
  local function define_last_quote_obj(q)
    local function select_around()
      local curr = Pos.from_pos('.'):find_next(-1, q)
      if not curr then return end
      -- Reset visual selection to current context:
      curr:save_to_pos '.'
      return Range.from_txtobj('a' .. q)
    end

    txtobj.define('al' .. q, function() return select_around() end)
    txtobj.define('il' .. q, function()
      local range = select_around()
      if range == nil or range:is_empty() then return range end

      local start_next = range.start:next(1) or range.start
      local stop_prev = range.stop:next(-1)
      if start_next > stop_prev then return Range.new(start_next) end

      return range:shrink(1) or range
    end)
  end
  define_last_quote_obj [["]]
  define_last_quote_obj [[']]
  define_last_quote_obj [[`]]

  -- Selects the "last" bracket object (searches backward):
  local function define_last_bracket_obj(b, ...)
    local function select_around()
      local curr = Pos.from_pos('.'):find_next(-1, b)
      if not curr then return end

      local other = curr:find_match(1000)
      if not other then return end

      return Range.new(curr, other)
    end

    local keybinds = { ... }
    table.insert(keybinds, b)
    for _, k in ipairs(keybinds) do
      txtobj.define('al' .. k, function() return select_around() end)
      txtobj.define('il' .. k, function()
        local range = select_around()
        return range and range:shrink(1)
      end)
    end
  end
  define_last_bracket_obj('}', 'B')
  define_last_bracket_obj ']'
  define_last_bracket_obj(')', 'b')
  define_last_bracket_obj '>'
end
return M