1 import * as Bluebird from 'bluebird'
2 import { extname } from 'path'
3 import { Op, Transaction } from 'sequelize'
4 import { URL } from 'url'
5 import { v4 as uuidv4 } from 'uuid'
6 import { getServerActor } from '@server/models/application/application'
7 import { ActorImageType } from '@shared/models'
8 import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes'
9 import { ActivityPubActor, ActivityPubActorType, ActivityPubOrderedCollection } from '../../../shared/models/activitypub'
10 import { ActivityPubAttributedTo } from '../../../shared/models/activitypub/objects'
11 import { checkUrlsSameHost, getAPId } from '../../helpers/activitypub'
12 import { ActorFetchByUrlType, fetchActorByUrl } from '../../helpers/actor'
13 import { sanitizeAndCheckActorObject } from '../../helpers/custom-validators/activitypub/actor'
14 import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
15 import { retryTransactionWrapper, updateInstanceWithAnother } from '../../helpers/database-utils'
16 import { logger } from '../../helpers/logger'
17 import { createPrivateAndPublicKeys } from '../../helpers/peertube-crypto'
18 import { doJSONRequest, PeerTubeRequestError } from '../../helpers/requests'
19 import { getUrlFromWebfinger } from '../../helpers/webfinger'
20 import { MIMETYPES, WEBSERVER } from '../../initializers/constants'
21 import { sequelizeTypescript } from '../../initializers/database'
22 import { AccountModel } from '../../models/account/account'
23 import { ActorImageModel } from '../../models/account/actor-image'
24 import { ActorModel } from '../../models/activitypub/actor'
25 import { ServerModel } from '../../models/server/server'
26 import { VideoChannelModel } from '../../models/video/video-channel'
31 MActorAccountChannelId,
32 MActorAccountChannelIdActor,
40 } from '../../types/models'
41 import { JobQueue } from '../job-queue'
43 // Set account keys, this could be long so process after the account creation and do not block the client
44 async function generateAndSaveActorKeys <T extends MActor> (actor: T) {
45 const { publicKey, privateKey } = await createPrivateAndPublicKeys()
47 actor.publicKey = publicKey
48 actor.privateKey = privateKey
53 function getOrCreateActorAndServerAndModel (
54 activityActor: string | ActivityPubActor,
56 recurseIfNeeded?: boolean,
57 updateCollections?: boolean
58 ): Promise<MActorFullActor>
60 function getOrCreateActorAndServerAndModel (
61 activityActor: string | ActivityPubActor,
62 fetchType?: 'association-ids',
63 recurseIfNeeded?: boolean,
64 updateCollections?: boolean
65 ): Promise<MActorAccountChannelId>
67 async function getOrCreateActorAndServerAndModel (
68 activityActor: string | ActivityPubActor,
69 fetchType: ActorFetchByUrlType = 'association-ids',
70 recurseIfNeeded = true,
71 updateCollections = false
72 ): Promise<MActorFullActor | MActorAccountChannelId> {
73 const actorUrl = getAPId(activityActor)
75 let accountPlaylistsUrl: string
77 let actor = await fetchActorByUrl(actorUrl, fetchType)
78 // Orphan actor (not associated to an account of channel) so recreate it
79 if (actor && (!actor.Account && !actor.VideoChannel)) {
84 // We don't have this actor in our database, fetch it on remote
86 const { result } = await fetchRemoteActor(actorUrl)
87 if (result === undefined) throw new Error('Cannot fetch remote actor ' + actorUrl)
89 // Create the attributed to actor
90 // In PeerTube a video channel is owned by an account
91 let ownerActor: MActorFullActor
92 if (recurseIfNeeded === true && result.actor.type === 'Group') {
93 const accountAttributedTo = result.attributedTo.find(a => a.type === 'Person')
94 if (!accountAttributedTo) throw new Error('Cannot find account attributed to video channel ' + actor.url)
96 if (checkUrlsSameHost(accountAttributedTo.id, actorUrl) !== true) {
97 throw new Error(`Account attributed to ${accountAttributedTo.id} does not have the same host than actor url ${actorUrl}`)
101 // Don't recurse another time
102 const recurseIfNeeded = false
103 ownerActor = await getOrCreateActorAndServerAndModel(accountAttributedTo.id, 'all', recurseIfNeeded)
105 logger.error('Cannot get or create account attributed to video channel ' + actorUrl)
110 actor = await retryTransactionWrapper(saveActorAndServerAndModelIfNotExist, result, ownerActor)
112 accountPlaylistsUrl = result.playlists
115 if (actor.Account) (actor as MActorAccountChannelIdActor).Account.Actor = actor
116 if (actor.VideoChannel) (actor as MActorAccountChannelIdActor).VideoChannel.Actor = actor
118 const { actor: actorRefreshed, refreshed } = await retryTransactionWrapper(refreshActorIfNeeded, actor, fetchType)
119 if (!actorRefreshed) throw new Error('Actor ' + actor.url + ' does not exist anymore.')
121 if ((created === true || refreshed === true) && updateCollections === true) {
122 const payload = { uri: actor.outboxUrl, type: 'activity' as 'activity' }
123 await JobQueue.Instance.createJobWithPromise({ type: 'activitypub-http-fetcher', payload })
126 // We created a new account: fetch the playlists
127 if (created === true && actor.Account && accountPlaylistsUrl) {
128 const payload = { uri: accountPlaylistsUrl, accountId: actor.Account.id, type: 'account-playlists' as 'account-playlists' }
129 await JobQueue.Instance.createJobWithPromise({ type: 'activitypub-http-fetcher', payload })
132 return actorRefreshed
135 function buildActorInstance (type: ActivityPubActorType, url: string, preferredUsername: string, uuid?: string) {
136 return new ActorModel({
145 inboxUrl: url + '/inbox',
146 outboxUrl: url + '/outbox',
147 sharedInboxUrl: WEBSERVER.URL + '/inbox',
148 followersUrl: url + '/followers',
149 followingUrl: url + '/following'
153 async function updateActorInstance (actorInstance: ActorModel, attributes: ActivityPubActor) {
154 const followersCount = await fetchActorTotalItems(attributes.followers)
155 const followingCount = await fetchActorTotalItems(attributes.following)
157 actorInstance.type = attributes.type
158 actorInstance.preferredUsername = attributes.preferredUsername
159 actorInstance.url = attributes.id
160 actorInstance.publicKey = attributes.publicKey.publicKeyPem
161 actorInstance.followersCount = followersCount
162 actorInstance.followingCount = followingCount
163 actorInstance.inboxUrl = attributes.inbox
164 actorInstance.outboxUrl = attributes.outbox
165 actorInstance.followersUrl = attributes.followers
166 actorInstance.followingUrl = attributes.following
168 if (attributes.endpoints?.sharedInbox) {
169 actorInstance.sharedInboxUrl = attributes.endpoints.sharedInbox
180 async function updateActorImageInstance (actor: MActorImages, type: ActorImageType, imageInfo: ImageInfo | null, t: Transaction) {
181 const oldImageModel = type === ActorImageType.AVATAR
186 // Don't update the avatar if the file URL did not change
187 if (imageInfo?.fileUrl && oldImageModel.fileUrl === imageInfo.fileUrl) return actor
190 await oldImageModel.destroy({ transaction: t })
192 setActorImage(actor, type, null)
194 logger.error('Cannot remove old actor image of actor %s.', actor.url, { err })
199 const imageModel = await ActorImageModel.create({
200 filename: imageInfo.name,
201 onDisk: imageInfo.onDisk ?? false,
202 fileUrl: imageInfo.fileUrl,
203 height: imageInfo.height,
204 width: imageInfo.width,
206 }, { transaction: t })
208 setActorImage(actor, type, imageModel)
214 async function deleteActorImageInstance (actor: MActorImages, type: ActorImageType, t: Transaction) {
216 if (type === ActorImageType.AVATAR) {
217 await actor.Avatar.destroy({ transaction: t })
219 actor.avatarId = null
222 await actor.Banner.destroy({ transaction: t })
224 actor.bannerId = null
228 logger.error('Cannot remove old image of actor %s.', actor.url, { err })
234 async function fetchActorTotalItems (url: string) {
236 const { body } = await doJSONRequest<ActivityPubOrderedCollection<unknown>>(url, { activityPub: true })
238 return body.totalItems || 0
240 logger.warn('Cannot fetch remote actor count %s.', url, { err })
245 function getImageInfoIfExists (actorJSON: ActivityPubActor, type: ActorImageType) {
246 const mimetypes = MIMETYPES.IMAGE
247 const icon = type === ActorImageType.AVATAR
251 if (!icon || icon.type !== 'Image' || !isActivityPubUrlValid(icon.url)) return undefined
253 let extension: string
255 if (icon.mediaType) {
256 extension = mimetypes.MIMETYPE_EXT[icon.mediaType]
258 const tmp = extname(icon.url)
260 if (mimetypes.EXT_MIMETYPE[tmp] !== undefined) extension = tmp
263 if (!extension) return undefined
266 name: uuidv4() + extension,
274 async function addFetchOutboxJob (actor: Pick<ActorModel, 'id' | 'outboxUrl'>) {
275 // Don't fetch ourselves
276 const serverActor = await getServerActor()
277 if (serverActor.id === actor.id) {
278 logger.error('Cannot fetch our own outbox!')
283 uri: actor.outboxUrl,
284 type: 'activity' as 'activity'
287 return JobQueue.Instance.createJob({ type: 'activitypub-http-fetcher', payload })
290 async function refreshActorIfNeeded <T extends MActorFull | MActorAccountChannelId> (
292 fetchedType: ActorFetchByUrlType
293 ): Promise<{ actor: T | MActorFull, refreshed: boolean }> {
294 if (!actorArg.isOutdated()) return { actor: actorArg, refreshed: false }
296 // We need more attributes
297 const actor = fetchedType === 'all'
298 ? actorArg as MActorFull
299 : await ActorModel.loadByUrlAndPopulateAccountAndChannel(actorArg.url)
304 actorUrl = await getUrlFromWebfinger(actor.preferredUsername + '@' + actor.getHost())
306 logger.warn('Cannot get actor URL from webfinger, keeping the old one.', err)
310 const { result } = await fetchRemoteActor(actorUrl)
312 if (result === undefined) {
313 logger.warn('Cannot fetch remote actor in refresh actor.')
314 return { actor, refreshed: false }
317 return sequelizeTypescript.transaction(async t => {
318 updateInstanceWithAnother(actor, result.actor)
320 await updateActorImageInstance(actor, ActorImageType.AVATAR, result.avatar, t)
321 await updateActorImageInstance(actor, ActorImageType.BANNER, result.banner, t)
324 actor.setDataValue('updatedAt', new Date())
325 await actor.save({ transaction: t })
328 actor.Account.name = result.name
329 actor.Account.description = result.summary
331 await actor.Account.save({ transaction: t })
332 } else if (actor.VideoChannel) {
333 actor.VideoChannel.name = result.name
334 actor.VideoChannel.description = result.summary
335 actor.VideoChannel.support = result.support
337 await actor.VideoChannel.save({ transaction: t })
340 return { refreshed: true, actor }
343 if ((err as PeerTubeRequestError).statusCode === HttpStatusCode.NOT_FOUND_404) {
344 logger.info('Deleting actor %s because there is a 404 in refresh actor.', actor.url)
346 ? await actor.Account.destroy()
347 : await actor.VideoChannel.destroy()
349 return { actor: undefined, refreshed: false }
352 logger.warn('Cannot refresh actor %s.', actor.url, { err })
353 return { actor, refreshed: false }
358 getOrCreateActorAndServerAndModel,
360 generateAndSaveActorKeys,
361 fetchActorTotalItems,
362 getImageInfoIfExists,
364 deleteActorImageInstance,
365 refreshActorIfNeeded,
366 updateActorImageInstance,
370 // ---------------------------------------------------------------------------
372 function setActorImage (actorModel: MActorImages, type: ActorImageType, imageModel: MActorImage) {
373 const id = imageModel
377 if (type === ActorImageType.AVATAR) {
378 actorModel.avatarId = id
379 actorModel.Avatar = imageModel
381 actorModel.bannerId = id
382 actorModel.Banner = imageModel
388 function saveActorAndServerAndModelIfNotExist (
389 result: FetchRemoteActorResult,
390 ownerActor?: MActorFullActor,
392 ): Bluebird<MActorFullActor> | Promise<MActorFullActor> {
393 const actor = result.actor
395 if (t !== undefined) return save(t)
397 return sequelizeTypescript.transaction(t => save(t))
399 async function save (t: Transaction) {
400 const actorHost = new URL(actor.url).host
402 const serverOptions = {
411 const [ server ] = await ServerModel.findOrCreate(serverOptions)
413 // Save our new account in database
414 actor.serverId = server.id
418 const avatar = await ActorImageModel.create({
419 filename: result.avatar.name,
420 fileUrl: result.avatar.fileUrl,
421 width: result.avatar.width,
422 height: result.avatar.height,
424 type: ActorImageType.AVATAR
425 }, { transaction: t })
427 actor.avatarId = avatar.id
432 const banner = await ActorImageModel.create({
433 filename: result.banner.name,
434 fileUrl: result.banner.fileUrl,
435 width: result.banner.width,
436 height: result.banner.height,
438 type: ActorImageType.BANNER
439 }, { transaction: t })
441 actor.bannerId = banner.id
444 // Force the actor creation, sometimes Sequelize skips the save() when it thinks the instance already exists
445 // (which could be false in a retried query)
446 const [ actorCreated, created ] = await ActorModel.findOrCreate<MActorFullActor>({
447 defaults: actor.toJSON(),
454 serverId: actor.serverId,
455 preferredUsername: actor.preferredUsername
462 // Try to fix non HTTPS accounts of remote instances that fixed their URL afterwards
463 if (created !== true && actorCreated.url !== actor.url) {
464 // Only fix http://example.com/account/djidane to https://example.com/account/djidane
465 if (actorCreated.url.replace(/^http:\/\//, '') !== actor.url.replace(/^https:\/\//, '')) {
466 throw new Error(`Actor from DB with URL ${actorCreated.url} does not correspond to actor ${actor.url}`)
469 actorCreated.url = actor.url
470 await actorCreated.save({ transaction: t })
473 if (actorCreated.type === 'Person' || actorCreated.type === 'Application') {
474 actorCreated.Account = await saveAccount(actorCreated, result, t) as MAccountDefault
475 actorCreated.Account.Actor = actorCreated
476 } else if (actorCreated.type === 'Group') { // Video channel
477 const channel = await saveVideoChannel(actorCreated, result, ownerActor, t)
478 actorCreated.VideoChannel = Object.assign(channel, { Actor: actorCreated, Account: ownerActor.Account })
481 actorCreated.Server = server
494 type FetchRemoteActorResult = {
502 attributedTo: ActivityPubAttributedTo[]
504 async function fetchRemoteActor (actorUrl: string): Promise<{ statusCode?: number, result: FetchRemoteActorResult }> {
505 logger.info('Fetching remote actor %s.', actorUrl)
507 const requestResult = await doJSONRequest<ActivityPubActor>(actorUrl, { activityPub: true })
508 const actorJSON = requestResult.body
510 if (sanitizeAndCheckActorObject(actorJSON) === false) {
511 logger.debug('Remote actor JSON is not valid.', { actorJSON })
512 return { result: undefined, statusCode: requestResult.statusCode }
515 if (checkUrlsSameHost(actorJSON.id, actorUrl) !== true) {
516 logger.warn('Actor url %s has not the same host than its AP id %s', actorUrl, actorJSON.id)
517 return { result: undefined, statusCode: requestResult.statusCode }
520 const followersCount = await fetchActorTotalItems(actorJSON.followers)
521 const followingCount = await fetchActorTotalItems(actorJSON.following)
523 const actor = new ActorModel({
524 type: actorJSON.type,
525 preferredUsername: actorJSON.preferredUsername,
527 publicKey: actorJSON.publicKey.publicKeyPem,
529 followersCount: followersCount,
530 followingCount: followingCount,
531 inboxUrl: actorJSON.inbox,
532 outboxUrl: actorJSON.outbox,
533 followersUrl: actorJSON.followers,
534 followingUrl: actorJSON.following,
536 sharedInboxUrl: actorJSON.endpoints?.sharedInbox
537 ? actorJSON.endpoints.sharedInbox
541 const avatarInfo = getImageInfoIfExists(actorJSON, ActorImageType.AVATAR)
542 const bannerInfo = getImageInfoIfExists(actorJSON, ActorImageType.BANNER)
544 const name = actorJSON.name || actorJSON.preferredUsername
546 statusCode: requestResult.statusCode,
552 summary: actorJSON.summary,
553 support: actorJSON.support,
554 playlists: actorJSON.playlists,
555 attributedTo: actorJSON.attributedTo
560 async function saveAccount (actor: MActorId, result: FetchRemoteActorResult, t: Transaction) {
561 const [ accountCreated ] = await AccountModel.findOrCreate({
564 description: result.summary,
573 return accountCreated as MAccount
576 async function saveVideoChannel (actor: MActorId, result: FetchRemoteActorResult, ownerActor: MActorAccountId, t: Transaction) {
577 const [ videoChannelCreated ] = await VideoChannelModel.findOrCreate({
580 description: result.summary,
581 support: result.support,
583 accountId: ownerActor.Account.id
591 return videoChannelCreated as MChannel