import * as Sequelize from 'sequelize' import { isVideoChannelNameValid, isVideoChannelDescriptionValid } from '../../helpers' import { removeVideoChannelToFriends } from '../../lib' import { addMethodsToModel, getSort } from '../utils' import { VideoChannelInstance, VideoChannelAttributes, VideoChannelMethods } from './video-channel-interface' let VideoChannel: Sequelize.Model let toFormattedJSON: VideoChannelMethods.ToFormattedJSON let toActivityPubObject: VideoChannelMethods.ToActivityPubObject let isOwned: VideoChannelMethods.IsOwned let countByAccount: VideoChannelMethods.CountByAccount let listOwned: VideoChannelMethods.ListOwned let listForApi: VideoChannelMethods.ListForApi let listByAccount: VideoChannelMethods.ListByAccount let loadByIdAndAccount: VideoChannelMethods.LoadByIdAndAccount let loadByUUID: VideoChannelMethods.LoadByUUID let loadAndPopulateAccount: VideoChannelMethods.LoadAndPopulateAccount let loadByUUIDAndPopulateAccount: VideoChannelMethods.LoadByUUIDAndPopulateAccount let loadByHostAndUUID: VideoChannelMethods.LoadByHostAndUUID let loadAndPopulateAccountAndVideos: VideoChannelMethods.LoadAndPopulateAccountAndVideos export default function (sequelize: Sequelize.Sequelize, DataTypes: Sequelize.DataTypes) { VideoChannel = sequelize.define('VideoChannel', { uuid: { type: DataTypes.UUID, defaultValue: DataTypes.UUIDV4, allowNull: false, validate: { isUUID: 4 } }, name: { type: DataTypes.STRING, allowNull: false, validate: { nameValid: value => { const res = isVideoChannelNameValid(value) if (res === false) throw new Error('Video channel name is not valid.') } } }, description: { type: DataTypes.STRING, allowNull: true, validate: { descriptionValid: value => { const res = isVideoChannelDescriptionValid(value) if (res === false) throw new Error('Video channel description is not valid.') } } }, remote: { type: DataTypes.BOOLEAN, allowNull: false, defaultValue: false }, url: { type: DataTypes.STRING, allowNull: false, validate: { isUrl: true } } }, { indexes: [ { fields: [ 'accountId' ] } ], hooks: { afterDestroy } } ) const classMethods = [ associate, listForApi, listByAccount, listOwned, loadByIdAndAccount, loadAndPopulateAccount, loadByUUIDAndPopulateAccount, loadByUUID, loadByHostAndUUID, loadAndPopulateAccountAndVideos, countByAccount ] const instanceMethods = [ isOwned, toFormattedJSON, toActivityPubObject, ] addMethodsToModel(VideoChannel, classMethods, instanceMethods) return VideoChannel } // ------------------------------ METHODS ------------------------------ isOwned = function (this: VideoChannelInstance) { return this.remote === false } toFormattedJSON = function (this: VideoChannelInstance) { const json = { id: this.id, uuid: this.uuid, name: this.name, description: this.description, isLocal: this.isOwned(), createdAt: this.createdAt, updatedAt: this.updatedAt } if (this.Account !== undefined) { json['owner'] = { name: this.Account.name, uuid: this.Account.uuid } } if (Array.isArray(this.Videos)) { json['videos'] = this.Videos.map(v => v.toFormattedJSON()) } return json } toActivityPubObject = function (this: VideoChannelInstance) { const json = { uuid: this.uuid, name: this.name, description: this.description, createdAt: this.createdAt, updatedAt: this.updatedAt, ownerUUID: this.Account.uuid } return json } // ------------------------------ STATICS ------------------------------ function associate (models) { VideoChannel.belongsTo(models.Account, { foreignKey: { name: 'accountId', allowNull: false }, onDelete: 'CASCADE' }) VideoChannel.hasMany(models.Video, { foreignKey: { name: 'channelId', allowNull: false }, onDelete: 'CASCADE' }) } function afterDestroy (videoChannel: VideoChannelInstance) { if (videoChannel.isOwned()) { const removeVideoChannelToFriendsParams = { uuid: videoChannel.uuid } return removeVideoChannelToFriends(removeVideoChannelToFriendsParams) } return undefined } countByAccount = function (accountId: number) { const query = { where: { accountId } } return VideoChannel.count(query) } listOwned = function () { const query = { where: { remote: false }, include: [ VideoChannel['sequelize'].models.Account ] } return VideoChannel.findAll(query) } listForApi = function (start: number, count: number, sort: string) { const query = { offset: start, limit: count, order: [ getSort(sort) ], include: [ { model: VideoChannel['sequelize'].models.Account, required: true, include: [ { model: VideoChannel['sequelize'].models.Pod, required: false } ] } ] } return VideoChannel.findAndCountAll(query).then(({ rows, count }) => { return { total: count, data: rows } }) } listByAccount = function (accountId: number) { const query = { order: [ getSort('createdAt') ], include: [ { model: VideoChannel['sequelize'].models.Account, where: { id: accountId }, required: true, include: [ { model: VideoChannel['sequelize'].models.Pod, required: false } ] } ] } return VideoChannel.findAndCountAll(query).then(({ rows, count }) => { return { total: count, data: rows } }) } loadByUUID = function (uuid: string, t?: Sequelize.Transaction) { const query: Sequelize.FindOptions = { where: { uuid } } if (t !== undefined) query.transaction = t return VideoChannel.findOne(query) } loadByHostAndUUID = function (fromHost: string, uuid: string, t?: Sequelize.Transaction) { const query: Sequelize.FindOptions = { where: { uuid }, include: [ { model: VideoChannel['sequelize'].models.Account, include: [ { model: VideoChannel['sequelize'].models.Pod, required: true, where: { host: fromHost } } ] } ] } if (t !== undefined) query.transaction = t return VideoChannel.findOne(query) } loadByIdAndAccount = function (id: number, accountId: number) { const options = { where: { id, accountId }, include: [ { model: VideoChannel['sequelize'].models.Account, include: [ { model: VideoChannel['sequelize'].models.Pod, required: false } ] } ] } return VideoChannel.findOne(options) } loadAndPopulateAccount = function (id: number) { const options = { include: [ { model: VideoChannel['sequelize'].models.Account, include: [ { model: VideoChannel['sequelize'].models.Pod, required: false } ] } ] } return VideoChannel.findById(id, options) } loadByUUIDAndPopulateAccount = function (uuid: string) { const options = { where: { uuid }, include: [ { model: VideoChannel['sequelize'].models.Account, include: [ { model: VideoChannel['sequelize'].models.Pod, required: false } ] } ] } return VideoChannel.findOne(options) } loadAndPopulateAccountAndVideos = function (id: number) { const options = { include: [ { model: VideoChannel['sequelize'].models.Account, include: [ { model: VideoChannel['sequelize'].models.Pod, required: false } ] }, VideoChannel['sequelize'].models.Video ] } return VideoChannel.findById(id, options) }