219 lines
4.7 KiB
TypeScript
219 lines
4.7 KiB
TypeScript
import * as Bluebird from 'bluebird'
|
|
import { AllowNull, BelongsTo, Column, CreatedAt, DataType, ForeignKey, Is, Model, Scopes, Table, UpdatedAt } from 'sequelize-typescript'
|
|
import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
|
|
import { CONSTRAINTS_FIELDS } from '../../initializers/constants'
|
|
import { AccountModel } from '../account/account'
|
|
import { ActorModel } from '../activitypub/actor'
|
|
import { buildLocalActorIdsIn, throwIfNotValid } from '../utils'
|
|
import { VideoModel } from './video'
|
|
import { VideoChannelModel } from './video-channel'
|
|
import { Op, Transaction } from 'sequelize'
|
|
|
|
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<VideoShareModel> {
|
|
|
|
@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: ActorModel
|
|
|
|
@ForeignKey(() => VideoModel)
|
|
@Column
|
|
videoId: number
|
|
|
|
@BelongsTo(() => VideoModel, {
|
|
foreignKey: {
|
|
allowNull: false
|
|
},
|
|
onDelete: 'cascade'
|
|
})
|
|
Video: VideoModel
|
|
|
|
static load (actorId: number, videoId: number, t?: Transaction) {
|
|
return VideoShareModel.scope(ScopeNames.WITH_ACTOR).findOne({
|
|
where: {
|
|
actorId,
|
|
videoId
|
|
},
|
|
transaction: t
|
|
})
|
|
}
|
|
|
|
static loadByUrl (url: string, t: Transaction) {
|
|
return VideoShareModel.scope(ScopeNames.FULL).findOne({
|
|
where: {
|
|
url
|
|
},
|
|
transaction: t
|
|
})
|
|
}
|
|
|
|
static loadActorsByShare (videoId: number, t: 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))
|
|
}
|
|
|
|
static loadActorsWhoSharedVideosOf (actorOwnerId: number, t: Transaction): Bluebird<ActorModel[]> {
|
|
const query = {
|
|
attributes: [],
|
|
include: [
|
|
{
|
|
model: ActorModel,
|
|
required: true
|
|
},
|
|
{
|
|
attributes: [],
|
|
model: VideoModel,
|
|
required: true,
|
|
include: [
|
|
{
|
|
attributes: [],
|
|
model: VideoChannelModel.unscoped(),
|
|
required: true,
|
|
include: [
|
|
{
|
|
attributes: [],
|
|
model: AccountModel.unscoped(),
|
|
required: true,
|
|
where: {
|
|
actorId: actorOwnerId
|
|
}
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
],
|
|
transaction: t
|
|
}
|
|
|
|
return VideoShareModel.scope(ScopeNames.FULL).findAll(query)
|
|
.then(res => res.map(r => r.Actor))
|
|
}
|
|
|
|
static loadActorsByVideoChannel (videoChannelId: number, t: Transaction): Bluebird<ActorModel[]> {
|
|
const query = {
|
|
attributes: [],
|
|
include: [
|
|
{
|
|
model: ActorModel,
|
|
required: true
|
|
},
|
|
{
|
|
attributes: [],
|
|
model: VideoModel,
|
|
required: true,
|
|
where: {
|
|
channelId: videoChannelId
|
|
}
|
|
}
|
|
],
|
|
transaction: t
|
|
}
|
|
|
|
return VideoShareModel.scope(ScopeNames.FULL)
|
|
.findAll(query)
|
|
.then(res => res.map(r => r.Actor))
|
|
}
|
|
|
|
static listAndCountByVideoId (videoId: number, start: number, count: number, t?: Transaction) {
|
|
const query = {
|
|
offset: start,
|
|
limit: count,
|
|
where: {
|
|
videoId
|
|
},
|
|
transaction: t
|
|
}
|
|
|
|
return VideoShareModel.findAndCountAll(query)
|
|
}
|
|
|
|
static cleanOldSharesOf (videoId: number, beforeUpdatedAt: Date) {
|
|
const query = {
|
|
where: {
|
|
updatedAt: {
|
|
[Op.lt]: beforeUpdatedAt
|
|
},
|
|
videoId,
|
|
actorId: {
|
|
[Op.notIn]: buildLocalActorIdsIn()
|
|
}
|
|
}
|
|
}
|
|
|
|
return VideoShareModel.destroy(query)
|
|
}
|
|
}
|