Add auto-certbot.

1 files changed, 27 insertions, 13 deletions

diff --git a/tool/aio.py b/tool/aio.py
index 94e4460..cdb616c 100755
--- a/tool/aio.py
+++ b/tool/aio.py
@@ -408,21 +408,35 @@ if to_gen_nginx_conf:
     domain = config["CRUPEST_DOMAIN"]
     generate_nginx_config(domain)
 
-
-if not os.path.exists(os.path.join(data_dir, "code-server")):
-    os.mkdir(os.path.join(data_dir, "code-server"))
+if not os.path.exists(data_dir):
     console.print(
-        "I also create data dir for code-server. Because letting docker create it would result in permission problem.", style="green")
-else:
-    code_server_stat = os.stat(os.path.join(data_dir, "code-server"))
-    if code_server_stat.st_uid == 0 or code_server_stat.st_gid == 0:
+        "Looks like you haven't generated data dir. I'll create it for you.", style="green")
+    os.mkdir(data_dir)
+elif not os.path.isdir(data_dir):
+    console.print(
+        "ERROR: data dir is not a dir! Everything will be broken! Please delete it manually", style="red")
+
+if os.path.isdir(data_dir):
+    if not os.path.exists(os.path.join(data_dir, "certbot")):
+        console.print(
+            "Looks like you haven't run certbot to get the init ssl certificates. You may want to run following code to get one:", style="cyan")
+        console.print(certbot_command_gen(domain, "create"),
+                      soft_wrap=True, highlight=False)
+
+    if not os.path.exists(os.path.join(data_dir, "code-server")):
+        os.mkdir(os.path.join(data_dir, "code-server"))
         console.print(
-            "WARNING: The owner of data dir for code-server is root. This may cause permission problem. You had better change it.", style="yellow")
-        to_fix = Confirm.ask(
-            "Do you want me to help you fix it?", console=console, default=True)
-        if to_fix:
-            os.system(
-                f"sudo chown -R {os.getuid()}:{os.getgid()} {os.path.join(data_dir, 'code-server')}")
+            "I also create data dir for code-server. Because letting docker create it would result in permission problem.", style="green")
+    else:
+        code_server_stat = os.stat(os.path.join(data_dir, "code-server"))
+        if code_server_stat.st_uid == 0 or code_server_stat.st_gid == 0:
+            console.print(
+                "WARNING: The owner of data dir for code-server is root. This may cause permission problem. You had better change it.", style="yellow")
+            to_fix = Confirm.ask(
+                "Do you want me to help you fix it?", console=console, default=True)
+            if to_fix:
+                os.system(
+                    f"sudo chown -R {os.getuid()}:{os.getgid()} {os.path.join(data_dir, 'code-server')}")
 
 console.print(":beers: All done!", style="green")
 to_download_tools = Confirm.ask(