63 lines
2.8 KiB
Python
63 lines
2.8 KiB
Python
"""Give adam ownership of the active node's tracked checkout and Git metadata."""
|
|
import inspect
|
|
import json
|
|
from pathlib import Path
|
|
import re
|
|
from engine.descriptor import Meta
|
|
|
|
METADATA = Meta(summary="Set ownership of tracked files and Git metadata under /srv to adam.")
|
|
|
|
|
|
def _repair(root, tracked, mounts, uid, gid):
|
|
import os
|
|
import stat
|
|
|
|
directories = {""}
|
|
for item in tracked:
|
|
directory = os.path.dirname(item)
|
|
while directory and directory not in directories:
|
|
directories.add(directory)
|
|
directory = os.path.dirname(directory)
|
|
metadata = os.path.join(root, ".git")
|
|
|
|
def fail(error):
|
|
raise error
|
|
|
|
for parent, dirs, files, fd in os.fwalk(root, onerror=fail, follow_symlinks=False):
|
|
os.fchown(fd, uid, gid)
|
|
for name in dirs.copy() + files:
|
|
path = os.path.join(parent, name)
|
|
relative = os.path.relpath(path, root)
|
|
if path in mounts or not (relative in tracked or relative in directories or
|
|
path == metadata or path.startswith(metadata + "/")):
|
|
if name in dirs:
|
|
dirs.remove(name)
|
|
continue
|
|
info = os.stat(name, dir_fd=fd, follow_symlinks=False)
|
|
if stat.S_ISDIR(info.st_mode):
|
|
continue
|
|
if (stat.S_ISREG(info.st_mode) or stat.S_ISLNK(info.st_mode)) and info.st_nlink == 1:
|
|
os.chown(name, uid, gid, dir_fd=fd, follow_symlinks=False)
|
|
|
|
|
|
def run(ctx, args):
|
|
root = Path(ctx.paths.repo_root)
|
|
if (root.parent != Path("/srv") or root != root.resolve(strict=True) or
|
|
not (root / "node.env").is_file() or (root / ".git").is_symlink() or
|
|
not (root / ".git").is_dir()):
|
|
raise RuntimeError("Expected the current node checkout directly under /srv.")
|
|
git = ["git", "-c", f"safe.directory={root}", "-c", f"safe.directory={root}/commons",
|
|
"-C", str(root)]
|
|
entries = ctx.sys.run([*git, "ls-files", "--cached", "--recurse-submodules", "-z"],
|
|
capture=True).stdout
|
|
gitfiles = ctx.sys.run([*git, "submodule", "foreach", "--quiet", "--recursive",
|
|
'printf "%s\\0" "$displaypath/.git"'], capture=True).stdout
|
|
mounts = [re.sub(r"\\([0-7]{3})", lambda match: chr(int(match[1], 8)), line.split()[4])
|
|
for line in Path("/proc/self/mountinfo").read_text().splitlines()]
|
|
script = (inspect.getsource(_repair) + "\nimport json, pwd, sys\n"
|
|
"root, entries, mounts = json.load(sys.stdin)\n"
|
|
"adam = pwd.getpwnam('adam')\n"
|
|
"_repair(root, set(entries), set(mounts), adam.pw_uid, adam.pw_gid)")
|
|
ctx.sys.sudo(["python3", "-I", "-c", script],
|
|
input=json.dumps([str(root), (entries + gitfiles).split("\0"), mounts]))
|