feat(cli): make register interactive stages working

cmd/cmd.go

@@ -14,8 +14,14 @@ import (
 
 const version = "0.1.5"
 
+type globalArgs struct {
+	EnvFile string
+}
+
 func Execute(ctx context.Context) {
-	task := runtime.NewTask("gitea", 0, nil, nil)
+	// task := runtime.NewTask("gitea", 0, nil, nil)
+
+	var gArgs globalArgs
 
 	// ./act_runner
 	rootCmd := &cobra.Command{
@@ -26,10 +32,10 @@ func Execute(ctx context.Context) {
 		Version:      version,
 		SilenceUsage: true,
 	}
-	rootCmd.Flags().BoolP("run", "r", false, "run workflows")
-	rootCmd.Flags().StringP("job", "j", "", "run job")
-	rootCmd.PersistentFlags().StringVarP(&task.Input.ForgeInstance, "forge-instance", "", "github.com", "Forge instance to use.")
-	rootCmd.PersistentFlags().StringVarP(&task.Input.EnvFile, "env-file", "", ".env", "Read in a file of environment variables.")
+	//rootCmd.Flags().BoolP("run", "r", false, "run workflows")
+	//rootCmd.Flags().StringP("job", "j", "", "run job")
+	//rootCmd.PersistentFlags().StringVarP(&task.Input.ForgeInstance, "forge-instance", "", "github.com", "Forge instance to use.")
+	rootCmd.PersistentFlags().StringVarP(&gArgs.EnvFile, "env-file", "", ".env", "Read in a file of environment variables.")
 
 	// ./act_runner register
 	var regArgs registerArgs
@@ -37,7 +43,7 @@ func Execute(ctx context.Context) {
 		Use:   "register",
 		Short: "Register a runner to the server",
 		Args:  cobra.MaximumNArgs(0),
-		RunE:  runRegister(ctx, &regArgs), // must use a pointer to regArgs
+		RunE:  runRegister(ctx, &regArgs, gArgs.EnvFile), // must use a pointer to regArgs
 	}
 	registerCmd.Flags().BoolVarP(&regArgs.NoInteractive, "no-interactive", "", false, "Disable interactive mode")
 	registerCmd.Flags().StringVarP(&regArgs.InstanceAddr, "instance", "", "", "Gitea instance address")
@@ -49,7 +55,7 @@ func Execute(ctx context.Context) {
 		Use:   "daemon",
 		Short: "Run as a runner daemon",
 		Args:  cobra.MaximumNArgs(1),
-		RunE:  runDaemon(ctx, task.Input.EnvFile),
+		RunE:  runDaemon(ctx, gArgs.EnvFile),
 	}
 	// add all command
 	rootCmd.AddCommand(daemonCmd)

cmd/daemon.go

@@ -3,15 +3,12 @@ package cmd
 import (
 	"context"
 	"os"
-	"time"
 
 	"gitea.com/gitea/act_runner/client"
 	"gitea.com/gitea/act_runner/config"
 	"gitea.com/gitea/act_runner/engine"
 	"gitea.com/gitea/act_runner/poller"
-	"gitea.com/gitea/act_runner/register"
 	"gitea.com/gitea/act_runner/runtime"
-	pingv1 "gitea.com/gitea/proto-go/ping/v1"
 	runnerv1 "gitea.com/gitea/proto-go/runner/v1"
 
 	"github.com/bufbuild/connect-go"
@@ -35,60 +32,6 @@ func runDaemon(ctx context.Context, envFile string) func(cmd *cobra.Command, arg
 
 		initLogging(cfg)
 
-		// initial http client
-		cli := client.New(
-			cfg.Client.Address,
-			client.WithSkipVerify(cfg.Client.SkipVerify),
-			client.WithGRPC(cfg.Client.GRPC),
-			client.WithGRPCWeb(cfg.Client.GRPCWeb),
-		)
-
-		for {
-			_, err := cli.Ping(ctx, connect.NewRequest(&pingv1.PingRequest{
-				Data: cfg.Runner.Name,
-			}))
-			select {
-			case <-ctx.Done():
-				return nil
-			default:
-			}
-			if ctx.Err() != nil {
-				break
-			}
-			if err != nil {
-				log.WithError(err).
-					Errorln("cannot ping the remote server")
-				// TODO: if ping failed, retry or exit
-				time.Sleep(time.Second)
-			} else {
-				log.Infoln("successfully pinged the remote server")
-				break
-			}
-		}
-
-		// register new runner
-		if cfg.Runner.UUID == "" {
-			register := register.New(
-				cli,
-				&client.Filter{
-					OS:     cfg.Platform.OS,
-					Arch:   cfg.Platform.Arch,
-					Labels: cfg.Runner.Labels,
-				},
-			)
-
-			data, err := register.Register(ctx, cfg.Runner)
-			if err != nil {
-				return err
-			}
-			if data.UUID != "" {
-				cfg.Runner.UUID = data.UUID
-			}
-			if data.Token != "" {
-				cfg.Runner.Token = data.Token
-			}
-		}
-
 		// try to connect to docker daemon
 		// if failed, exit with error
 		if err := engine.Start(ctx); err != nil {
@@ -97,7 +40,7 @@ func runDaemon(ctx context.Context, envFile string) func(cmd *cobra.Command, arg
 
 		var g errgroup.Group
 
-		cli = client.New(
+		cli := client.New(
 			cfg.Client.Address,
 			client.WithSkipVerify(cfg.Client.SkipVerify),
 			client.WithGRPC(cfg.Client.GRPC),

cmd/register.go

@@ -7,14 +7,22 @@ import (
 	"os/signal"
 	"runtime"
 	"strings"
+	"time"
 
+	"gitea.com/gitea/act_runner/client"
+	"gitea.com/gitea/act_runner/config"
+	"gitea.com/gitea/act_runner/register"
+	pingv1 "gitea.com/gitea/proto-go/ping/v1"
+	"github.com/appleboy/com/file"
+	"github.com/bufbuild/connect-go"
+	"github.com/joho/godotenv"
 	"github.com/mattn/go-isatty"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
 // runRegister registers a runner to the server
-func runRegister(ctx context.Context, regArgs *registerArgs) func(*cobra.Command, []string) error {
+func runRegister(ctx context.Context, regArgs *registerArgs, envFile string) func(*cobra.Command, []string) error {
 	return func(cmd *cobra.Command, args []string) error {
 		log.SetReportCaller(false)
 		isTerm := isatty.IsTerminal(os.Stdout.Fd())
@@ -22,6 +30,7 @@ func runRegister(ctx context.Context, regArgs *registerArgs) func(*cobra.Command
 			DisableColors:    !isTerm,
 			DisableTimestamp: true,
 		})
+		log.SetLevel(log.DebugLevel)
 
 		log.Infof("Registering runner, arch=%s, os=%s, version=%s.",
 			runtime.GOARCH, runtime.GOOS, version)
@@ -33,7 +42,7 @@ func runRegister(ctx context.Context, regArgs *registerArgs) func(*cobra.Command
 		}
 
 		go func() {
-			if err := registerInteractive(); err != nil {
+			if err := registerInteractive(envFile); err != nil {
 				// log.Errorln(err)
 				os.Exit(2)
 				return
@@ -59,12 +68,14 @@ type registerArgs struct {
 type registerStage int8
 
 const (
-	StageUnknown       registerStage = -1
-	StageInputInstance registerStage = iota + 1
+	StageUnknown              registerStage = -1
+	StageOverwriteLocalConfig registerStage = iota + 1
+	StageInputInstance
 	StageInputToken
 	StageInputRunnerName
 	StageInputCustomLabels
 	StageWaitingForRegistration
+	StageExit
 )
 
 type registerInputs struct {
@@ -90,6 +101,11 @@ func (r *registerInputs) assignToNext(stage registerStage, value string) registe
 	}
 
 	switch stage {
+	case StageOverwriteLocalConfig:
+		if value == "Y" || value == "y" {
+			return StageInputInstance
+		}
+		return StageExit
 	case StageInputInstance:
 		r.InstanceAddr = value
 		return StageInputToken
@@ -106,15 +122,30 @@ func (r *registerInputs) assignToNext(stage registerStage, value string) registe
 	return StageUnknown
 }
 
-func registerInteractive() error {
+func getLocalConfigFile(envFile string) (string, bool) {
+	_ = godotenv.Load(envFile)
+	cfg, _ := config.FromEnviron()
+	return cfg.Runner.File, file.IsFile(cfg.Runner.File)
+}
+
+func registerInteractive(envFile string) error {
+
 	var (
 		reader = bufio.NewReader(os.Stdin)
 		stage  = StageInputInstance
 		inputs = new(registerInputs)
 	)
 
+	// check if overwrite local config
+	_ = godotenv.Load(envFile)
+	cfg, _ := config.FromEnviron()
+	if file.IsFile(cfg.Runner.File) {
+		stage = StageOverwriteLocalConfig
+	}
+
 	for {
 		printStageHelp(stage)
+
 		cmdString, err := reader.ReadString('\n')
 		if err != nil {
 			return err
@@ -123,6 +154,15 @@ func registerInteractive() error {
 
 		if stage == StageWaitingForRegistration {
 			log.Infof("Registering runner, name=%s, instance=%s, labels=%v.", inputs.RunnerName, inputs.InstanceAddr, inputs.CustomLabels)
+			if err := doRegister(&cfg, inputs); err != nil {
+				log.Errorf("Failed to register runner: %v", err)
+			} else {
+				log.Infof("Runner registered successfully.")
+			}
+			return nil
+		}
+
+		if stage == StageExit {
 			return nil
 		}
 
@@ -135,6 +175,8 @@ func registerInteractive() error {
 
 func printStageHelp(stage registerStage) {
 	switch stage {
+	case StageOverwriteLocalConfig:
+		log.Infoln("Runner is already registered, overwrite local config? [Y/n]")
 	case StageInputInstance:
 		log.Infoln("Enter the Gitea instance URL (for example, https://gitea.com/):")
 	case StageInputToken:
@@ -148,3 +190,57 @@ func printStageHelp(stage registerStage) {
 		log.Infoln("Waiting for registration...")
 	}
 }
+
+func doRegister(cfg *config.Config, inputs *registerInputs) error {
+
+	ctx := context.Background()
+
+	// initial http client
+	cli := client.New(
+		inputs.InstanceAddr,
+		client.WithSkipVerify(cfg.Client.SkipVerify),
+		client.WithGRPC(cfg.Client.GRPC),
+		client.WithGRPCWeb(cfg.Client.GRPCWeb),
+	)
+
+	for {
+		_, err := cli.Ping(ctx, connect.NewRequest(&pingv1.PingRequest{
+			Data: inputs.RunnerName,
+		}))
+		select {
+		case <-ctx.Done():
+			return nil
+		default:
+		}
+		if ctx.Err() != nil {
+			break
+		}
+		if err != nil {
+			log.WithError(err).
+				Errorln("Cannot ping the Gitea instance server")
+			// TODO: if ping failed, retry or exit
+			time.Sleep(time.Second)
+		} else {
+			log.Debugln("Successfully pinged the Gitea instance server")
+			break
+		}
+	}
+
+	register := register.New(
+		cli,
+		&client.Filter{
+			OS:     cfg.Platform.OS,
+			Arch:   cfg.Platform.Arch,
+			Labels: inputs.CustomLabels,
+		},
+	)
+	cfg.Runner.Name = inputs.RunnerName
+	cfg.Runner.Token = inputs.Token
+	_, err := register.Register(ctx, cfg.Runner)
+	if err != nil {
+		log.WithError(err).Errorln("Cannot register the runner")
+		return nil
+	}
+
+	return nil
+}