-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 { database as db, CONFIG, CACHE } from '../../initializers'
-import { logger, writeFilePromise, unlinkPromise } from '../../helpers'
-import { VideoInstance } from '../../models'
-import { fetchRemotePreview } from '../../lib'
+import { join } from 'path'
+import { unlinkPromise } from '../../helpers/core-utils'
+import { logger } from '../../helpers/logger'
+import { CACHE, CONFIG } from '../../initializers'
+import { VideoModel } from '../../models/video/video'
+import { fetchRemoteVideoPreview } from '../activitypub'
class VideosPreviewCache {
})
}
- getPreviewPath (key: string) {
+ async getPreviewPath (key: string) {
+ const video = await VideoModel.loadByUUID(key)
+ if (!video) return undefined
+
+ if (video.isOwned()) return join(CONFIG.STORAGE.PREVIEWS_DIR, video.getPreviewName())
+
return new Promise<string>((res, rej) => {
this.lru.get(key, (err, value) => {
err ? rej(err) : res(value)
})
}
- private loadPreviews (key: string) {
- return db.Video.loadByUUIDAndPopulateAuthorAndPodAndTags(key)
- .then(video => {
- if (!video) return undefined
+ private async loadPreviews (key: string) {
+ const video = await VideoModel.loadByUUIDAndPopulateAccountAndServerAndTags(key)
+ if (!video) return undefined
- if (video.isOwned()) return join(CONFIG.STORAGE.PREVIEWS_DIR, video.getPreviewName())
+ if (video.isOwned()) throw new Error('Cannot load preview of owned video.')
- return this.saveRemotePreviewAndReturnPath(video)
- })
+ return this.saveRemotePreviewAndReturnPath(video)
}
- private saveRemotePreviewAndReturnPath (video: VideoInstance) {
- const req = fetchRemotePreview(video.Author.Pod, video)
-
+ private saveRemotePreviewAndReturnPath (video: VideoModel) {
return new Promise<string>((res, rej) => {
+ const req = fetchRemoteVideoPreview(video, rej)
const path = join(CACHE.DIRECTORIES.PREVIEWS, video.getPreviewName())
const stream = createWriteStream(path)
req.pipe(stream)
- .on('finish', () => res(path))
- .on('error', (err) => rej(err))
+ .on('error', (err) => rej(err))
+ .on('finish', () => res(path))
})
}
}