require("luasnip.session.snippet_collection").clear_snippets("norg")

local ls = require("luasnip")
local s = ls.snippet
local sn = ls.snippet_node
local i = ls.insert_node
local f = ls.function_node
local d = ls.dynamic_node
local t = ls.text_node
local k = require("luasnip.nodes.key_indexer").new_key
local rep = require("luasnip.extras").rep
local fmt = require("luasnip.extras.fmt").fmt

local DATE_FORMAT = "!%Y-%m-%dT%H:%M:%S+09:00"

local function today_fn()
    return os.date(DATE_FORMAT)
end

-- TODO: reimplement these with git
local function btime_fn()
    local path = vim.fn.expand("%")
    local stat = vim.uv.fs_stat(path)
    assert(stat)
    local btime = (stat.birthtime or stat.mtime).sec
    return os.date(DATE_FORMAT, btime)
end

local function mtime_fn()
    local path = vim.fn.expand("%")
    local stat = vim.uv.fs_stat(path)
    assert(stat)
    local mtime = stat.mtime.sec
    return os.date(DATE_FORMAT, mtime)
end

local function file_index_fn(_, snip)
    local filename = vim.fn.expand("%:t:r")
    local title
    -- TODO: rewrite this with `vim.fs` lua api
    if filename == "index" then
        title = vim.fn.expand("%"):match("([^/]+)/[^/]+$")
    else
        title = vim.fn.substitute(snip.env.TM_FILENAME, "\\..*$", "", "g")
    end
    title = vim.fn.substitute(title, "-", " ", "g")
    return sn(nil, { i(1, title) })
end

-- stylua: ignore
ls.add_snippets("norg", {
    s("!", fmt(
        [[
            @document.meta
            title: {}
            created: {}
            updated: {}
            @end

        ]],
        {
            d(1, file_index_fn),
            f(btime_fn),
            f(mtime_fn),
        }
    )),
    -- cool ranged-tag snippet
    s(
        { trig = "(@+)([^ ]+)(.*)", regTrig = true, wordTrig = false, name = "Dynamic Code Trigger" },
        {
            f(function(_, snip)
                return snip.captures[1]
            end, {}, { key = "prefix" }),
            f(function(_, snip)
                return snip.captures[2]
            end),
            d(1, function(_, snip)
                local args = snip.captures[3]
                return sn(nil, { i(1, args) })
            end),
            t{"", ""},
            f(function(_, snip)
                local current_line = snip.env.TM_CURRENT_LINE
                local trigger_column = current_line:find(snip.trigger, 1, true)
                return string.rep(" ", trigger_column - 1)
            end, {}, { key = "indent" }),
            i(2),
            t{"", ""},
            rep(k("indent")),
            rep(k("prefix")),
            t{"end", ""},
        }
    ),
})