feat(shellconfig): implement PlanForce
This commit is contained in:
@@ -3,7 +3,6 @@ package shellconfig
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -91,5 +90,20 @@ func (m Module) Verify(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
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")
|
var actions []module.Action
|
||||||
|
for _, b := range m.cfg.Blocks {
|
||||||
|
b := b
|
||||||
|
actions = append(actions, module.Action{
|
||||||
|
Description: fmt.Sprintf("rewrite block %s into %s (force)", b.Name, b.File),
|
||||||
|
Kind: module.ActionFileWrite,
|
||||||
|
Exec: func(ctx context.Context) error {
|
||||||
|
p, err := resolveFile(b.File)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return platform.EnsureBlock(p, b.Name, b.Content)
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return module.Plan{Actions: actions}, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -55,4 +55,42 @@ func TestShellConfigIdempotent(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestPlanForceRewritesBlocks(t *testing.T) {
|
||||||
|
home := t.TempDir()
|
||||||
|
t.Setenv("HOME", home)
|
||||||
|
cfg := config.ShellConfigSection{Blocks: []config.ShellConfigBlock{
|
||||||
|
{Name: "path", File: ".zshrc", Content: `export PATH="$HOME/.local/bin:$PATH"`},
|
||||||
|
}}
|
||||||
|
m := New(cfg)
|
||||||
|
st, _ := m.Detect(context.Background())
|
||||||
|
p, _ := m.Plan(context.Background(), st)
|
||||||
|
if err := m.Apply(context.Background(), p, nop{}); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
st2, _ := m.Detect(context.Background())
|
||||||
|
p2, _ := m.Plan(context.Background(), st2)
|
||||||
|
if !p2.IsEmpty() {
|
||||||
|
t.Fatalf("precondition: regular plan should be empty")
|
||||||
|
}
|
||||||
|
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: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
data, _ := os.ReadFile(filepath.Join(home, ".zshrc"))
|
||||||
|
if !strings.Contains(string(data), `export PATH="$HOME/.local/bin:$PATH"`) {
|
||||||
|
t.Fatalf("content missing after force: %q", data)
|
||||||
|
}
|
||||||
|
if !strings.Contains(string(data), ">>> path >>>") {
|
||||||
|
t.Fatalf("block marker missing: %q", data)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
var _ module.Module = Module{}
|
var _ module.Module = Module{}
|
||||||
|
|||||||
Reference in New Issue
Block a user