Improve torrent search and ranking algorithm

Enhanced the ranking algorithm to better distinguish complete title matches from partial matches, reducing series confusion. Updated the torrent search modal to allow users to customize the search title for new requests, improving search flexibility. Also refined request lookup to ignore deleted requests.
This commit is contained in:
kikootwo
2025-12-24 02:52:29 -05:00
parent f043688a71
commit 1374e66f13
4 changed files with 80 additions and 12 deletions
@@ -96,11 +96,12 @@ export async function POST(request: NextRequest) {
});
}
// Check if user already has an active request for this audiobook
// Check if user already has an active (non-deleted) request for this audiobook
const existingRequest = await prisma.request.findFirst({
where: {
userId: req.user.id,
audiobookId: audiobookRecord.id,
deletedAt: null, // Only check active requests
},
});
@@ -43,6 +43,7 @@ export function InteractiveTorrentSearchModal({
const [results, setResults] = useState<(TorrentResult & { rank: number; qualityScore?: number })[]>([]);
const [confirmTorrent, setConfirmTorrent] = useState<TorrentResult | null>(null);
const [searchTitle, setSearchTitle] = useState(audiobook.title);
// Determine which mode we're in
const hasRequestId = !!requestId;
@@ -52,6 +53,12 @@ export function InteractiveTorrentSearchModal({
? (searchByRequestError || selectTorrentError)
: (searchByAudiobookError || requestWithTorrentError);
// Reset search title when modal opens/closes or audiobook changes
React.useEffect(() => {
setSearchTitle(audiobook.title);
setResults([]);
}, [isOpen, audiobook.title]);
// Perform search when modal opens
React.useEffect(() => {
if (isOpen && results.length === 0) {
@@ -60,14 +67,17 @@ export function InteractiveTorrentSearchModal({
}, [isOpen]);
const performSearch = async () => {
// Clear existing results while searching
setResults([]);
try {
let data;
if (hasRequestId) {
// Existing flow: search by requestId
// Existing flow: search by requestId (cannot customize search term)
data = await searchByRequestId(requestId);
} else {
// New flow: search by audiobook title/author
data = await searchByAudiobook(audiobook.title, audiobook.author);
// New flow: search by custom title + original author
data = await searchByAudiobook(searchTitle, audiobook.author);
}
setResults(data || []);
} catch (err) {
@@ -76,6 +86,12 @@ export function InteractiveTorrentSearchModal({
}
};
const handleSearchKeyPress = (e: React.KeyboardEvent<HTMLInputElement>) => {
if (e.key === 'Enter') {
performSearch();
}
};
const handleDownloadClick = (torrent: TorrentResult) => {
setConfirmTorrent(torrent);
};
@@ -124,10 +140,42 @@ export function InteractiveTorrentSearchModal({
<>
<Modal isOpen={isOpen} onClose={onClose} title="Select Torrent" size="full">
<div className="space-y-4">
{/* Audiobook info */}
{/* Search customization */}
<div className="bg-gray-50 dark:bg-gray-900 p-4 rounded-lg">
<h3 className="font-semibold text-gray-900 dark:text-gray-100">{audiobook.title}</h3>
<p className="text-sm text-gray-600 dark:text-gray-400">By {audiobook.author}</p>
{hasRequestId ? (
// Existing request: show static title (cannot customize)
<>
<h3 className="font-semibold text-gray-900 dark:text-gray-100">{audiobook.title}</h3>
<p className="text-sm text-gray-600 dark:text-gray-400">By {audiobook.author}</p>
</>
) : (
// New search: allow title customization
<>
<label className="block text-sm font-medium text-gray-700 dark:text-gray-300 mb-2">
Search Title
</label>
<div className="flex gap-2">
<input
type="text"
value={searchTitle}
onChange={(e) => setSearchTitle(e.target.value)}
onKeyPress={handleSearchKeyPress}
placeholder="Enter book title to search..."
disabled={isSearching}
className="flex-1 px-3 py-2 border border-gray-300 dark:border-gray-600 rounded-md shadow-sm focus:outline-none focus:ring-blue-500 focus:border-blue-500 bg-white dark:bg-gray-800 text-gray-900 dark:text-gray-100 disabled:opacity-50"
/>
<Button
onClick={performSearch}
disabled={isSearching || !searchTitle.trim()}
variant="primary"
size="sm"
>
Search
</Button>
</div>
<p className="text-sm text-gray-600 dark:text-gray-400 mt-2">By {audiobook.author}</p>
</>
)}
</div>
{/* Error message */}
+19 -3
View File
@@ -201,10 +201,26 @@ export class RankingAlgorithm {
// Title matching (0-35 points)
let titleScore = 0;
if (torrentTitle.includes(requestTitle)) {
// Exact substring match → full points
titleScore = 35;
// Found the title, but is it the complete title or part of a longer one?
const titleIndex = torrentTitle.indexOf(requestTitle);
const afterTitle = torrentTitle.substring(titleIndex + requestTitle.length);
// Title is complete if followed by clear metadata markers
// (not followed by more title words like "'s Secret" or " Is Watching")
const metadataMarkers = [' by ', ' - ', ' [', ' (', ' {', ' :', ','];
const isCompleteTitle = afterTitle === '' ||
metadataMarkers.some(marker => afterTitle.startsWith(marker));
if (isCompleteTitle) {
// Complete title match → full points
titleScore = 35;
} else {
// Title continues with more words (e.g., "The Housemaid" + "'s Secret")
// This is likely a different book in a series → use fuzzy similarity
titleScore = compareTwoStrings(requestTitle, torrentTitle) * 35;
}
} else {
// No exact match → use fuzzy similarity for partial credit
// No substring match at all → use fuzzy similarity
titleScore = compareTwoStrings(requestTitle, torrentTitle) * 35;
}