viye @ 486dcc18817486f28b23ca2488dd1dfac80724c4

if vim.g.loaded_viye_plugin ~= nil then return end

vim.g.loaded_viye_plugin = true

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

local function tokenize(s)

  local out, buf, q = {}, "", nil

  for i = 1, #s do

    local c = s:sub(i, i)

    if q then

      if c == q then q = nil else buf = buf .. c end

    elseif c == "'" or c == '"' then

      q = c

    elseif c:match("%s") then

      if #buf > 0 then

        table.insert(out, buf); buf = ""

      end

    else

      buf = buf .. c

    end

  end

  if #buf > 0 then table.insert(out, buf) end

  return out

end

local function children_of(cmd_lnum)

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

  local first, last = cmd_lnum + 1, nil

  for i = first, vim.fn.line("$") do

    local l = vim.fn.getline(i)

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

    last = i

  end

  return first, last

end

local function parent_of(lnum)

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

  for i = lnum - 1, 1, -1 do

    local l = vim.fn.getline(i)

    if l:find("%S") and not l:match("^%s*[|:]") and indent(l) < d then return i end

  end

end

local function ancestors(cmd_lnum)

  local parts, lnum = {}, cmd_lnum

  while lnum do

    local l = vim.fn.getline(lnum)

    local line = (l:match("^%s*[+-] (.*)") or vim.trim(l)):gsub("/$", "")

    local words = tokenize(line)

    for i = #words, 1, -1 do table.insert(parts, 1, words[i]) end

    lnum = parent_of(lnum)

  end

  return parts

end

local function body_of(cmd_lnum)

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

  local first, last = children_of(cmd_lnum)

  local body = {}

  if last then

    for i = first, last do

      local l = vim.fn.getline(i)

      local val = indent(l) == depth + 2 and l:match("^%s*: (.*)")

      if val then

        table.insert(body, ": " .. val)

      end

    end

  end

  return body

end

local function insert_output(cmd_lnum, out)

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

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

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

  vim.api.nvim_buf_set_lines(0, cmd_lnum, cmd_lnum, false, lines)

end

local function toggle_bullet(cmd_lnum)

  vim.fn.setline(cmd_lnum, (vim.fn.getline(cmd_lnum):gsub("^(%s*)([+-]) ", function(ws, c)

    return ws .. (c == "+" and "-" or "+") .. " "

  end, 1)))

end

local function run(cmd_lnum)

  local cmd = ancestors(cmd_lnum)

  if #cmd == 0 then return end

  local body = body_of(cmd_lnum)

  if #body > 0 then vim.list_extend(vim.list_extend(cmd, { "--" }), body) end

  vim.system({ "viye", unpack(cmd) }, { text = true, env = { GOGC = "off" } }, function(res)

    vim.schedule(function()

      if res.code ~= 0 then

        vim.notify(vim.trim(res.stderr), vim.log.levels.ERROR)

        return

      end

      local pos = vim.api.nvim_win_get_cursor(0)

      local first, last = children_of(cmd_lnum)

      if last then vim.api.nvim_buf_set_lines(0, first - 1, last, false, {}) end

      if res.stdout ~= "" then insert_output(cmd_lnum, res.stdout) end

      toggle_bullet(cmd_lnum)

      local row = math.min(pos[1], vim.api.nvim_buf_line_count(0))

      local line = vim.api.nvim_buf_get_lines(0, row - 1, row, false)[1] or ""

      vim.api.nvim_win_set_cursor(0, { row, math.min(pos[2], #line) })

    end)

  end)

end

local function exec()

  local lnum = vim.fn.line(".")

  local line = vim.fn.getline(lnum)

  local is_out = line:match("^%s*|")

  local cmd_lnum = (line:match("^%s*[|:]") and parent_of(lnum)) or lnum

  if not cmd_lnum or vim.fn.getline(cmd_lnum):match("^==%s") then return end

  if is_out then

    local pos = vim.api.nvim_win_get_cursor(0)

    local first, last = children_of(cmd_lnum)

    if last then vim.api.nvim_buf_set_lines(0, first - 1, last, false, {}) end

    local row = math.min(pos[1], vim.api.nvim_buf_line_count(0))

    local line = vim.api.nvim_buf_get_lines(0, row - 1, row, false)[1] or ""

    vim.api.nvim_win_set_cursor(0, { row, math.min(pos[2], #line) })

    return

  end

  run(cmd_lnum)

end

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

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

vim.api.nvim_create_autocmd("BufEnter", {

  pattern = "*",

  group = group,

  callback = function()

    if vim.api.nvim_buf_get_name(0) == "" then

      vim.bo.filetype = "viye"

    end

  end,

})

vim.api.nvim_create_autocmd("FileType", {

  pattern = "viye",

  group = group,

  callback = function()

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

  end,

})