M main.go → main.go

@@ -22,15 +22,28 @@ // TODO: make arg parsing less shit
 
 	if len(args) <= 1 {
 		fmt.Println(helpStr)
+        return
 	}
 
 	switch args[1] {
 	case "init":
+        if len(args) <= 2 {
+            fmt.Println(helpStr)
+            return
+        }
 		initPath := args[2]
 		viteInit(initPath)
 	case "build":
+        _, err := os.Stat("config.yaml") 
+        if err != nil {
+            return
+        }
 		viteBuild()
 	case "new":
+        if len(args) <= 2 {
+            fmt.Println(helpStr)
+            return
+        }
 		newPath := args[2]
 		viteNew(newPath)
 	}