Add support for isorec flashing in recovery installer (#609)

pmb/flasher/variables.py

@@ -0,0 +1,46 @@
+"""
+Copyright 2017 Oliver Smith
+
+This file is part of pmbootstrap.
+
+pmbootstrap is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+pmbootstrap is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with pmbootstrap.  If not, see <http://www.gnu.org/licenses/>.
+"""
+
+
+def variables(args, flavor, method):
+    _cmdline = args.deviceinfo["kernel_cmdline"]
+    if "cmdline" in args and args.cmdline:
+        _cmdline = args.cmdline
+
+    if method == "fastboot":
+        _partition_system = "system"
+    else:
+        _partition_system = args.deviceinfo["flash_heimdall_partition_system"] or "SYSTEM"
+    if "partition" in args and args.partition:
+        _partition_system = args.partition
+
+    vars = {
+        "$BOOT": "/mnt/rootfs_" + args.device + "/boot",
+        "$FLAVOR": flavor if flavor is not None else "",
+        "$IMAGE": "/home/user/rootfs/" + args.device + ".img",
+        "$KERNEL_CMDLINE": _cmdline,
+        "$PARTITION_KERNEL": args.deviceinfo["flash_heimdall_partition_kernel"] or "KERNEL",
+        "$PARTITION_INITFS": args.deviceinfo["flash_heimdall_partition_initfs"] or "RECOVERY",
+        "$PARTITION_SYSTEM": _partition_system,
+        "$RECOVERY_ZIP": "/mnt/buildroot_" + args.deviceinfo["arch"] +
+                         "/var/lib/postmarketos-android-recovery-installer"
+                         "/pmos-" + args.device + ".zip",
+    }
+
+    return vars