update

cmd/main/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"fmt"
 	"kube-forge/internal/config"
 	"kube-forge/internal/kubespray"
 	"kube-forge/internal/templates"
@@ -32,12 +33,17 @@ func main() {
 		case "apply":
 			kubespray.InstallCluster("")
 			kubespray.CopyK8SAdminConfig("k8s-admin.conf")
+			return
 		case "apply-modules":
 			kubespray.InstallCluster("helm-apps")
+			return
 		case "upgrade":
 			kubespray.UpgradeCluster("")
+			return
 		case "scale":
 			kubespray.ScaleCluster()
+			return
 		}
 	}
+	fmt.Println("No such command\nAvailable commands: apply, apply-modules, upgrade, scale")
 }