From 07e844d0f6a104e68c36a914dbe5e3d2034aea1e Mon Sep 17 00:00:00 2001
From: Late Night Defender <pongpeera054@protonmail.com>
Date: Tue, 4 Jun 2024 09:10:46 +0700
Subject: [PATCH] Change behaviour of 1-arg and 3-arg commands

---
 main.go | 32 ++++++++++++++++++++------------
 1 file changed, 20 insertions(+), 12 deletions(-)

diff --git a/main.go b/main.go
index 6d94b0a..ab59cd5 100644
--- a/main.go
+++ b/main.go
@@ -68,7 +68,6 @@ func ExecSystem(command string) {
 }
 
 
-
 func main() {
 
 	flag.Parse()
@@ -80,12 +79,21 @@ func main() {
         os.Exit(1)
     }
 
+	// ./compose-runner should run docker compose up -d in current working directory
 	if len(os.Args) == 1 {
-		fmt.Println("Usage: compose-runner [-v] [path]")
-        os.Exit(2)
+		workDir, err := os.Getwd()
+
+		if err != nil {
+			panic(err)
+		}
+		os.Chdir(workDir)
+		fmt.Println("Running docker compose up -d in", workDir)
+		ExecSystem("docker compose up -d")
 	}
 
 	if len(os.Args) == 2 {
+
+		// ./compose-runner -v should run docker compose up -d in current working directory, verbose output
 		if *verbosePtr {
 			workDir, err := os.Getwd()
 
@@ -94,25 +102,25 @@ func main() {
 			}
 
 			os.Chdir(workDir)
+			fmt.Println("Running docker compose up -d in", workDir)
 			ExecSystem("docker compose up -d")
 		} else {
+			// ./compose-runner /path should run docker compose up -d in /path
+
 			workDir := os.Args[1]
 
 			os.Chdir(workDir)
+			fmt.Println("Running docker compose up -d in", workDir)
 			ExecSystem("docker compose up -d")
 		}
 	}
 
+	// ./compose-runner -v /path should run docker compose up -d in /path, verbose output
 	if len(os.Args) == 3 {
-		if os.Args[1] == "-v" {
-			workDir := os.Args[2]
-			os.Chdir(workDir)
-			ExecSystem("docker compose up -d")
-		} else {
-			workDir := os.Args[1]
-			os.Chdir(workDir)
-			ExecSystem("docker compose up -d")
-		}
+		workDir := os.Args[2]
+		os.Chdir(workDir)
+		fmt.Println("Running docker compose up -d in", workDir)
+		ExecSystem("docker compose up -d")
 	}
 
 }
\ No newline at end of file