import * as express from 'express'
-import * as Promise from 'bluebird'
-
-import { database as db } from '../../../initializers/database'
-import {
- logger,
- retryTransactionWrapper
-} from '../../../helpers'
-import {
- VIDEO_RATE_TYPES,
- REQUEST_VIDEO_EVENT_TYPES,
- REQUEST_VIDEO_QADU_TYPES
-} from '../../../initializers'
-import {
- addEventsToRemoteVideo,
- quickAndDirtyUpdatesVideoToFriends
-} from '../../../lib'
-import {
- authenticate,
- videoRateValidator
-} from '../../../middlewares'
-import { UserVideoRateUpdate, VideoRateType } from '../../../../shared'
+import { UserVideoRateUpdate } from '../../../../shared'
+import { logger } from '../../../helpers/logger'
+import { VIDEO_RATE_TYPES } from '../../../initializers/constants'
+import { getLocalRateUrl, sendVideoRateChange } from '../../../lib/activitypub/video-rates'
+import { asyncMiddleware, asyncRetryTransactionMiddleware, authenticate, videoUpdateRateValidator } from '../../../middlewares'
+import { AccountModel } from '../../../models/account/account'
+import { AccountVideoRateModel } from '../../../models/account/account-video-rate'
+import { sequelizeTypescript } from '../../../initializers/database'
const rateVideoRouter = express.Router()
rateVideoRouter.put('/:id/rate',
authenticate,
- videoRateValidator,
- rateVideoRetryWrapper
+ asyncMiddleware(videoUpdateRateValidator),
+ asyncRetryTransactionMiddleware(rateVideo)
)
// ---------------------------------------------------------------------------
// ---------------------------------------------------------------------------
-function rateVideoRetryWrapper (req: express.Request, res: express.Response, next: express.NextFunction) {
- const options = {
- arguments: [ req, res ],
- errorMessage: 'Cannot update the user video rate.'
- }
-
- retryTransactionWrapper(rateVideo, options)
- .then(() => res.type('json').status(204).end())
- .catch(err => next(err))
-}
-
-function rateVideo (req: express.Request, res: express.Response) {
+async function rateVideo (req: express.Request, res: express.Response) {
const body: UserVideoRateUpdate = req.body
const rateType = body.rating
- const videoInstance = res.locals.video
- const userInstance = res.locals.oauth.token.User
-
- return db.sequelize.transaction(t => {
- return db.UserVideoRate.load(userInstance.id, videoInstance.id, t)
- .then(previousRate => {
- const options = { transaction: t }
-
- let likesToIncrement = 0
- let dislikesToIncrement = 0
-
- if (rateType === VIDEO_RATE_TYPES.LIKE) likesToIncrement++
- else if (rateType === VIDEO_RATE_TYPES.DISLIKE) dislikesToIncrement++
-
- let promise: Promise<any>
-
- // There was a previous rate, update it
- if (previousRate) {
- // We will remove the previous rate, so we will need to update the video count attribute
- if (previousRate.type === VIDEO_RATE_TYPES.LIKE) likesToIncrement--
- else if (previousRate.type === VIDEO_RATE_TYPES.DISLIKE) dislikesToIncrement--
-
- if (rateType === 'none') { // Destroy previous rate
- promise = previousRate.destroy()
- } else { // Update previous rate
- previousRate.type = rateType as VideoRateType
-
- promise = previousRate.save()
- }
- } else if (rateType !== 'none') { // There was not a previous rate, insert a new one if there is a rate
- const query = {
- userId: userInstance.id,
- videoId: videoInstance.id,
- type: rateType
- }
-
- promise = db.UserVideoRate.create(query, options)
- } else {
- promise = Promise.resolve()
- }
-
- return promise.then(() => ({ likesToIncrement, dislikesToIncrement }))
- })
- .then(({ likesToIncrement, dislikesToIncrement }) => {
- const options = { transaction: t }
- const incrementQuery = {
- likes: likesToIncrement,
- dislikes: dislikesToIncrement
- }
-
- // Even if we do not own the video we increment the attributes
- // It is usefull for the user to have a feedback
- return videoInstance.increment(incrementQuery, options).then(() => ({ likesToIncrement, dislikesToIncrement }))
- })
- .then(({ likesToIncrement, dislikesToIncrement }) => {
- // No need for an event type, we own the video
- if (videoInstance.isOwned()) return { likesToIncrement, dislikesToIncrement }
-
- const eventsParams = []
-
- if (likesToIncrement !== 0) {
- eventsParams.push({
- videoId: videoInstance.id,
- type: REQUEST_VIDEO_EVENT_TYPES.LIKES,
- count: likesToIncrement
- })
- }
-
- if (dislikesToIncrement !== 0) {
- eventsParams.push({
- videoId: videoInstance.id,
- type: REQUEST_VIDEO_EVENT_TYPES.DISLIKES,
- count: dislikesToIncrement
- })
- }
-
- return addEventsToRemoteVideo(eventsParams, t).then(() => ({ likesToIncrement, dislikesToIncrement }))
- })
- .then(({ likesToIncrement, dislikesToIncrement }) => {
- // We do not own the video, there is no need to send a quick and dirty update to friends
- // Our rate was already sent by the addEvent function
- if (videoInstance.isOwned() === false) return undefined
-
- const qadusParams = []
-
- if (likesToIncrement !== 0) {
- qadusParams.push({
- videoId: videoInstance.id,
- type: REQUEST_VIDEO_QADU_TYPES.LIKES
- })
- }
-
- if (dislikesToIncrement !== 0) {
- qadusParams.push({
- videoId: videoInstance.id,
- type: REQUEST_VIDEO_QADU_TYPES.DISLIKES
- })
- }
-
- return quickAndDirtyUpdatesVideoToFriends(qadusParams, t)
- })
- })
- .then(() => logger.info('User video rate for video %s of user %s updated.', videoInstance.name, userInstance.username))
- .catch(err => {
- // This is just a debug because we will retry the insert
- logger.debug('Cannot add the user video rate.', err)
- throw err
+ const videoInstance = res.locals.videoAll
+ const userAccount = res.locals.oauth.token.User.Account
+
+ await sequelizeTypescript.transaction(async t => {
+ const sequelizeOptions = { transaction: t }
+
+ const accountInstance = await AccountModel.load(userAccount.id, t)
+ const previousRate = await AccountVideoRateModel.load(accountInstance.id, videoInstance.id, t)
+
+ let likesToIncrement = 0
+ let dislikesToIncrement = 0
+
+ if (rateType === VIDEO_RATE_TYPES.LIKE) likesToIncrement++
+ else if (rateType === VIDEO_RATE_TYPES.DISLIKE) dislikesToIncrement++
+
+ // There was a previous rate, update it
+ if (previousRate) {
+ // We will remove the previous rate, so we will need to update the video count attribute
+ if (previousRate.type === 'like') likesToIncrement--
+ else if (previousRate.type === 'dislike') dislikesToIncrement--
+
+ if (rateType === 'none') { // Destroy previous rate
+ await previousRate.destroy(sequelizeOptions)
+ } else { // Update previous rate
+ previousRate.type = rateType
+ previousRate.url = getLocalRateUrl(rateType, userAccount.Actor, videoInstance)
+ await previousRate.save(sequelizeOptions)
+ }
+ } else if (rateType !== 'none') { // There was not a previous rate, insert a new one if there is a rate
+ const query = {
+ accountId: accountInstance.id,
+ videoId: videoInstance.id,
+ type: rateType,
+ url: getLocalRateUrl(rateType, userAccount.Actor, videoInstance)
+ }
+
+ await AccountVideoRateModel.create(query, sequelizeOptions)
+ }
+
+ const incrementQuery = {
+ likes: likesToIncrement,
+ dislikes: dislikesToIncrement
+ }
+
+ await videoInstance.increment(incrementQuery, sequelizeOptions)
+
+ await sendVideoRateChange(accountInstance, videoInstance, likesToIncrement, dislikesToIncrement, t)
+
+ logger.info('Account video rate for video %s of account %s updated.', videoInstance.name, accountInstance.name)
})
+
+ return res.type('json').status(204).end()
}
projects / github / Chocobozzz / PeerTube.git / blobdiff