import { literal, Op, QueryTypes, Transaction } from 'sequelize' import { AllowNull, BelongsTo, Column, CreatedAt, DataType, ForeignKey, Is, Model, Scopes, Table, UpdatedAt } from 'sequelize-typescript' import { forceNumber } from '@peertube/peertube-core-utils' import { AttributesOnly } from '@peertube/peertube-typescript-utils' import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc.js' import { CONSTRAINTS_FIELDS } from '../../initializers/constants.js' import { MActorDefault, MActorFollowersUrl, MActorId } from '../../types/models/index.js' import { MVideoShareActor, MVideoShareFull } from '../../types/models/video/index.js' import { ActorModel } from '../actor/actor.js' import { buildLocalActorIdsIn, throwIfNotValid } from '../shared/index.js' import { VideoModel } from './video.js' 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' ] }, { fields: [ 'url' ], unique: true } ] }) export class VideoShareModel extends Model>> { @AllowNull(false) @Is('VideoShareUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'url')) @Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEO_SHARE.URL.max)) url: string @CreatedAt createdAt: Date @UpdatedAt updatedAt: Date @ForeignKey(() => ActorModel) @Column actorId: number @BelongsTo(() => ActorModel, { foreignKey: { allowNull: false }, onDelete: 'cascade' }) Actor: Awaited @ForeignKey(() => VideoModel) @Column videoId: number @BelongsTo(() => VideoModel, { foreignKey: { allowNull: false }, onDelete: 'cascade' }) Video: Awaited static load (actorId: number | string, videoId: number | string, t?: Transaction): Promise { return VideoShareModel.scope(ScopeNames.WITH_ACTOR).findOne({ where: { actorId, videoId }, transaction: t }) } static loadByUrl (url: string, t: Transaction): Promise { return VideoShareModel.scope(ScopeNames.FULL).findOne({ where: { url }, transaction: t }) } static listActorIdsAndFollowerUrlsByShare (videoId: number, t: Transaction) { const query = `SELECT "actor"."id" AS "id", "actor"."followersUrl" AS "followersUrl" ` + `FROM "videoShare" ` + `INNER JOIN "actor" ON "actor"."id" = "videoShare"."actorId" ` + `WHERE "videoShare"."videoId" = :videoId` const options = { type: QueryTypes.SELECT as QueryTypes.SELECT, replacements: { videoId }, transaction: t } return VideoShareModel.sequelize.query(query, options) } static loadActorsWhoSharedVideosOf (actorOwnerId: number, t: Transaction): Promise { const safeOwnerId = forceNumber(actorOwnerId) // /!\ On actor model const query = { where: { [Op.and]: [ literal( `EXISTS (` + ` SELECT 1 FROM "videoShare" ` + ` INNER JOIN "video" ON "videoShare"."videoId" = "video"."id" ` + ` INNER JOIN "videoChannel" ON "videoChannel"."id" = "video"."channelId" ` + ` INNER JOIN "account" ON "account"."id" = "videoChannel"."accountId" ` + ` WHERE "videoShare"."actorId" = "ActorModel"."id" AND "account"."actorId" = ${safeOwnerId} ` + ` LIMIT 1` + `)` ) ] }, transaction: t } return ActorModel.findAll(query) } static loadActorsByVideoChannel (videoChannelId: number, t: Transaction): Promise { const safeChannelId = forceNumber(videoChannelId) // /!\ On actor model const query = { where: { [Op.and]: [ literal( `EXISTS (` + ` SELECT 1 FROM "videoShare" ` + ` INNER JOIN "video" ON "videoShare"."videoId" = "video"."id" ` + ` WHERE "videoShare"."actorId" = "ActorModel"."id" AND "video"."channelId" = ${safeChannelId} ` + ` LIMIT 1` + `)` ) ] }, transaction: t } return ActorModel.findAll(query) } static listAndCountByVideoId (videoId: number, start: number, count: number, t?: Transaction) { const query = { offset: start, limit: count, where: { videoId }, transaction: t } return Promise.all([ VideoShareModel.count(query), VideoShareModel.findAll(query) ]).then(([ total, data ]) => ({ total, data })) } static listRemoteShareUrlsOfLocalVideos () { const query = `SELECT "videoShare".url FROM "videoShare" ` + `INNER JOIN actor ON actor.id = "videoShare"."actorId" AND actor."serverId" IS NOT NULL ` + `INNER JOIN video ON video.id = "videoShare"."videoId" AND video.remote IS FALSE` return VideoShareModel.sequelize.query<{ url: string }>(query, { type: QueryTypes.SELECT, raw: true }).then(rows => rows.map(r => r.url)) } static cleanOldSharesOf (videoId: number, beforeUpdatedAt: Date) { const query = { where: { updatedAt: { [Op.lt]: beforeUpdatedAt }, videoId, actorId: { [Op.notIn]: buildLocalActorIdsIn() } } } return VideoShareModel.destroy(query) } }