mirror of https://github.com/Chocobozzz/PeerTube
69 lines
2.2 KiB
TypeScript
69 lines
2.2 KiB
TypeScript
import { Transaction } from 'sequelize'
|
|
import { AccountModel } from '../../models/account/account'
|
|
import { VideoModel } from '../../models/video/video'
|
|
import { sendCreateDislike, sendLike, sendUndoDislike, sendUndoLike } from './send'
|
|
import { VideoRateType } from '../../../shared/models/videos'
|
|
import * as Bluebird from 'bluebird'
|
|
import { getOrCreateActorAndServerAndModel } from './actor'
|
|
import { AccountVideoRateModel } from '../../models/account/account-video-rate'
|
|
import { logger } from '../../helpers/logger'
|
|
import { CRAWL_REQUEST_CONCURRENCY } from '../../initializers'
|
|
|
|
async function createRates (actorUrls: string[], video: VideoModel, rate: VideoRateType) {
|
|
let rateCounts = 0
|
|
|
|
await Bluebird.map(actorUrls, async actorUrl => {
|
|
try {
|
|
const actor = await getOrCreateActorAndServerAndModel(actorUrl)
|
|
const [ , created ] = await AccountVideoRateModel
|
|
.findOrCreate({
|
|
where: {
|
|
videoId: video.id,
|
|
accountId: actor.Account.id
|
|
},
|
|
defaults: {
|
|
videoId: video.id,
|
|
accountId: actor.Account.id,
|
|
type: rate
|
|
}
|
|
})
|
|
|
|
if (created) rateCounts += 1
|
|
} catch (err) {
|
|
logger.warn('Cannot add rate %s for actor %s.', rate, actorUrl, { err })
|
|
}
|
|
}, { concurrency: CRAWL_REQUEST_CONCURRENCY })
|
|
|
|
logger.info('Adding %d %s to video %s.', rateCounts, rate, video.uuid)
|
|
|
|
// This is "likes" and "dislikes"
|
|
if (rateCounts !== 0) await video.increment(rate + 's', { by: rateCounts })
|
|
|
|
return
|
|
}
|
|
|
|
async function sendVideoRateChange (account: AccountModel,
|
|
video: VideoModel,
|
|
likes: number,
|
|
dislikes: number,
|
|
t: Transaction) {
|
|
const actor = account.Actor
|
|
|
|
// Keep the order: first we undo and then we create
|
|
|
|
// Undo Like
|
|
if (likes < 0) await sendUndoLike(actor, video, t)
|
|
// Undo Dislike
|
|
if (dislikes < 0) await sendUndoDislike(actor, video, t)
|
|
|
|
// Like
|
|
if (likes > 0) await sendLike(actor, video, t)
|
|
// Dislike
|
|
if (dislikes > 0) await sendCreateDislike(actor, video, t)
|
|
}
|
|
|
|
export {
|
|
createRates,
|
|
sendVideoRateChange
|
|
}
|