Fix user notifications on new follow
[oweals/peertube.git] / server / lib / activitypub / process / process-follow.ts
1 import { ActivityFollow } from '../../../../shared/models/activitypub'
2 import { retryTransactionWrapper } from '../../../helpers/database-utils'
3 import { logger } from '../../../helpers/logger'
4 import { sequelizeTypescript } from '../../../initializers'
5 import { ActorModel } from '../../../models/activitypub/actor'
6 import { ActorFollowModel } from '../../../models/activitypub/actor-follow'
7 import { sendAccept, sendReject } from '../send'
8 import { Notifier } from '../../notifier'
9 import { getAPId } from '../../../helpers/activitypub'
10 import { getServerActor } from '../../../helpers/utils'
11 import { CONFIG } from '../../../initializers/config'
12 import { APProcessorOptions } from '../../../typings/activitypub-processor.model'
13
14 async function processFollowActivity (options: APProcessorOptions<ActivityFollow>) {
15   const { activity, byActor } = options
16   const activityObject = getAPId(activity.object)
17
18   return retryTransactionWrapper(processFollow, byActor, activityObject)
19 }
20
21 // ---------------------------------------------------------------------------
22
23 export {
24   processFollowActivity
25 }
26
27 // ---------------------------------------------------------------------------
28
29 async function processFollow (actor: ActorModel, targetActorURL: string) {
30   const { actorFollow, created, isFollowingInstance } = await sequelizeTypescript.transaction(async t => {
31     const targetActor = await ActorModel.loadByUrlAndPopulateAccountAndChannel(targetActorURL, t)
32
33     if (!targetActor) throw new Error('Unknown actor')
34     if (targetActor.isOwned() === false) throw new Error('This is not a local actor.')
35
36     const serverActor = await getServerActor()
37     const isFollowingInstance = targetActor.id === serverActor.id
38
39     if (isFollowingInstance && CONFIG.FOLLOWERS.INSTANCE.ENABLED === false) {
40       logger.info('Rejecting %s because instance followers are disabled.', targetActor.url)
41
42       await sendReject(actor, targetActor)
43
44       return { actorFollow: undefined }
45     }
46
47     const [ actorFollow, created ] = await ActorFollowModel.findOrCreate({
48       where: {
49         actorId: actor.id,
50         targetActorId: targetActor.id
51       },
52       defaults: {
53         actorId: actor.id,
54         targetActorId: targetActor.id,
55         state: CONFIG.FOLLOWERS.INSTANCE.MANUAL_APPROVAL ? 'pending' : 'accepted'
56       },
57       transaction: t
58     })
59
60     if (actorFollow.state !== 'accepted' && CONFIG.FOLLOWERS.INSTANCE.MANUAL_APPROVAL === false) {
61       actorFollow.state = 'accepted'
62       await actorFollow.save({ transaction: t })
63     }
64
65     actorFollow.ActorFollower = actor
66     actorFollow.ActorFollowing = targetActor
67
68     // Target sends to actor he accepted the follow request
69     if (actorFollow.state === 'accepted') await sendAccept(actorFollow)
70
71     return { actorFollow, created, isFollowingInstance }
72   })
73
74   // Rejected
75   if (!actorFollow) return
76
77   if (created) {
78     if (isFollowingInstance) Notifier.Instance.notifyOfNewInstanceFollow(actorFollow)
79     else Notifier.Instance.notifyOfNewUserFollow(actorFollow)
80   }
81
82   logger.info('Actor %s is followed by actor %s.', targetActorURL, actor.url)
83 }