Merge pull request #51 from sgerner/codex/prowlarr-audiobook-source-label

separate Prowlarr audiobook results from ABB

Author: JeremiahM37

internal/search/prowlarr.go

@@ -167,7 +167,7 @@ func (p *Prowlarr) doSearch(ctx context.Context, params prowlarrSearchParams) ([
 	for _, item := range items {
 		source := "torrent"
 		if p.tab == "audiobook" {
-			source = "audiobook"
+			source = "prowlarr_audiobooks"
 		} else if p.tab == "manga" {
 			source = "prowlarr_manga"
 		}

internal/search/prowlarr_test.go

@@ -149,8 +149,8 @@ func TestProwlarr_SearchAudiobook(t *testing.T) {
 	}
 
 	for _, r := range results {
-		if r.Source != "audiobook" {
-			t.Errorf("expected source audiobook, got %s", r.Source)
+		if r.Source != "prowlarr_audiobooks" {
+			t.Errorf("expected source prowlarr_audiobooks, got %s", r.Source)
 		}
 	}
 }

internal/search/scorer.go

@@ -263,7 +263,7 @@ func extractFormatFromTitle(title string) string {
 // guessMediaType guesses the media type from the result source.
 func guessMediaType(result models.SearchResult) string {
 	switch result.Source {
-	case "audiobook":
+	case "audiobook", "prowlarr_audiobooks":
 		return "audiobook"
 	case "prowlarr_manga", "nyaa_manga", "mangadex", "annas_manga":
 		return "manga"

internal/search/scorer_test.go

@@ -128,6 +128,18 @@ func TestScoreResult_SizeAudiobook(t *testing.T) {
 	}
 }
 
+func TestScoreResult_SizeProwlarrAudiobookWithoutMediaType(t *testing.T) {
+	result := models.SearchResult{
+		Title:  "Test Audiobook",
+		Source: "prowlarr_audiobooks",
+		Size:   500e6, // 500MB
+	}
+	sb := ScoreResult(result, "test audiobook", "")
+	if sb.SizeScore != 10 {
+		t.Errorf("500MB prowlarr audiobook should score 10, got %f", sb.SizeScore)
+	}
+}
+
 func TestScoreResult_ConfidenceLevels(t *testing.T) {
 	tests := []struct {
 		name     string