Add docker command to aio.

1 files changed, 40 insertions, 8 deletions

diff --git a/tool/aio.py b/tool/aio.py
index 27404f1..debcd68 100755
--- a/tool/aio.py
+++ b/tool/aio.py
@@ -53,7 +53,7 @@ list_domain_parser = subparsers.add_parser(
     "list-domain", help="Misc things about domains.")
 
 nginx_parser = subparsers.add_parser(
-    "nginx", help="Generate nginx config for a domain.")
+    "nginx", help="Generate nginx config.")
 
 certbot_parser = subparsers.add_parser(
     "certbot", help="Get some common certbot commands.")
@@ -78,16 +78,23 @@ clear_parser.add_argument("-D", "--include-data-dir", action="store_true",
 install_docker_parser = subparsers.add_parser(
     "install-docker", help="Install docker and docker-compose.")
 
-backup_docker_parser = subparsers.add_parser(
+backup_parser = subparsers.add_parser(
     "backup", help="Backup related things."
 )
 
-backup_command_group = backup_docker_parser.add_mutually_exclusive_group()
+backup_command_group = backup_parser.add_mutually_exclusive_group()
 backup_command_group.add_argument(
     "-R", "--restore", action="append", nargs="?", default=None, help="Restore data from url.")
 backup_command_group.add_argument(
     "-B", "--backup", action="append", nargs="?", default=None, help="Backup data to specified path.")
 
+docker_parser = subparsers.add_parser("docker", help="Docker related things.")
+docker_subparsers = docker_parser.add_subparsers(dest="docker_action")
+docker_subparsers.add_parser("up", help="Run docker compose up -d.")
+docker_subparsers.add_parser("down", help="Run docker compose down.")
+docker_subparsers.add_parser(
+    "prune", help="Run docker system prune -a -f.")
+
 args = parser.parse_args()
 
 if args.yes:
@@ -126,11 +133,6 @@ if not args.no_check_system:
     if not check_ubuntu():
         console.print("This script works well on Ubuntu 22.04. Otherwise you may encounter some problems. But I would like to improve some rational compatibility.", style="yellow")
 
-
-def print_order(number: int, total: int, *, console=console) -> None:
-    console.print(f"\[{number}/{total}]", end=" ", style="green")
-
-
 if args.action == "certbot":
     if args.create or args.renew or args.expand:
         args.no_hello = True
@@ -138,6 +140,11 @@ if args.action == "certbot":
 if not args.no_hello:
     console.print("Nice to see you! :waving_hand:", style="cyan")
 
+
+def print_order(number: int, total: int, *, console=console) -> None:
+    console.print(f"\[{number}/{total}]", end=" ", style="green")
+
+
 if args.action == "install-docker":
     ensure_tmp_dir()
     get_docker_path = os.path.join(tmp_dir, "get-docker.sh")
@@ -152,6 +159,31 @@ if args.action == "install-docker":
         "Succeeded to install docker. Please re-login to take effect.", style="green")
     exit(0)
 
+
+if args.action == "docker":
+    def run_in_dir(dir: str, func: callable):
+        old_dir = os.path.abspath(os.getcwd())
+        os.chdir(dir)
+        func()
+        os.chdir(old_dir)
+    match args.docker_action:
+        case "up":
+            def docker_compose_up():
+                subprocess.run(["docker", "compose", "up", "-d"], check=True)
+            run_in_dir(project_abs_path, docker_compose_up)
+        case "down":
+            def docker_compose_down():
+                subprocess.run(["docker", "compose", "down"], check=True)
+            run_in_dir(project_abs_path, docker_compose_down)
+        case "prune":
+            to_do = Confirm.ask("[yellow]Are you sure to prune docker?[/]")
+            if to_do:
+                subprocess.run(
+                    ["docker", "system", "prune", "-a", "-f"], check=True)
+        case _:
+            raise ValueError("Unknown docker action.")
+    exit(0)
+
 if args.action == "backup":
     if not args.restore is None:
         if args.restore[0] is None: