mirror of https://github.com/Chocobozzz/PeerTube
378 lines
9.0 KiB
TypeScript
378 lines
9.0 KiB
TypeScript
import { values } from 'lodash'
|
|
import { extname } from 'path'
|
|
import * as Sequelize from 'sequelize'
|
|
import {
|
|
AllowNull, BelongsTo, Column, CreatedAt, DataType, Default, DefaultScope, ForeignKey, HasMany, HasOne, Is, IsUUID, Model, Scopes,
|
|
Table, UpdatedAt
|
|
} from 'sequelize-typescript'
|
|
import { ActivityPubActorType } from '../../../shared/models/activitypub'
|
|
import { Avatar } from '../../../shared/models/avatars/avatar.model'
|
|
import { activityPubContextify } from '../../helpers/activitypub'
|
|
import {
|
|
isActorFollowersCountValid, isActorFollowingCountValid, isActorPreferredUsernameValid, isActorPrivateKeyValid,
|
|
isActorPublicKeyValid
|
|
} from '../../helpers/custom-validators/activitypub/actor'
|
|
import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
|
|
import { ACTIVITY_PUB_ACTOR_TYPES, CONFIG, CONSTRAINTS_FIELDS } from '../../initializers'
|
|
import { AccountModel } from '../account/account'
|
|
import { AvatarModel } from '../avatar/avatar'
|
|
import { ServerModel } from '../server/server'
|
|
import { throwIfNotValid } from '../utils'
|
|
import { VideoChannelModel } from '../video/video-channel'
|
|
import { ActorFollowModel } from './actor-follow'
|
|
|
|
enum ScopeNames {
|
|
FULL = 'FULL'
|
|
}
|
|
|
|
@DefaultScope({
|
|
include: [
|
|
{
|
|
model: () => ServerModel,
|
|
required: false
|
|
},
|
|
{
|
|
model: () => AvatarModel,
|
|
required: false
|
|
}
|
|
]
|
|
})
|
|
@Scopes({
|
|
[ScopeNames.FULL]: {
|
|
include: [
|
|
{
|
|
model: () => AccountModel,
|
|
required: false
|
|
},
|
|
{
|
|
model: () => VideoChannelModel,
|
|
required: false
|
|
},
|
|
{
|
|
model: () => ServerModel,
|
|
required: false
|
|
},
|
|
{
|
|
model: () => AvatarModel,
|
|
required: false
|
|
}
|
|
]
|
|
}
|
|
})
|
|
@Table({
|
|
tableName: 'actor',
|
|
indexes: [
|
|
{
|
|
fields: [ 'preferredUsername', 'serverId' ],
|
|
unique: true
|
|
}
|
|
]
|
|
})
|
|
export class ActorModel extends Model<ActorModel> {
|
|
|
|
@AllowNull(false)
|
|
@Column(DataType.ENUM(values(ACTIVITY_PUB_ACTOR_TYPES)))
|
|
type: ActivityPubActorType
|
|
|
|
@AllowNull(false)
|
|
@Default(DataType.UUIDV4)
|
|
@IsUUID(4)
|
|
@Column(DataType.UUID)
|
|
uuid: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorPreferredUsername', value => throwIfNotValid(value, isActorPreferredUsernameValid, 'actor preferred username'))
|
|
@Column
|
|
preferredUsername: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'url'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.URL.max))
|
|
url: string
|
|
|
|
@AllowNull(true)
|
|
@Is('ActorPublicKey', value => throwIfNotValid(value, isActorPublicKeyValid, 'public key'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.PUBLIC_KEY.max))
|
|
publicKey: string
|
|
|
|
@AllowNull(true)
|
|
@Is('ActorPublicKey', value => throwIfNotValid(value, isActorPrivateKeyValid, 'private key'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.PRIVATE_KEY.max))
|
|
privateKey: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorFollowersCount', value => throwIfNotValid(value, isActorFollowersCountValid, 'followers count'))
|
|
@Column
|
|
followersCount: number
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorFollowersCount', value => throwIfNotValid(value, isActorFollowingCountValid, 'following count'))
|
|
@Column
|
|
followingCount: number
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorInboxUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'inbox url'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.URL.max))
|
|
inboxUrl: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorOutboxUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'outbox url'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.URL.max))
|
|
outboxUrl: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorSharedInboxUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'shared inbox url'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.URL.max))
|
|
sharedInboxUrl: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorFollowersUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'followers url'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.URL.max))
|
|
followersUrl: string
|
|
|
|
@AllowNull(false)
|
|
@Is('ActorFollowingUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'following url'))
|
|
@Column(DataType.STRING(CONSTRAINTS_FIELDS.ACTORS.URL.max))
|
|
followingUrl: string
|
|
|
|
@CreatedAt
|
|
createdAt: Date
|
|
|
|
@UpdatedAt
|
|
updatedAt: Date
|
|
|
|
@ForeignKey(() => AvatarModel)
|
|
@Column
|
|
avatarId: number
|
|
|
|
@BelongsTo(() => AvatarModel, {
|
|
foreignKey: {
|
|
allowNull: true
|
|
},
|
|
onDelete: 'set null'
|
|
})
|
|
Avatar: AvatarModel
|
|
|
|
@HasMany(() => ActorFollowModel, {
|
|
foreignKey: {
|
|
name: 'actorId',
|
|
allowNull: false
|
|
},
|
|
onDelete: 'cascade'
|
|
})
|
|
AccountFollowing: ActorFollowModel[]
|
|
|
|
@HasMany(() => ActorFollowModel, {
|
|
foreignKey: {
|
|
name: 'targetActorId',
|
|
allowNull: false
|
|
},
|
|
as: 'followers',
|
|
onDelete: 'cascade'
|
|
})
|
|
AccountFollowers: ActorFollowModel[]
|
|
|
|
@ForeignKey(() => ServerModel)
|
|
@Column
|
|
serverId: number
|
|
|
|
@BelongsTo(() => ServerModel, {
|
|
foreignKey: {
|
|
allowNull: true
|
|
},
|
|
onDelete: 'cascade'
|
|
})
|
|
Server: ServerModel
|
|
|
|
@HasOne(() => AccountModel, {
|
|
foreignKey: {
|
|
allowNull: true
|
|
},
|
|
onDelete: 'cascade'
|
|
})
|
|
Account: AccountModel
|
|
|
|
@HasOne(() => VideoChannelModel, {
|
|
foreignKey: {
|
|
allowNull: true
|
|
},
|
|
onDelete: 'cascade'
|
|
})
|
|
VideoChannel: VideoChannelModel
|
|
|
|
static load (id: number) {
|
|
return ActorModel.scope(ScopeNames.FULL).findById(id)
|
|
}
|
|
|
|
static listByFollowersUrls (followersUrls: string[], transaction?: Sequelize.Transaction) {
|
|
const query = {
|
|
where: {
|
|
followersUrl: {
|
|
[ Sequelize.Op.in ]: followersUrls
|
|
}
|
|
},
|
|
transaction
|
|
}
|
|
|
|
return ActorModel.scope(ScopeNames.FULL).findAll(query)
|
|
}
|
|
|
|
static loadLocalByName (preferredUsername: string) {
|
|
const query = {
|
|
where: {
|
|
preferredUsername,
|
|
serverId: null
|
|
}
|
|
}
|
|
|
|
return ActorModel.scope(ScopeNames.FULL).findOne(query)
|
|
}
|
|
|
|
static loadByNameAndHost (preferredUsername: string, host: string) {
|
|
const query = {
|
|
where: {
|
|
preferredUsername
|
|
},
|
|
include: [
|
|
{
|
|
model: ServerModel,
|
|
required: true,
|
|
where: {
|
|
host
|
|
}
|
|
}
|
|
]
|
|
}
|
|
|
|
return ActorModel.scope(ScopeNames.FULL).findOne(query)
|
|
}
|
|
|
|
static loadByUrl (url: string, transaction?: Sequelize.Transaction) {
|
|
const query = {
|
|
where: {
|
|
url
|
|
},
|
|
transaction
|
|
}
|
|
|
|
return ActorModel.scope(ScopeNames.FULL).findOne(query)
|
|
}
|
|
|
|
toFormattedJSON () {
|
|
let avatar: Avatar = null
|
|
if (this.Avatar) {
|
|
avatar = this.Avatar.toFormattedJSON()
|
|
}
|
|
|
|
let score: number
|
|
if (this.Server) {
|
|
score = this.Server.score
|
|
}
|
|
|
|
return {
|
|
id: this.id,
|
|
uuid: this.uuid,
|
|
host: this.getHost(),
|
|
score,
|
|
followingCount: this.followingCount,
|
|
followersCount: this.followersCount,
|
|
avatar
|
|
}
|
|
}
|
|
|
|
toActivityPubObject (name: string, type: 'Account' | 'Application' | 'VideoChannel') {
|
|
let activityPubType
|
|
if (type === 'Account') {
|
|
activityPubType = 'Person' as 'Person'
|
|
} else if (type === 'Application') {
|
|
activityPubType = 'Application' as 'Application'
|
|
} else { // VideoChannel
|
|
activityPubType = 'Group' as 'Group'
|
|
}
|
|
|
|
let icon = undefined
|
|
if (this.avatarId) {
|
|
const extension = extname(this.Avatar.filename)
|
|
icon = {
|
|
type: 'Image',
|
|
mediaType: extension === '.png' ? 'image/png' : 'image/jpeg',
|
|
url: this.getAvatarUrl()
|
|
}
|
|
}
|
|
|
|
const json = {
|
|
type: activityPubType,
|
|
id: this.url,
|
|
following: this.getFollowingUrl(),
|
|
followers: this.getFollowersUrl(),
|
|
inbox: this.inboxUrl,
|
|
outbox: this.outboxUrl,
|
|
preferredUsername: this.preferredUsername,
|
|
url: this.url,
|
|
name,
|
|
endpoints: {
|
|
sharedInbox: this.sharedInboxUrl
|
|
},
|
|
uuid: this.uuid,
|
|
publicKey: {
|
|
id: this.getPublicKeyUrl(),
|
|
owner: this.url,
|
|
publicKeyPem: this.publicKey
|
|
},
|
|
icon
|
|
}
|
|
|
|
return activityPubContextify(json)
|
|
}
|
|
|
|
getFollowerSharedInboxUrls (t: Sequelize.Transaction) {
|
|
const query = {
|
|
attributes: [ 'sharedInboxUrl' ],
|
|
include: [
|
|
{
|
|
model: ActorFollowModel,
|
|
required: true,
|
|
as: 'followers',
|
|
where: {
|
|
targetActorId: this.id
|
|
}
|
|
}
|
|
],
|
|
transaction: t
|
|
}
|
|
|
|
return ActorModel.findAll(query)
|
|
.then(accounts => accounts.map(a => a.sharedInboxUrl))
|
|
}
|
|
|
|
getFollowingUrl () {
|
|
return this.url + '/following'
|
|
}
|
|
|
|
getFollowersUrl () {
|
|
return this.url + '/followers'
|
|
}
|
|
|
|
getPublicKeyUrl () {
|
|
return this.url + '#main-key'
|
|
}
|
|
|
|
isOwned () {
|
|
return this.serverId === null
|
|
}
|
|
|
|
getWebfingerUrl () {
|
|
return 'acct:' + this.preferredUsername + '@' + this.getHost()
|
|
}
|
|
|
|
getHost () {
|
|
return this.Server ? this.Server.host : CONFIG.WEBSERVER.HOST
|
|
}
|
|
|
|
getAvatarUrl () {
|
|
if (!this.avatarId) return undefined
|
|
|
|
return CONFIG.WEBSERVER.URL + this.Avatar.getWebserverPath
|
|
}
|
|
}
|