124 lines
2.9 KiB
Groovy
Executable File
124 lines
2.9 KiB
Groovy
Executable File
#!/usr/bin/env filebot -script
|
|
|
|
import static org.apache.commons.io.FileUtils.*
|
|
|
|
|
|
scaleFactor = [1, 2]
|
|
thumbnailSize = [48, 48]
|
|
|
|
|
|
void ls(f) {
|
|
if (f.exists()) {
|
|
log.info "$f (${byteCountToDisplaySize(f.length())})"
|
|
} else {
|
|
log.warning "[FILE NOT FOUND] $f"
|
|
}
|
|
}
|
|
|
|
|
|
File getOriginalPath(db, id) {
|
|
return _args.outputPath.resolve("images/${db}/original/poster/${id}.jpg")
|
|
}
|
|
|
|
|
|
File getThumbnailPath(db, id, scale) {
|
|
// e.g. 42.png or 42@2x.png
|
|
def n = scale == 1 ? id : id + '@' + scale + 'x'
|
|
|
|
return _args.outputPath.resolve("images/${db}/thumb/poster/${n}.png")
|
|
}
|
|
|
|
|
|
void createThumbnail(original, thumb, scale) {
|
|
thumb.dir.mkdirs()
|
|
|
|
def size = thumbnailSize*.multiply(scale).join('x')
|
|
execute 'convert', '-strip', original, '-thumbnail', size, '-gravity', 'center', '-background', 'transparent', '-extent', size, 'PNG8:' + thumb
|
|
}
|
|
|
|
|
|
void createIndexFile(db) {
|
|
def indexFile = _args.outputPath.resolve("images/${db}/thumb/poster/index.txt")
|
|
def index = indexFile.dir.listFiles()*.getNameWithoutExtension().findAll{ it ==~ /\d+/ }*.toInteger() as TreeSet
|
|
|
|
index.join('\n').saveAs(indexFile)
|
|
execute 'xz', indexFile, '--force', '--keep'
|
|
|
|
println "Index: ${index.size()}"
|
|
indexFile.dir.listFiles{ !it.image }.each{ ls it }
|
|
}
|
|
|
|
|
|
|
|
|
|
void build(ids, section, db, query) {
|
|
def files = []
|
|
|
|
ids.each{ id ->
|
|
def original = getOriginalPath(section, id)
|
|
|
|
if (original.length() == 0 && original.exists() && System.currentTimeMillis() - original.lastModified() > 90 * 24 * 60 * 60 * 1000) {
|
|
log.finest "[SKIP] $id"
|
|
return
|
|
}
|
|
|
|
scaleFactor.each{ scale ->
|
|
def thumb = getThumbnailPath(section, id, scale)
|
|
|
|
if (thumb.exists()) {
|
|
log.finest "[SKIP] $id @ $scale"
|
|
return
|
|
}
|
|
|
|
if (original.length() == 0 || !original.exists()) {
|
|
def artwork = retry(2, 60000) {
|
|
try {
|
|
return db.getArtwork(id, query, Locale.ENGLISH)
|
|
} catch (FileNotFoundException e) {
|
|
log.warning "[ARTWORK NOT FOUND] $e"
|
|
return null
|
|
}
|
|
}
|
|
|
|
artwork?.findResult{ a ->
|
|
return retry(2, 60000) {
|
|
try {
|
|
log.fine "Fetch $a"
|
|
return a.url.saveAs(original)
|
|
} catch (FileNotFoundException e) {
|
|
log.warning "[IMAGE NOT FOUND] $e"
|
|
return null
|
|
} finally {
|
|
ls original
|
|
}
|
|
}
|
|
}
|
|
|
|
// create empty placeholder if there is no artwork
|
|
if (original.length() == 0 || !original.exists()) {
|
|
original.createNewFile()
|
|
original.setLastModified(System.currentTimeMillis())
|
|
}
|
|
}
|
|
|
|
if (original.length() > 0 && !thumb.exists()) {
|
|
createThumbnail(original, thumb, scale)
|
|
files << thumb
|
|
|
|
ls thumb
|
|
}
|
|
}
|
|
}
|
|
|
|
if (files) {
|
|
createIndexFile(section)
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
build(MediaDetection.animeIndex.object.id as HashSet, 'anidb', AniDB, 'poster')
|
|
build(MediaDetection.seriesIndex.object.id as HashSet, 'thetvdb', TheTVDB, 'poster')
|
|
build(MediaDetection.movieIndex.object.tmdbId as HashSet, 'themoviedb', TheMovieDB, 'posters')
|