From 9de863a1af3ad27afa44354403fac38e61a20cf1 Mon Sep 17 00:00:00 2001
From: Philip <philm@manjaro.org>
Date: Tue, 19 Aug 2014 15:54:02 +0200
Subject: [PATCH] New initcpiocfg module, ported from Thus.

---
 settings.conf                       |  1 +
 src/modules/initcpiocfg/main.py     | 92 +++++++++++++++++++++++++++++
 src/modules/initcpiocfg/module.desc |  5 ++
 3 files changed, 98 insertions(+)
 create mode 100644 src/modules/initcpiocfg/main.py
 create mode 100644 src/modules/initcpiocfg/module.desc

diff --git a/settings.conf b/settings.conf
index 7640314d6..eed8b7565 100644
--- a/settings.conf
+++ b/settings.conf
@@ -35,6 +35,7 @@ install:
 - fstab
 - locale
 - keyboard
+- initcpiocfg
 - initcpio
 - users
 - networkcfg
diff --git a/src/modules/initcpiocfg/main.py b/src/modules/initcpiocfg/main.py
new file mode 100644
index 000000000..5bd33a05f
--- /dev/null
+++ b/src/modules/initcpiocfg/main.py
@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+# encoding: utf-8
+# === This file is part of Calamares - <http://github.com/calamares> ===
+#
+#   Copyright 2014, Philip Müller <philm@manjaro.org>
+#
+#   Calamares 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.
+#
+#   Calamares 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 Calamares. If not, see <http://www.gnu.org/licenses/>.
+
+import libcalamares
+import os
+import subprocess
+from libcalamares.utils import check_chroot_call
+
+
+def get_cpu():
+    """ Check if system is an intel system. """
+
+    process1 = subprocess.Popen(["hwinfo", "--cpu"], stdout=subprocess.PIPE)
+    process2 = subprocess.Popen(["grep", "Model:[[:space:]]"],
+                                stdin=process1.stdout, stdout=subprocess.PIPE)
+    process1.stdout.close()
+    out, err = process2.communicate()
+    return out.decode().lower()
+
+
+def set_mkinitcpio_hooks_and_modules(hooks, modules, root_mount_point):
+    """ Set up mkinitcpio.conf """
+
+    with open("/etc/mkinitcpio.conf", "r") as mkinitcpio_file:
+        mklins = [x.strip() for x in mkinitcpio_file.readlines()]
+
+    for i in range(len(mklins)):
+        if mklins[i].startswith("HOOKS"):
+            mklins[i] = 'HOOKS="%s"' % ' '.join(hooks)
+        elif mklins[i].startswith("MODULES"):
+            mklins[i] = 'MODULES="%s"' % ' '.join(modules)
+
+    path = os.path.join(root_mount_point, "etc/mkinitcpio.conf")
+    with open(path, "w") as mkinitcpio_file:
+        mkinitcpio_file.write("\n".join(mklins) + "\n")
+
+def modify_mkinitcpio_conf(partitions, root_mount_point):
+    """ Modifies mkinitcpio.conf """
+
+    cpu = get_cpu()
+    swap_uuid = ""
+    btrfs = ""
+    hooks = ["base", "udev", "autodetect", "modconf", "block", "keyboard", "keymap"]
+    modules = []
+
+    # It is important that the plymouth hook comes before any encrypt hook
+    plymouth_bin = os.path.join(root_mount_point, "usr/bin/plymouth")
+    if os.path.exists(plymouth_bin):
+        hooks.append("plymouth")
+
+    for partition in partitions:
+        if partition["fs"] == "linuxswap":
+            swap_uuid = partition["uuid"]
+        if partition["fs"] == "btrfs":
+            btrfs = "yes"
+
+    if swap_uuid is not "":
+        hooks.extend(["resume", "filesystems"])
+    else:
+        hooks.extend(["filesystems"])
+
+    if btrfs is "yes" and cpu is not "genuineintel":
+        modules.append("crc32c")
+    elif btrfs is "yes" and cpu is "genuineintel":
+        modules.append("crc32c-intel")
+    else:
+        hooks.append("fsck")
+
+    set_mkinitcpio_hooks_and_modules(hooks, modules, root_mount_point)
+
+
+def run():
+    partitions = libcalamares.globalstorage.value("partitions")
+    root_mount_point = libcalamares.globalstorage.value("rootMountPoint")
+    modify_mkinitcpio_conf(partitions, root_mount_point)
+    return None
diff --git a/src/modules/initcpiocfg/module.desc b/src/modules/initcpiocfg/module.desc
new file mode 100644
index 000000000..936a99bbe
--- /dev/null
+++ b/src/modules/initcpiocfg/module.desc
@@ -0,0 +1,5 @@
+---
+type:       "job"
+name:       "initcpiocfg"
+interface:  "python"
+script:     "main.py"