PeerTube/server/models/video/video-comment.ts

411 lines
10 KiB
TypeScript
Raw Normal View History

import * as Sequelize from 'sequelize'
import {
AllowNull, BeforeDestroy, BelongsTo, Column, CreatedAt, DataType, ForeignKey, IFindOptions, Is, Model, Scopes, Table,
UpdatedAt
} from 'sequelize-typescript'
2018-01-05 11:19:25 +01:00
import { ActivityTagObject } from '../../../shared/models/activitypub/objects/common-objects'
import { VideoCommentObject } from '../../../shared/models/activitypub/objects/video-comment-object'
2017-12-22 10:50:07 +01:00
import { VideoComment } from '../../../shared/models/videos/video-comment.model'
2017-12-28 11:16:08 +01:00
import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
import { CONSTRAINTS_FIELDS } from '../../initializers'
2018-01-04 11:19:16 +01:00
import { sendDeleteVideoComment } from '../../lib/activitypub/send'
2017-12-22 12:10:40 +01:00
import { AccountModel } from '../account/account'
2017-12-27 16:11:53 +01:00
import { ActorModel } from '../activitypub/actor'
2018-01-03 17:25:47 +01:00
import { AvatarModel } from '../avatar/avatar'
2017-12-27 16:11:53 +01:00
import { ServerModel } from '../server/server'
2017-12-22 10:50:07 +01:00
import { getSort, throwIfNotValid } from '../utils'
import { VideoModel } from './video'
2018-01-04 11:19:16 +01:00
import { VideoChannelModel } from './video-channel'
2017-12-22 10:50:07 +01:00
enum ScopeNames {
WITH_ACCOUNT = 'WITH_ACCOUNT',
2017-12-27 16:11:53 +01:00
WITH_IN_REPLY_TO = 'WITH_IN_REPLY_TO',
2017-12-28 11:16:08 +01:00
WITH_VIDEO = 'WITH_VIDEO',
2017-12-27 16:11:53 +01:00
ATTRIBUTES_FOR_API = 'ATTRIBUTES_FOR_API'
2017-12-22 10:50:07 +01:00
}
@Scopes({
2017-12-27 16:11:53 +01:00
[ScopeNames.ATTRIBUTES_FOR_API]: {
attributes: {
include: [
[
Sequelize.literal(
'(SELECT COUNT("replies"."id") ' +
'FROM "videoComment" AS "replies" ' +
'WHERE "replies"."originCommentId" = "VideoCommentModel"."id")'
),
'totalReplies'
]
]
}
},
2017-12-22 12:10:40 +01:00
[ScopeNames.WITH_ACCOUNT]: {
2017-12-22 10:50:07 +01:00
include: [
2017-12-27 16:11:53 +01:00
{
model: () => AccountModel,
include: [
{
model: () => ActorModel,
include: [
{
model: () => ServerModel,
required: false
2018-01-03 17:25:47 +01:00
},
{
model: () => AvatarModel,
required: false
2017-12-27 16:11:53 +01:00
}
]
}
]
}
2017-12-22 10:50:07 +01:00
]
},
[ScopeNames.WITH_IN_REPLY_TO]: {
include: [
{
model: () => VideoCommentModel,
2017-12-28 11:16:08 +01:00
as: 'InReplyToVideoComment'
}
]
},
[ScopeNames.WITH_VIDEO]: {
include: [
{
model: () => VideoModel,
2018-01-04 11:19:16 +01:00
required: true,
include: [
{
model: () => VideoChannelModel.unscoped(),
required: true,
include: [
{
model: () => AccountModel,
required: true,
include: [
{
model: () => ActorModel,
required: true
}
]
}
]
}
]
}
]
2017-12-22 10:50:07 +01:00
}
})
@Table({
tableName: 'videoComment',
indexes: [
{
fields: [ 'videoId' ]
2017-12-22 10:50:07 +01:00
},
{
fields: [ 'videoId', 'originCommentId' ]
2018-01-26 14:14:43 +01:00
},
{
fields: [ 'url' ],
unique: true
}
]
})
export class VideoCommentModel extends Model<VideoCommentModel> {
@CreatedAt
createdAt: Date
@UpdatedAt
updatedAt: Date
@AllowNull(false)
@Is('VideoCommentUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'url'))
@Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEOS.URL.max))
url: string
@AllowNull(false)
@Column(DataType.TEXT)
text: string
@ForeignKey(() => VideoCommentModel)
@Column
originCommentId: number
@BelongsTo(() => VideoCommentModel, {
foreignKey: {
2017-12-28 11:45:10 +01:00
name: 'originCommentId',
allowNull: true
},
2017-12-28 11:45:10 +01:00
as: 'OriginVideoComment',
onDelete: 'CASCADE'
})
OriginVideoComment: VideoCommentModel
@ForeignKey(() => VideoCommentModel)
@Column
inReplyToCommentId: number
@BelongsTo(() => VideoCommentModel, {
foreignKey: {
2017-12-28 11:45:10 +01:00
name: 'inReplyToCommentId',
allowNull: true
},
2017-12-28 11:16:08 +01:00
as: 'InReplyToVideoComment',
onDelete: 'CASCADE'
})
InReplyToVideoComment: VideoCommentModel
@ForeignKey(() => VideoModel)
@Column
videoId: number
@BelongsTo(() => VideoModel, {
foreignKey: {
allowNull: false
},
onDelete: 'CASCADE'
})
Video: VideoModel
2017-12-22 12:10:40 +01:00
@ForeignKey(() => AccountModel)
@Column
2017-12-22 12:10:40 +01:00
accountId: number
2017-12-22 12:10:40 +01:00
@BelongsTo(() => AccountModel, {
foreignKey: {
allowNull: false
},
onDelete: 'CASCADE'
})
2017-12-22 12:10:40 +01:00
Account: AccountModel
@BeforeDestroy
static async sendDeleteIfOwned (instance: VideoCommentModel, options) {
if (!instance.Account || !instance.Account.Actor) {
instance.Account = await instance.$get('Account', {
include: [ ActorModel ],
transaction: options.transaction
}) as AccountModel
}
2018-02-13 13:35:10 +01:00
if (!instance.Video) {
instance.Video = await instance.$get('Video', {
include: [
{
model: VideoChannelModel,
include: [
{
model: AccountModel,
include: [
{
model: ActorModel
}
]
}
]
}
],
transaction: options.transaction
}) as VideoModel
}
2018-01-04 11:19:16 +01:00
if (instance.isOwned()) {
await sendDeleteVideoComment(instance, options.transaction)
2018-01-04 11:19:16 +01:00
}
2017-12-22 10:50:07 +01:00
}
static loadById (id: number, t?: Sequelize.Transaction) {
const query: IFindOptions<VideoCommentModel> = {
where: {
id
}
}
if (t !== undefined) query.transaction = t
return VideoCommentModel.findOne(query)
}
2017-12-28 11:16:08 +01:00
static loadByIdAndPopulateVideoAndAccountAndReply (id: number, t?: Sequelize.Transaction) {
const query: IFindOptions<VideoCommentModel> = {
where: {
id
}
}
if (t !== undefined) query.transaction = t
return VideoCommentModel
.scope([ ScopeNames.WITH_VIDEO, ScopeNames.WITH_ACCOUNT, ScopeNames.WITH_IN_REPLY_TO ])
.findOne(query)
}
2018-01-10 17:18:12 +01:00
static loadByUrlAndPopulateAccount (url: string, t?: Sequelize.Transaction) {
const query: IFindOptions<VideoCommentModel> = {
where: {
url
}
}
if (t !== undefined) query.transaction = t
2018-01-10 17:18:12 +01:00
return VideoCommentModel.scope([ ScopeNames.WITH_ACCOUNT ]).findOne(query)
}
2017-12-22 10:50:07 +01:00
2018-01-10 17:18:12 +01:00
static loadByUrlAndPopulateReplyAndVideo (url: string, t?: Sequelize.Transaction) {
2018-01-04 11:19:16 +01:00
const query: IFindOptions<VideoCommentModel> = {
where: {
url
}
}
if (t !== undefined) query.transaction = t
2018-01-10 17:18:12 +01:00
return VideoCommentModel.scope([ ScopeNames.WITH_IN_REPLY_TO, ScopeNames.WITH_VIDEO ]).findOne(query)
2018-01-04 11:19:16 +01:00
}
2017-12-22 10:50:07 +01:00
static listThreadsForApi (videoId: number, start: number, count: number, sort: string) {
const query = {
offset: start,
limit: count,
2018-02-19 09:41:03 +01:00
order: getSort(sort),
2017-12-22 10:50:07 +01:00
where: {
2017-12-22 12:10:40 +01:00
videoId,
inReplyToCommentId: null
2017-12-22 10:50:07 +01:00
}
}
return VideoCommentModel
2017-12-27 16:11:53 +01:00
.scope([ ScopeNames.WITH_ACCOUNT, ScopeNames.ATTRIBUTES_FOR_API ])
2017-12-22 10:50:07 +01:00
.findAndCountAll(query)
.then(({ rows, count }) => {
return { total: count, data: rows }
})
}
static listThreadCommentsForApi (videoId: number, threadId: number) {
const query = {
order: [ [ 'createdAt', 'ASC' ], [ 'updatedAt', 'ASC' ] ],
2017-12-22 10:50:07 +01:00
where: {
videoId,
[ Sequelize.Op.or ]: [
{ id: threadId },
{ originCommentId: threadId }
]
}
}
return VideoCommentModel
2017-12-27 16:11:53 +01:00
.scope([ ScopeNames.WITH_ACCOUNT, ScopeNames.ATTRIBUTES_FOR_API ])
2017-12-22 10:50:07 +01:00
.findAndCountAll(query)
.then(({ rows, count }) => {
return { total: count, data: rows }
})
}
2018-01-10 17:18:12 +01:00
static listThreadParentComments (comment: VideoCommentModel, t: Sequelize.Transaction, order: 'ASC' | 'DESC' = 'ASC') {
2018-01-05 11:19:25 +01:00
const query = {
2018-01-10 17:18:12 +01:00
order: [ [ 'createdAt', order ] ],
2018-01-05 11:19:25 +01:00
where: {
id: {
2018-03-21 11:17:01 +01:00
[ Sequelize.Op.in ]: Sequelize.literal('(' +
'WITH RECURSIVE children (id, "inReplyToCommentId") AS ( ' +
'SELECT id, "inReplyToCommentId" FROM "videoComment" WHERE id = ' + comment.id + ' UNION ' +
'SELECT p.id, p."inReplyToCommentId" from "videoComment" p ' +
'INNER JOIN children c ON c."inReplyToCommentId" = p.id) ' +
'SELECT id FROM children' +
')'),
2018-01-05 11:19:25 +01:00
[ Sequelize.Op.ne ]: comment.id
}
},
transaction: t
}
return VideoCommentModel
.scope([ ScopeNames.WITH_ACCOUNT ])
.findAll(query)
}
2018-02-28 18:04:46 +01:00
static async getStats () {
const totalLocalVideoComments = await VideoCommentModel.count({
include: [
{
model: AccountModel,
required: true,
include: [
{
model: ActorModel,
required: true,
where: {
serverId: null
}
}
]
}
]
})
const totalVideoComments = await VideoCommentModel.count()
return {
totalLocalVideoComments,
totalVideoComments
}
}
2018-01-05 11:19:25 +01:00
getThreadId (): number {
return this.originCommentId || this.id
}
2018-01-04 11:19:16 +01:00
isOwned () {
return this.Account.isOwned()
}
2017-12-22 10:50:07 +01:00
toFormattedJSON () {
return {
id: this.id,
url: this.url,
text: this.text,
threadId: this.originCommentId || this.id,
2017-12-27 20:03:37 +01:00
inReplyToCommentId: this.inReplyToCommentId || null,
2017-12-22 10:50:07 +01:00
videoId: this.videoId,
createdAt: this.createdAt,
2017-12-22 12:10:40 +01:00
updatedAt: this.updatedAt,
2017-12-27 16:11:53 +01:00
totalReplies: this.get('totalReplies') || 0,
2018-01-03 17:25:47 +01:00
account: this.Account.toFormattedJSON()
2017-12-22 10:50:07 +01:00
} as VideoComment
}
2018-01-05 11:19:25 +01:00
toActivityPubObject (threadParentComments: VideoCommentModel[]): VideoCommentObject {
let inReplyTo: string
// New thread, so in AS we reply to the video
if (this.inReplyToCommentId === null) {
inReplyTo = this.Video.url
} else {
inReplyTo = this.InReplyToVideoComment.url
}
2018-01-05 11:19:25 +01:00
const tag: ActivityTagObject[] = []
for (const parentComment of threadParentComments) {
const actor = parentComment.Account.Actor
tag.push({
type: 'Mention',
href: actor.url,
name: `@${actor.preferredUsername}@${actor.getHost()}`
})
}
return {
type: 'Note' as 'Note',
id: this.url,
content: this.text,
inReplyTo,
2017-12-28 11:16:08 +01:00
updated: this.updatedAt.toISOString(),
published: this.createdAt.toISOString(),
2017-12-28 11:16:08 +01:00
url: this.url,
2018-01-05 11:19:25 +01:00
attributedTo: this.Account.Actor.url,
tag
}
}
}