feat(arduinocli): implement PlanForce
This commit is contained in:
@@ -4,7 +4,6 @@ package arduinocli
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -61,6 +60,30 @@ func (m Module) Detect(ctx context.Context) (module.State, error) {
|
|||||||
return s, nil
|
return s, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// downloadBinary fetches, extracts, and installs arduino-cli to bp.
|
||||||
|
func (m Module) downloadBinary(ctx context.Context, bp string) error {
|
||||||
|
tmpDir, err := os.MkdirTemp("", "arduinocli-")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer os.RemoveAll(tmpDir)
|
||||||
|
archive := filepath.Join(tmpDir, "arduino-cli.tar.gz")
|
||||||
|
if err := platform.Download(ctx, m.cfg.URL, archive, 0o644); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := platform.ExtractTarGz(ctx, archive, tmpDir); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := os.MkdirAll(filepath.Dir(bp), 0o755); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
data, err := os.ReadFile(filepath.Join(tmpDir, "arduino-cli"))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return os.WriteFile(bp, data, 0o755)
|
||||||
|
}
|
||||||
|
|
||||||
func (m Module) Plan(ctx context.Context, cur module.State) (module.Plan, error) {
|
func (m Module) Plan(ctx context.Context, cur module.State) (module.Plan, error) {
|
||||||
s, _ := cur.(state)
|
s, _ := cur.(state)
|
||||||
var actions []module.Action
|
var actions []module.Action
|
||||||
@@ -69,28 +92,7 @@ func (m Module) Plan(ctx context.Context, cur module.State) (module.Plan, error)
|
|||||||
actions = append(actions, module.Action{
|
actions = append(actions, module.Action{
|
||||||
Description: "download + install arduino-cli " + m.cfg.Version,
|
Description: "download + install arduino-cli " + m.cfg.Version,
|
||||||
Kind: module.ActionFileWrite,
|
Kind: module.ActionFileWrite,
|
||||||
Exec: func(ctx context.Context) error {
|
Exec: func(ctx context.Context) error { return m.downloadBinary(ctx, bp) },
|
||||||
tmpDir, err := os.MkdirTemp("", "arduinocli-")
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer os.RemoveAll(tmpDir)
|
|
||||||
archive := filepath.Join(tmpDir, "arduino-cli.tar.gz")
|
|
||||||
if err := platform.Download(ctx, m.cfg.URL, archive, 0o644); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := platform.ExtractTarGz(ctx, archive, tmpDir); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := os.MkdirAll(filepath.Dir(bp), 0o755); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
data, err := os.ReadFile(filepath.Join(tmpDir, "arduino-cli"))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return os.WriteFile(bp, data, 0o755)
|
|
||||||
},
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
for _, core := range s.MissingCores {
|
for _, core := range s.MissingCores {
|
||||||
@@ -138,6 +140,33 @@ func (m Module) Verify(ctx context.Context) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// PlanForce redownloads the arduino-cli binary, refreshes the core index,
|
||||||
|
// and reinstalls every configured core unconditionally. arduino-cli's
|
||||||
|
// `core install` is idempotent, so reissuing it is safe.
|
||||||
func (m Module) PlanForce(ctx context.Context) (module.Plan, error) {
|
func (m Module) PlanForce(ctx context.Context) (module.Plan, error) {
|
||||||
return module.Plan{}, errors.New("PlanForce not implemented")
|
bp, err := binaryPath()
|
||||||
|
if err != nil {
|
||||||
|
return module.Plan{}, err
|
||||||
|
}
|
||||||
|
actions := []module.Action{
|
||||||
|
{
|
||||||
|
Description: "redownload + reinstall arduino-cli " + m.cfg.Version + " (force)",
|
||||||
|
Kind: module.ActionFileWrite,
|
||||||
|
Exec: func(ctx context.Context) error { return m.downloadBinary(ctx, bp) },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Description: "arduino-cli core update-index (force)",
|
||||||
|
Kind: module.ActionCommand,
|
||||||
|
Exec: func(ctx context.Context) error { return m.runner.Run(ctx, bp, "core", "update-index") },
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, core := range m.cfg.Cores {
|
||||||
|
core := core
|
||||||
|
actions = append(actions, module.Action{
|
||||||
|
Description: "arduino-cli core install " + core + " (force)",
|
||||||
|
Kind: module.ActionCommand,
|
||||||
|
Exec: func(ctx context.Context) error { return m.runner.Run(ctx, bp, "core", "install", core) },
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return module.Plan{Actions: actions}, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -91,4 +91,64 @@ func TestArduinoCLIDownloadsAndInstallsCore(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestPlanForceRedownloadsAndReinstallsCores(t *testing.T) {
|
||||||
|
home := t.TempDir()
|
||||||
|
t.Setenv("HOME", home)
|
||||||
|
// Pre-populate a stale binary to prove force overwrites.
|
||||||
|
os.MkdirAll(filepath.Join(home, ".local", "bin"), 0o755)
|
||||||
|
bp := filepath.Join(home, ".local", "bin", "arduino-cli")
|
||||||
|
os.WriteFile(bp, []byte("stale"), 0o755)
|
||||||
|
|
||||||
|
archive := makeTarGz(t, map[string][]byte{"arduino-cli": []byte("#!/bin/sh\nfresh\n")})
|
||||||
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
data, _ := os.ReadFile(archive)
|
||||||
|
w.Write(data)
|
||||||
|
}))
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
r := &fakeRunner{coreListOut: []byte("arduino:avr\nesp32:esp32\n")}
|
||||||
|
m := New(config.ArduinoCLIConfig{
|
||||||
|
Version: "1.0.0",
|
||||||
|
URL: srv.URL + "/arduino-cli.tgz",
|
||||||
|
Cores: []string{"arduino:avr", "esp32:esp32"},
|
||||||
|
}, r)
|
||||||
|
|
||||||
|
plan, err := m.PlanForce(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("PlanForce: %v", err)
|
||||||
|
}
|
||||||
|
if plan.IsEmpty() {
|
||||||
|
t.Fatal("force plan must not be empty")
|
||||||
|
}
|
||||||
|
for _, a := range plan.Actions {
|
||||||
|
if err := a.Exec(context.Background()); err != nil {
|
||||||
|
t.Fatalf("Exec %q: %v", a.Description, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Binary got overwritten with fresh payload.
|
||||||
|
got, _ := os.ReadFile(bp)
|
||||||
|
if string(got) == "stale" {
|
||||||
|
t.Fatalf("binary not overwritten by force")
|
||||||
|
}
|
||||||
|
|
||||||
|
// update-index and both `core install` calls fired.
|
||||||
|
sawUpdate := false
|
||||||
|
installed := map[string]bool{}
|
||||||
|
for _, c := range r.runCalls {
|
||||||
|
if filepath.Base(c[0]) == "arduino-cli" && len(c) >= 3 && c[1] == "core" && c[2] == "update-index" {
|
||||||
|
sawUpdate = true
|
||||||
|
}
|
||||||
|
if filepath.Base(c[0]) == "arduino-cli" && len(c) >= 4 && c[1] == "core" && c[2] == "install" {
|
||||||
|
installed[c[3]] = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !sawUpdate {
|
||||||
|
t.Errorf("no core update-index call; calls=%v", r.runCalls)
|
||||||
|
}
|
||||||
|
if !installed["arduino:avr"] || !installed["esp32:esp32"] {
|
||||||
|
t.Errorf("expected both cores force-installed; got=%v calls=%v", installed, r.runCalls)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
var _ module.Module = Module{}
|
var _ module.Module = Module{}
|
||||||
|
|||||||
Reference in New Issue
Block a user