torrentclaw/torrentclaw-mcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(mcp): add season filtering and presentation guide for better UX

Browse source
This commit is contained in:
Deivid Soto 2026-02-12 18:44:22 +01:00
parent b6f0af707c
commit d48b91f554
9 changed files with 941 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

372
tests/formatters/content.test.ts
View file

@ -745,9 +745,27 @@ describe("formatSearchResults", () => {
season: null,
episode: null,
audioTracks: [
{ lang: "en", codec: "aac", channels: "5.1", title: "English", default: true },
{ lang: "es", codec: "aac", channels: "5.1", title: "Spanish", default: false },
{ lang: "en", codec: "ac3", channels: "2.0", title: "Commentary", default: false },
{
lang: "en",
codec: "aac",
channels: "5.1",
title: "English",
default: true,
},
{
lang: "es",
codec: "aac",
channels: "5.1",
title: "Spanish",
default: false,
},
{
lang: "en",
codec: "ac3",
channels: "2.0",
title: "Commentary",
default: false,
},
],
subtitleTracks: null,
videoInfo: null,
@ -809,7 +827,13 @@ describe("formatSearchResults", () => {
season: null,
episode: null,
audioTracks: [
{ lang: null, codec: null, channels: null, title: null, default: null },
{
lang: null,
codec: null,
channels: null,
title: null,
default: null,
},
],
subtitleTracks: null,
videoInfo: null,
@ -1021,6 +1045,346 @@ describe("formatSearchResults", () => {
expect(text).toContain("Stream: Netflix, Disney+");
expect(text).toContain("Free: Tubi");
});
it("filters torrents by season when specified", () => {
const response: SearchResponse = {
total: 1,
page: 1,
pageSize: 10,
results: [
{
id: 1,
imdbId: null,
tmdbId: null,
contentType: "show",
title: "Test Show",
titleOriginal: null,
year: 2024,
overview: null,
posterUrl: null,
backdropUrl: null,
genres: null,
ratingImdb: null,
ratingTmdb: null,
contentUrl: null,
hasTorrents: true,
torrents: [
{
infoHash: "a".repeat(40),
rawTitle: null,
quality: "1080p",
codec: "x264",
sourceType: "WEB-DL",
sizeBytes: "1073741824",
seeders: 100,
leechers: 5,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 90,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 1,
episode: null,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
{
infoHash: "b".repeat(40),
rawTitle: null,
quality: "720p",
codec: "x264",
sourceType: "WEB-DL",
sizeBytes: "536870912",
seeders: 50,
leechers: 2,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 70,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 4,
episode: null,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
{
infoHash: "c".repeat(40),
rawTitle: null,
quality: "1080p",
codec: "x265",
sourceType: "WEB-DL",
sizeBytes: "805306368",
seeders: 75,
leechers: 3,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 85,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 4,
episode: null,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
],
},
],
};
// Filter by season 4
const text = formatSearchResults(response, { season: 4 });
expect(text).toContain("2 matching, 3 total");
// Should show season 4 torrents (quality scores 70 and 85)
expect(text).toContain("Score: 85");
expect(text).toContain("Score: 70");
// Should NOT show season 1 torrent (quality score 90)
expect(text).not.toContain("Score: 90");
});
it("filters torrents by season and episode when both specified", () => {
const response: SearchResponse = {
total: 1,
page: 1,
pageSize: 10,
results: [
{
id: 1,
imdbId: null,
tmdbId: null,
contentType: "show",
title: "Test Show",
titleOriginal: null,
year: 2024,
overview: null,
posterUrl: null,
backdropUrl: null,
genres: null,
ratingImdb: null,
ratingTmdb: null,
contentUrl: null,
hasTorrents: true,
torrents: [
{
infoHash: "a".repeat(40),
rawTitle: null,
quality: "1080p",
codec: "x264",
sourceType: "WEB-DL",
sizeBytes: "1073741824",
seeders: 100,
leechers: 5,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 90,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 2,
episode: 5,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
{
infoHash: "b".repeat(40),
rawTitle: null,
quality: "720p",
codec: "x264",
sourceType: "WEB-DL",
sizeBytes: "536870912",
seeders: 50,
leechers: 2,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 70,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 2,
episode: 3,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
],
},
],
};
// Filter by season 2, episode 5
const text = formatSearchResults(response, { season: 2, episode: 5 });
expect(text).toContain("1 matching, 2 total");
// Should show only S02E05 torrent
expect(text).toContain("Score: 90");
expect(text).toContain("S02E05");
// Should NOT show S02E03
expect(text).not.toContain("Score: 70");
});
it("shows message when no torrents match the season filter", () => {
const response: SearchResponse = {
total: 1,
page: 1,
pageSize: 10,
results: [
{
id: 1,
imdbId: null,
tmdbId: null,
contentType: "show",
title: "Test Show",
titleOriginal: null,
year: 2024,
overview: null,
posterUrl: null,
backdropUrl: null,
genres: null,
ratingImdb: null,
ratingTmdb: null,
contentUrl: null,
hasTorrents: true,
torrents: [
{
infoHash: "a".repeat(40),
rawTitle: null,
quality: "1080p",
codec: "x264",
sourceType: "WEB-DL",
sizeBytes: "1073741824",
seeders: 100,
leechers: 5,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 90,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 1,
episode: null,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
],
},
],
};
// Filter by season 4 (not available)
const text = formatSearchResults(response, { season: 4 });
expect(text).toContain("No torrents available for season 4");
expect(text).toContain("1 torrents available for other seasons");
});
it("shows message when no torrents match the season+episode filter", () => {
const response: SearchResponse = {
total: 1,
page: 1,
pageSize: 10,
results: [
{
id: 1,
imdbId: null,
tmdbId: null,
contentType: "show",
title: "Test Show",
titleOriginal: null,
year: 2024,
overview: null,
posterUrl: null,
backdropUrl: null,
genres: null,
ratingImdb: null,
ratingTmdb: null,
contentUrl: null,
hasTorrents: true,
torrents: [
{
infoHash: "a".repeat(40),
rawTitle: null,
quality: "1080p",
codec: "x264",
sourceType: "WEB-DL",
sizeBytes: "1073741824",
seeders: 100,
leechers: 5,
magnetUrl: null,
torrentUrl: null,
source: "test",
qualityScore: 90,
uploadedAt: null,
languages: [],
audioCodec: null,
hdrType: null,
releaseGroup: null,
isProper: null,
isRepack: null,
isRemastered: null,
season: 2,
episode: 5,
audioTracks: null,
subtitleTracks: null,
videoInfo: null,
scanStatus: null,
},
],
},
],
};
// Filter by S04E01 (not available)
const text = formatSearchResults(response, { season: 4, episode: 1 });
expect(text).toContain("No torrents available for S04E01");
expect(text).toContain("1 torrents available for other seasons");
});
});
describe("formatPopularResults", () => {

5
tests/prompts.test.ts
View file

@ -26,10 +26,11 @@ describe("registerPrompts", () => {
return { server, prompts };
}
it("registers 4 prompts", () => {
it("registers 5 prompts", () => {
const { server, prompts } = createMockServer();
registerPrompts(server);
expect(prompts.size).toBe(4);
expect(prompts.size).toBe(5);
expect(prompts.has("presentation_guide")).toBe(true);
expect(prompts.has("search_movie")).toBe(true);
expect(prompts.has("search_show")).toBe(true);
expect(prompts.has("whats_new")).toBe(true);

78
tests/resources/presentation-guide.test.ts Normal file
View file

@ -0,0 +1,78 @@
import { describe, it, expect } from "vitest";
import { createMockServer } from "../helpers.js";
import { registerPresentationGuideResource } from "../../src/resources/presentation-guide.js";
describe("presentation-guide resource", () => {
it("returns markdown guide with best practices", async () => {
const { server, getResourceHandler } = createMockServer();
registerPresentationGuideResource(server);
const handler = getResourceHandler("torrentclaw://presentation-guide");
const result = await handler({ href: "torrentclaw://presentation-guide" });
expect(result.contents).toHaveLength(1);
expect(result.contents[0].mimeType).toBe("text/markdown");
const text = result.contents[0].text;
// Check for key sections
expect(text).toContain("# TorrentClaw Results Presentation Guide");
expect(text).toContain("## Critical Requirements");
expect(text).toContain("### 1. Clickable Magnet Links");
expect(text).toContain("### 2. Content URL for Browsing");
expect(text).toContain("### 3. User-Friendly Presentation Format");
// Check for markdown examples
expect(text).toContain("[📥 Download](magnet:");
expect(text).toContain("[🔗 View");
// Check for good vs bad examples
expect(text).toContain("❌ BAD");
expect(text).toContain("✅ GOOD");
// Check for warnings about seeders
expect(text).toContain("⚠️ No active seeders");
expect(text).toContain("⭐ Recommended");
});
it("provides guidance for TV shows", async () => {
const { server, getResourceHandler } = createMockServer();
registerPresentationGuideResource(server);
const handler = getResourceHandler("torrentclaw://presentation-guide");
const result = await handler({ href: "torrentclaw://presentation-guide" });
const text = result.contents[0].text;
expect(text).toContain("**For TV Shows**");
expect(text).toContain("S04E01");
expect(text).toContain("Entrevías");
});
it("provides guidance for movies", async () => {
const { server, getResourceHandler } = createMockServer();
registerPresentationGuideResource(server);
const handler = getResourceHandler("torrentclaw://presentation-guide");
const result = await handler({ href: "torrentclaw://presentation-guide" });
const text = result.contents[0].text;
expect(text).toContain("**For Movies**");
expect(text).toContain("Inception");
expect(text).toContain("BluRay");
});
it("warns against bad practices", async () => {
const { server, getResourceHandler } = createMockServer();
registerPresentationGuideResource(server);
const handler = getResourceHandler("torrentclaw://presentation-guide");
const result = await handler({ href: "torrentclaw://presentation-guide" });
const text = result.contents[0].text;
// Check for warnings
expect(text).toContain("### 5. What NOT to Do");
expect(text).toContain("without clickable links");
expect(text).toContain("truncated magnet links");
expect(text).toContain("omit the content URL");
});
});