81 lines
1.9 KiB
Lua
81 lines
1.9 KiB
Lua
local COLORS = { red = "red", yellow = "yellow", orange = "magenta", green = "green", blue = "blue" }
|
|
|
|
local update = ya.sync(function(st, tags)
|
|
for path, tag in pairs(tags) do
|
|
st.tags[path] = #tag == 0 and nil or tag
|
|
end
|
|
ya.render()
|
|
end)
|
|
|
|
local selected_or_hovered = ya.sync(function()
|
|
local tab, urls = cx.active, {}
|
|
for _, u in pairs(tab.selected) do
|
|
urls[#urls + 1] = u
|
|
end
|
|
if #urls == 0 and tab.current.hovered then
|
|
urls[1] = tab.current.hovered.url
|
|
end
|
|
return urls
|
|
end)
|
|
|
|
local function setup(st)
|
|
st.tags = {}
|
|
|
|
Linemode:children_add(function(self)
|
|
local url = tostring(self._file.url)
|
|
local spans = {}
|
|
for _, tag in ipairs(st.tags[url] or {}) do
|
|
spans[#spans + 1] = ui.Span("⬤"):fg(COLORS[tag:lower()] or "reset")
|
|
end
|
|
return ui.Line(spans)
|
|
end, 500)
|
|
end
|
|
|
|
local function fetch(self)
|
|
local paths = {}
|
|
for _, file in ipairs(self.files) do
|
|
paths[#paths + 1] = tostring(file.url)
|
|
end
|
|
|
|
local output, err = Command("tag"):args(paths):stdout(Command.PIPED):output()
|
|
if not output then
|
|
return ya.err("Cannot spawn tag command, error code " .. tostring(err))
|
|
end
|
|
|
|
local i, tags = 1, {}
|
|
for line in output.stdout:gmatch("[^\r\n]+") do
|
|
if i > #paths then
|
|
break
|
|
end
|
|
tags[paths[i]] = tags[paths[i]] or {}
|
|
|
|
local joint = line:match("\t(.+)$") or ""
|
|
for s in joint:gmatch("[^,]+") do
|
|
table.insert(tags[paths[i]], s)
|
|
end
|
|
i = i + 1
|
|
end
|
|
|
|
update(tags)
|
|
return 1
|
|
end
|
|
|
|
local function entry(_, args)
|
|
assert(args[1] == "add" or args[1] == "remove", "Invalid action")
|
|
assert(args[2], "No tag specified")
|
|
|
|
local t = { args[1] == "remove" and "-r" or "-a", args[2] }
|
|
local files = {}
|
|
for _, url in ipairs(selected_or_hovered()) do
|
|
t[#t + 1] = tostring(url)
|
|
files[#files + 1] = { url = url }
|
|
end
|
|
|
|
local status = Command("tag"):args(t):status()
|
|
if status.success then
|
|
fetch { files = files }
|
|
end
|
|
end
|
|
|
|
return { setup = setup, fetch = fetch, entry = entry }
|