mirror of https://github.com/Chocobozzz/PeerTube
251 lines
8.6 KiB
TypeScript
251 lines
8.6 KiB
TypeScript
import * as Bluebird from 'bluebird'
|
|
import { ActivityCreate, VideoTorrentObject } from '../../../../shared'
|
|
import { DislikeObject, VideoAbuseObject, ViewObject } from '../../../../shared/models/activitypub/objects'
|
|
import { VideoCommentObject } from '../../../../shared/models/activitypub/objects/video-comment-object'
|
|
import { VideoRateType } from '../../../../shared/models/videos'
|
|
import { retryTransactionWrapper } from '../../../helpers/database-utils'
|
|
import { logger } from '../../../helpers/logger'
|
|
import { sequelizeTypescript } from '../../../initializers'
|
|
import { AccountVideoRateModel } from '../../../models/account/account-video-rate'
|
|
import { ActorModel } from '../../../models/activitypub/actor'
|
|
import { VideoModel } from '../../../models/video/video'
|
|
import { VideoAbuseModel } from '../../../models/video/video-abuse'
|
|
import { VideoCommentModel } from '../../../models/video/video-comment'
|
|
import { getOrCreateActorAndServerAndModel } from '../actor'
|
|
import { forwardActivity, getActorsInvolvedInVideo } from '../send/misc'
|
|
import { addVideoComments, resolveThread } from '../video-comments'
|
|
import { addVideoShares, getOrCreateAccountAndVideoAndChannel } from '../videos'
|
|
|
|
async function processCreateActivity (activity: ActivityCreate) {
|
|
const activityObject = activity.object
|
|
const activityType = activityObject.type
|
|
const actor = await getOrCreateActorAndServerAndModel(activity.actor)
|
|
|
|
if (activityType === 'View') {
|
|
return processCreateView(actor, activity)
|
|
} else if (activityType === 'Dislike') {
|
|
return processCreateDislike(actor, activity)
|
|
} else if (activityType === 'Video') {
|
|
return processCreateVideo(actor, activity)
|
|
} else if (activityType === 'Flag') {
|
|
return processCreateVideoAbuse(actor, activityObject as VideoAbuseObject)
|
|
} else if (activityType === 'Note') {
|
|
return processCreateVideoComment(actor, activity)
|
|
}
|
|
|
|
logger.warn('Unknown activity object type %s when creating activity.', activityType, { activity: activity.id })
|
|
return Promise.resolve(undefined)
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
processCreateActivity
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
async function processCreateVideo (
|
|
actor: ActorModel,
|
|
activity: ActivityCreate
|
|
) {
|
|
const videoToCreateData = activity.object as VideoTorrentObject
|
|
|
|
const { video } = await getOrCreateAccountAndVideoAndChannel(videoToCreateData, actor)
|
|
|
|
// Process outside the transaction because we could fetch remote data
|
|
if (videoToCreateData.likes && Array.isArray(videoToCreateData.likes.orderedItems)) {
|
|
logger.info('Adding likes of video %s.', video.uuid)
|
|
await createRates(videoToCreateData.likes.orderedItems, video, 'like')
|
|
}
|
|
|
|
if (videoToCreateData.dislikes && Array.isArray(videoToCreateData.dislikes.orderedItems)) {
|
|
logger.info('Adding dislikes of video %s.', video.uuid)
|
|
await createRates(videoToCreateData.dislikes.orderedItems, video, 'dislike')
|
|
}
|
|
|
|
if (videoToCreateData.shares && Array.isArray(videoToCreateData.shares.orderedItems)) {
|
|
logger.info('Adding shares of video %s.', video.uuid)
|
|
await addVideoShares(video, videoToCreateData.shares.orderedItems)
|
|
}
|
|
|
|
if (videoToCreateData.comments && Array.isArray(videoToCreateData.comments.orderedItems)) {
|
|
logger.info('Adding comments of video %s.', video.uuid)
|
|
await addVideoComments(video, videoToCreateData.comments.orderedItems)
|
|
}
|
|
|
|
return video
|
|
}
|
|
|
|
async function createRates (actorUrls: string[], video: VideoModel, rate: VideoRateType) {
|
|
let rateCounts = 0
|
|
const tasks: Bluebird<any>[] = []
|
|
|
|
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"
|
|
await video.increment(rate + 's', { by: rateCounts })
|
|
|
|
return
|
|
}
|
|
|
|
async function processCreateDislike (byActor: ActorModel, activity: ActivityCreate) {
|
|
const options = {
|
|
arguments: [ byActor, activity ],
|
|
errorMessage: 'Cannot dislike the video with many retries.'
|
|
}
|
|
|
|
return retryTransactionWrapper(createVideoDislike, options)
|
|
}
|
|
|
|
async function createVideoDislike (byActor: ActorModel, activity: ActivityCreate) {
|
|
const dislike = activity.object as DislikeObject
|
|
const byAccount = byActor.Account
|
|
|
|
if (!byAccount) throw new Error('Cannot create dislike with the non account actor ' + byActor.url)
|
|
|
|
const { video } = await getOrCreateAccountAndVideoAndChannel(dislike.object)
|
|
|
|
return sequelizeTypescript.transaction(async t => {
|
|
const rate = {
|
|
type: 'dislike' as 'dislike',
|
|
videoId: video.id,
|
|
accountId: byAccount.id
|
|
}
|
|
const [ , created ] = await AccountVideoRateModel.findOrCreate({
|
|
where: rate,
|
|
defaults: rate,
|
|
transaction: t
|
|
})
|
|
if (created === true) await video.increment('dislikes', { transaction: t })
|
|
|
|
if (video.isOwned() && created === true) {
|
|
// Don't resend the activity to the sender
|
|
const exceptions = [ byActor ]
|
|
await forwardActivity(activity, t, exceptions)
|
|
}
|
|
})
|
|
}
|
|
|
|
async function processCreateView (byActor: ActorModel, activity: ActivityCreate) {
|
|
const view = activity.object as ViewObject
|
|
|
|
const { video } = await getOrCreateAccountAndVideoAndChannel(view.object)
|
|
|
|
const actor = await ActorModel.loadByUrl(view.actor)
|
|
if (!actor) throw new Error('Unknown actor ' + view.actor)
|
|
|
|
await video.increment('views')
|
|
|
|
if (video.isOwned()) {
|
|
// Don't resend the activity to the sender
|
|
const exceptions = [ byActor ]
|
|
await forwardActivity(activity, undefined, exceptions)
|
|
}
|
|
}
|
|
|
|
function processCreateVideoAbuse (actor: ActorModel, videoAbuseToCreateData: VideoAbuseObject) {
|
|
const options = {
|
|
arguments: [ actor, videoAbuseToCreateData ],
|
|
errorMessage: 'Cannot insert the remote video abuse with many retries.'
|
|
}
|
|
|
|
return retryTransactionWrapper(addRemoteVideoAbuse, options)
|
|
}
|
|
|
|
async function addRemoteVideoAbuse (actor: ActorModel, videoAbuseToCreateData: VideoAbuseObject) {
|
|
logger.debug('Reporting remote abuse for video %s.', videoAbuseToCreateData.object)
|
|
|
|
const account = actor.Account
|
|
if (!account) throw new Error('Cannot create dislike with the non account actor ' + actor.url)
|
|
|
|
const { video } = await getOrCreateAccountAndVideoAndChannel(videoAbuseToCreateData.object)
|
|
|
|
return sequelizeTypescript.transaction(async t => {
|
|
const videoAbuseData = {
|
|
reporterAccountId: account.id,
|
|
reason: videoAbuseToCreateData.content,
|
|
videoId: video.id
|
|
}
|
|
|
|
await VideoAbuseModel.create(videoAbuseData)
|
|
|
|
logger.info('Remote abuse for video uuid %s created', videoAbuseToCreateData.object)
|
|
})
|
|
}
|
|
|
|
function processCreateVideoComment (byActor: ActorModel, activity: ActivityCreate) {
|
|
const options = {
|
|
arguments: [ byActor, activity ],
|
|
errorMessage: 'Cannot create video comment with many retries.'
|
|
}
|
|
|
|
return retryTransactionWrapper(createVideoComment, options)
|
|
}
|
|
|
|
async function createVideoComment (byActor: ActorModel, activity: ActivityCreate) {
|
|
const comment = activity.object as VideoCommentObject
|
|
const byAccount = byActor.Account
|
|
|
|
if (!byAccount) throw new Error('Cannot create video comment with the non account actor ' + byActor.url)
|
|
|
|
const { video, parents } = await resolveThread(comment.inReplyTo)
|
|
|
|
return sequelizeTypescript.transaction(async t => {
|
|
let originCommentId = null
|
|
let inReplyToCommentId = null
|
|
|
|
if (parents.length !== 0) {
|
|
const parent = parents[0]
|
|
|
|
originCommentId = parent.getThreadId()
|
|
inReplyToCommentId = parent.id
|
|
}
|
|
|
|
// This is a new thread
|
|
const objectToCreate = {
|
|
url: comment.id,
|
|
text: comment.content,
|
|
originCommentId,
|
|
inReplyToCommentId,
|
|
videoId: video.id,
|
|
accountId: byAccount.id
|
|
}
|
|
|
|
const options = {
|
|
where: {
|
|
url: objectToCreate.url
|
|
},
|
|
defaults: objectToCreate,
|
|
transaction: t
|
|
}
|
|
const [ ,created ] = await VideoCommentModel.findOrCreate(options)
|
|
|
|
if (video.isOwned() && created === true) {
|
|
// Don't resend the activity to the sender
|
|
const exceptions = [ byActor ]
|
|
|
|
// Mastodon does not add our announces in audience, so we forward to them manually
|
|
const additionalActors = await getActorsInvolvedInVideo(video, t)
|
|
const additionalFollowerUrls = additionalActors.map(a => a.followersUrl)
|
|
|
|
await forwardActivity(activity, t, exceptions, additionalFollowerUrls)
|
|
}
|
|
})
|
|
}
|