wiki: Add generate_variants script

Change-Id: If0ff99fc12d936326cc304404757d62094d28e6d

scripts/generate_variants.py

@@ -0,0 +1,130 @@
+import os
+import re
+import shutil
+import sys
+
+from pathlib import Path
+
+WIKI_PATH = str(Path(__file__).parents[1])
+DEVICE_PATH = WIKI_PATH + "/_data/devices"
+INFO_PATH = WIKI_PATH + "/pages/info"
+INSTALL_PATH = WIKI_PATH + "/pages/install"
+BUILD_PATH = WIKI_PATH + "/pages/build"
+UPDATE_PATH = WIKI_PATH + "/pages/update"
+UPGRADE_PATH = WIKI_PATH + "/pages/upgrade"
+FW_UPDATE_PATH = WIKI_PATH + "/pages/fw_update"
+
+
+def main(args):
+    if len(args) != 2:
+        device = input("Which device? ")
+        amount = input("How many? ")
+    else:
+        device = args[0]
+        amount = args[1]
+
+    try:
+        amount = int(amount)
+    except (TypeError, ValueError):
+        print("Could not convert to number!")
+        exit(-1)
+
+    orig_yaml = f"{DEVICE_PATH}/{device}.yml"
+    orig_info = f"{INFO_PATH}/{device}.md"
+    orig_install = f"{INSTALL_PATH}/{device}.md"
+    orig_build = f"{BUILD_PATH}/{device}.md"
+    orig_update = f"{UPDATE_PATH}/{device}.md"
+    orig_upgrade = f"{UPGRADE_PATH}/{device}.md"
+    orig_fw = f"{FW_UPDATE_PATH}/{device}.md"
+
+    if not os.path.isfile(orig_yaml):
+        print("Could not find yaml!")
+        exit(-1)
+
+    for i in range(amount):
+        new_device = f"{device}_variant{i+1}"
+
+        # Copy and modify yaml
+        new_yaml = f"{DEVICE_PATH}/{new_device}.yml"
+        shutil.copy2(orig_yaml, new_yaml)
+        with open(new_yaml, "r+") as f:
+            lines = f.readlines()
+            f.seek(0)
+            for line in lines:
+                if line.startswith("vendor:"):
+                    f.writelines(f"variant: {i+1}\n")
+                f.writelines(line)
+            f.truncate()
+            f.close()
+
+        # Info page
+        new_info = f"{INFO_PATH}/{new_device}.md"
+        replace_pages(orig_info, new_info, device, new_device, i + 1)
+
+        # Install page
+        new_install = f"{INSTALL_PATH}/{new_device}.md"
+        replace_pages(orig_install, new_install, device, new_device, i + 1, "install")
+
+        # Build page
+        new_build = f"{BUILD_PATH}/{new_device}.md"
+        replace_pages(orig_build, new_build, device, new_device, i + 1, "build")
+
+        # Update page
+        new_update = f"{UPDATE_PATH}/{new_device}.md"
+        replace_pages(orig_update, new_update, device, new_device, i + 1, "update")
+
+        # Upgrade page
+        new_upgrade = f"{UPGRADE_PATH}/{new_device}.md"
+        replace_pages(orig_upgrade, new_upgrade, device, new_device, i + 1, "upgrade")
+
+        if os.path.isfile(orig_fw):
+            new_fw = f"{FW_UPDATE_PATH}/{new_device}.md"
+            replace_pages(orig_fw, new_fw, device, new_device, i + 1, "fw_update")
+
+    # Original info and upgrade pages
+    make_common_page("templates/variant_info.md", orig_info, device)
+    make_common_page("templates/variant_upgrade.md", orig_upgrade, device)
+
+    os.remove(orig_install)
+    os.remove(orig_build)
+    os.remove(orig_update)
+    os.remove(orig_yaml)
+    if os.path.isfile(orig_fw):
+        os.remove(orig_fw)
+
+
+def replace_pages(orig, new, device, new_device, variant_nr, target=None):
+    shutil.copy2(orig, new)
+    with open(new, "r+") as f:
+        content = f.read()
+        if target:
+            content = re.sub(
+                r"permalink: (.*)",
+                f"permalink: /devices/{device}/{target}/variant{variant_nr}",
+                content,
+            )
+        else:
+            content = re.sub(
+                r"permalink: (.*)",
+                f"permalink: /devices/{device}/variant{variant_nr}/",
+                content,
+            )
+        content = re.sub(r"device: (.*)", f"device: {new_device}", content)
+        f.seek(0)
+        f.write(content)
+        f.close()
+
+
+def make_common_page(template, path, device):
+    shutil.copy2(template, path)
+    with open(path, "r+") as f:
+        content = f.read()
+        content = content.replace("{codename}", device)
+        f.seek(0)
+        f.truncate()
+        f.write(content)
+        f.close()
+
+
+if __name__ == "__main__":
+    main(sys.argv[1:])

scripts/templates/variant_info.md

@@ -0,0 +1,12 @@
+---
+sidebar: home_sidebar
+title: Info about {codename} variants
+folder: info
+permalink: /devices/{codename}/
+redirect_from:
+- /devices/{codename}/install
+- /devices/{codename}/update
+device: {codename}
+toc: false
+---
+{% include templates/device_variants.md info=true %}

scripts/templates/variant_upgrade.md

@@ -0,0 +1,9 @@
+---
+sidebar: home_sidebar
+title: Upgrade LineageOS on {codename}
+folder: upgrade
+permalink: /devices/{codename}/upgrade
+device: {codename}
+toc: false
+---
+{% include templates/device_variants.md %}