1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
import { logger } from '../../helpers/logger'
import { AbstractScheduler } from './abstract-scheduler'
import { INSTANCES_INDEX, SCHEDULER_INTERVALS_MS, SERVER_ACTOR_NAME } from '../../initializers/constants'
import { CONFIG } from '../../initializers/config'
import { chunk } from 'lodash'
import { doRequest } from '@server/helpers/requests'
import { ActorFollowModel } from '@server/models/activitypub/actor-follow'
import { JobQueue } from '@server/lib/job-queue'
import { getServerActor } from '@server/models/application/application'
export class AutoFollowIndexInstances extends AbstractScheduler {
private static instance: AbstractScheduler
protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.autoFollowIndexInstances
private lastCheck: Date
private constructor () {
super()
}
protected async internalExecute () {
return this.autoFollow()
}
private async autoFollow () {
if (CONFIG.FOLLOWINGS.INSTANCE.AUTO_FOLLOW_INDEX.ENABLED === false) return
const indexUrl = CONFIG.FOLLOWINGS.INSTANCE.AUTO_FOLLOW_INDEX.INDEX_URL
logger.info('Auto follow instances of index %s.', indexUrl)
try {
const serverActor = await getServerActor()
const uri = indexUrl + INSTANCES_INDEX.HOSTS_PATH
const qs = { count: 1000 }
if (this.lastCheck) Object.assign(qs, { since: this.lastCheck.toISOString() })
this.lastCheck = new Date()
const { body } = await doRequest<any>({ uri, qs, json: true })
if (!body.data || Array.isArray(body.data) === false) {
logger.error('Cannot auto follow instances of index %s: bad URL format. Please check the auto follow URL.', indexUrl)
return
}
const hosts: string[] = body.data.map(o => o.host)
const chunks = chunk(hosts, 20)
for (const chunk of chunks) {
const unfollowedHosts = await ActorFollowModel.keepUnfollowedInstance(chunk)
for (const unfollowedHost of unfollowedHosts) {
const payload = {
host: unfollowedHost,
name: SERVER_ACTOR_NAME,
followerActorId: serverActor.id,
isAutoFollow: true
}
JobQueue.Instance.createJob({ type: 'activitypub-follow', payload })
}
}
} catch (err) {
logger.error('Cannot auto follow hosts of index %s.', indexUrl, { err })
}
}
static get Instance () {
return this.instance || (this.instance = new this())
}
}
|
