package engine

import (
	"os"
	"path/filepath"
	"testing"
)

func TestOrganizeDisabled(t *testing.T) {
	r := &Result{FilePath: "/tmp/file.mkv", FileName: "file.mkv"}
	task := &Task{Title: "Movie"}
	path, err := organize(r, task, OrganizeConfig{Enabled: false})
	if err != nil {
		t.Fatal(err)
	}
	if path != "/tmp/file.mkv" {
		t.Errorf("path = %q, want original path when disabled", path)
	}
}

func TestOrganizeMovie(t *testing.T) {
	tmp := t.TempDir()
	srcDir := filepath.Join(tmp, "src")
	os.MkdirAll(srcDir, 0o755)
	srcFile := filepath.Join(srcDir, "Movie.2023.1080p.mkv")
	os.WriteFile(srcFile, []byte("data"), 0o644)

	moviesDir := filepath.Join(tmp, "Movies")

	r := &Result{FilePath: srcFile, FileName: "Movie.2023.1080p.mkv"}
	task := &Task{Title: "Movie 2023"}

	path, err := organize(r, task, OrganizeConfig{
		Enabled:   true,
		MoviesDir: moviesDir,
	})
	if err != nil {
		t.Fatal(err)
	}

	// Should be in Movies/Movie (2023)/
	if path == srcFile {
		t.Error("file should have moved")
	}
	if _, err := os.Stat(path); err != nil {
		t.Errorf("organized file should exist at %s: %v", path, err)
	}
}

func TestOrganizeTVShow(t *testing.T) {
	tmp := t.TempDir()
	srcFile := filepath.Join(tmp, "Show.S02E05.1080p.mkv")
	os.WriteFile(srcFile, []byte("data"), 0o644)

	tvDir := filepath.Join(tmp, "TV Shows")

	r := &Result{FilePath: srcFile, FileName: "Show.S02E05.1080p.mkv"}
	task := &Task{Title: "Show S02E05"}

	path, err := organize(r, task, OrganizeConfig{
		Enabled:    true,
		TVShowsDir: tvDir,
	})
	if err != nil {
		t.Fatal(err)
	}

	// Should detect season from filename S02
	if _, err := os.Stat(path); err != nil {
		t.Errorf("organized file should exist at %s: %v", path, err)
	}
}

func TestCleanTitle(t *testing.T) {
	tests := []struct {
		input string
		want  string
	}{
		{"The Matrix (1999)", "The Matrix"},
		{"Oppenheimer 2023 1080p BluRay", "Oppenheimer 2023"},
		{"Movie", "Movie"},
	}

	for _, tt := range tests {
		t.Run(tt.input, func(t *testing.T) {
			got := cleanTitle(tt.input)
			if got != tt.want {
				t.Errorf("cleanTitle(%q) = %q, want %q", tt.input, got, tt.want)
			}
		})
	}
}