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()
 

cli/cmd/runner_start.go

@@ -5,15 +5,35 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var runnerStartArgs struct {
+	register bool
+}
+
 func init() {
+	runnerRegisterCmd.PersistentFlags().BoolVar(&runnerStartArgs.register, "register", false, "Register new runner if not registered")
 	runnerCmd.AddCommand(runnerStartCmd)
 }
 
 func runRunner() {
-	util.ConfigInit(persistentFlags.configPath, persistentFlags.noConfig)
+
+	configFile := util.ConfigInit(persistentFlags.configPath, persistentFlags.noConfig)
 
 	taskPool := createRunnerJobPool()
 
+	if runnerStartArgs.register {
+
+		initRunnerRegistrationToken()
+
+		if util.Config.Runner.Token == "" {
+
+			err := taskPool.Register(configFile)
+
+			if err != nil {
+				panic(err)
+			}
+		}
+	}
+
 	taskPool.Run()
 }
 

deployment/docker/runner/runner-wrapper

@@ -15,7 +15,11 @@ fi
 
 echoerr "Starting semaphore runner"
 if test "$#" -ne 1; then
-    exec /usr/local/bin/semaphore runner start --no-config
+    if [ -n "${SEMAPHORE_RUNNER_REGISTRATION_TOKEN:-}" ]; then
+      exec /usr/local/bin/semaphore runner start --no-config --register
+    else
+      exec /usr/local/bin/semaphore runner start --no-config
+    fi
 else
     exec "$@"
 fi