PeerTube/server/core/lib/redundancy.ts
2023-10-04 15:13:25 +02:00

60 lines
2.1 KiB
TypeScript

import { Transaction } from 'sequelize'
import { logger, loggerTagsFactory } from '@server/helpers/logger.js'
import { CONFIG } from '@server/initializers/config.js'
import { ActorFollowModel } from '@server/models/actor/actor-follow.js'
import { getServerActor } from '@server/models/application/application.js'
import { MActorSignature, MVideoRedundancyVideo } from '@server/types/models/index.js'
import { Activity } from '@peertube/peertube-models'
import { VideoRedundancyModel } from '../models/redundancy/video-redundancy.js'
import { sendUndoCacheFile } from './activitypub/send/index.js'
const lTags = loggerTagsFactory('redundancy')
async function removeVideoRedundancy (videoRedundancy: MVideoRedundancyVideo, t?: Transaction) {
const serverActor = await getServerActor()
// Local cache, send undo to remote instances
if (videoRedundancy.actorId === serverActor.id) await sendUndoCacheFile(serverActor, videoRedundancy, t)
await videoRedundancy.destroy({ transaction: t })
}
async function removeRedundanciesOfServer (serverId: number) {
const redundancies = await VideoRedundancyModel.listLocalOfServer(serverId)
for (const redundancy of redundancies) {
await removeVideoRedundancy(redundancy)
}
}
async function isRedundancyAccepted (activity: Activity, byActor: MActorSignature) {
const configAcceptFrom = CONFIG.REMOTE_REDUNDANCY.VIDEOS.ACCEPT_FROM
if (configAcceptFrom === 'nobody') {
logger.info('Do not accept remote redundancy %s due instance accept policy.', activity.id, lTags())
return false
}
if (configAcceptFrom === 'followings') {
const serverActor = await getServerActor()
const allowed = await ActorFollowModel.isFollowedBy(byActor.id, serverActor.id)
if (allowed !== true) {
logger.info(
'Do not accept remote redundancy %s because actor %s is not followed by our instance.',
activity.id, byActor.url, lTags()
)
return false
}
}
return true
}
// ---------------------------------------------------------------------------
export {
isRedundancyAccepted,
removeRedundanciesOfServer,
removeVideoRedundancy
}