package artifactpolicy

import (
	"strings"
	"testing"
)

func TestClassifySource(t *testing.T) {
	tests := []struct {
		name        string
		source      string
		wantKind    SourceKind
		wantKey     string
		wantErrLike string
	}{
		{name: "built in transcript", source: "narratio.transcript.final_trimmed", wantKind: SourceKindBuiltIn},
		{name: "built in bounds", source: "narratio.bounds.session", wantKind: SourceKindBuiltIn},
		{name: "configured artifact", source: "narratio.artifact.session_recap", wantKind: SourceKindConfiguredArtifact, wantKey: "session_recap"},
		{name: "previous session configured", source: "narratio.previous_session.artifact.session_recap", wantKind: SourceKindPreviousArtifact, wantKey: "session_recap"},
		{name: "unsupported", source: "narratio.unknown", wantErrLike: "unsupported artifact source"},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := ClassifySource(tt.source)
			if tt.wantErrLike != "" {
				if err == nil || !strings.Contains(err.Error(), tt.wantErrLike) {
					t.Fatalf("ClassifySource() error = %v, want like %q", err, tt.wantErrLike)
				}
				return
			}
			if err != nil {
				t.Fatalf("ClassifySource() error = %v", err)
			}
			if got.Kind != tt.wantKind {
				t.Fatalf("ClassifySource().Kind = %q, want %q", got.Kind, tt.wantKind)
			}
			if got.ConfiguredKey != tt.wantKey {
				t.Fatalf("ClassifySource().ConfiguredKey = %q, want %q", got.ConfiguredKey, tt.wantKey)
			}
		})
	}
}

func TestValidatePublishSource(t *testing.T) {
	configured := map[string]string{"session_recap": "artifacts/session_recap.md"}
	if _, err := ValidatePublishSource("narratio.artifact.session_recap", configured); err != nil {
		t.Fatalf("ValidatePublishSource(configured) error = %v", err)
	}
	if _, err := ValidatePublishSource("narratio.previous_session.artifact.session_recap", configured); err == nil {
		t.Fatal("ValidatePublishSource(previous) error = nil, want error")
	}
	if _, err := ValidatePublishSource("narratio.artifact.missing", configured); err == nil {
		t.Fatal("ValidatePublishSource(missing configured) error = nil, want error")
	}
}

func TestResolvePublishedDestination(t *testing.T) {
	configured := map[string]string{"session_recap": "artifacts/session_recap.md"}

	got, err := ResolvePublishedDestination("narratio.transcript.final_trimmed", "", configured)
	if err != nil {
		t.Fatalf("ResolvePublishedDestination(built-in) error = %v", err)
	}
	if got != "transcripts/final.trimmed.json" {
		t.Fatalf("built-in destination = %q, want transcripts/final.trimmed.json", got)
	}

	got, err = ResolvePublishedDestination("narratio.artifact.session_recap", "", configured)
	if err != nil {
		t.Fatalf("ResolvePublishedDestination(configured) error = %v", err)
	}
	if got != "artifacts/session_recap.md" {
		t.Fatalf("configured destination = %q, want artifacts/session_recap.md", got)
	}

	got, err = ResolvePublishedDestination("narratio.transcript.final_trimmed", "published/../published/final.json", configured)
	if err != nil {
		t.Fatalf("ResolvePublishedDestination(explicit) error = %v", err)
	}
	if got != "published/final.json" {
		t.Fatalf("explicit destination = %q, want published/final.json", got)
	}
}

func TestResolvePublishedDestinationRejectsTraversal(t *testing.T) {
	configured := map[string]string{"session_recap": "artifacts/session_recap.md"}
	_, err := ResolvePublishedDestination("narratio.transcript.final_trimmed", "../escape.txt", configured)
	if err == nil {
		t.Fatal("ResolvePublishedDestination() error = nil, want traversal rejection")
	}
}