package contract

import (
	"bytes"
	"context"
	"errors"
	"os"
	"path/filepath"
	"strings"
	"testing"

	renamercmd "github.com/rogeecn/renamer/cmd"
	"github.com/rogeecn/renamer/internal/extension"
	"github.com/rogeecn/renamer/internal/listing"
)

func TestExtensionApplyMetadataCaptured(t *testing.T) {
	t.Parallel()

	tmp := t.TempDir()
	writeTestFile(t, filepath.Join(tmp, "clip.jpeg"))
	writeTestFile(t, filepath.Join(tmp, "poster.JPG"))
	writeTestFile(t, filepath.Join(tmp, "flyer.jpg"))

	scope := &listing.ListingRequest{
		WorkingDir:         tmp,
		IncludeDirectories: false,
		Recursive:          false,
		IncludeHidden:      false,
		Extensions:         nil,
		Format:             listing.FormatTable,
	}
	if err := scope.Validate(); err != nil {
		t.Fatalf("validate scope: %v", err)
	}

	req := extension.NewRequest(scope)
	req.SetExecutionMode(true, false)

	parsed, err := extension.ParseArgs([]string{".jpeg", ".JPG", ".jpg"})
	if err != nil {
		t.Fatalf("parse args: %v", err)
	}
	req.SetExtensions(parsed.SourcesCanonical, parsed.SourcesDisplay, parsed.Target)
	req.SetWarnings(parsed.Duplicates, parsed.NoOps)

	summary, planned, err := extension.Preview(context.Background(), req, nil)
	if err != nil {
		t.Fatalf("preview error: %v", err)
	}
	req.SetExecutionMode(false, true)

	entry, err := extension.Apply(context.Background(), req, planned, summary)
	if err != nil {
		t.Fatalf("apply error: %v", err)
	}

	if entry.Metadata == nil {
		t.Fatalf("expected metadata to be recorded")
	}

	sourcesMeta, ok := entry.Metadata["sourceExtensions"].([]string)
	if !ok || len(sourcesMeta) != len(parsed.SourcesDisplay) {
		t.Fatalf("sourceExtensions metadata mismatch: %#v", entry.Metadata["sourceExtensions"])
	}
	if sourcesMeta[0] != parsed.SourcesDisplay[0] || sourcesMeta[1] != parsed.SourcesDisplay[1] {
		t.Fatalf("expected display sources preserved in metadata, got %v", sourcesMeta)
	}

	targetMeta, ok := entry.Metadata["targetExtension"].(string)
	if !ok || targetMeta != parsed.Target {
		t.Fatalf("targetExtension metadata mismatch: %v", targetMeta)
	}

	if changed, ok := entry.Metadata["totalChanged"].(int); !ok || changed != summary.TotalChanged {
		t.Fatalf("totalChanged metadata mismatch: %v", entry.Metadata["totalChanged"])
	}
	if noChange, ok := entry.Metadata["noChange"].(int); !ok || noChange != summary.NoChange {
		t.Fatalf("noChange metadata mismatch: %v", entry.Metadata["noChange"])
	}

	counts, ok := entry.Metadata["perExtensionCounts"].(map[string]int)
	if !ok {
		t.Fatalf("perExtensionCounts metadata missing: %#v", entry.Metadata["perExtensionCounts"])
	}
	if counts[".jpeg"] == 0 || counts[".jpg"] == 0 {
		t.Fatalf("expected counts for .jpeg and .jpg, got %#v", counts)
	}

	scopeMeta, ok := entry.Metadata["scope"].(map[string]any)
	if !ok {
		t.Fatalf("scope metadata missing: %#v", entry.Metadata["scope"])
	}
	if includeHidden, _ := scopeMeta["includeHidden"].(bool); includeHidden {
		t.Fatalf("includeHidden should be false, got %v", includeHidden)
	}

	if _, ok := entry.Metadata["warnings"]; ok {
		t.Fatalf("did not expect warnings metadata when no duplicates were provided: %#v", entry.Metadata["warnings"])
	}

	ledger := filepath.Join(tmp, ".renamer")
	if _, err := os.Stat(ledger); err != nil {
		t.Fatalf("ledger not created: %v", err)
	}

	if err := os.Remove(ledger); err != nil {
		t.Fatalf("cleanup ledger: %v", err)
	}
}

func TestExtensionCommandExitCodes(t *testing.T) {
	t.Parallel()

	tmp := t.TempDir()

	var out bytes.Buffer
	cmd := renamercmd.NewRootCommand()
	cmd.SetOut(&out)
	cmd.SetErr(&out)
	cmd.SetArgs([]string{"extension", ".jpeg", ".jpg", "--dry-run", "--path", tmp})

	if err := cmd.Execute(); err != nil {
		t.Fatalf("expected dry-run to exit successfully, err=%v output=%s", err, out.String())
	}

	if !strings.Contains(out.String(), "No candidates found.") {
		t.Fatalf("expected no candidates notice, output=%s", out.String())
	}

	writeTestFile(t, filepath.Join(tmp, "clip.jpeg"))

	out.Reset()
	cmd = renamercmd.NewRootCommand()
	cmd.SetOut(&out)
	cmd.SetErr(&out)
	cmd.SetArgs([]string{"extension", ".jpeg", ".jpg", "--yes", "--path", tmp})

	if err := cmd.Execute(); err != nil {
		t.Fatalf("expected apply to exit successfully, err=%v output=%s", err, out.String())
	}

	if _, err := os.Stat(filepath.Join(tmp, "clip.jpg")); err != nil {
		t.Fatalf("expected clip.jpg after apply: %v", err)
	}

	if _, err := os.Stat(filepath.Join(tmp, "clip.jpeg")); !errors.Is(err, os.ErrNotExist) {
		t.Fatalf("expected clip.jpeg to be renamed, err=%v", err)
	}
}