Generating platform specific installers

Author: Heshdude

generate.py

@@ -7,14 +7,15 @@
 import re
 
 methods = {
-   "curl": "$CURL",
-   "apt": "$APT_GET",
-   "yum": "$YUM",
-   "dnf": "$DNF",
-   "apk": "$APK",
-   "pacman": "$PACMAN",
-   "git": "$GIT"
+    "curl": "$CURL",
+    "apt": "$APT_GET",
+    "yum": "$YUM",
+    "dnf": "$DNF",
+    "apk": "$APK",
+    "pacman": "$PACMAN",
+    "git": "$GIT"
 }
+
 def update_readme(summary):
     writer = pytablewriter.MarkdownTableWriter()
     writer.headers = ["Tool", "Apt", "Yum", "Packman", "APK", "DNF", "CURL", "URL"]
@@ -29,7 +30,7 @@ def update_readme(summary):
         apk = "Yes" if "apk" in installers else "No"
         dnf = "Yes" if "dnf" in installers else "No"
         curl = "Yes" if "curl" in installers else "No"
-        url = "https://installer.to/"+tool_shortname
+        url = "https://installer.to/" + tool_shortname
         value_matrix.append([name, apt, yum, pacman, apk, dnf, curl, url])
 
     writer.value_matrix = value_matrix
@@ -39,14 +40,14 @@ def update_readme(summary):
             readme = readme_md.read()
             beggining = "<!-- beginning of tools list -->"
             end = "<!-- end of tools list -->"
-            regex = r""+beggining+"\n(.*)\n"+end
-            readme = re.sub(regex, beggining+"\n"+table_md+"\n"+end, readme, flags=re.S)
+            regex = r"" + beggining + "\n(.*)\n" + end
+            readme = re.sub(regex, beggining + "\n" + table_md + "\n" + end, readme, flags=re.S)
             readme_md.seek(0)  # sets  point at the beginning of the file
             readme_md.truncate()  # Clear previous content
             readme_md.write(readme)
             readme_md.close()
     except Error as e:
-        logging.error('Error occurred when trying to update README.md, error: '+ e)
+        logging.error('Error occurred when trying to update README.md, error: ' + e)
 
 
 def update_summary(name, shortname, description, installers):
@@ -67,43 +68,40 @@ def update_summary(name, shortname, description, installers):
 
             update_readme(parsed_summary_toml)
     except IOError as e:
-        logging.error('Error occurred when trying to update installers.toml, error: '+ e)
+        logging.error('Error occurred when trying to update installers.toml, error: ' + e)
 
 def get_method_case(method):
-   if method in methods:
-      return "[ ! -z "+methods[method]+"_CMD ]; then\n"
-   else:
-      logging.error('Unpupported method in the TOML file, method: '+method)
-      exit(1)
+    if method in methods:
+        return "[ ! -z " + methods[method] + "_CMD ]; then\n"
+    else:
+        logging.error('Unpupported method in the TOML file, method: ' + method)
+        exit(1)
+
 
 def parse_line(line):
-    line = line\
-        .replace('@sudo', '$SUDO')\
-        .replace('@log', 'info')\
-        .replace('@info', 'info')\
-        .replace('@warn', 'warn')\
+    line = line \
+        .replace('@sudo', '$SUDO') \
+        .replace('@log', 'info') \
+        .replace('@info', 'info') \
+        .replace('@warn', 'warn') \
         .replace('@error', 'error')
     return line
 
-def generate(path):
-   installer_methods = [ ]
-   installer_toml_path = path+"/installer.toml"
-   installer_sh_path = path+"/installer.sh"
 
-   installer_toml = open(installer_toml_path, "r")
-   parsed_toml = toml.loads(installer_toml.read())
-   try:
-    with open(installer_sh_path, "w") as installer_sh:
-
-      installer_sh.write("""#!/bin/sh
-      
+def write_sniffer_commands(sh_file):
+    sh_file.write("""
 CURL_CMD=$(which curl) 
 YUM_CMD=$(which yum) 
 DNF_CMD=$(which dnf) 
 APT_GET_CMD=$(which apt-get) 
 PACMAN_CMD=$(which pacman) 
 APK_CMD=$(which apk) 
 GIT_CMD=$(which git) 
+""")
+
+
+def write_sudo_fix_commands(sh_file):
+    sh_file.write("""
 SUDO_CMD=$(which sudo) 
 
 USER="$(id -un 2>/dev/null || true)"
@@ -119,7 +117,11 @@ def generate(path):
 		exit 1
 	fi
 fi
+""")
+
 
+def write_logger_commands(sh_file):
+    sh_file.write("""
 RESET='\033[0m'
 RED='\033[0;31m'
 GREEN='\033[0;32m'
@@ -139,42 +141,80 @@ def generate(path):
 
 """)
 
-      seperator = "if"
-
-      for section in parsed_toml:
-         if not isinstance(parsed_toml[section], dict):
-            continue
-         if parsed_toml[section]['sh'] is  "":
-            continue
-         installer_methods.append(section)
-         lines = parsed_toml[section]['sh']
-         installer_sh.write(seperator+" "+get_method_case(section))
-         for line in lines.split("\n"):
-             step = parse_line(line)
-             installer_sh.write("   "+step+"\n")
-         seperator = "elif"
-
-      installer_sh.write("""
+
+def write_installer_commands(sh_file, lines, indent=""):
+    for line in lines.split("\n"):
+        step = parse_line(line)
+        sh_file.write(indent + step + "\n")
+
+def generate_individual_installers(method, lines):
+    installer_sh_path = path + "/installer."+method+".sh"
+    try:
+        with open(installer_sh_path, "w") as installer_sh:
+            write_sudo_fix_commands(installer_sh)
+            write_logger_commands(installer_sh)
+            write_installer_commands(installer_sh, lines)
+
+    except IOError as x:
+        if x.errno == errno.EACCES:
+            logging.error('No enough permissions to write to ' + installer_sh_path)
+            exit(1)
+        else:
+            logging.error('Something went wrong when trying to write to ' + installer_sh_path, x)
+            exit(1)
+
+def generate(path):
+    installer_methods = []
+    installer_toml_path = path + "/installer.toml"
+    installer_sh_path = path + "/installer.sh"
+
+    installer_toml = open(installer_toml_path, "r")
+    parsed_toml = toml.loads(installer_toml.read())
+    try:
+        with open(installer_sh_path, "w") as installer_sh:
+
+            installer_sh.write("""#!/bin/sh
+      
+""")
+            write_sniffer_commands(installer_sh)
+            write_sudo_fix_commands(installer_sh)
+            write_logger_commands(installer_sh)
+
+            seperator = "if"
+
+            for section in parsed_toml:
+                if not isinstance(parsed_toml[section], dict):
+                    continue
+                if parsed_toml[section]['sh'] is "":
+                    continue
+                installer_methods.append(section)
+                lines = parsed_toml[section]['sh']
+                installer_sh.write(seperator + " " + get_method_case(section))
+                write_installer_commands(installer_sh, lines, "   ")
+                generate_individual_installers(section, lines)
+                seperator = "elif"
+
+            installer_sh.write("""
 else
    echo "Couldn't install package"
    exit 1;
 fi
-      """.strip())
+              """.strip())
 
-      installer_sh.close()
-      update_summary(parsed_toml['name'], parsed_toml['shortname'], parsed_toml['description'], installer_methods)
+        installer_sh.close()
+        update_summary(parsed_toml['name'], parsed_toml['shortname'], parsed_toml['description'], installer_methods)
 
-   except IOError as x:
-      if x.errno == errno.EACCES:
-         logging.error('No enough permissions to write to '+installer_sh_path)
-         exit(1)
-      else:
-         logging.error('Something went wrong when trying to write to '+installer_sh_path)
-         exit(1)
+    except IOError as x:
+        if x.errno == errno.EACCES:
+            logging.error('No enough permissions to write to ' + installer_sh_path)
+            exit(1)
+        else:
+            logging.error('Something went wrong when trying to write to ' + installer_sh_path, x)
+            exit(1)
 
 for path in sys.argv[1:]:
-   if os.path.exists(path+'/installer.toml'):
-         logging.info('Generating installer.sh for '+path)
-         generate(path)
-   else:
-         logging.warn('Could not find an installer.toml in '+path)
+    if os.path.exists(path + '/installer.toml'):
+        logging.info('Generating installer.sh for ' + path)
+        generate(path)
+    else:
+        logging.warn('Could not find an installer.toml in ' + path)