logic

create_site.py

@@ -1,19 +1,28 @@
-#!/bin/python3
+#!/usr/bin/python3
 
 import argparse
 import subprocess
 import util.color_io as color_io
+import sys
 
-parser = argparse.ArgumentParser(prog='Site Generator', description='Apache/Nginx Site Generator', epilog='Currently Only For Nginx')
+parser = argparse.ArgumentParser(prog='Site Generator', description='Apache/Nginx Site Generator',
+                                 epilog='Currently Only For Nginx')
 
 parse_type_group = parser.add_mutually_exclusive_group()
 
 parse_type_group.add_argument('-p', '--proxy', action='store_true', help="Generate a site which will act as a proxy")
-parse_type_group.add_argument('-r', '--root', action='store_true', help="Generate a site which serves from a document root")
+parse_type_group.add_argument('-r', '--root', action='store_true',
+                              help="Generate a site which serves from a document root")
 
-parser.add_argument('-i', '--install', default="/etc/nginx/sites-available/", help="Set the install location for the generated config, defaults to /etc/nginx/sites-available/")
-parser.add_argument('-l', '--log', default="/var/log/nginx/", help="Set the default log location for the generated config, defaults to /var/log/nginx/")
-parser.add_argument('-s', '--symb', default="/etc/nginx/sites-enabled/", help="Set the location for the symbolic link to enable the config")
+parser.add_argument('-i', '--install', default=None, const="/etc/nginx/sites-available/",
+                    help="Set the install location for the generated config, defaults to /etc/nginx/sites-available/",
+                    nargs="?")
+parser.add_argument('-l', '--log', default="/var/log/nginx/",
+                    help="Set the default log location for the generated config, defaults to /var/log/nginx/")
+parser.add_argument('-s', '--symb', default=None, const="/etc/nginx/sites-enabled/", nargs="?",
+                    help="Set the location for the symbolic link to enable the config")
+parser.add_argument("--no-ssl", action="store_const", default="ssl", const="", help="Disable SSL", dest="ssl")
+parser.add_argument("--http", default="", required=False, help="Provide HTTP version")
 
 args = parser.parse_args()
 
@@ -21,73 +30,96 @@ install_location = args.install
 link_location = args.symb
 log_location = args.log
 
-if not install_location.endswith('/'):
-	install_location = install_location + "/"
-if not log_location.endswith('/'):
-	log_location = log_location + "/"
+if install_location is None:
+    print("Hint: you can provide install location or use the default (nginx dir) via command-line args")
+    install_location = color_io.input_print("Enter install location (default: stdout):", "stdout")
+    if install_location.lower() == "stdout":
+        install_location = None
+
+if install_location is not None and not install_location.endswith('/'):
+    install_location = install_location + "/"
+if not log_location.endswith('/'):
+    log_location = log_location + "/"
+
+site = {
+    'name': color_io.input_print("Enter site address (eg: 'tpgc.me')")
+}
 
-site = {}
-site['name'] = color_io.input_print("Enter site address (eg: 'tpgc.me')")
 
 def get_config_name():
-	return site['name']
+    return site['name']
+
 
 def get_config_path():
-	return install_location + get_config_name()
+    return install_location + get_config_name()
+
 
 def create_file():
-	return open(get_config_path(), 'a')
+    if install_location is None:
+        return sys.stdout
+    return open(get_config_path(), 'a')
+
 
 def create_nginx():
-	f = create_file()
-	f.write("#--- --- --- {Begin Site " + site['name'] + "} --- --- ---#\n")
-	f.write("server {\n")
-	f.write("	listen 443 ssl http2;\n\n")
-	f.write("	server_name " + site['name'] + ";\n\n")
-	f.write("	access_log " + log_location + site['name'] + "-access.log;\n")
-	f.write("	error_log " + log_location + site['name'] + "-error.log error;\n\n")
-	return f
+    f = create_file()
+    f.write("#--- --- --- {Begin Site " + site['name'] + "} --- --- ---#\n")
+    f.write("server {\n")
+    port = "443"
+    if len(args.ssl) == 0:
+        port = "80"
+    f.write("   listen " + port + " " + args.ssl + " " + args.http + ";\n\n")
+    f.write("	server_name " + site['name'] + ";\n\n")
+    f.write("	access_log " + log_location + site['name'] + "-access.log;\n")
+    f.write("	error_log " + log_location + site['name'] + "-error.log error;\n\n")
+    return f
+
 
 def end_nginx(f):
-	f.write("}\n")
-	f.write("#--- --- --- {End Site " + site['name'] + "} --- --- ---#\n\n")
-	f.close()
-	subprocess.run(["ln", "-s", get_config_path(), link_location + get_config_name()])
+    f.write("}\n")
+    f.write("#--- --- --- {End Site " + site['name'] + "} --- --- ---#\n\n")
+    if f is not sys.stdout:
+        f.close()
+    if link_location is not None:
+        subprocess.run(["ln", "-s", get_config_path(), link_location + get_config_name()])
+
 
 def write_location(f, loc, mod):
-	f.write("	location " + loc + " {\n")
-	for m in mod:
-		f.write("		" + m + ";\n")
-	f.write("	}\n\n")
+    f.write("	location " + loc + " {\n")
+    for m in mod:
+        f.write("		" + m + ";\n")
+    f.write("	}\n\n")
+
 
 def create_proxy():
-	color_io.green_print("Creating proxy site")
-	site['location'] = color_io.input_print("Enter site location (default '/')", "/")
-	site['address'] = color_io.input_print("Enter address to proxy to (default: http://127.0.0.1)", "http://127.0.0.1")
-	site['port'] = color_io.input_print("Enter port")
-	f = create_nginx()
-	write_location(f, site['location'], ["proxy_pass " + site['address'] + ":" + site['port'],
-				"proxy_set_header Upgrade $http_upgrade",
-				'proxy_set_header Connection "Upgrade"',
-				"proxy_set_header Host $host"])
-	end_nginx(f)
+    color_io.green_print("Creating proxy site")
+    site['location'] = color_io.input_print("Enter site location (default: /)", "/")
+    site['address'] = color_io.input_print("Enter address to proxy to (eg: http://127.0.0.1:)")
+    f = create_nginx()
+    write_location(f, site['location'], ["proxy_pass " + site['address'],
+                                         "proxy_set_header Upgrade $http_upgrade",
+                                         'proxy_set_header Connection "Upgrade"',
+                                         "proxy_set_header Host $host"])
+    end_nginx(f)
+
 
 def create_root():
-	color_io.green_print("Creating root site")
-	site['path'] = color_io.input_print("Enter site files path")
-	f = create_nginx()
-	f.write("	root " + site['path'] + ";\n\n");
-	f.write("	include php-fpm;\n\n");
-	end_nginx(f)
+    color_io.green_print("Creating root site")
+    site['path'] = color_io.input_print("Enter site files path")
+    f = create_nginx()
+    f.write("	root " + site['path'] + ";\n\n")
+    f.write("	include php-fpm;\n\n")
+    end_nginx(f)
+
 
 # Process user input based on args provided
 def execute_input():
-	if args.proxy:
-		create_proxy()
-	elif args.root:
-		create_root()
-	else:
-		create_proxy()
+    if args.proxy:
+        create_proxy()
+    elif args.root:
+        create_root()
+    else:
+        create_proxy()
+
 
 if __name__ == "__main__":
-	execute_input()
+    execute_input()