package contract

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

	renamercmd "github.com/rogeecn/renamer/cmd"
	"github.com/rogeecn/renamer/internal/ai/genkit"
	"github.com/rogeecn/renamer/internal/ai/prompt"
)

type captureWorkflow struct {
	request genkit.Request
}

func (c *captureWorkflow) Run(ctx context.Context, req genkit.Request) (genkit.Result, error) {
	c.request = req
	return genkit.Result{
		Response: prompt.RenameResponse{
			Items: []prompt.RenameItem{
				{
					Original: "alpha.txt",
					Proposed: "proj_001_sample_file.txt",
					Sequence: 1,
				},
			},
			Model: "test-model",
		},
	}, nil
}

func TestAICommandUsesDefaultPoliciesInPrompt(t *testing.T) {
	genkit.ResetWorkflowFactory()
	stub := &captureWorkflow{}
	genkit.OverrideWorkflowFactory(func(ctx context.Context, opts genkit.Options) (genkit.WorkflowRunner, error) {
		return stub, nil
	})
	t.Cleanup(genkit.ResetWorkflowFactory)

	rootDir := t.TempDir()
	createPolicyTestFile(t, filepath.Join(rootDir, "alpha.txt"))

	rootCmd := renamercmd.NewRootCommand()
	var stdout, stderr bytes.Buffer
	rootCmd.SetOut(&stdout)
	rootCmd.SetErr(&stderr)
	rootCmd.SetArgs([]string{
		"ai",
		"--path", rootDir,
		"--dry-run",
	})

	if err := rootCmd.Execute(); err != nil {
		if stdout.Len() > 0 {
			t.Logf("stdout: %s", stdout.String())
		}
		if stderr.Len() > 0 {
			t.Logf("stderr: %s", stderr.String())
		}
		t.Fatalf("command execute: %v", err)
	}

	req := stub.request
	policies := req.Payload.Policies
	if policies.Prefix != "" {
		t.Fatalf("expected empty prefix, got %q", policies.Prefix)
	}
	if policies.Casing != "kebab" {
		t.Fatalf("expected default casing kebab, got %q", policies.Casing)
	}
	if policies.AllowSpaces {
		t.Fatalf("expected allow spaces default false")
	}
	if policies.KeepOriginalOrder {
		t.Fatalf("expected keep original order default false")
	}

	banned := req.Payload.BannedTerms
	containsDefault := false
	for _, term := range banned {
		if term == "clickbait" {
			containsDefault = true
			break
		}
	}
	if !containsDefault {
		t.Fatalf("expected default banned terms propagated, got %#v", banned)
	}
}

func createPolicyTestFile(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("demo"), 0o644); err != nil {
		t.Fatalf("write file %s: %v", path, err)
	}
}