feat: add replace subcommand with multi-pattern support

This commit is contained in:
Rogee
2025-10-29 17:46:54 +08:00
parent fa57af8a26
commit ceea09f7be
42 changed files with 1848 additions and 14 deletions

View File

@@ -0,0 +1,102 @@
package contract
import (
"bytes"
"context"
"os"
"path/filepath"
"strings"
"testing"
"github.com/rogeecn/renamer/internal/replace"
)
func TestPreviewSummaryCounts(t *testing.T) {
tmp := t.TempDir()
createFile(t, filepath.Join(tmp, "draft.txt"))
createFile(t, filepath.Join(tmp, "Draft.md"))
createFile(t, filepath.Join(tmp, "notes", "DRAFT.log"))
args := []string{"draft", "Draft", "DRAFT", "final"}
parsed, err := replace.ParseArgs(args)
if err != nil {
t.Fatalf("ParseArgs error: %v", err)
}
req := &replace.ReplaceRequest{
WorkingDir: tmp,
Patterns: parsed.Patterns,
Replacement: parsed.Replacement,
Recursive: true,
}
var buf bytes.Buffer
summary, planned, err := replace.Preview(context.Background(), req, parsed, &buf)
if err != nil {
t.Fatalf("Preview error: %v", err)
}
if summary.TotalCandidates == 0 {
t.Fatalf("expected candidates to be processed")
}
if summary.ChangedCount != len(planned) {
t.Fatalf("changed count mismatch: %d vs %d", summary.ChangedCount, len(planned))
}
for _, pattern := range []string{"draft", "Draft", "DRAFT"} {
if summary.PatternMatches[pattern] == 0 {
t.Fatalf("expected matches recorded for %s", pattern)
}
}
output := buf.String()
if !strings.Contains(output, "draft.txt -> final.txt") {
t.Fatalf("expected preview output to list replacements, got: %s", output)
}
if summary.ReplacementWasEmpty(parsed.Replacement) {
t.Fatalf("replacement should not be empty warning for this test")
}
}
func TestPreviewWarnsOnEmptyReplacement(t *testing.T) {
tmp := t.TempDir()
createFile(t, filepath.Join(tmp, "foo.txt"))
args := []string{"foo", ""}
parsed, err := replace.ParseArgs(args)
if err != nil {
t.Fatalf("ParseArgs error: %v", err)
}
req := &replace.ReplaceRequest{
WorkingDir: tmp,
Patterns: parsed.Patterns,
Replacement: parsed.Replacement,
}
var buf bytes.Buffer
summary, _, err := replace.Preview(context.Background(), req, parsed, &buf)
if err != nil {
t.Fatalf("Preview error: %v", err)
}
if !summary.EmptyReplacement {
t.Fatalf("expected empty replacement flag to be set")
}
if !strings.Contains(buf.String(), "Warning: replacement string is empty") {
t.Fatalf("expected empty replacement warning in preview output")
}
}
func createFile(t *testing.T, path string) {
t.Helper()
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
t.Fatalf("mkdir %s: %v", path, err)
}
if err := os.WriteFile(path, []byte("test"), 0o644); err != nil {
t.Fatalf("write file %s: %v", path, err)
}
}

View File

@@ -0,0 +1,9 @@
# Replace Command Fixtures
Use this directory to store sample file trees referenced by replace command tests. Keep fixtures
minimal and platform-neutral (ASCII names, small files). Create subdirectories per scenario, e.g.:
- `basic/` — Simple files demonstrating multi-pattern replacements.
- `conflicts/` — Cases that intentionally trigger name collisions for validation.
File contents are typically irrelevant; create empty files unless a test requires data.

View File

@@ -0,0 +1,101 @@
package integration
import (
"bytes"
"context"
"os"
"path/filepath"
"testing"
renamercmd "github.com/rogeecn/renamer/cmd"
"github.com/rogeecn/renamer/internal/history"
"github.com/rogeecn/renamer/internal/replace"
)
func TestReplaceApplyAndUndo(t *testing.T) {
tmp := t.TempDir()
createFile(t, filepath.Join(tmp, "foo_draft.txt"))
createFile(t, filepath.Join(tmp, "bar_draft.txt"))
parsed, err := replace.ParseArgs([]string{"draft", "final"})
if err != nil {
t.Fatalf("ParseArgs error: %v", err)
}
req := &replace.ReplaceRequest{
WorkingDir: tmp,
Patterns: parsed.Patterns,
Replacement: parsed.Replacement,
}
summary, planned, err := replace.Preview(context.Background(), req, parsed, nil)
if err != nil {
t.Fatalf("Preview error: %v", err)
}
if summary.ChangedCount != 2 {
t.Fatalf("expected 2 changes, got %d", summary.ChangedCount)
}
entry, err := replace.Apply(context.Background(), req, planned, summary)
if err != nil {
t.Fatalf("apply error: %v", err)
}
if len(entry.Operations) != 2 {
t.Fatalf("expected 2 operations recorded, got %d", len(entry.Operations))
}
if entry.Metadata == nil {
t.Fatalf("expected metadata to be recorded")
}
counts, ok := entry.Metadata["patterns"].(map[string]int)
if !ok {
t.Fatalf("patterns metadata missing or wrong type: %#v", entry.Metadata)
}
if counts["draft"] != 2 {
t.Fatalf("expected pattern count for 'draft' to be 2, got %d", counts["draft"])
}
if _, err := os.Stat(filepath.Join(tmp, "foo_final.txt")); err != nil {
t.Fatalf("expected renamed file exists: %v", err)
}
if _, err := os.Stat(filepath.Join(tmp, "bar_final.txt")); err != nil {
t.Fatalf("expected renamed file exists: %v", err)
}
_, err = history.Undo(tmp)
if err != nil {
t.Fatalf("undo error: %v", err)
}
if _, err := os.Stat(filepath.Join(tmp, "foo_draft.txt")); err != nil {
t.Fatalf("expected original file after undo: %v", err)
}
if _, err := os.Stat(filepath.Join(tmp, "bar_draft.txt")); err != nil {
t.Fatalf("expected original file after undo: %v", err)
}
}
func TestReplaceCommandInvalidArgs(t *testing.T) {
root := renamercmd.NewRootCommand()
var out bytes.Buffer
root.SetOut(&out)
root.SetErr(&out)
root.SetArgs([]string{"replace", "onlyone"})
err := root.Execute()
if err == nil {
t.Fatalf("expected error for insufficient arguments")
}
}
func createFile(t *testing.T, path string) {
t.Helper()
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
t.Fatalf("mkdir %s: %v", path, err)
}
if err := os.WriteFile(path, []byte("x"), 0o644); err != nil {
t.Fatalf("write file %s: %v", path, err)
}
}

View File

@@ -0,0 +1,31 @@
package replace_test
import (
"testing"
"github.com/rogeecn/renamer/internal/replace"
)
func TestParseArgsHandlesWhitespaceAndDuplicates(t *testing.T) {
args := []string{" draft ", "Draft", "draft", "final"}
result, err := replace.ParseArgs(args)
if err != nil {
t.Fatalf("ParseArgs returned error: %v", err)
}
if len(result.Patterns) != 2 {
t.Fatalf("expected 2 unique patterns, got %d", len(result.Patterns))
}
if len(result.Duplicates) != 1 {
t.Fatalf("expected duplicate reported, got %d", len(result.Duplicates))
}
if result.Replacement != "final" {
t.Fatalf("replacement mismatch: %s", result.Replacement)
}
}
func TestParseArgsRequiresSufficientTokens(t *testing.T) {
if _, err := replace.ParseArgs([]string{"onlyone"}); err == nil {
t.Fatalf("expected error when replacement missing")
}
}