save config / factory reset

pkg/agent/clsp/processor.go

@@ -7,6 +7,7 @@ import (
 	"context"
 	"fmt"
 	"os"
+	"os/exec"
 
 	agentapi "go.githedgehog.com/fabric/api/agent/v1beta1"
 	"go.githedgehog.com/fabric/pkg/agent/dozer"
@@ -33,9 +34,16 @@ func (c *CelesticaPlusProcessor) Reinstall(ctx context.Context) error {
 	return bcm.Processor().Reinstall(ctx) //nolint:wrapcheck
 }
 
-// TODO
 func (c *CelesticaPlusProcessor) FactoryReset(ctx context.Context) error {
-	return fmt.Errorf("not implemented") //nolint:err113
+	cmd := exec.CommandContext(ctx, "sonic-cli", "-c", "\"write erase\"")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to reset config: %w", err)
+	}
+
+	return c.Reboot(ctx, true)
 }
 
 // TODO
@@ -109,3 +117,15 @@ func (c *CelesticaPlusProcessor) ApplyActions(ctx context.Context, actions []doz
 func (c *CelesticaPlusProcessor) CalculateActions(ctx context.Context, actual *dozer.Spec, desired *dozer.Spec) ([]dozer.Action, error) {
 	return nil, fmt.Errorf("unsupported operation") //nolint:err113
 }
+
+func (c *CelesticaPlusProcessor) SaveConfig(ctx context.Context) error {
+	cmd := exec.CommandContext(ctx, "sonic-cli", "-c", "\"copy running-config startup-config\"")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to save config: %w", err)
+	}
+
+	return nil
+}

pkg/agent/cmls/processor.go

@@ -72,3 +72,8 @@ func (c *CumulusProcessor) ApplyActions(ctx context.Context, actions []dozer.Act
 func (c *CumulusProcessor) CalculateActions(ctx context.Context, actual *dozer.Spec, desired *dozer.Spec) ([]dozer.Action, error) {
 	return nil, fmt.Errorf("unsupported operation") //nolint:err113
 }
+
+// TODO
+func (c *CumulusProcessor) SaveConfig(ctx context.Context) error {
+	return fmt.Errorf("unsupported operation") //nolint:err113
+}

pkg/agent/dozer/bcm/processor.go

@@ -154,22 +154,22 @@ func (p *BroadcomProcessor) Reinstall(ctx context.Context) error {
 	return p.Reboot(ctx, true)
 }
 
-func (p *BroadcomProcessor) FactoryReset(_ context.Context) error {
-	// TODO use sonic-cli for it and then switch to GNOI
-	// write erase boot
-
-	// stdin, err := cmd.StdinPipe()
-	// if err != nil {
-	// 	log.Fatal(err)
-	// }
+func (p *BroadcomProcessor) FactoryReset(ctx context.Context) error {
+	// "write erase boot" prompts for confirmation ("...continue? [y/N]:") and
+	// reads the answer from the controlling TTY, not stdin, so a plain stdin
+	// pipe is ignored and the command defaults to N. Wrap it in `script`, which
+	// allocates a pty; `script` forwards its own stdin into the pty, so feeding
+	// "y\n" here reaches the prompt.
+	cmd := exec.CommandContext(ctx, "script", "-qec", `sonic-cli -c "write erase boot"`, "/dev/null")
+	cmd.Stdin = strings.NewReader("y\n")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
 
-	// go func() {
-	// 	defer stdin.Close()
-	// 	// todo test it
-	// 	io.WriteString(stdin, "y\n")
-	// }()
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("failed to reset config: %w", err)
+	}
 
-	return fmt.Errorf("not supported") //nolint:err113
+	return p.Reboot(ctx, true)
 }
 
 func (p *BroadcomProcessor) LoadActualState(ctx context.Context, agent *agentapi.Agent) (*dozer.Spec, error) {
@@ -413,3 +413,11 @@ func (p *BroadcomProcessor) SetRoCE(ctx context.Context, val bool) error {
 
 	return nil
 }
+
+func (p *BroadcomProcessor) SaveConfig(ctx context.Context) error {
+	cmd := exec.CommandContext(ctx, "sonic-cli", "-c", "\"write memory\"")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	return errors.Wrap(cmd.Run(), "failed to save config")
+}

pkg/agent/dozer/dozer.go

@@ -40,6 +40,7 @@ type Processor interface {
 	FactoryReset(ctx context.Context) error
 	GetRoCE(ctx context.Context) (bool, error)
 	SetRoCE(ctx context.Context, enable bool) error
+	SaveConfig(ctx context.Context) error
 }
 
 type Action interface {