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

import { isActivityPubUrlValid } from '@server/helpers/custom-validators/activitypub/misc'
import { MIMETYPES } from '@server/initializers/constants'
import { ActorModel } from '@server/models/actor/actor'
import { FilteredModelAttributes } from '@server/types'
import { getLowercaseExtension } from '@shared/core-utils'
import { buildUUID } from '@shared/extra-utils'
import { ActivityIconObject, ActivityPubActor, ActorImageType } from '@shared/models'

function getActorAttributesFromObject (
  actorObject: ActivityPubActor,
  followersCount: number,
  followingCount: number
): FilteredModelAttributes<ActorModel> {
  return {
    type: actorObject.type,
    preferredUsername: actorObject.preferredUsername,
    url: actorObject.id,
    publicKey: actorObject.publicKey.publicKeyPem,
    privateKey: null,
    followersCount,
    followingCount,
    inboxUrl: actorObject.inbox,
    outboxUrl: actorObject.outbox,
    followersUrl: actorObject.followers,
    followingUrl: actorObject.following,

    sharedInboxUrl: actorObject.endpoints?.sharedInbox
      ? actorObject.endpoints.sharedInbox
      : null
  }
}

function getImagesInfoFromObject (actorObject: ActivityPubActor, type: ActorImageType) {
  const iconsOrImages = type === ActorImageType.AVATAR
    ? actorObject.icons || actorObject.icon
    : actorObject.image

  return normalizeIconOrImage(iconsOrImages).map(iconOrImage => {
    const mimetypes = MIMETYPES.IMAGE

    if (iconOrImage.type !== 'Image' || !isActivityPubUrlValid(iconOrImage.url)) return undefined

    let extension: string

    if (iconOrImage.mediaType) {
      extension = mimetypes.MIMETYPE_EXT[iconOrImage.mediaType]
    } else {
      const tmp = getLowercaseExtension(iconOrImage.url)

      if (mimetypes.EXT_MIMETYPE[tmp] !== undefined) extension = tmp
    }

    if (!extension) return undefined

    return {
      name: buildUUID() + extension,
      fileUrl: iconOrImage.url,
      height: iconOrImage.height,
      width: iconOrImage.width,
      type
    }
  })
}

function getActorDisplayNameFromObject (actorObject: ActivityPubActor) {
  return actorObject.name || actorObject.preferredUsername
}

export {
  getActorAttributesFromObject,
  getImagesInfoFromObject,
  getActorDisplayNameFromObject
}

// ---------------------------------------------------------------------------

function normalizeIconOrImage (icon: ActivityIconObject | ActivityIconObject[]): ActivityIconObject[] {
  if (Array.isArray(icon)) return icon
  if (icon) return [ icon ]

  return []
}