import * as Sequelize from 'sequelize' import { BelongsTo, Column, CreatedAt, ForeignKey, Model, Scopes, Table, UpdatedAt } from 'sequelize-typescript' import { ActorModel } from '../activitypub/actor' import { VideoModel } from './video' enum ScopeNames { FULL = 'FULL', WITH_ACTOR = 'WITH_ACTOR' } @Scopes({ [ScopeNames.FULL]: { include: [ { model: () => ActorModel, required: true }, { model: () => VideoModel, required: true } ] }, [ScopeNames.WITH_ACTOR]: { include: [ { model: () => ActorModel, required: true } ] } }) @Table({ tableName: 'videoShare', indexes: [ { fields: [ 'actorId' ] }, { fields: [ 'videoId' ] } ] }) export class VideoShareModel extends Model { @CreatedAt createdAt: Date @UpdatedAt updatedAt: Date @ForeignKey(() => ActorModel) @Column actorId: number @BelongsTo(() => ActorModel, { foreignKey: { allowNull: false }, onDelete: 'cascade' }) Actor: ActorModel @ForeignKey(() => VideoModel) @Column videoId: number @BelongsTo(() => VideoModel, { foreignKey: { allowNull: false }, onDelete: 'cascade' }) Video: VideoModel static load (actorId: number, videoId: number, t: Sequelize.Transaction) { return VideoShareModel.scope(ScopeNames.WITH_ACTOR).findOne({ where: { actorId, videoId }, transaction: t }) } static loadActorsByShare (videoId: number, t: Sequelize.Transaction) { const query = { where: { videoId }, include: [ { model: ActorModel, required: true } ], transaction: t } return VideoShareModel.scope(ScopeNames.FULL).findAll(query) .then(res => res.map(r => r.Actor)) } }