mirror of https://github.com/Chocobozzz/PeerTube
76 lines
2.3 KiB
TypeScript
76 lines
2.3 KiB
TypeScript
import { chunk } from 'lodash'
|
|
import { doJSONRequest } from '@server/helpers/requests'
|
|
import { JobQueue } from '@server/lib/job-queue'
|
|
import { ActorFollowModel } from '@server/models/actor/actor-follow'
|
|
import { getServerActor } from '@server/models/application/application'
|
|
import { logger } from '../../helpers/logger'
|
|
import { CONFIG } from '../../initializers/config'
|
|
import { SCHEDULER_INTERVALS_MS, SERVER_ACTOR_NAME } from '../../initializers/constants'
|
|
import { AbstractScheduler } from './abstract-scheduler'
|
|
|
|
export class AutoFollowIndexInstances extends AbstractScheduler {
|
|
|
|
private static instance: AbstractScheduler
|
|
|
|
protected schedulerIntervalMs = SCHEDULER_INTERVALS_MS.AUTO_FOLLOW_INDEX_INSTANCES
|
|
|
|
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 searchParams = { count: 1000 }
|
|
if (this.lastCheck) Object.assign(searchParams, { since: this.lastCheck.toISOString() })
|
|
|
|
this.lastCheck = new Date()
|
|
|
|
const { body } = await doJSONRequest<any>(indexUrl, { searchParams })
|
|
if (!body.data || Array.isArray(body.data) === false) {
|
|
logger.error('Cannot auto follow instances of index %s. Please check the auto follow URL.', indexUrl, { body })
|
|
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())
|
|
}
|
|
}
|