viye @ 0da3343

local M = {}

local function indent(text) return #text:match "^%s*" end

local function is_data(text) return text:find "^%s*[|:] " ~= nil end

local function strip_bullet(text) return (text:gsub("^[+-] ", "", 1)) end

local function build_path()

  local lnum = vim.fn.line "."

  local line = vim.fn.getline(lnum)

  local depth = indent(line)

  local parts = { strip_bullet(vim.trim(line)) }

  local i = lnum - 1

  while i >= 1 do

    local l = vim.fn.getline(i)

    local d = indent(l)

    if d == 0 and l:find "%S" and depth > 0 then

      table.insert(parts, 1, strip_bullet(vim.trim(l)))

      break

    end

    if d < depth and l:find "%S" then

      table.insert(parts, 1, strip_bullet(vim.trim(l)))

      depth = d

    end

    i = i - 1

  end

  for idx = 1, #parts - 1 do

    parts[idx] = parts[idx]:gsub("/+$", "")

  end

  return table.concat(parts, "/")

end

local function has_children()

  local lnum = vim.fn.line "."

  local depth = indent(vim.fn.getline(lnum))

  local next_line = vim.fn.getline(lnum + 1)

  return next_line ~= "" and indent(next_line) > depth

end

-- Collect : body lines under the current line (siblings at depth+2)

-- Returns list of stripped lines (with ": " prefix preserved)

local function collect_body()

  local lnum = vim.fn.line "."

  local depth = indent(vim.fn.getline(lnum))

  local body = {}

  local i = lnum + 1

  while i <= vim.fn.line "$" do

    local l = vim.fn.getline(i)

    if l == "" then break end

    local d = indent(l)

    if d <= depth then break end

    if d == depth + 2 and l:find("^%s*: ") then

      table.insert(body, vim.trim(l))

    end

    i = i + 1

  end

  return body

end

-- Remove | output lines under the current line (siblings at depth+2)

-- Returns the line number after the removed block, or nil if nothing removed

local function remove_output()

  local lnum = vim.fn.line "."

  local depth = indent(vim.fn.getline(lnum))

  local first = nil

  local last = nil

  local i = lnum + 1

  while i <= vim.fn.line "$" do

    local l = vim.fn.getline(i)

    if l == "" then break end

    local d = indent(l)

    if d <= depth then break end

    if d == depth + 2 and l:find("^%s*| ") then

      if first == nil then first = i end

      last = i

    end

    i = i + 1

  end

  if first then

    vim.cmd(string.format("%d,%ddelete _", first, last))

    return first

  end

  return nil

end

local function collapse()

  local lnum = vim.fn.line "."

  local depth = indent(vim.fn.getline(lnum))

  local s = lnum + 1

  local e = s

  while e <= vim.fn.line "$" do

    local l = vim.fn.getline(e)

    if l == "" or indent(l) <= depth then break end

    e = e + 1

  end

  if e > s then vim.cmd(string.format("%d,%ddelete _", s, e - 1)) end

end

local function toggle_bullet()

  local lnum = vim.fn.line "."

  local line = vim.fn.getline(lnum)

  local toggled = line:gsub("^%+ ", "- ", 1)

  if toggled ~= line then

    vim.fn.setline(lnum, toggled)

    return

  end

  toggled = line:gsub("^%- ", "+ ", 1)

  if toggled ~= line then vim.fn.setline(lnum, toggled) end

end

local function insert_output(output, after_lnum)

  after_lnum = after_lnum or vim.fn.line "."

  local pad = string.rep(" ", indent(vim.fn.getline(after_lnum)) + 2)

  local lines = vim.split(output, "\n", { trimempty = true })

  for i, line in ipairs(lines) do lines[i] = pad .. line end

  vim.fn.append(after_lnum, lines)

end

-- Find the last child line at depth+2 under the current line

local function find_last_child()

  local lnum = vim.fn.line "."

  local depth = indent(vim.fn.getline(lnum))

  local last = lnum

  local i = lnum + 1

  while i <= vim.fn.line "$" do

    local l = vim.fn.getline(i)

    if l == "" then break end

    local d = indent(l)

    if d <= depth then break end

    if d == depth + 2 then

      last = i

    end

    i = i + 1

  end

  return last

end

local function exec(reenter)

  local line = vim.fn.getline "."

  if is_data(line) then

    local lnum = vim.fn.line "."

    local depth = indent(line)

    local i = lnum - 1

    while i >= 1 do

      local l = vim.fn.getline(i)

      if not is_data(l) and indent(l) < depth and l:find "%S" then

        vim.fn.cursor(i, 1)

        exec(true)

        return

      end

      i = i - 1

    end

    return

  end

  if not reenter and has_children() then

    collapse()

    toggle_bullet()

    return

  end

  local path = build_path()

  if path == "" then return end

  local body = reenter and collect_body() or {}

  local cmd = "viye " .. vim.fn.shellescape(path)

  if #body > 0 then

    cmd = cmd .. " --"

    for _, line in ipairs(body) do

      cmd = cmd .. " " .. vim.fn.shellescape(line)

    end

  end

  vim.notify("viye: " .. cmd, vim.log.levels.INFO)

  local out = vim.fn.system(cmd)

  if vim.v.shell_error ~= 0 then

    vim.notify("viye: " .. out:gsub("\n$", ""), vim.log.levels.ERROR)

    return

  end

  if out ~= "" then

    remove_output()

    local target = reenter and find_last_child() or vim.fn.line "."

    insert_output(out, target)

    if not reenter then

      toggle_bullet()

    end

  end

end

function M.setup()

  vim.filetype.add { extension = { viye = "viye" } }

  vim.api.nvim_create_autocmd("FileType", {

    pattern = "viye",

    group = vim.api.nvim_create_augroup("viye_plugin", { clear = true }),

    callback = function() vim.keymap.set("n", "<CR>", exec, { buffer = true, silent = true }) end,

  })

end

return M