server/lib/activitypub/actors/shared/object-to-model-attributes.ts

   1 import { isActivityPubUrlValid } from '@server/helpers/custom-validators/activitypub/misc'
   2 import { MIMETYPES } from '@server/initializers/constants'
   3 import { ActorModel } from '@server/models/actor/actor'
   4 import { FilteredModelAttributes } from '@server/types'
   5 import { getLowercaseExtension } from '@shared/core-utils'
   6 import { buildUUID } from '@shared/extra-utils'
   7 import { ActivityIconObject, ActivityPubActor, ActorImageType } from '@shared/models'
   8
   9 function getActorAttributesFromObject (
  10   actorObject: ActivityPubActor,
  11   followersCount: number,
  12   followingCount: number
  13 ): FilteredModelAttributes<ActorModel> {
  14   return {
  15     type: actorObject.type,
  16     preferredUsername: actorObject.preferredUsername,
  17     url: actorObject.id,
  18     publicKey: actorObject.publicKey.publicKeyPem,
  19     privateKey: null,
  20     followersCount,
  21     followingCount,
  22     inboxUrl: actorObject.inbox,
  23     outboxUrl: actorObject.outbox,
  24     followersUrl: actorObject.followers,
  25     followingUrl: actorObject.following,
  26
  27     sharedInboxUrl: actorObject.endpoints?.sharedInbox
  28       ? actorObject.endpoints.sharedInbox
  29       : null
  30   }
  31 }
  32
  33 function getImagesInfoFromObject (actorObject: ActivityPubActor, type: ActorImageType) {
  34   const iconsOrImages = type === ActorImageType.AVATAR
  35     ? actorObject.icons || actorObject.icon
  36     : actorObject.image
  37
  38   return normalizeIconOrImage(iconsOrImages)
  39     .map(iconOrImage => {
  40       const mimetypes = MIMETYPES.IMAGE
  41
  42       if (iconOrImage.type !== 'Image' || !isActivityPubUrlValid(iconOrImage.url)) return undefined
  43
  44       let extension: string
  45
  46       if (iconOrImage.mediaType) {
  47         extension = mimetypes.MIMETYPE_EXT[iconOrImage.mediaType]
  48       } else {
  49         const tmp = getLowercaseExtension(iconOrImage.url)
  50
  51         if (mimetypes.EXT_MIMETYPE[tmp] !== undefined) extension = tmp
  52       }
  53
  54       if (!extension) return undefined
  55
  56       return {
  57         name: buildUUID() + extension,
  58         fileUrl: iconOrImage.url,
  59         height: iconOrImage.height,
  60         width: iconOrImage.width,
  61         type
  62       }
  63     })
  64     .filter(i => !!i)
  65 }
  66
  67 function getActorDisplayNameFromObject (actorObject: ActivityPubActor) {
  68   return actorObject.name || actorObject.preferredUsername
  69 }
  70
  71 export {
  72   getActorAttributesFromObject,
  73   getImagesInfoFromObject,
  74   getActorDisplayNameFromObject
  75 }
  76
  77 // ---------------------------------------------------------------------------
  78
  79 function normalizeIconOrImage (icon: ActivityIconObject | ActivityIconObject[]): ActivityIconObject[] {
  80   if (Array.isArray(icon)) return icon
  81   if (icon) return [ icon ]
  82
  83   return []
  84 }