Skip to content

Add season and episode search (#x##) TorznabApi.kt #94

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
MrOrange9-JCT wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Add season and episode search (#x##) TorznabApi.kt #94

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
dcc899b
Add season and episode search (#x##) TorznabApi.kt
MrOrange9-JCT Sep 28, 2024
4abdf5e
Modified episode search to only work with t=tvsearch
MrOrange9-JCT Sep 28, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 20 additions & 6 deletions app/src/main/kotlin/amarr/torznab/TorznabApi.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,30 +36,44 @@ private suspend fun ApplicationCall.handleRequests(indexer: Indexer) {
xmlDeclMode = XmlDeclMode.Charset
xmlVersion = XmlVersion.XML10
} // This API uses XML instead of JSON

request.queryParameters["t"]?.let {
when (it) {
"caps" -> {
application.log.debug("Handling caps request")
respondText(xmlFormat.encodeToString(indexer.capabilities()), contentType = ContentType.Application.Xml)
}

"tvsearch" -> performSearch(indexer, xmlFormat)
"search" -> performSearch(indexer, xmlFormat)
"tvsearch" -> performSearch(indexer, xmlFormat, isTvSearch = true)

"movie" -> performSearch(indexer, xmlFormat, isTvSearch = false)

else -> throw IllegalArgumentException("Unknown action: $it")
}
} ?: throw IllegalArgumentException("Missing action")
}

private suspend fun ApplicationCall.performSearch(indexer: Indexer, xmlFormat: XML) {
private suspend fun ApplicationCall.performSearch(indexer: Indexer, xmlFormat: XML, isTvSearch: Boolean) {
val query = request.queryParameters["q"].orEmpty()
val finalQuery = if (isTvSearch) {
// Handle season and episode
val season = request.queryParameters["season"].orEmpty()
val episode = request.queryParameters["episode"].orEmpty().padStart(2, '0')
"$query ${season}x$episode"
} else {
// Only use the query for movies
query
}

val offset = request.queryParameters["offset"]?.toIntOrNull() ?: 0
val limit = request.queryParameters["limit"]?.toIntOrNull() ?: 100
val cat = request.queryParameters["cat"]?.split(",")?.map { cat -> cat.toInt() } ?: emptyList()
application.log.debug("Handling search request: {}, {}, {}, {}", query, offset, limit, cat)

application.log.debug("Handling search request: {}, {}, {}, {}", finalQuery, offset, limit, cat)

try {
respondText(
xmlFormat.encodeToString(indexer.search(query, offset, limit, cat)),
xmlFormat.encodeToString(indexer.search(finalQuery, offset, limit, cat)),
contentType = ContentType.Application.Xml
)
} catch (e: ThrottledException) {
Expand All @@ -69,4 +83,4 @@ private suspend fun ApplicationCall.performSearch(indexer: Indexer, xmlFormat: X
application.log.warn("Unauthorized, returning 401")
respondText("Unauthorized, check your credentials.", status = HttpStatusCode.Unauthorized)
}
}
}