From 158d6da41536f3eafc7970f44ab2ade7032f5b70 Mon Sep 17 00:00:00 2001
From: Achilleas Koutsou <achilleas@koutsou.net>
Date: Mon, 14 Jun 2021 17:41:20 +0200
Subject: [PATCH] stages: add org.osbuild.chmod

Add new stage to change the mode bits of files in the tree.
---
 stages/org.osbuild.chmod | 67 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100755 stages/org.osbuild.chmod

diff --git a/stages/org.osbuild.chmod b/stages/org.osbuild.chmod
new file mode 100755
index 00000000..72c529be
--- /dev/null
+++ b/stages/org.osbuild.chmod
@@ -0,0 +1,67 @@
+#!/usr/bin/python3
+"""
+Change file mode bits
+
+Change the file mode bits of one or more files or directories inside the tree.
+"""
+
+import os
+import subprocess
+import sys
+
+import osbuild.api
+from osbuild.util.path import in_tree
+
+SCHEMA = r"""
+"additionalProperties": false,
+"properties": {
+  "items": {
+    "type": "object",
+    "additionalProperties": false,
+    "patternProperties": {
+      "^\\/(?!\\.\\.)((?!\\/\\.\\.\\/).)+$": {
+        "type": "object",
+        "required": ["mode"],
+        "properties": {
+          "mode": {
+            "type": "string",
+            "description": "Symbolic or numeric octal mode"
+          },
+          "recursive": {
+            "type": "boolean",
+            "description": "Change modes recursively",
+            "default": false
+          }
+        }
+      }
+    }
+  }
+}
+"""
+
+
+def chmod(path: str, mode: str, recursive: bool):
+    arguments = [mode]
+    if recursive:
+        arguments.append("--recursive")
+    arguments.append("--")
+    arguments.append(path)
+
+    subprocess.run(["chmod", *arguments], check=True)
+
+
+def main(tree, options):
+    for path, cmdargs in options["items"].items():
+        mode = cmdargs["mode"]
+        recursive = cmdargs.get("recursive", False)
+        realpath = os.path.join(tree, path.lstrip("/"))
+        if not in_tree(realpath, tree, must_exist=True):
+            raise ValueError(f"path {path} not in tree")
+        chmod(realpath, mode, recursive)
+
+    return 0
+
+
+if __name__ == "__main__":
+    args = osbuild.api.arguments()
+    sys.exit(main(args["tree"], args["options"]))