PeerTube/server/models/account/account.ts

237 lines
4.7 KiB
TypeScript
Raw Normal View History

2017-11-09 10:51:58 -06:00
import * as Sequelize from 'sequelize'
2017-12-12 10:53:50 -06:00
import {
AllowNull,
BeforeDestroy,
BelongsTo,
Column,
CreatedAt,
Default,
DefaultScope,
ForeignKey,
HasMany,
Is,
Model,
Table,
2017-12-12 10:53:50 -06:00
UpdatedAt
} from 'sequelize-typescript'
2017-12-29 12:10:13 -06:00
import { Account } from '../../../shared/models/actors'
import { isAccountDescriptionValid } from '../../helpers/custom-validators/accounts'
import { logger } from '../../helpers/logger'
2017-12-14 10:38:41 -06:00
import { sendDeleteActor } from '../../lib/activitypub/send'
2017-12-14 04:18:49 -06:00
import { ActorModel } from '../activitypub/actor'
2017-12-12 10:53:50 -06:00
import { ApplicationModel } from '../application/application'
2018-01-03 09:38:50 -06:00
import { AvatarModel } from '../avatar/avatar'
2017-12-12 10:53:50 -06:00
import { ServerModel } from '../server/server'
import { getSort, throwIfNotValid } from '../utils'
2017-12-12 10:53:50 -06:00
import { VideoChannelModel } from '../video/video-channel'
import { VideoCommentModel } from '../video/video-comment'
2017-12-12 10:53:50 -06:00
import { UserModel } from './user'
2017-12-14 10:38:41 -06:00
@DefaultScope({
include: [
2017-12-12 10:53:50 -06:00
{
2017-12-14 10:38:41 -06:00
model: () => ActorModel,
required: true,
include: [
{
model: () => ServerModel,
required: false
2018-01-03 09:38:50 -06:00
},
{
model: () => AvatarModel,
required: false
2017-12-14 10:38:41 -06:00
}
]
2017-11-09 10:51:58 -06:00
}
]
2017-12-12 10:53:50 -06:00
})
2017-12-14 10:38:41 -06:00
@Table({
tableName: 'account'
})
2017-12-14 04:18:49 -06:00
export class AccountModel extends Model<AccountModel> {
2017-12-12 10:53:50 -06:00
2017-12-14 10:38:41 -06:00
@AllowNull(false)
@Column
name: string
@AllowNull(true)
@Default(null)
@Is('AccountDescription', value => throwIfNotValid(value, isAccountDescriptionValid, 'description'))
@Column
description: string
2017-12-12 10:53:50 -06:00
@CreatedAt
createdAt: Date
@UpdatedAt
updatedAt: Date
2017-12-14 04:18:49 -06:00
@ForeignKey(() => ActorModel)
2017-12-12 10:53:50 -06:00
@Column
2017-12-14 04:18:49 -06:00
actorId: number
2017-11-09 10:51:58 -06:00
2017-12-14 04:18:49 -06:00
@BelongsTo(() => ActorModel, {
2017-11-09 10:51:58 -06:00
foreignKey: {
2017-12-14 04:18:49 -06:00
allowNull: false
2017-11-09 10:51:58 -06:00
},
onDelete: 'cascade'
})
2017-12-14 04:18:49 -06:00
Actor: ActorModel
2017-11-09 10:51:58 -06:00
2017-12-12 10:53:50 -06:00
@ForeignKey(() => UserModel)
@Column
userId: number
@BelongsTo(() => UserModel, {
2017-11-09 10:51:58 -06:00
foreignKey: {
allowNull: true
},
onDelete: 'cascade'
})
2017-12-12 10:53:50 -06:00
User: UserModel
@ForeignKey(() => ApplicationModel)
@Column
applicationId: number
2017-11-09 10:51:58 -06:00
2017-12-12 10:53:50 -06:00
@BelongsTo(() => ApplicationModel, {
2017-11-09 10:51:58 -06:00
foreignKey: {
allowNull: true
},
onDelete: 'cascade'
})
Application: ApplicationModel
2017-11-09 10:51:58 -06:00
2017-12-12 10:53:50 -06:00
@HasMany(() => VideoChannelModel, {
2017-11-09 10:51:58 -06:00
foreignKey: {
allowNull: false
},
onDelete: 'cascade',
hooks: true
})
2017-12-12 10:53:50 -06:00
VideoChannels: VideoChannelModel[]
2017-11-09 10:51:58 -06:00
@HasMany(() => VideoCommentModel, {
foreignKey: {
allowNull: false
},
onDelete: 'cascade',
hooks: true
})
VideoComments: VideoCommentModel[]
@BeforeDestroy
static async sendDeleteIfOwned (instance: AccountModel, options) {
if (!instance.Actor) {
instance.Actor = await instance.$get('Actor', { transaction: options.transaction }) as ActorModel
}
2017-12-12 10:53:50 -06:00
if (instance.isOwned()) {
logger.debug('Sending delete of actor of account %s.', instance.Actor.url)
return sendDeleteActor(instance.Actor, options.transaction)
2017-12-12 10:53:50 -06:00
}
2017-11-09 10:51:58 -06:00
2017-12-12 10:53:50 -06:00
return undefined
2017-11-09 10:51:58 -06:00
}
2017-12-12 10:53:50 -06:00
static load (id: number) {
return AccountModel.findById(id)
}
2017-12-04 03:34:40 -06:00
2017-12-12 10:53:50 -06:00
static loadByUUID (uuid: string) {
const query = {
2017-12-14 10:38:41 -06:00
include: [
{
model: ActorModel,
required: true,
where: {
uuid
}
}
]
2017-12-04 03:34:40 -06:00
}
2017-11-15 04:00:25 -06:00
2017-12-12 10:53:50 -06:00
return AccountModel.findOne(query)
2017-11-15 04:00:25 -06:00
}
2017-12-12 10:53:50 -06:00
static loadLocalByName (name: string) {
const query = {
where: {
name,
[ Sequelize.Op.or ]: [
{
userId: {
[ Sequelize.Op.ne ]: null
}
},
{
applicationId: {
[ Sequelize.Op.ne ]: null
}
}
]
}
}
2017-11-13 10:39:41 -06:00
2017-12-12 10:53:50 -06:00
return AccountModel.findOne(query)
}
2017-11-13 10:39:41 -06:00
2017-12-12 10:53:50 -06:00
static loadByUrl (url: string, transaction?: Sequelize.Transaction) {
const query = {
2017-12-14 04:18:49 -06:00
include: [
{
model: ActorModel,
required: true,
where: {
url
}
}
],
2017-12-12 10:53:50 -06:00
transaction
}
2017-11-09 10:51:58 -06:00
2017-12-12 10:53:50 -06:00
return AccountModel.findOne(query)
}
2017-11-09 10:51:58 -06:00
2018-01-03 09:38:50 -06:00
static listForApi (start: number, count: number, sort: string) {
const query = {
offset: start,
limit: count,
2018-02-19 04:31:50 -06:00
order: getSort(sort)
2018-01-03 09:38:50 -06:00
}
return AccountModel.findAndCountAll(query)
.then(({ rows, count }) => {
return {
data: rows,
total: count
}
})
}
2017-12-29 12:10:13 -06:00
toFormattedJSON (): Account {
2017-12-14 04:18:49 -06:00
const actor = this.Actor.toFormattedJSON()
const account = {
2017-12-12 10:53:50 -06:00
id: this.id,
2017-12-29 12:10:13 -06:00
displayName: this.name,
description: this.description,
2017-12-12 10:53:50 -06:00
createdAt: this.createdAt,
2017-12-14 04:18:49 -06:00
updatedAt: this.updatedAt
2017-12-12 10:53:50 -06:00
}
2017-12-14 04:18:49 -06:00
return Object.assign(actor, account)
2017-12-12 10:53:50 -06:00
}
2017-11-09 10:51:58 -06:00
2017-12-12 10:53:50 -06:00
toActivityPubObject () {
const obj = this.Actor.toActivityPubObject(this.name, 'Account')
return Object.assign(obj, {
summary: this.description
})
2017-11-09 10:51:58 -06:00
}
2017-12-12 10:53:50 -06:00
isOwned () {
2017-12-14 04:18:49 -06:00
return this.Actor.isOwned()
2017-12-12 10:53:50 -06:00
}
}