X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Flib%2Factivitypub%2Fvideos.ts;h=91231a187f386e6a04e7e429ecca2ba4d99340bb;hb=d9bdd007d7a1368d2a13127ecb5c0a81a18a8c04;hp=d1888556c0df182cc7c5d8c571533834acca2921;hpb=965c4b22d0e4d2f853501e844e6ebbb861bd389d;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/lib/activitypub/videos.ts b/server/lib/activitypub/videos.ts index d1888556c..91231a187 100644 --- a/server/lib/activitypub/videos.ts +++ b/server/lib/activitypub/videos.ts @@ -3,29 +3,32 @@ import * as sequelize from 'sequelize' import * as magnetUtil from 'magnet-uri' import { join } from 'path' import * as request from 'request' -import { ActivityIconObject, VideoState } from '../../../shared/index' +import { ActivityIconObject, ActivityUrlObject, ActivityVideoUrlObject, VideoState } from '../../../shared/index' import { VideoTorrentObject } from '../../../shared/models/activitypub/objects' -import { VideoPrivacy, VideoRateType } from '../../../shared/models/videos' +import { VideoPrivacy } from '../../../shared/models/videos' import { sanitizeAndCheckVideoTorrentObject } from '../../helpers/custom-validators/activitypub/videos' import { isVideoFileInfoHashValid } from '../../helpers/custom-validators/videos' -import { retryTransactionWrapper } from '../../helpers/database-utils' +import { resetSequelizeInstance, retryTransactionWrapper } from '../../helpers/database-utils' import { logger } from '../../helpers/logger' import { doRequest, doRequestAndSaveToFile } from '../../helpers/requests' import { ACTIVITY_PUB, CONFIG, REMOTE_SCHEME, sequelizeTypescript, VIDEO_MIMETYPE_EXT } from '../../initializers' -import { AccountVideoRateModel } from '../../models/account/account-video-rate' import { ActorModel } from '../../models/activitypub/actor' import { TagModel } from '../../models/video/tag' import { VideoModel } from '../../models/video/video' import { VideoChannelModel } from '../../models/video/video-channel' import { VideoFileModel } from '../../models/video/video-file' -import { VideoShareModel } from '../../models/video/video-share' import { getOrCreateActorAndServerAndModel } from './actor' import { addVideoComments } from './video-comments' import { crawlCollectionPage } from './crawl' import { sendCreateVideo, sendUpdateVideo } from './send' -import { shareVideoByServerAndChannel } from './index' import { isArray } from '../../helpers/custom-validators/misc' import { VideoCaptionModel } from '../../models/video/video-caption' +import { JobQueue } from '../job-queue' +import { ActivitypubHttpFetcherPayload } from '../job-queue/handlers/activitypub-http-fetcher' +import { createRates } from './video-rates' +import { addVideoShares, shareVideoByServerAndChannel } from './share' +import { AccountModel } from '../../models/account/account' +import { fetchVideoByUrl, VideoFetchByUrlType } from '../../helpers/video' async function federateVideoIfNeeded (video: VideoModel, isNewVideo: boolean, transaction?: sequelize.Transaction) { // If the video is not private and published, we federate it @@ -48,18 +51,29 @@ async function federateVideoIfNeeded (video: VideoModel, isNewVideo: boolean, tr } } -function fetchRemoteVideoStaticFile (video: VideoModel, path: string, reject: Function) { - const host = video.VideoChannel.Account.Actor.Server.host +async function fetchRemoteVideo (videoUrl: string): Promise<{ response: request.RequestResponse, videoObject: VideoTorrentObject }> { + const options = { + uri: videoUrl, + method: 'GET', + json: true, + activityPub: true + } - // We need to provide a callback, if no we could have an uncaught exception - return request.get(REMOTE_SCHEME.HTTP + '://' + host + path, err => { - if (err) reject(err) - }) + logger.info('Fetching remote video %s.', videoUrl) + + const { response, body } = await doRequest(options) + + if (sanitizeAndCheckVideoTorrentObject(body) === false) { + logger.debug('Remote video JSON is not valid.', { body }) + return { response, videoObject: undefined } + } + + return { response, videoObject: body } } async function fetchRemoteVideoDescription (video: VideoModel) { const host = video.VideoChannel.Account.Actor.Server.host - const path = video.getDescriptionPath() + const path = video.getDescriptionAPIPath() const options = { uri: REMOTE_SCHEME.HTTP + '://' + host + path, json: true @@ -69,6 +83,15 @@ async function fetchRemoteVideoDescription (video: VideoModel) { return body.description ? body.description : '' } +function fetchRemoteVideoStaticFile (video: VideoModel, path: string, reject: Function) { + const host = video.VideoChannel.Account.Actor.Server.host + + // We need to provide a callback, if no we could have an uncaught exception + return request.get(REMOTE_SCHEME.HTTP + '://' + host + path, err => { + if (err) reject(err) + }) +} + function generateThumbnailFromUrl (video: VideoModel, icon: ActivityIconObject) { const thumbnailName = video.getThumbnailName() const thumbnailPath = join(CONFIG.STORAGE.THUMBNAILS_DIR, thumbnailName) @@ -80,6 +103,293 @@ function generateThumbnailFromUrl (video: VideoModel, icon: ActivityIconObject) return doRequestAndSaveToFile(options, thumbnailPath) } +function getOrCreateVideoChannelFromVideoObject (videoObject: VideoTorrentObject) { + const channel = videoObject.attributedTo.find(a => a.type === 'Group') + if (!channel) throw new Error('Cannot find associated video channel to video ' + videoObject.url) + + return getOrCreateActorAndServerAndModel(channel.id, 'all') +} + +type SyncParam = { + likes: boolean + dislikes: boolean + shares: boolean + comments: boolean + thumbnail: boolean + refreshVideo: boolean +} +async function syncVideoExternalAttributes (video: VideoModel, fetchedVideo: VideoTorrentObject, syncParam: SyncParam) { + logger.info('Adding likes/dislikes/shares/comments of video %s.', video.uuid) + + const jobPayloads: ActivitypubHttpFetcherPayload[] = [] + + if (syncParam.likes === true) { + await crawlCollectionPage(fetchedVideo.likes, items => createRates(items, video, 'like')) + .catch(err => logger.error('Cannot add likes of video %s.', video.uuid, { err })) + } else { + jobPayloads.push({ uri: fetchedVideo.likes, videoId: video.id, type: 'video-likes' as 'video-likes' }) + } + + if (syncParam.dislikes === true) { + await crawlCollectionPage(fetchedVideo.dislikes, items => createRates(items, video, 'dislike')) + .catch(err => logger.error('Cannot add dislikes of video %s.', video.uuid, { err })) + } else { + jobPayloads.push({ uri: fetchedVideo.dislikes, videoId: video.id, type: 'video-dislikes' as 'video-dislikes' }) + } + + if (syncParam.shares === true) { + await crawlCollectionPage(fetchedVideo.shares, items => addVideoShares(items, video)) + .catch(err => logger.error('Cannot add shares of video %s.', video.uuid, { err })) + } else { + jobPayloads.push({ uri: fetchedVideo.shares, videoId: video.id, type: 'video-shares' as 'video-shares' }) + } + + if (syncParam.comments === true) { + await crawlCollectionPage(fetchedVideo.comments, items => addVideoComments(items, video)) + .catch(err => logger.error('Cannot add comments of video %s.', video.uuid, { err })) + } else { + jobPayloads.push({ uri: fetchedVideo.shares, videoId: video.id, type: 'video-shares' as 'video-shares' }) + } + + await Bluebird.map(jobPayloads, payload => JobQueue.Instance.createJob({ type: 'activitypub-http-fetcher', payload })) +} + +async function getOrCreateVideoAndAccountAndChannel (options: { + videoObject: VideoTorrentObject | string, + syncParam?: SyncParam, + fetchType?: VideoFetchByUrlType, + refreshViews?: boolean +}) { + // Default params + const syncParam = options.syncParam || { likes: true, dislikes: true, shares: true, comments: true, thumbnail: true, refreshVideo: false } + const fetchType = options.fetchType || 'all' + const refreshViews = options.refreshViews || false + + // Get video url + const videoUrl = typeof options.videoObject === 'string' ? options.videoObject : options.videoObject.id + + let videoFromDatabase = await fetchVideoByUrl(videoUrl, fetchType) + if (videoFromDatabase) { + const refreshOptions = { + video: videoFromDatabase, + fetchedType: fetchType, + syncParam, + refreshViews + } + const p = retryTransactionWrapper(refreshVideoIfNeeded, refreshOptions) + if (syncParam.refreshVideo === true) videoFromDatabase = await p + + return { video: videoFromDatabase } + } + + const { videoObject: fetchedVideo } = await fetchRemoteVideo(videoUrl) + if (!fetchedVideo) throw new Error('Cannot fetch remote video with url: ' + videoUrl) + + const channelActor = await getOrCreateVideoChannelFromVideoObject(fetchedVideo) + const video = await retryTransactionWrapper(createVideo, fetchedVideo, channelActor, syncParam.thumbnail) + + await syncVideoExternalAttributes(video, fetchedVideo, syncParam) + + return { video } +} + +async function updateVideoFromAP (options: { + video: VideoModel, + videoObject: VideoTorrentObject, + account: AccountModel, + channel: VideoChannelModel, + updateViews: boolean, + overrideTo?: string[] +}) { + logger.debug('Updating remote video "%s".', options.videoObject.uuid) + let videoFieldsSave: any + + try { + const updatedVideo: VideoModel = await sequelizeTypescript.transaction(async t => { + const sequelizeOptions = { + transaction: t + } + + videoFieldsSave = options.video.toJSON() + + // Check actor has the right to update the video + const videoChannel = options.video.VideoChannel + if (videoChannel.Account.id !== options.account.id) { + throw new Error('Account ' + options.account.Actor.url + ' does not own video channel ' + videoChannel.Actor.url) + } + + const to = options.overrideTo ? options.overrideTo : options.videoObject.to + const videoData = await videoActivityObjectToDBAttributes(options.channel, options.videoObject, to) + options.video.set('name', videoData.name) + options.video.set('uuid', videoData.uuid) + options.video.set('url', videoData.url) + options.video.set('category', videoData.category) + options.video.set('licence', videoData.licence) + options.video.set('language', videoData.language) + options.video.set('description', videoData.description) + options.video.set('support', videoData.support) + options.video.set('nsfw', videoData.nsfw) + options.video.set('commentsEnabled', videoData.commentsEnabled) + options.video.set('waitTranscoding', videoData.waitTranscoding) + options.video.set('state', videoData.state) + options.video.set('duration', videoData.duration) + options.video.set('createdAt', videoData.createdAt) + options.video.set('publishedAt', videoData.publishedAt) + options.video.set('privacy', videoData.privacy) + options.video.set('channelId', videoData.channelId) + + if (options.updateViews === true) options.video.set('views', videoData.views) + await options.video.save(sequelizeOptions) + + // Don't block on request + generateThumbnailFromUrl(options.video, options.videoObject.icon) + .catch(err => logger.warn('Cannot generate thumbnail of %s.', options.videoObject.id, { err })) + + // Remove old video files + const videoFileDestroyTasks: Bluebird[] = [] + for (const videoFile of options.video.VideoFiles) { + videoFileDestroyTasks.push(videoFile.destroy(sequelizeOptions)) + } + await Promise.all(videoFileDestroyTasks) + + const videoFileAttributes = videoFileActivityUrlToDBAttributes(options.video, options.videoObject) + const tasks = videoFileAttributes.map(f => VideoFileModel.create(f, sequelizeOptions)) + await Promise.all(tasks) + + // Update Tags + const tags = options.videoObject.tag.map(tag => tag.name) + const tagInstances = await TagModel.findOrCreateTags(tags, t) + await options.video.$set('Tags', tagInstances, sequelizeOptions) + + // Update captions + await VideoCaptionModel.deleteAllCaptionsOfRemoteVideo(options.video.id, t) + + const videoCaptionsPromises = options.videoObject.subtitleLanguage.map(c => { + return VideoCaptionModel.insertOrReplaceLanguage(options.video.id, c.identifier, t) + }) + await Promise.all(videoCaptionsPromises) + }) + + logger.info('Remote video with uuid %s updated', options.videoObject.uuid) + + return updatedVideo + } catch (err) { + if (options.video !== undefined && videoFieldsSave !== undefined) { + resetSequelizeInstance(options.video, videoFieldsSave) + } + + // This is just a debug because we will retry the insert + logger.debug('Cannot update the remote video.', { err }) + throw err + } +} + +export { + updateVideoFromAP, + federateVideoIfNeeded, + fetchRemoteVideo, + getOrCreateVideoAndAccountAndChannel, + fetchRemoteVideoStaticFile, + fetchRemoteVideoDescription, + generateThumbnailFromUrl, + getOrCreateVideoChannelFromVideoObject +} + +// --------------------------------------------------------------------------- + +function isActivityVideoUrlObject (url: ActivityUrlObject): url is ActivityVideoUrlObject { + const mimeTypes = Object.keys(VIDEO_MIMETYPE_EXT) + + return mimeTypes.indexOf(url.mimeType) !== -1 && url.mimeType.startsWith('video/') +} + +async function createVideo (videoObject: VideoTorrentObject, channelActor: ActorModel, waitThumbnail = false) { + logger.debug('Adding remote video %s.', videoObject.id) + + const videoCreated: VideoModel = await sequelizeTypescript.transaction(async t => { + const sequelizeOptions = { transaction: t } + + const videoData = await videoActivityObjectToDBAttributes(channelActor.VideoChannel, videoObject, videoObject.to) + const video = VideoModel.build(videoData) + + const videoCreated = await video.save(sequelizeOptions) + + // Process files + const videoFileAttributes = videoFileActivityUrlToDBAttributes(videoCreated, videoObject) + if (videoFileAttributes.length === 0) { + throw new Error('Cannot find valid files for video %s ' + videoObject.url) + } + + const videoFilePromises = videoFileAttributes.map(f => VideoFileModel.create(f, { transaction: t })) + await Promise.all(videoFilePromises) + + // Process tags + const tags = videoObject.tag.map(t => t.name) + const tagInstances = await TagModel.findOrCreateTags(tags, t) + await videoCreated.$set('Tags', tagInstances, sequelizeOptions) + + // Process captions + const videoCaptionsPromises = videoObject.subtitleLanguage.map(c => { + return VideoCaptionModel.insertOrReplaceLanguage(videoCreated.id, c.identifier, t) + }) + await Promise.all(videoCaptionsPromises) + + logger.info('Remote video with uuid %s inserted.', videoObject.uuid) + + videoCreated.VideoChannel = channelActor.VideoChannel + return videoCreated + }) + + const p = generateThumbnailFromUrl(videoCreated, videoObject.icon) + .catch(err => logger.warn('Cannot generate thumbnail of %s.', videoObject.id, { err })) + + if (waitThumbnail === true) await p + + return videoCreated +} + +async function refreshVideoIfNeeded (options: { + video: VideoModel, + fetchedType: VideoFetchByUrlType, + syncParam: SyncParam, + refreshViews: boolean +}): Promise { + // We need more attributes if the argument video was fetched with not enough joints + const video = options.fetchedType === 'all' ? options.video : await VideoModel.loadByUrlAndPopulateAccount(options.video.url) + + if (!video.isOutdated()) return video + + try { + const { response, videoObject } = await fetchRemoteVideo(video.url) + if (response.statusCode === 404) { + // Video does not exist anymore + await video.destroy() + return undefined + } + + if (videoObject === undefined) { + logger.warn('Cannot refresh remote video: invalid body.') + return video + } + + const channelActor = await getOrCreateVideoChannelFromVideoObject(videoObject) + const account = await AccountModel.load(channelActor.VideoChannel.accountId) + + const updateOptions = { + video, + videoObject, + account, + channel: channelActor.VideoChannel, + updateViews: options.refreshViews + } + await updateVideoFromAP(updateOptions) + await syncVideoExternalAttributes(video, videoObject, options.syncParam) + } catch (err) { + logger.warn('Cannot refresh video.', { err }) + return video + } +} + async function videoActivityObjectToDBAttributes ( videoChannel: VideoChannelModel, videoObject: VideoTorrentObject, @@ -134,10 +444,7 @@ async function videoActivityObjectToDBAttributes ( } function videoFileActivityUrlToDBAttributes (videoCreated: VideoModel, videoObject: VideoTorrentObject) { - const mimeTypes = Object.keys(VIDEO_MIMETYPE_EXT) - const fileUrls = videoObject.url.filter(u => { - return mimeTypes.indexOf(u.mimeType) !== -1 && u.mimeType.startsWith('video/') - }) + const fileUrls = videoObject.url.filter(u => isActivityVideoUrlObject(u)) as ActivityVideoUrlObject[] if (fileUrls.length === 0) { throw new Error('Cannot find video files for ' + videoCreated.url) @@ -170,190 +477,3 @@ function videoFileActivityUrlToDBAttributes (videoCreated: VideoModel, videoObje return attributes } - -function getOrCreateVideoChannel (videoObject: VideoTorrentObject) { - const channel = videoObject.attributedTo.find(a => a.type === 'Group') - if (!channel) throw new Error('Cannot find associated video channel to video ' + videoObject.url) - - return getOrCreateActorAndServerAndModel(channel.id) -} - -async function getOrCreateVideo (videoObject: VideoTorrentObject, channelActor: ActorModel) { - logger.debug('Adding remote video %s.', videoObject.id) - - return sequelizeTypescript.transaction(async t => { - const sequelizeOptions = { - transaction: t - } - const videoFromDatabase = await VideoModel.loadByUUIDOrURLAndPopulateAccount(videoObject.uuid, videoObject.id, t) - if (videoFromDatabase) return videoFromDatabase - - const videoData = await videoActivityObjectToDBAttributes(channelActor.VideoChannel, videoObject, videoObject.to) - const video = VideoModel.build(videoData) - - // Don't block on remote HTTP request (we are in a transaction!) - generateThumbnailFromUrl(video, videoObject.icon) - .catch(err => logger.warn('Cannot generate thumbnail of %s.', videoObject.id, { err })) - - const videoCreated = await video.save(sequelizeOptions) - - // Process files - const videoFileAttributes = videoFileActivityUrlToDBAttributes(videoCreated, videoObject) - if (videoFileAttributes.length === 0) { - throw new Error('Cannot find valid files for video %s ' + videoObject.url) - } - - const videoFilePromises = videoFileAttributes.map(f => VideoFileModel.create(f, { transaction: t })) - await Promise.all(videoFilePromises) - - // Process tags - const tags = videoObject.tag.map(t => t.name) - const tagInstances = await TagModel.findOrCreateTags(tags, t) - await videoCreated.$set('Tags', tagInstances, sequelizeOptions) - - // Process captions - const videoCaptionsPromises = videoObject.subtitleLanguage.map(c => { - return VideoCaptionModel.insertOrReplaceLanguage(videoCreated.id, c.identifier, t) - }) - await Promise.all(videoCaptionsPromises) - - logger.info('Remote video with uuid %s inserted.', videoObject.uuid) - - videoCreated.VideoChannel = channelActor.VideoChannel - return videoCreated - }) -} - -async function getOrCreateAccountAndVideoAndChannel (videoObject: VideoTorrentObject | string, actor?: ActorModel) { - const videoUrl = typeof videoObject === 'string' ? videoObject : videoObject.id - - const videoFromDatabase = await VideoModel.loadByUrlAndPopulateAccount(videoUrl) - if (videoFromDatabase) { - return { - video: videoFromDatabase, - actor: videoFromDatabase.VideoChannel.Account.Actor, - channelActor: videoFromDatabase.VideoChannel.Actor - } - } - - videoObject = await fetchRemoteVideo(videoUrl) - if (!videoObject) throw new Error('Cannot fetch remote video with url: ' + videoUrl) - - if (!actor) { - const actorObj = videoObject.attributedTo.find(a => a.type === 'Person') - if (!actorObj) throw new Error('Cannot find associated actor to video ' + videoObject.url) - - actor = await getOrCreateActorAndServerAndModel(actorObj.id) - } - - const channelActor = await getOrCreateVideoChannel(videoObject) - - const video = await retryTransactionWrapper(getOrCreateVideo, videoObject, channelActor) - - // Process outside the transaction because we could fetch remote data - logger.info('Adding likes of video %s.', video.uuid) - await crawlCollectionPage(videoObject.likes, (items) => createRates(items, video, 'like')) - - logger.info('Adding dislikes of video %s.', video.uuid) - await crawlCollectionPage(videoObject.dislikes, (items) => createRates(items, video, 'dislike')) - - logger.info('Adding shares of video %s.', video.uuid) - await crawlCollectionPage(videoObject.shares, (items) => addVideoShares(items, video)) - - logger.info('Adding comments of video %s.', video.uuid) - await crawlCollectionPage(videoObject.comments, (items) => addVideoComments(items, video)) - - return { actor, channelActor, video } -} - -async function createRates (actorUrls: string[], video: VideoModel, rate: VideoRateType) { - let rateCounts = 0 - const tasks: Bluebird[] = [] - - for (const actorUrl of actorUrls) { - const actor = await getOrCreateActorAndServerAndModel(actorUrl) - const p = AccountVideoRateModel - .create({ - videoId: video.id, - accountId: actor.Account.id, - type: rate - }) - .then(() => rateCounts += 1) - - tasks.push(p) - } - - await Promise.all(tasks) - - logger.info('Adding %d %s to video %s.', rateCounts, rate, video.uuid) - - // This is "likes" and "dislikes" - if (rateCounts !== 0) await video.increment(rate + 's', { by: rateCounts }) - - return -} - -async function addVideoShares (shareUrls: string[], instance: VideoModel) { - for (const shareUrl of shareUrls) { - // Fetch url - const { body } = await doRequest({ - uri: shareUrl, - json: true, - activityPub: true - }) - if (!body || !body.actor) { - logger.warn('Cannot add remote share with url: %s, skipping...', shareUrl) - continue - } - - const actorUrl = body.actor - const actor = await getOrCreateActorAndServerAndModel(actorUrl) - - const entry = { - actorId: actor.id, - videoId: instance.id, - url: shareUrl - } - - await VideoShareModel.findOrCreate({ - where: { - url: shareUrl - }, - defaults: entry - }) - } -} - -async function fetchRemoteVideo (videoUrl: string): Promise { - const options = { - uri: videoUrl, - method: 'GET', - json: true, - activityPub: true - } - - logger.info('Fetching remote video %s.', videoUrl) - - const { body } = await doRequest(options) - - if (sanitizeAndCheckVideoTorrentObject(body) === false) { - logger.debug('Remote video JSON is not valid.', { body }) - return undefined - } - - return body -} - -export { - federateVideoIfNeeded, - fetchRemoteVideo, - getOrCreateAccountAndVideoAndChannel, - fetchRemoteVideoStaticFile, - fetchRemoteVideoDescription, - generateThumbnailFromUrl, - videoActivityObjectToDBAttributes, - videoFileActivityUrlToDBAttributes, - getOrCreateVideo, - getOrCreateVideoChannel, - addVideoShares -}