123 lines
2.8 KiB
Lua
123 lines
2.8 KiB
Lua
local PATS = {
|
|
{ "!$", "." }, -- Ignored
|
|
{ "?$", "?" }, -- Untracked
|
|
{ "U", "U" }, -- Updated
|
|
{ "[AD][AD]", "U" }, -- Updated
|
|
{ "[MT]", "M" }, -- Modified
|
|
{ "[AC]", "A" }, -- Added
|
|
}
|
|
|
|
local PRIOS = {
|
|
[""] = 1,
|
|
["."] = 2,
|
|
["?"] = 3,
|
|
["U"] = 4,
|
|
["M"] = 5,
|
|
["A"] = 6,
|
|
}
|
|
|
|
local function match(s)
|
|
for _, p in ipairs(PATS) do
|
|
if s:find(p[1]) then
|
|
return p[2]
|
|
end
|
|
end
|
|
end
|
|
|
|
local function root(cwd)
|
|
repeat
|
|
local cha = fs.cha(cwd:join(".git"))
|
|
if cha and cha.is_dir then
|
|
return string.format("%s%s", cwd, ya.target_family() == "windows" and "\\" or "/")
|
|
end
|
|
cwd = cwd:parent()
|
|
until not cwd
|
|
end
|
|
|
|
local save = ya.sync(function(st, states)
|
|
st.states = st.states or {}
|
|
for k, v in pairs(states) do
|
|
st.states[k] = v ~= "" and v or nil
|
|
end
|
|
ya.render()
|
|
end)
|
|
|
|
local function setup(st, opts)
|
|
st.states = {}
|
|
local styles = {
|
|
["."] = THEME.git_ignored and ui.Style(THEME.git_ignored) or ui.Style():fg("gray"),
|
|
["?"] = THEME.git_staged and ui.Style(THEME.git_staged) or ui.Style():fg("yellow"),
|
|
["U"] = THEME.git_untracked and ui.Style(THEME.git_untracked) or ui.Style():fg("blue"),
|
|
["M"] = THEME.git_modified and ui.Style(THEME.git_modified) or ui.Style():fg("red"),
|
|
["A"] = THEME.git_deleted and ui.Style(THEME.git_deleted) or ui.Style():fg("green"),
|
|
}
|
|
|
|
Linemode:children_add(function(self)
|
|
local state = st.states[tostring(self._file.url)]
|
|
if state then
|
|
return ui.Line { ui.Span(" "), ui.Span(state):style(styles[state]), ui.Span(" ") }
|
|
else
|
|
return ui.Line {}
|
|
end
|
|
end, 5000)
|
|
end
|
|
|
|
local function fetch(self)
|
|
local paths = {}
|
|
for _, file in ipairs(self.files) do
|
|
paths[#paths + 1] = tostring(file.url)
|
|
end
|
|
|
|
local cwd = self.files[1].url:parent()
|
|
local output, err = Command("git")
|
|
:cwd(tostring(cwd))
|
|
:args({ "-c", "core.quotePath=", "status", "--porcelain", "-unormal", "--no-renames", "--ignored=matching" })
|
|
:args(paths)
|
|
:stdout(Command.PIPED)
|
|
:output()
|
|
if not output then
|
|
ya.err("Cannot spawn git command, error code " .. tostring(err))
|
|
return 0
|
|
end
|
|
|
|
local prefix = root(cwd)
|
|
if not prefix then
|
|
return 1
|
|
end
|
|
|
|
local states = {}
|
|
for line in output.stdout:gmatch("[^\r\n]+") do
|
|
local s = match(line:sub(1, 2))
|
|
if s and line:find("[/\\]$") then
|
|
states[prefix .. line:sub(4, -2)] = s
|
|
else
|
|
states[prefix .. line:sub(4)] = s
|
|
end
|
|
end
|
|
|
|
prefix = Url(prefix)
|
|
if self.files[1].cha.is_dir then
|
|
local parents = {}
|
|
for k, v in pairs(states) do
|
|
local url = Url(k):parent()
|
|
while url and url ~= prefix do
|
|
local s = tostring(url)
|
|
parents[s] = (PRIOS[parents[s]] or 0) > PRIOS[v] and parents[s] or v
|
|
url = url:parent()
|
|
end
|
|
end
|
|
for k, v in pairs(parents) do
|
|
states[k] = v
|
|
end
|
|
end
|
|
|
|
for _, p in ipairs(paths) do
|
|
states[p] = states[p] or ""
|
|
end
|
|
save(states)
|
|
|
|
return 3
|
|
end
|
|
|
|
return { setup = setup, fetch = fetch }
|