]> git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - server/lib/activitypub/process/process-undo.ts
Fix like/dislike federation
[github/Chocobozzz/PeerTube.git] / server / lib / activitypub / process / process-undo.ts
1 import { ActivityAnnounce, ActivityFollow, ActivityLike, ActivityUndo, CacheFileObject } from '../../../../shared/models/activitypub'
2 import { DislikeObject } from '../../../../shared/models/activitypub/objects'
3 import { retryTransactionWrapper } from '../../../helpers/database-utils'
4 import { logger } from '../../../helpers/logger'
5 import { sequelizeTypescript } from '../../../initializers'
6 import { AccountVideoRateModel } from '../../../models/account/account-video-rate'
7 import { ActorModel } from '../../../models/activitypub/actor'
8 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
9 import { forwardVideoRelatedActivity } from '../send/utils'
10 import { getOrCreateVideoAndAccountAndChannel } from '../videos'
11 import { VideoShareModel } from '../../../models/video/video-share'
12 import { VideoRedundancyModel } from '../../../models/redundancy/video-redundancy'
13
14 async function processUndoActivity (activity: ActivityUndo, byActor: ActorModel) {
15 const activityToUndo = activity.object
16
17 if (activityToUndo.type === 'Like') {
18 return retryTransactionWrapper(processUndoLike, byActor, activity)
19 }
20
21 if (activityToUndo.type === 'Create') {
22 if (activityToUndo.object.type === 'Dislike') {
23 return retryTransactionWrapper(processUndoDislike, byActor, activity)
24 } else if (activityToUndo.object.type === 'CacheFile') {
25 return retryTransactionWrapper(processUndoCacheFile, byActor, activity)
26 }
27 }
28
29 if (activityToUndo.type === 'Dislike') {
30 return retryTransactionWrapper(processUndoDislike, byActor, activity)
31 }
32
33 if (activityToUndo.type === 'Follow') {
34 return retryTransactionWrapper(processUndoFollow, byActor, activityToUndo)
35 }
36
37 if (activityToUndo.type === 'Announce') {
38 return retryTransactionWrapper(processUndoAnnounce, byActor, activityToUndo)
39 }
40
41 logger.warn('Unknown activity object type %s -> %s when undo activity.', activityToUndo.type, { activity: activity.id })
42
43 return undefined
44 }
45
46 // ---------------------------------------------------------------------------
47
48 export {
49 processUndoActivity
50 }
51
52 // ---------------------------------------------------------------------------
53
54 async function processUndoLike (byActor: ActorModel, activity: ActivityUndo) {
55 const likeActivity = activity.object as ActivityLike
56
57 const { video } = await getOrCreateVideoAndAccountAndChannel({ videoObject: likeActivity.object })
58
59 return sequelizeTypescript.transaction(async t => {
60 if (!byActor.Account) throw new Error('Unknown account ' + byActor.url)
61
62 const rate = await AccountVideoRateModel.loadByAccountAndVideoOrUrl(byActor.Account.id, video.id, likeActivity.id, t)
63 if (!rate || rate.type !== 'like') throw new Error(`Unknown like by account ${byActor.Account.id} for video ${video.id}.`)
64
65 await rate.destroy({ transaction: t })
66 await video.decrement('likes', { transaction: t })
67
68 if (video.isOwned()) {
69 // Don't resend the activity to the sender
70 const exceptions = [ byActor ]
71
72 await forwardVideoRelatedActivity(activity, t, exceptions, video)
73 }
74 })
75 }
76
77 async function processUndoDislike (byActor: ActorModel, activity: ActivityUndo) {
78 const dislike = activity.object.type === 'Dislike'
79 ? activity.object
80 : activity.object.object as DislikeObject
81
82 const { video } = await getOrCreateVideoAndAccountAndChannel({ videoObject: dislike.object })
83
84 return sequelizeTypescript.transaction(async t => {
85 if (!byActor.Account) throw new Error('Unknown account ' + byActor.url)
86
87 const rate = await AccountVideoRateModel.loadByAccountAndVideoOrUrl(byActor.Account.id, video.id, dislike.id, t)
88 if (!rate || rate.type !== 'dislike') throw new Error(`Unknown dislike by account ${byActor.Account.id} for video ${video.id}.`)
89
90 await rate.destroy({ transaction: t })
91 await video.decrement('dislikes', { transaction: t })
92
93 if (video.isOwned()) {
94 // Don't resend the activity to the sender
95 const exceptions = [ byActor ]
96
97 await forwardVideoRelatedActivity(activity, t, exceptions, video)
98 }
99 })
100 }
101
102 async function processUndoCacheFile (byActor: ActorModel, activity: ActivityUndo) {
103 const cacheFileObject = activity.object.object as CacheFileObject
104
105 const { video } = await getOrCreateVideoAndAccountAndChannel({ videoObject: cacheFileObject.object })
106
107 return sequelizeTypescript.transaction(async t => {
108 const cacheFile = await VideoRedundancyModel.loadByUrl(cacheFileObject.id)
109 if (!cacheFile) {
110 logger.debug('Cannot undo unknown video cache %s.', cacheFileObject.id)
111 return
112 }
113
114 if (cacheFile.actorId !== byActor.id) throw new Error('Cannot delete redundancy ' + cacheFile.url + ' of another actor.')
115
116 await cacheFile.destroy()
117
118 if (video.isOwned()) {
119 // Don't resend the activity to the sender
120 const exceptions = [ byActor ]
121
122 await forwardVideoRelatedActivity(activity, t, exceptions, video)
123 }
124 })
125 }
126
127 function processUndoFollow (follower: ActorModel, followActivity: ActivityFollow) {
128 return sequelizeTypescript.transaction(async t => {
129 const following = await ActorModel.loadByUrlAndPopulateAccountAndChannel(followActivity.object, t)
130 const actorFollow = await ActorFollowModel.loadByActorAndTarget(follower.id, following.id, t)
131
132 if (!actorFollow) throw new Error(`'Unknown actor follow ${follower.id} -> ${following.id}.`)
133
134 await actorFollow.destroy({ transaction: t })
135
136 return undefined
137 })
138 }
139
140 function processUndoAnnounce (byActor: ActorModel, announceActivity: ActivityAnnounce) {
141 return sequelizeTypescript.transaction(async t => {
142 const share = await VideoShareModel.loadByUrl(announceActivity.id, t)
143 if (!share) throw new Error(`Unknown video share ${announceActivity.id}.`)
144
145 if (share.actorId !== byActor.id) throw new Error(`${share.url} is not shared by ${byActor.url}.`)
146
147 await share.destroy({ transaction: t })
148
149 if (share.Video.isOwned()) {
150 // Don't resend the activity to the sender
151 const exceptions = [ byActor ]
152
153 await forwardVideoRelatedActivity(announceActivity, t, exceptions, share.Video)
154 }
155 })
156 }