-import * as request from 'request'
-import * as asyncLRU from 'async-lru'
import { join } from 'path'
-import { createWriteStream } from 'fs'
-import * as Promise from 'bluebird'
+import { CACHE, CONFIG, STATIC_PATHS } from '../../initializers'
+import { VideoModel } from '../../models/video/video'
+import { AbstractVideoStaticFileCache } from './abstract-video-static-file-cache'
-import { database as db, CONFIG, CACHE } from '../../initializers'
-import { logger, writeFilePromise, unlinkPromise } from '../../helpers'
-import { VideoInstance } from '../../models'
-import { fetchRemotePreview } from '../../lib'
-
-class VideosPreviewCache {
+class VideosPreviewCache extends AbstractVideoStaticFileCache <string> {
private static instance: VideosPreviewCache
- private lru
-
- private constructor () { }
+ private constructor () {
+ super()
+ }
static get Instance () {
return this.instance || (this.instance = new this())
}
- init (max: number) {
- this.lru = new asyncLRU({
- max,
- load: (key, cb) => {
- this.loadPreviews(key)
- .then(res => cb(null, res))
- .catch(err => cb(err))
- }
- })
+ async getFilePath (videoUUID: string) {
+ const video = await VideoModel.loadByUUIDWithFile(videoUUID)
+ if (!video) return undefined
- this.lru.on('evict', (obj: { key: string, value: string }) => {
- unlinkPromise(obj.value).then(() => logger.debug('%s evicted from VideosPreviewCache', obj.value))
- })
- }
+ if (video.isOwned()) return join(CONFIG.STORAGE.PREVIEWS_DIR, video.getPreviewName())
- getPreviewPath (key: string) {
- return new Promise<string>((res, rej) => {
- this.lru.get(key, (err, value) => {
- err ? rej(err) : res(value)
- })
- })
+ return this.loadFromLRU(videoUUID)
}
- private loadPreviews (key: string) {
- return db.Video.loadByUUIDAndPopulateAuthorAndPodAndTags(key)
- .then(video => {
- if (!video) return undefined
-
- if (video.isOwned()) return join(CONFIG.STORAGE.PREVIEWS_DIR, video.getPreviewName())
-
- return this.saveRemotePreviewAndReturnPath(video)
- })
- }
+ protected async loadRemoteFile (key: string) {
+ const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(key)
+ if (!video) return undefined
- private saveRemotePreviewAndReturnPath (video: VideoInstance) {
- const req = fetchRemotePreview(video.Author.Pod, video)
+ if (video.isOwned()) throw new Error('Cannot load remote preview of owned video.')
- return new Promise<string>((res, rej) => {
- const path = join(CACHE.DIRECTORIES.PREVIEWS, video.getPreviewName())
- const stream = createWriteStream(path)
+ const remoteStaticPath = join(STATIC_PATHS.PREVIEWS, video.getPreviewName())
+ const destPath = join(CACHE.PREVIEWS.DIRECTORY, video.getPreviewName())
- req.pipe(stream)
- .on('finish', () => res(path))
- .on('error', (err) => rej(err))
- })
+ return this.saveRemoteVideoFileAndReturnPath(video, remoteStaticPath, destPath)
}
}