Invoke cluster reset function when only reset flag is passed (k3s-io#…

…3276)

Signed-off-by: galal-hussein <[email protected]>

pkg/agent/run.go

@@ -361,6 +361,11 @@ func setupTunnelAndRunAgent(ctx context.Context, nodeConfig *daemonconfig.Node,
 		case <-ctx.Done():
 			return ctx.Err()
 		}
+	} else if cfg.ClusterReset && proxy.IsAPIServerLBEnabled() {
+		if err := agent.Agent(&nodeConfig.AgentConfig); err != nil {
+			return err
+		}
+		agentRan = true
 	}
 
 	if err := tunnel.Setup(ctx, nodeConfig, proxy); err != nil {

pkg/cli/cmds/agent.go

@@ -43,6 +43,7 @@ type Agent struct {
 	Labels                   cli.StringSlice
 	Taints                   cli.StringSlice
 	AgentShared
+	ClusterReset bool
 }
 
 type AgentShared struct {

pkg/cli/server/server.go

@@ -404,6 +404,7 @@ func run(app *cli.Context, cfg *cmds.Server, leaderControllers server.CustomCont
 	agentConfig.Token = token
 	agentConfig.DisableLoadBalancer = !serverConfig.ControlConfig.DisableAPIServer
 	agentConfig.ETCDAgent = serverConfig.ControlConfig.DisableAPIServer
+	agentConfig.ClusterReset = serverConfig.ControlConfig.ClusterReset
 
 	agentConfig.Rootless = cfg.Rootless
 

pkg/cluster/managed.go

@@ -70,9 +70,12 @@ func (c *Cluster) start(ctx context.Context) error {
 			if !os.IsNotExist(err) {
 				return err
 			}
-		} else {
-			return fmt.Errorf("cluster-reset was successfully performed, please remove the cluster-reset flag and start %s normally, if you need to perform another cluster reset, you must first manually delete the %s file", version.Program, resetFile)
+			rebootstrap := func() error {
+				return c.storageBootstrap(ctx)
+			}
+			return c.managedDB.Reset(ctx, rebootstrap)
 		}
+		return fmt.Errorf("cluster-reset was successfully performed, please remove the cluster-reset flag and start %s normally, if you need to perform another cluster reset, you must first manually delete the %s file", version.Program, resetFile)
 	}
 
 	// removing the reset file and ignore error if the file doesn't exist