if not pcall(require, "fzf") then
  return
end

local action = require("fzf.actions").action
local core = require "fzf-lua.core"
local path = require "fzf-lua.path"
local utils = require "fzf-lua.utils"
local config = require "fzf-lua.config"
local actions = require "fzf-lua.actions"
local fn, api = vim.fn, vim.api

local M = {}

local function getbufnumber(line)
  return tonumber(string.match(line, "%[(%d+)"))
end

local function getfilename(line)
  -- return string.match(line, "%[.*%] (.+)")
  -- greedy match anything after last nbsp
  return line:match("[^" .. utils.nbsp .. "]*$")
end

M.buffers = function(opts)

  opts = config.normalize_opts(opts, config.globals.buffers)
  if not opts then return end

    local act = action(function (items, fzf_lines, _)
      -- only preview first item
      local item = items[1]
      local buf = getbufnumber(item)
      if api.nvim_buf_is_loaded(buf) then
        return api.nvim_buf_get_lines(buf, 0, fzf_lines, false)
      else
        local name = getfilename(item)
        if fn.filereadable(name) ~= 0 then
          return fn.readfile(name, "", fzf_lines)
        end
        return "UNLOADED: " .. name
      end
    end)

  coroutine.wrap(function ()
    local items = {}

    local bufnrs = vim.tbl_filter(function(b)
      if 1 ~= vim.fn.buflisted(b) then
          return false
      end
      -- only hide unloaded buffers if opts.show_all_buffers is false, keep them listed if true or nil
      if opts.show_all_buffers == false and not vim.api.nvim_buf_is_loaded(b) then
        return false
      end
      if opts.ignore_current_buffer and b == vim.api.nvim_get_current_buf() then
        return false
      end
      if opts.cwd_only and not path.is_relative(vim.api.nvim_buf_get_name(b), vim.loop.cwd()) then
        return false
      end
      return true
    end, opts._list_bufs and opts._list_bufs() or vim.api.nvim_list_bufs())
    if not next(bufnrs) then return end

    local header_line = false
    local buffers = {}
    for _, bufnr in ipairs(bufnrs) do
      local flag = bufnr == vim.fn.bufnr('') and '%' or (bufnr == vim.fn.bufnr('#') and '#' or ' ')

      local element = {
        bufnr = bufnr,
        flag = flag,
        info = vim.fn.getbufinfo(bufnr)[1],
      }

      if opts.sort_lastused and (flag == "#" or flag == "%") then
        if flag == "%" then header_line = true end
        local idx = ((buffers[1] ~= nil and buffers[1].flag == "%") and 2 or 1)
        table.insert(buffers, idx, element)
      else
        table.insert(buffers, element)
      end
    end

    for _, buf in pairs(buffers) do
      -- local hidden = buf.info.hidden == 1 and 'h' or 'a'
      local hidden = ''
      local readonly = vim.api.nvim_buf_get_option(buf.bufnr, 'readonly') and '=' or ' '
      local changed = buf.info.changed == 1 and '+' or ' '
      local flags = hidden .. readonly .. changed
      local leftbr = utils.ansi_codes.clear('[')
      local rightbr = utils.ansi_codes.clear(']')
      local bufname = string.format("%s:%s",
        utils._if(#buf.info.name>0, path.relative(buf.info.name, vim.loop.cwd()), "[No Name]"),
        utils._if(buf.info.lnum>0, buf.info.lnum, ""))
      if buf.flag == '%' then
        flags = utils.ansi_codes.red(buf.flag) .. flags
        leftbr, rightbr = '[', ']'
        if not header_line then
          leftbr = utils.ansi_codes.green(leftbr)
          rightbr = utils.ansi_codes.green(rightbr)
          bufname = utils.ansi_codes.green(bufname)
        end
      elseif buf.flag == '#' then
        flags = utils.ansi_codes.cyan(buf.flag) .. flags
      else
        flags = utils.nbsp .. flags
      end
      local bufnrstr = string.format("%s%s%s", leftbr,
        utils.ansi_codes.yellow(string.format(buf.bufnr)), rightbr)
      local buficon = ''
      local hl = ''
      if opts.file_icons then
        if utils.is_term_bufname(buf.info.name) then
          -- get shell-like icon for terminal buffers
          buficon, hl = core.get_devicon(buf.info.name, "sh")
        else
          local filename = path.tail(buf.info.name)
          local extension = path.extension(filename)
          buficon, hl = core.get_devicon(filename, extension)
        end
        if opts.color_icons then
          buficon = utils.ansi_codes[hl](buficon)
        end
      end
      local item_str = string.format("%s%s%s%s%s%s%s",
        utils._if(header_line and vim.tbl_isempty(items),
          string.format("%-16s", bufnrstr),
          string.format("%-32s", bufnrstr)),
        utils.nbsp,
        flags,
        utils.nbsp,
        buficon,
        utils.nbsp,
        bufname)
      table.insert(items, item_str)
    end

    opts.preview = act
    opts._fzf_cli_args = utils._if(
      header_line and not opts.ignore_current_buffer,
      '--header-lines=1', ''
    )

    local selected = core.fzf(opts, items)

    if not selected then return end

    if #selected > 1 then
      for i = 2, #selected do
          selected[i] = tostring(getbufnumber(selected[i]))
      end
    end

    actions.act(opts.actions, selected)

  end)()
end

M.lines = function(opts)
  opts = config.normalize_opts(opts, config.globals.lines)
  M.buffer_lines(opts)
end

M.blines = function(opts)
  opts = config.normalize_opts(opts, config.globals.blines)
  opts.current_buffer_only = true
  M.buffer_lines(opts)
end


M.buffer_lines = function(opts)
  if not opts then return end

  local buffers = {}
  if opts.current_buffer_only then
    table.insert(buffers, vim.api.nvim_get_current_buf())
  else
    buffers = vim.tbl_filter(function(b)
      if 1 ~= vim.fn.buflisted(b) or utils.is_term_buffer(b) then
          return false
      end
      -- only hide unloaded buffers if opts.show_all_buffers is false, keep them listed if true or nil
      if opts.show_all_buffers == false and not vim.api.nvim_buf_is_loaded(b) then
        return false
      end
      if opts.ignore_current_buffer and b == vim.api.nvim_get_current_buf() then
        return false
      end
      if opts.cwd_only and not path.is_relative(vim.api.nvim_buf_get_name(b), vim.loop.cwd()) then
        return false
      end
      return true
    end, vim.api.nvim_list_bufs())
  end

  coroutine.wrap(function()
    local items = {}

    for _, bufnr in ipairs(buffers) do
      if api.nvim_buf_is_loaded(bufnr) then
        local bufname = path.basename(api.nvim_buf_get_name(bufnr))
        local data = api.nvim_buf_get_lines(bufnr, 0, -1, false)
        local buficon, hl
        if opts.file_icons then
          local filename = path.tail(bufname)
          local extension = path.extension(filename)
          buficon, hl = core.get_devicon(filename, extension)
          if opts.color_icons then
            buficon = utils.ansi_codes[hl](buficon)
          end
        end
        for l, text in ipairs(data) do
          table.insert(items, ("[%s]%s%s%s%s:%s: %s"):format(
            utils.ansi_codes.yellow(tostring(bufnr)),
            utils.nbsp,
            buficon or '',
            buficon and utils.nbsp or '',
            utils.ansi_codes.magenta(bufname),
            utils.ansi_codes.green(tostring(l)),
            text))
        end
      end
    end

    -- ignore bufnr when searching
    -- disable multi-select
    opts.nomulti = true
    opts.preview_window = 'hidden:right:0'
    opts._fzf_cli_args = "--delimiter=']' --nth 2,-1"

    local selected = core.fzf(opts, items)

    if not selected then return end

    -- get the line number
    local line = tonumber(selected[2]:match(":(%d+):"))

    if #selected > 1 then
      for i = 2, #selected do
          selected[i] = tostring(getbufnumber(selected[i]))
      end
    end

    actions.act(opts.actions, selected)

    if line then vim.api.nvim_win_set_cursor(0, {line, 0}) end

  end)()
end

M.tabs = function(opts)

  opts = config.normalize_opts(opts, config.globals.tabs)
  if not opts then return end

  local buf_to_tab = {}
  opts._list_bufs = function()
    local res = {}
    for _, t in ipairs(vim.api.nvim_list_tabpages()) do
      local w = vim.api.nvim_tabpage_get_win(t)
      local b = vim.api.nvim_win_get_buf(w)
      buf_to_tab[b] = t
      table.insert(res, b)
    end
    return res
  end

  opts.nomulti = true
  opts.preview_window = 'hidden:right:0'
  opts.fzf_cli_args = "--delimiter=']' --nth 2,-1" --with-nth 2"

  opts.actions["default"] = function(selected)
      local bufnr = tonumber(selected[2])
      local tabnr = buf_to_tab[bufnr]
      vim.cmd("tabn " .. tabnr)
  end

  M.buffers(opts)
end

return M