feat(cli): add flag --register for runner

Author: fiftin

cli/cmd/runner_register.go

@@ -18,22 +18,28 @@ func init() {
 	runnerCmd.AddCommand(runnerRegisterCmd)
 }
 
-func registerRunner() {
+func initRunnerRegistrationToken() {
+	if !runnerRegisterArgs.stdinRegistrationToken {
+		return
+	}
 
-	configFile := util.ConfigInit(persistentFlags.configPath, persistentFlags.noConfig)
+	tokenBytes, err := io.ReadAll(os.Stdin)
+	if err != nil {
+		panic(err)
+	}
 
-	if runnerRegisterArgs.stdinRegistrationToken {
-		tokenBytes, err := io.ReadAll(os.Stdin)
-		if err != nil {
-			panic(err)
-		}
+	if len(tokenBytes) == 0 {
+		panic("Empty token")
+	}
 
-		if len(tokenBytes) == 0 {
-			panic("Empty token")
-		}
+	util.Config.Runner.RegistrationToken = strings.TrimSpace(string(tokenBytes))
+}
 
-		util.Config.Runner.RegistrationToken = strings.TrimSpace(string(tokenBytes))
-	}
+func registerRunner() {
+
+	configFile := util.ConfigInit(persistentFlags.configPath, persistentFlags.noConfig)
+
+	initRunnerRegistrationToken()
 
 	taskPool := createRunnerJobPool()