feat(neovim): implement PlanForce

This commit is contained in:
MDietermann
2026-04-14 22:15:32 +02:00
parent 77b767aa0e
commit 25b4eefecd
2 changed files with 198 additions and 2 deletions

View File

@@ -6,7 +6,6 @@ package neovim
import ( import (
"bytes" "bytes"
"context" "context"
"errors"
"embed" "embed"
"fmt" "fmt"
"os" "os"
@@ -463,6 +462,114 @@ func (m Module) Verify(ctx context.Context) error {
return nil return nil
} }
// PlanForce rewrites chadrc.lua and reinstalls packages but does NOT
// delete an existing NvChad clone, to preserve any user modifications.
// If NvChad is missing, it's cloned fresh.
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") cfg, err := nvimCfg()
if err != nil {
return module.Plan{}, err
}
rendered, err := m.renderChadrc()
if err != nil {
return module.Plan{}, err
}
var actions []module.Action
if m.pm != nil {
actions = append(actions, module.Action{
Description: "pacman -S (force) neovim",
Kind: module.ActionPackageInstall,
Exec: func(ctx context.Context) error { return m.pm.Reinstall(ctx, []string{"neovim"}) },
})
}
// Clone NvChad only if absent; preserve any user modifications otherwise.
repo := m.cfg.NvChadStarterRepo
actions = append(actions, module.Action{
Description: "ensure NvChad starter cloned (skip if present)",
Kind: module.ActionGitClone,
Exec: func(ctx context.Context) error {
if _, err := os.Stat(filepath.Join(cfg, "lua", "plugins", "init.lua")); err == nil {
return nil
}
if err := os.MkdirAll(filepath.Dir(cfg), 0o755); err != nil {
return err
}
if err := m.git.Clone(ctx, repo, cfg); err != nil {
return err
}
return os.RemoveAll(filepath.Join(cfg, ".git"))
},
})
actions = append(actions, module.Action{
Description: "strip nvim-lspconfig block from starter init.lua (force)",
Kind: module.ActionFileWrite,
Exec: func(ctx context.Context) error {
p := filepath.Join(cfg, "lua", "plugins", "init.lua")
data, err := os.ReadFile(p)
if err != nil {
if os.IsNotExist(err) {
return nil
}
return err
}
return platform.WriteUserFile(p, stripLspconfigBlock(data), 0o644)
},
})
actions = append(actions, module.Action{
Description: "remove starter lspconfig.lua (force)",
Kind: module.ActionFileWrite,
Exec: func(ctx context.Context) error {
err := os.Remove(filepath.Join(cfg, "lua", "configs", "lspconfig.lua"))
if err != nil && !os.IsNotExist(err) {
return err
}
return nil
},
})
renderedCopy := append([]byte{}, rendered...)
themeName := m.resolvedTheme()
actions = append(actions, module.Action{
Description: "rewrite chadrc.lua with theme " + themeName + " (force)",
Kind: module.ActionFileWrite,
Exec: func(ctx context.Context) error {
if err := os.MkdirAll(filepath.Join(cfg, "lua"), 0o755); err != nil {
return err
}
return platform.WriteUserFile(filepath.Join(cfg, "lua", "chadrc.lua"), renderedCopy, 0o644)
},
})
for _, a := range assetMap {
a := a
actions = append(actions, module.Action{
Description: "rewrite asset " + a.DstRel + " (force)",
Kind: module.ActionFileWrite,
Exec: func(ctx context.Context) error {
data, err := assets.ReadFile(a.SrcPath)
if err != nil {
return err
}
return platform.WriteUserFile(filepath.Join(cfg, a.DstRel), data, 0o644)
},
})
}
actions = append(actions, module.Action{
Description: "clear NvChad base46 theme cache (force)",
Kind: module.ActionFileWrite,
Exec: func(ctx context.Context) error {
home, _ := os.UserHomeDir()
_ = os.RemoveAll(filepath.Join(home, ".local", "share", "nvim", "lazy", "base46", "lua", "base46", "themes"))
_ = os.RemoveAll(filepath.Join(home, ".cache", "nvim", "base46"))
return nil
},
})
return module.Plan{Actions: actions}, nil
} }

View File

@@ -207,4 +207,93 @@ func TestWriteChadrcNoopWithoutConfigDir(t *testing.T) {
} }
} }
func TestPlanForceReinstallsAndRewrites(t *testing.T) {
home := t.TempDir()
t.Setenv("HOME", home)
// Pre-populate an "installed" NvChad clone with a local user tweak so we
// can verify PlanForce does NOT wipe the clone.
cfg := filepath.Join(home, ".config", "nvim")
os.MkdirAll(filepath.Join(cfg, "lua", "plugins"), 0o755)
os.WriteFile(filepath.Join(cfg, "lua", "plugins", "init.lua"),
[]byte("return {\n \"other\",\n},\n"), 0o644)
userTweak := filepath.Join(cfg, "lua", "user_tweak.lua")
os.WriteFile(userTweak, []byte("-- my custom stuff"), 0o644)
r := &fakeRunner{installed: map[string]bool{"neovim": true}}
m := New(config.NeovimConfig{NvChadStarterRepo: "https://x/starter"}, "dracula", platform.NewPacman(r), platform.GitClient{Runner: 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)
}
}
// User tweak must survive.
if _, err := os.Stat(userTweak); err != nil {
t.Fatalf("PlanForce wiped user NvChad modification: %v", err)
}
// No `git clone` call because starter init.lua was already present.
for _, c := range r.runCalls {
if c[0] == "git" && len(c) >= 2 && c[1] == "clone" {
t.Fatalf("PlanForce re-cloned existing NvChad: %v", c)
}
}
// pacman reinstall must have fired.
sawReinstall := false
for _, c := range r.runCalls {
if len(c) >= 4 && c[0] == "sudo" && c[1] == "pacman" && c[2] == "-S" && c[3] == "--noconfirm" {
sawReinstall = true
}
}
if !sawReinstall {
t.Fatalf("no pacman reinstall call; calls=%v", r.runCalls)
}
// chadrc.lua was rewritten with the requested theme.
chadrc, err := os.ReadFile(filepath.Join(cfg, "lua", "chadrc.lua"))
if err != nil {
t.Fatalf("chadrc missing after force: %v", err)
}
if !strings.Contains(string(chadrc), `theme = "dracula"`) {
t.Fatalf("chadrc theme not applied: %s", chadrc)
}
// assets rewritten
for _, name := range []string{"arduino.lua", "dap.lua"} {
if _, err := os.Stat(filepath.Join(cfg, "lua", "plugins", name)); err != nil {
t.Errorf("asset %s missing after force: %v", name, err)
}
}
}
func TestPlanForceClonesIfMissing(t *testing.T) {
home := t.TempDir()
t.Setenv("HOME", home)
r := &fakeRunner{installed: map[string]bool{"neovim": true}}
m := New(config.NeovimConfig{NvChadStarterRepo: "https://x/starter"}, "gruvbox", platform.NewPacman(r), platform.GitClient{Runner: r})
plan, err := m.PlanForce(context.Background())
if err != nil {
t.Fatalf("PlanForce: %v", err)
}
for _, a := range plan.Actions {
if err := a.Exec(context.Background()); err != nil {
t.Fatalf("Exec %q: %v", a.Description, err)
}
}
sawClone := false
for _, c := range r.runCalls {
if c[0] == "git" && len(c) >= 2 && c[1] == "clone" {
sawClone = true
}
}
if !sawClone {
t.Fatalf("expected git clone when NvChad absent; calls=%v", r.runCalls)
}
}
var _ module.Module = Module{} var _ module.Module = Module{}