server/lib/video-comment.ts

   1 import * as Sequelize from 'sequelize'
   2 import { ResultList } from '../../shared/models'
   3 import { VideoCommentThreadTree } from '../../shared/models/videos/video-comment.model'
   4 import { AccountModel } from '../models/account/account'
   5 import { VideoModel } from '../models/video/video'
   6 import { VideoCommentModel } from '../models/video/video-comment'
   7 import { getVideoCommentActivityPubUrl } from './activitypub'
   8 import { sendCreateVideoComment } from './activitypub/send'
   9
  10 async function createVideoComment (obj: {
  11   text: string,
  12   inReplyToComment: VideoCommentModel,
  13   video: VideoModel
  14   account: AccountModel
  15 }, t: Sequelize.Transaction) {
  16   let originCommentId: number = null
  17   let inReplyToCommentId: number = null
  18
  19   if (obj.inReplyToComment) {
  20     originCommentId = obj.inReplyToComment.originCommentId || obj.inReplyToComment.id
  21     inReplyToCommentId = obj.inReplyToComment.id
  22   }
  23
  24   const comment = await VideoCommentModel.create({
  25     text: obj.text,
  26     originCommentId,
  27     inReplyToCommentId,
  28     videoId: obj.video.id,
  29     accountId: obj.account.id,
  30     url: 'fake url'
  31   }, { transaction: t, validate: false })
  32
  33   comment.set('url', getVideoCommentActivityPubUrl(obj.video, comment))
  34
  35   const savedComment = await comment.save({ transaction: t })
  36   savedComment.InReplyToVideoComment = obj.inReplyToComment
  37   savedComment.Video = obj.video
  38   savedComment.Account = obj.account
  39
  40   await sendCreateVideoComment(savedComment, t)
  41
  42   return savedComment
  43 }
  44
  45 function buildFormattedCommentTree (resultList: ResultList<VideoCommentModel>): VideoCommentThreadTree {
  46   // Comments are sorted by id ASC
  47   const comments = resultList.data
  48
  49   const comment = comments.shift()
  50   const thread: VideoCommentThreadTree = {
  51     comment: comment.toFormattedJSON(),
  52     children: []
  53   }
  54   const idx = {
  55     [comment.id]: thread
  56   }
  57
  58   while (comments.length !== 0) {
  59     const childComment = comments.shift()
  60
  61     const childCommentThread: VideoCommentThreadTree = {
  62       comment: childComment.toFormattedJSON(),
  63       children: []
  64     }
  65
  66     const parentCommentThread = idx[childComment.inReplyToCommentId]
  67     if (!parentCommentThread) {
  68       const msg = `Cannot format video thread tree, parent ${childComment.inReplyToCommentId} not found for child ${childComment.id}`
  69       throw new Error(msg)
  70     }
  71
  72     parentCommentThread.children.push(childCommentThread)
  73     idx[childComment.id] = childCommentThread
  74   }
  75
  76   return thread
  77 }
  78
  79 // ---------------------------------------------------------------------------
  80
  81 export {
  82   createVideoComment,
  83   buildFormattedCommentTree
  84 }