Now the program uses the flag std lib to parse the arguments

Author: Tom5521

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"os"
 
@@ -15,74 +16,73 @@ var sh = func() command.Cmd {
 	return cmd
 }()
 
-var HelpStr = `
-Usage:
-#[bin] [argument] -[option]
-
-Arguments:
-- help				Print this text
-
-- version			Print the current version of the binary
-
-- install			Run all the nesesary functions to install completely Arch Linux 
-
-- pacstrap			Only runs the pacstrap functions
-
-- grub				Only installs Grub
-
-- newconfig			Creates a new config overwriting the original
-
-- part				Open cfdisk to partitionate the install disk
-
-- passwd			Only changes the password of the new root 
-
-- mount				Only mounts the disks in her routes
-
-Options:
-Argument options will be applied before the config fil
-
--nopasswd			Skip the passwd set
-
--nopacstrap			Skip the pacstrap prosess
-
--nopart				Skip the partitionating prosess (not open cfdisk)
-
--nogrub				Don't install Grub
-
--noformat			Don't format the partitions
-
--nomount			Don't mount the partitions
-
--nopacmanconf			Don't paste custom pacman config
+func main() {
+	if len(os.Args) == 0 {
+		fmt.Println("Not enough arguments")
+		return
+	}
+	versionFlag := flag.Bool("version", false, "Show the version of the binary")
+	helpFlag := flag.Bool("help", false, "Print the help message")
 
--nowifi				Don't configure for wifi 
+	// Only individual functions
+	installFlag := flag.Bool("install", false, "Run all the nesesary functions to install completely Arch Linux")
+	pacstrapFlag := flag.Bool("pacstrap", false, "Only runs the pacstrap functions")
+	grubFlag := flag.Bool("grub", false, "Only installs Grub")
+	passwdFlag := flag.Bool("passwd", false, "Only changes the password of the new root")
+	partFlag := flag.Bool("part", false, "Only changes the password of the new root")
+	mountFlag := flag.Bool("mount", false, "Only mounts the disks in her routes")
+	newConfigFlag := flag.Bool("newconfig", false, "Creates a new config overwriting the original")
 
--nofstab			Don't generate a fstab for the new system
+	flag.Parse()
 
--nokeymap			Don't config the keymap for the new system
+	if *newConfigFlag {
+		data.NewYamlFile()
+		err := sh.SetAndRun("vim " + data.Pfilename)
+		if err != nil {
+			src.Error("Error oppening vim.\n" + err.Error())
+		}
+		return
+	}
 
--noreboot			Don't reboot the system after the prosess
-`
+	parserFlags := []bool{*installFlag, *pacstrapFlag, *grubFlag, *passwdFlag, *partFlag, *mountFlag}
+	catchBadFlags := func(flags []bool) (HaveBadFlags bool) {
+		var trueValues int
+		for _, parser := range parserFlags {
+			if parser {
+				trueValues++
+			}
+		}
+		if trueValues > 1 {
+			return true
+		} else {
+			return false
+		}
+	}
 
-func PrintHelp() {
-	fmt.Print(HelpStr)
-}
+	if catchBadFlags(parserFlags) {
+		fmt.Println("There are arguments that cannot be used with others!")
+		flag.PrintDefaults()
+		return
+	}
 
-func main() {
-	if len(os.Args) == 0 {
-		fmt.Println("Not enough arguments")
+	if *versionFlag {
+		fmt.Println("...")
 		return
 	}
-	switch os.Args[1] {
-	case "version":
-		fmt.Println()
-	case "help":
-		PrintHelp()
-	case "passwd":
-		src.ConfigRootPasswd()
-	case "part":
-		src.Partitioning()
-	case "install":
+	if *helpFlag {
+		fmt.Println(`
+Usage:
+#[bin] [argument] -[option]`)
+		flag.PrintDefaults()
+		return
+	}
+	if *pacstrapFlag {
+		src.Wifi()
+		src.PacmanConf()
+		src.Mount()
+		src.Pacstrap()
+	}
+	if *installFlag {
 		src.Partitioning()
 		src.Wifi()
 		src.PacmanConf()
@@ -95,22 +95,20 @@ func main() {
 		src.ConfigRootPasswd()
 		src.FinalCmds()
 		src.Reboot()
-	case "pacstrap":
-		src.Wifi()
-		src.PacmanConf()
-		src.Mount()
-		src.Pacstrap()
-	case "grub":
+	}
+	if *passwdFlag {
+		src.ConfigRootPasswd()
+	}
+	if *partFlag {
+		src.Partitioning()
+	}
+	if *grubFlag {
 		src.Mount()
 		src.Fstab()
 		src.Grub()
-	case "mount":
+	}
+	if *mountFlag {
 		src.Mount()
-	case "newconfig":
-		data.NewYamlFile()
-		err := sh.SetAndRun("vim " + data.Pfilename)
-		if err != nil {
-			src.Error("Error oppening vim.\n" + err.Error())
-		}
 	}
+
 }

src/src.go

@@ -1,6 +1,7 @@
 package src
 
 import (
+	"flag"
 	"fmt"
 	"io"
 	"os"
@@ -14,6 +15,20 @@ import (
 	"github.com/gookit/color"
 )
 
+var (
+	NoPart       *bool = flag.Bool("nopart", false, "Skip the partitionating prosess (not open cfdisk)")
+	NoGrub             = flag.Bool("nogrub", false, "Don't install Grub")
+	NoWifi             = flag.Bool("nowifi", false, "Don't configure for wifi")
+	NoFstab            = flag.Bool("nofstab", false, "Don't generate a fstab for the new system")
+	NoKeymap           = flag.Bool("nokeymap", false, "Don't config the keymap for the new system")
+	NoReboot           = flag.Bool("noreboot", false, "Don't reboot the system after the prosess")
+	NoPasswd           = flag.Bool("nopasswd", false, "Skip the passwd setting")
+	NoPacstrap         = flag.Bool("nopacstrap", false, "Skip the pacstrap process")
+	NoPacmanConf       = flag.Bool("nopacmanconf", false, "Don't copy the temporal pacman.conf for best performance in the pacstrap")
+	NoMount            = flag.Bool("nomount", false, "Don't mount the partitions")
+	NoFormat           = flag.Bool("noformat", false, "Don't format the partitions")
+)
+
 var (
 	// Functions
 	sh = func() command.Cmd {
@@ -27,7 +42,6 @@ var (
 	fmGreen  = color.Green.Println
 	fmYellow = color.Yellow.Println
 	// Data
-	args       = strings.Join(os.Args, " ")
 	conf       = data.GetYamldata()
 	partitions = conf.Partitions
 	wifi       = conf.Wifi
@@ -70,7 +84,7 @@ func Warn(err string) {
 }
 
 func Partitioning() {
-	if strings.Contains(args, "-nopart") {
+	if *NoPart {
 		return
 	}
 	var err error
@@ -81,7 +95,7 @@ func Partitioning() {
 }
 
 func Format() {
-	if strings.Contains(args, "-noformat") {
+	if *NoFormat {
 		return
 	}
 	var err error
@@ -135,7 +149,7 @@ func Format() {
 	}
 }
 func Mount() {
-	if strings.Contains(args, "-nomount") {
+	if *NoMount {
 		return
 	}
 	var err error
@@ -209,7 +223,7 @@ func Mount() {
 }
 
 func PacmanConf() {
-	if strings.Contains(args, "-nopacmanconf") {
+	if *NoPacmanConf {
 		return
 	}
 	if check_pacman_cfg, _ := file.CheckFile("pacman.conf"); conf.CustomPacmanConfig && !check_pacman_cfg {
@@ -235,7 +249,7 @@ func PacmanConf() {
 }
 
 func Wifi() {
-	if strings.Contains(args, "-nowifi") {
+	if *NoWifi {
 		return
 	}
 	var err error
@@ -271,7 +285,7 @@ func Wifi() {
 }
 
 func Pacstrap() {
-	if strings.Contains(args, "-nopacstrap") {
+	if *NoPacstrap {
 		return
 	}
 	if !conf.PacstrapSkip {
@@ -288,7 +302,7 @@ func Pacstrap() {
 }
 
 func Fstab() {
-	if strings.Contains(args, "-nofstab") {
+	if *NoFstab {
 		return
 	}
 	fmYellow("Generating Fstab...")
@@ -302,7 +316,7 @@ func Fstab() {
 }
 
 func Grub() {
-	if strings.Contains(args, "-nogrub") {
+	if *NoGrub {
 		return
 	}
 	var err error
@@ -334,7 +348,7 @@ func Grub() {
 }
 
 func Keymap() {
-	if strings.Contains(args, "-nokeymap") {
+	if *NoKeymap {
 		return
 	}
 	sh.UseBashShell(true)
@@ -354,7 +368,7 @@ func Keymap() {
 }
 
 func ConfigRootPasswd() {
-	if strings.Contains(args, "-nopasswd") {
+	if *NoPasswd {
 		return
 	}
 	fmYellow("Setting the root password:", conf.Password)
@@ -395,7 +409,7 @@ func FinalCmds() {
 }
 
 func Reboot() {
-	if strings.Contains(args, "-noreboot") {
+	if *NoReboot {
 		return
 	}
 	var err error