import { Transaction } from 'sequelize'
-import { AccountModel } from '../../models/account/account'
-import { VideoModel } from '../../models/video/video'
import { 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'
+import { CRAWL_REQUEST_CONCURRENCY } from '../../initializers/constants'
import { doRequest } from '../../helpers/requests'
import { checkUrlsSameHost, getAPId } from '../../helpers/activitypub'
-import { ActorModel } from '../../models/activitypub/actor'
-import { getVideoDislikeActivityPubUrl, getVideoLikeActivityPubUrl } from './url'
+import { getVideoDislikeActivityPubUrlByLocalActor, getVideoLikeActivityPubUrlByLocalActor } from './url'
import { sendDislike } from './send/send-dislike'
+import { MAccountActor, MActorUrl, MVideo, MVideoAccountLight, MVideoId } from '../../types/models'
-async function createRates (ratesUrl: string[], video: VideoModel, rate: VideoRateType) {
- let rateCounts = 0
-
+async function createRates (ratesUrl: string[], video: MVideo, rate: VideoRateType) {
await Bluebird.map(ratesUrl, async rateUrl => {
try {
// Fetch url
- const { body } = await doRequest({
+ const { body } = await doRequest<any>({
uri: rateUrl,
json: true,
activityPub: true
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,
- url: body.id
- }
- })
+ const entry = {
+ videoId: video.id,
+ accountId: actor.Account.id,
+ type: rate,
+ url: body.id
+ }
- if (created) rateCounts += 1
+ // Video "likes"/"dislikes" will be updated by the caller
+ await AccountVideoRateModel.upsert(entry)
} catch (err) {
logger.warn('Cannot add rate %s.', rateUrl, { 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) {
+async function sendVideoRateChange (
+ account: MAccountActor,
+ video: MVideoAccountLight,
+ likes: number,
+ dislikes: number,
+ t: Transaction
+) {
const actor = account.Actor
// Keep the order: first we undo and then we create
if (dislikes > 0) await sendDislike(actor, video, t)
}
-function getRateUrl (rateType: VideoRateType, actor: ActorModel, video: VideoModel) {
- return rateType === 'like' ? getVideoLikeActivityPubUrl(actor, video) : getVideoDislikeActivityPubUrl(actor, video)
+function getLocalRateUrl (rateType: VideoRateType, actor: MActorUrl, video: MVideoId) {
+ return rateType === 'like'
+ ? getVideoLikeActivityPubUrlByLocalActor(actor, video)
+ : getVideoDislikeActivityPubUrlByLocalActor(actor, video)
}
export {
- getRateUrl,
+ getLocalRateUrl,
createRates,
sendVideoRateChange
}