diff options
Diffstat (limited to 'server/lib/activitypub/process/process-follow.ts')
-rw-r--r-- | server/lib/activitypub/process/process-follow.ts | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/server/lib/activitypub/process/process-follow.ts b/server/lib/activitypub/process/process-follow.ts new file mode 100644 index 000000000..553639580 --- /dev/null +++ b/server/lib/activitypub/process/process-follow.ts | |||
@@ -0,0 +1,59 @@ | |||
1 | import { ActivityFollow } from '../../../../shared/models/activitypub/activity' | ||
2 | import { getOrCreateAccount, retryTransactionWrapper } from '../../../helpers' | ||
3 | import { database as db } from '../../../initializers' | ||
4 | import { AccountInstance } from '../../../models/account/account-interface' | ||
5 | import { logger } from '../../../helpers/logger' | ||
6 | import { sendAccept } from '../send/send-accept' | ||
7 | |||
8 | async function processFollowActivity (activity: ActivityFollow) { | ||
9 | const activityObject = activity.object | ||
10 | const account = await getOrCreateAccount(activity.actor) | ||
11 | |||
12 | return processFollow(account, activityObject) | ||
13 | } | ||
14 | |||
15 | // --------------------------------------------------------------------------- | ||
16 | |||
17 | export { | ||
18 | processFollowActivity | ||
19 | } | ||
20 | |||
21 | // --------------------------------------------------------------------------- | ||
22 | |||
23 | function processFollow (account: AccountInstance, targetAccountURL: string) { | ||
24 | const options = { | ||
25 | arguments: [ account, targetAccountURL ], | ||
26 | errorMessage: 'Cannot follow with many retries.' | ||
27 | } | ||
28 | |||
29 | return retryTransactionWrapper(follow, options) | ||
30 | } | ||
31 | |||
32 | async function follow (account: AccountInstance, targetAccountURL: string) { | ||
33 | await db.sequelize.transaction(async t => { | ||
34 | const targetAccount = await db.Account.loadByUrl(targetAccountURL, t) | ||
35 | |||
36 | if (!targetAccount) throw new Error('Unknown account') | ||
37 | if (targetAccount.isOwned() === false) throw new Error('This is not a local account.') | ||
38 | |||
39 | const [ accountFollow ] = await db.AccountFollow.findOrCreate({ | ||
40 | where: { | ||
41 | accountId: account.id, | ||
42 | targetAccountId: targetAccount.id | ||
43 | }, | ||
44 | defaults: { | ||
45 | accountId: account.id, | ||
46 | targetAccountId: targetAccount.id, | ||
47 | state: 'accepted' | ||
48 | }, | ||
49 | transaction: t | ||
50 | }) | ||
51 | accountFollow.AccountFollower = account | ||
52 | accountFollow.AccountFollowing = targetAccount | ||
53 | |||
54 | // Target sends to account he accepted the follow request | ||
55 | return sendAccept(accountFollow, t) | ||
56 | }) | ||
57 | |||
58 | logger.info('Account uuid %s is followed by account %s.', account.url, targetAccountURL) | ||
59 | } | ||