Mirrors/audiobookshelf
Mirrors/audiobookshelf
1
0
Fork 0
mirror of https://github.com/advplyr/audiobookshelf.git synced 2025-01-08 00:08:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

[enhancement] Improve candidate sorting

Browse Source
This commit is contained in:
mikiher 2023-10-05 10:28:55 +00:00
parent 752bfffb11
commit 8979586404
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
server/finders/BookFinder.js
View File

@ -189,9 +189,11 @@ class BookFinder {
this.bookFinder = bookFinder this.bookFinder = bookFinder
this.candidates = new Set() this.candidates = new Set()
this.cleanAuthor = cleanAuthor this.cleanAuthor = cleanAuthor
this.priorities = {}
this.positions = {}
} }
add(title) { add(title, position = 0) {
const titleTransformers = [ const titleTransformers = [
[/([,:;_]| by ).*/g, ''], // Remove subtitle [/([,:;_]| by ).*/g, ''], // Remove subtitle
[/^\d+ | \d+$/g, ''], // Remove preceding/trailing numbers [/^\d+ | \d+$/g, ''], // Remove preceding/trailing numbers
@ -203,14 +205,22 @@ class BookFinder {
const cleanTitle = this.bookFinder.cleanTitleForCompares(title).trim() const cleanTitle = this.bookFinder.cleanTitleForCompares(title).trim()
if (!cleanTitle) return if (!cleanTitle) return
this.candidates.add(cleanTitle) this.candidates.add(cleanTitle)
this.priorities[cleanTitle] = 0
this.positions[cleanTitle] = position
let candidate = cleanTitle let candidate = cleanTitle
for (const transformer of titleTransformers) for (const transformer of titleTransformers)
candidate = candidate.replace(transformer[0], transformer[1]).trim() candidate = candidate.replace(transformer[0], transformer[1]).trim()
if (candidate) if (candidate != cleanTitle) {
if (candidate) {
this.candidates.add(candidate) this.candidates.add(candidate)
this.priorities[candidate] = 0
this.positions[candidate] = position
}
this.priorities[cleanTitle] = 1
}
} }
get size() { get size() {
@ -227,6 +237,12 @@ class BookFinder {
const onlyDigits = /^\d+$/ const onlyDigits = /^\d+$/
const includesOnlyDigitsDiff = !onlyDigits.test(b) - !onlyDigits.test(a) const includesOnlyDigitsDiff = !onlyDigits.test(b) - !onlyDigits.test(a)
if (includesOnlyDigitsDiff) return includesOnlyDigitsDiff if (includesOnlyDigitsDiff) return includesOnlyDigitsDiff
// transformed candidates receive higher priority
const priorityDiff = this.priorities[a] - this.priorities[b]
if (priorityDiff) return priorityDiff
// if same priorirty, prefer candidates that are closer to the beginning (e.g. titles before subtitles)
const positionDiff = this.positions[a] - this.positions[b]
if (positionDiff) return positionDiff
// Start with longer candidaets, as they are likely more specific // Start with longer candidaets, as they are likely more specific
const lengthDiff = b.length - a.length const lengthDiff = b.length - a.length
if (lengthDiff) return lengthDiff if (lengthDiff) return lengthDiff
@ -280,8 +296,8 @@ class BookFinder {
const cleanTitle = title.replace(/\[.*?\]|\(.*?\)|{.*?}/g, " - ") const cleanTitle = title.replace(/\[.*?\]|\(.*?\)|{.*?}/g, " - ")
// Split title into hypen-separated parts // Split title into hypen-separated parts
const titleParts = cleanTitle.split(/ - | -|- /) const titleParts = cleanTitle.split(/ - | -|- /)
for (const titlePart of titleParts) { for (const [position, titlePart] of titleParts.entries()) {
titleCandidates.add(titlePart) titleCandidates.add(titlePart, position)
} }
if (titleCandidates.size > 0) { if (titleCandidates.size > 0) {
titleCandidates = titleCandidates.getCandidates() titleCandidates = titleCandidates.getCandidates()