Improve reboot function and add an arg for them

Author: Tom5521

main.go

@@ -55,6 +55,7 @@ Argument options will be applied before the config fil
 
 -nokeymap			Don't config the keymap for the new system
 
+-noreboot			Don't reboot the system after the prosess
 `
 
 func PrintHelp() {
@@ -85,6 +86,7 @@ func main() {
 		src.Keymap()
 		src.ConfigRootPasswd()
 		src.FinalCmds()
+		src.Reboot()
 	case "pacstrap":
 		src.Wifi()
 		src.PacmanConf()

src/src.go

@@ -6,6 +6,7 @@ import (
 	"os"
 	"os/exec"
 	"strings"
+	"time"
 
 	"github.com/Tom5521/ArchLinuxInstaller/data"
 	"github.com/Tom5521/CmdRunTools/command"
@@ -262,7 +263,7 @@ func Wifi() {
 				fmGreen("Connected to " + wifi.Name)
 			}
 		} else {
-			Error("Warning: Adaptator not found")
+			Warn("Adaptator not found")
 		}
 		wifi_pkg = "networkmanger iwd"
 	}
@@ -389,10 +390,25 @@ func FinalCmds() {
 			Error("Error using arch-chroot.")
 		}
 	}
-	if conf.Reboot {
+
+}
+
+func Reboot() {
+	if strings.Contains(args, "-noreboot") {
+		return
+	}
+	var err error
+	if !conf.Reboot {
 		err = sh.SetAndRun("reboot")
 		if err != nil {
 			Error("Error in... in... Reboot? wtf")
+			for i := 5; i != 0; i-- {
+				time.Sleep(1 * time.Second)
+				fmt.Print(f("\rRetrying in...%v seconds", i))
+				if i == 1 {
+					Reboot()
+				}
+			}
 		}
 	}
 }