123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- import { values } from 'lodash'
- import { FindOptions, Op, Transaction } from 'sequelize'
- import { AllowNull, BelongsTo, Column, CreatedAt, DataType, ForeignKey, Is, Model, Table, UpdatedAt } from 'sequelize-typescript'
- import { VideoRateType } from '../../../shared/models/videos'
- import { CONSTRAINTS_FIELDS, VIDEO_RATE_TYPES } from '../../initializers/constants'
- import { VideoModel } from '../video/video'
- import { AccountModel } from './account'
- import { ActorModel } from '../activitypub/actor'
- import { getSort, throwIfNotValid } from '../utils'
- import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
- import { AccountVideoRate } from '../../../shared'
- import { ScopeNames as VideoChannelScopeNames, VideoChannelModel } from '../video/video-channel'
- /*
- Account rates per video.
- */
- @Table({
- tableName: 'accountVideoRate',
- indexes: [
- {
- fields: [ 'videoId', 'accountId' ],
- unique: true
- },
- {
- fields: [ 'videoId' ]
- },
- {
- fields: [ 'accountId' ]
- },
- {
- fields: [ 'videoId', 'type' ]
- },
- {
- fields: [ 'url' ],
- unique: true
- }
- ]
- })
- export class AccountVideoRateModel extends Model<AccountVideoRateModel> {
- @AllowNull(false)
- @Column(DataType.ENUM(...values(VIDEO_RATE_TYPES)))
- type: VideoRateType
- @AllowNull(false)
- @Is('AccountVideoRateUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'url'))
- @Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEO_RATES.URL.max))
- url: string
- @CreatedAt
- createdAt: Date
- @UpdatedAt
- updatedAt: Date
- @ForeignKey(() => VideoModel)
- @Column
- videoId: number
- @BelongsTo(() => VideoModel, {
- foreignKey: {
- allowNull: false
- },
- onDelete: 'CASCADE'
- })
- Video: VideoModel
- @ForeignKey(() => AccountModel)
- @Column
- accountId: number
- @BelongsTo(() => AccountModel, {
- foreignKey: {
- allowNull: false
- },
- onDelete: 'CASCADE'
- })
- Account: AccountModel
- static load (accountId: number, videoId: number, transaction?: Transaction) {
- const options: FindOptions = {
- where: {
- accountId,
- videoId
- }
- }
- if (transaction) options.transaction = transaction
- return AccountVideoRateModel.findOne(options)
- }
- static listByAccountForApi (options: {
- start: number,
- count: number,
- sort: string,
- type?: string,
- accountId: number
- }) {
- const query: FindOptions = {
- offset: options.start,
- limit: options.count,
- order: getSort(options.sort),
- where: {
- accountId: options.accountId
- },
- include: [
- {
- model: VideoModel,
- required: true,
- include: [
- {
- model: VideoChannelModel.scope({ method: [VideoChannelScopeNames.SUMMARY, true] }),
- required: true
- }
- ]
- }
- ]
- }
- if (options.type) query.where['type'] = options.type
- return AccountVideoRateModel.findAndCountAll(query)
- }
- static loadLocalAndPopulateVideo (rateType: VideoRateType, accountName: string, videoId: number, transaction?: Transaction) {
- const options: FindOptions = {
- where: {
- videoId,
- type: rateType
- },
- include: [
- {
- model: AccountModel.unscoped(),
- required: true,
- include: [
- {
- attributes: [ 'id', 'url', 'preferredUsername' ],
- model: ActorModel.unscoped(),
- required: true,
- where: {
- preferredUsername: accountName
- }
- }
- ]
- },
- {
- model: VideoModel.unscoped(),
- required: true
- }
- ]
- }
- if (transaction) options.transaction = transaction
- return AccountVideoRateModel.findOne(options)
- }
- static loadByUrl (url: string, transaction: Transaction) {
- const options: FindOptions = {
- where: {
- url
- }
- }
- if (transaction) options.transaction = transaction
- return AccountVideoRateModel.findOne(options)
- }
- static listAndCountAccountUrlsByVideoId (rateType: VideoRateType, videoId: number, start: number, count: number, t?: Transaction) {
- const query = {
- offset: start,
- limit: count,
- where: {
- videoId,
- type: rateType
- },
- transaction: t,
- include: [
- {
- attributes: [ 'actorId' ],
- model: AccountModel.unscoped(),
- required: true,
- include: [
- {
- attributes: [ 'url' ],
- model: ActorModel.unscoped(),
- required: true
- }
- ]
- }
- ]
- }
- return AccountVideoRateModel.findAndCountAll(query)
- }
- static cleanOldRatesOf (videoId: number, type: VideoRateType, beforeUpdatedAt: Date) {
- return AccountVideoRateModel.sequelize.transaction(async t => {
- const query = {
- where: {
- updatedAt: {
- [Op.lt]: beforeUpdatedAt
- },
- videoId,
- type
- },
- transaction: t
- }
- const deleted = await AccountVideoRateModel.destroy(query)
- const options = {
- transaction: t,
- where: {
- id: videoId
- }
- }
- if (type === 'like') await VideoModel.increment({ likes: -deleted }, options)
- else if (type === 'dislike') await VideoModel.increment({ dislikes: -deleted }, options)
- })
- }
- toFormattedJSON (): AccountVideoRate {
- return {
- video: this.Video.toFormattedJSON(),
- rating: this.type
- }
- }
- }
|