Refactor vmm variables

1 files changed, 115 insertions, 0 deletions

diff --git a/roles/vmm/files/init_vm_alpine.py b/roles/vmm/files/init_vm_alpine.py
new file mode 100644
index 0000000..d9dc0d5
--- /dev/null
+++ b/roles/vmm/files/init_vm_alpine.py
@@ -0,0 +1,115 @@
+#!/bin/python3
+
+import serial
+import subprocess
+import sys
+import os
+
+from serial_macro import *
+
+USAGE = f"USAGE: {sys.argv[0]} vm_guest gate ip mask ssh_key"
+
+def init_network():
+    send_cmds(ser, [
+        [1, "setup-interfaces"],
+        [1, ""],
+        [1, f"{IP}"],
+        [1, f"{MASK}"],
+        [1, f"{GATE}"],
+        [1, "no"],
+        [1, "ifdown -a"],
+        [10, "ifup -a"],
+        [1, "rc-update add networking"]
+    ])
+
+
+def init_dns():
+    send_cmds(ser, [
+        [1, f"setup-dns"],
+        [1, ""],
+        [1, f"{DNS}"],
+    ])
+
+
+def init_disk():
+    send_cmds(ser, [
+        [10, "apk add e2fsprogs sfdisk syslinux"],
+        [1, "setup-disk"],
+        [1, ""],
+        [10, "sys"],
+        [30, "y"],
+    ])
+
+
+def init_ssh():
+    send_cmds(ser, [
+        [5, "apk add openssh"],
+        [1, "mkdir /root/.ssh"],
+        [1, f"echo '{SSHKEY}' > /root/.ssh/authorized_keys"],
+        [1, f"echo 'PermitRootLogin yes' >> /etc/ssh/sshd_config"],
+        [1, "/etc/init.d/sshd restart"]
+    ])
+
+def init_packages():
+    send_cmds(ser, [
+        [1, "echo https://mirror.ungleich.ch/mirror/packages/alpine/latest-stable/main/ > /etc/apk/repositories "],
+        [5, "apk update"]
+    ])
+
+
+def main():
+
+    global ser
+    global IP
+    global GATE
+    global MASK
+    global DNS
+    global SSHKEY
+
+    COM = "/dev/"
+    BAUD = 115200
+    TIMEOUT = 1
+
+    if len(sys.argv) != 7:
+        sys.stderr.write(USAGE)
+        sys.exit(1)
+
+    GUEST = "vm-tmp"
+    HOST = sys.argv[1]
+
+    cmd = f"vmctl show | grep {GUEST} | tr -s ' ' | cut -d ' ' -f7"
+    _buffer = subprocess.check_output(cmd, shell=True).decode().rstrip()
+    print(_buffer)
+
+    if _buffer == "":
+        sys.exit(1)
+    COM += _buffer
+
+    IP = sys.argv[2]
+    GATE = sys.argv[3]
+    MASK = sys.argv[4]
+    DNS = sys.argv[5]
+    SSHKEY = sys.argv[6]
+
+    ser = serial.Serial(COM, BAUD, timeout=TIMEOUT)
+    send_cmd(ser, 1, "root")
+    # first boot :: live
+    init_network()
+    init_dns()
+    init_packages()
+    init_disk()
+    send_cmd(ser, 70, "reboot")
+    ser.close()
+
+    ser = serial.Serial(COM, BAUD, timeout=TIMEOUT)
+    send_cmd(ser, 1, "root")
+    # second boot :: disk
+    init_network()
+    init_dns()
+    init_ssh()
+    init_packages()
+    ser.close()
+
+
+if __name__ == "__main__":
+    main()