1 import * as Bluebird from 'bluebird'
2 import { values } from 'lodash'
19 } from 'sequelize-typescript'
20 import { FollowState } from '../../../shared/models/actors'
21 import { ActorFollow } from '../../../shared/models/actors/follow.model'
22 import { logger } from '../../helpers/logger'
23 import { getServerActor } from '../../helpers/utils'
24 import { ACTOR_FOLLOW_SCORE, FOLLOW_STATES } from '../../initializers/constants'
25 import { ServerModel } from '../server/server'
26 import { createSafeIn, getSort } from '../utils'
27 import { ActorModel, unusedActorAttributesForAPI } from './actor'
28 import { VideoChannelModel } from '../video/video-channel'
29 import { AccountModel } from '../account/account'
30 import { IncludeOptions, Op, QueryTypes, Transaction } from 'sequelize'
32 MActorFollowActorsDefault,
33 MActorFollowActorsDefaultSubscription,
34 MActorFollowFollowingHost,
35 MActorFollowFormattable,
36 MActorFollowSubscriptions
37 } from '@server/typings/models'
40 tableName: 'actorFollow',
46 fields: [ 'targetActorId' ]
49 fields: [ 'actorId', 'targetActorId' ],
57 export class ActorFollowModel extends Model<ActorFollowModel> {
60 @Column(DataType.ENUM(...values(FOLLOW_STATES)))
64 @Default(ACTOR_FOLLOW_SCORE.BASE)
66 @Max(ACTOR_FOLLOW_SCORE.MAX)
76 @ForeignKey(() => ActorModel)
80 @BelongsTo(() => ActorModel, {
88 ActorFollower: ActorModel
90 @ForeignKey(() => ActorModel)
94 @BelongsTo(() => ActorModel, {
96 name: 'targetActorId',
102 ActorFollowing: ActorModel
106 static incrementFollowerAndFollowingCount (instance: ActorFollowModel) {
107 if (instance.state !== 'accepted') return undefined
110 ActorModel.incrementFollows(instance.actorId, 'followingCount', 1),
111 ActorModel.incrementFollows(instance.targetActorId, 'followersCount', 1)
116 static decrementFollowerAndFollowingCount (instance: ActorFollowModel) {
118 ActorModel.incrementFollows(instance.actorId, 'followingCount',-1),
119 ActorModel.incrementFollows(instance.targetActorId, 'followersCount', -1)
123 static removeFollowsOf (actorId: number, t?: Transaction) {
131 targetActorId: actorId
138 return ActorFollowModel.destroy(query)
141 // Remove actor follows with a score of 0 (too many requests where they were unreachable)
142 static async removeBadActorFollows () {
143 const actorFollows = await ActorFollowModel.listBadActorFollows()
145 const actorFollowsRemovePromises = actorFollows.map(actorFollow => actorFollow.destroy())
146 await Promise.all(actorFollowsRemovePromises)
148 const numberOfActorFollowsRemoved = actorFollows.length
150 if (numberOfActorFollowsRemoved) logger.info('Removed bad %d actor follows.', numberOfActorFollowsRemoved)
153 static loadByActorAndTarget (actorId: number, targetActorId: number, t?: Transaction): Bluebird<MActorFollowActorsDefault> {
157 targetActorId: targetActorId
174 return ActorFollowModel.findOne(query)
177 static loadByActorAndTargetNameAndHostForAPI (
182 ): Bluebird<MActorFollowActorsDefaultSubscription> {
183 const actorFollowingPartInclude: IncludeOptions = {
186 as: 'ActorFollowing',
188 preferredUsername: targetName
192 model: VideoChannelModel.unscoped(),
198 if (targetHost === null) {
199 actorFollowingPartInclude.where['serverId'] = null
201 actorFollowingPartInclude.include.push({
215 actorFollowingPartInclude,
225 return ActorFollowModel.findOne(query)
227 if (result && result.ActorFollowing.VideoChannel) {
228 result.ActorFollowing.VideoChannel.Actor = result.ActorFollowing
235 static listSubscribedIn (actorId: number, targets: { name: string, host?: string }[]): Bluebird<MActorFollowFollowingHost[]> {
236 const whereTab = targets
242 '$preferredUsername$': t.name
254 '$preferredUsername$': t.name
277 attributes: [ 'preferredUsername' ],
278 model: ActorModel.unscoped(),
280 as: 'ActorFollowing',
283 attributes: [ 'host' ],
284 model: ServerModel.unscoped(),
292 return ActorFollowModel.findAll(query)
295 static listFollowingForApi (id: number, start: number, count: number, sort: string, search?: string) {
300 order: getSort(sort),
312 as: 'ActorFollowing',
320 [Op.iLike]: '%' + search + '%'
329 return ActorFollowModel.findAndCountAll<MActorFollowActorsDefault>(query)
330 .then(({ rows, count }) => {
338 static listFollowersForApi (actorId: number, start: number, count: number, sort: string, search?: string) {
343 order: getSort(sort),
355 [ Op.iLike ]: '%' + search + '%'
363 as: 'ActorFollowing',
372 return ActorFollowModel.findAndCountAll<MActorFollowActorsDefault>(query)
373 .then(({ rows, count }) => {
381 static listSubscriptionsForApi (actorId: number, start: number, count: number, sort: string) {
387 order: getSort(sort),
393 attributes: [ 'id' ],
394 model: ActorModel.unscoped(),
395 as: 'ActorFollowing',
399 model: VideoChannelModel.unscoped(),
404 exclude: unusedActorAttributesForAPI
410 model: AccountModel.unscoped(),
415 exclude: unusedActorAttributesForAPI
429 return ActorFollowModel.findAndCountAll<MActorFollowSubscriptions>(query)
430 .then(({ rows, count }) => {
432 data: rows.map(r => r.ActorFollowing.VideoChannel),
438 static listAcceptedFollowerUrlsForAP (actorIds: number[], t: Transaction, start?: number, count?: number) {
439 return ActorFollowModel.createListAcceptedFollowForApiQuery('followers', actorIds, t, start, count)
442 static listAcceptedFollowerSharedInboxUrls (actorIds: number[], t: Transaction) {
443 return ActorFollowModel.createListAcceptedFollowForApiQuery(
454 static listAcceptedFollowingUrlsForApi (actorIds: number[], t: Transaction, start?: number, count?: number) {
455 return ActorFollowModel.createListAcceptedFollowForApiQuery('following', actorIds, t, start, count)
458 static async getStats () {
459 const serverActor = await getServerActor()
461 const totalInstanceFollowing = await ActorFollowModel.count({
463 actorId: serverActor.id
467 const totalInstanceFollowers = await ActorFollowModel.count({
469 targetActorId: serverActor.id
474 totalInstanceFollowing,
475 totalInstanceFollowers
479 static updateScore (inboxUrl: string, value: number, t?: Transaction) {
480 const query = `UPDATE "actorFollow" SET "score" = LEAST("score" + ${value}, ${ACTOR_FOLLOW_SCORE.MAX}) ` +
482 'SELECT "actorFollow"."id" FROM "actorFollow" ' +
483 'INNER JOIN "actor" ON "actor"."id" = "actorFollow"."actorId" ' +
484 `WHERE "actor"."inboxUrl" = '${inboxUrl}' OR "actor"."sharedInboxUrl" = '${inboxUrl}'` +
488 type: QueryTypes.BULKUPDATE,
492 return ActorFollowModel.sequelize.query(query, options)
495 static async updateScoreByFollowingServers (serverIds: number[], value: number, t?: Transaction) {
496 if (serverIds.length === 0) return
498 const me = await getServerActor()
499 const serverIdsString = createSafeIn(ActorFollowModel, serverIds)
501 const query = `UPDATE "actorFollow" SET "score" = LEAST("score" + ${value}, ${ACTOR_FOLLOW_SCORE.MAX}) ` +
503 'SELECT "actorFollow"."id" FROM "actorFollow" ' +
504 'INNER JOIN "actor" ON "actor"."id" = "actorFollow"."targetActorId" ' +
505 `WHERE "actorFollow"."actorId" = ${me.Account.actorId} ` + // I'm the follower
506 `AND "actor"."serverId" IN (${serverIdsString})` + // Criteria on followings
510 type: QueryTypes.BULKUPDATE,
514 return ActorFollowModel.sequelize.query(query, options)
517 private static async createListAcceptedFollowForApiQuery (
518 type: 'followers' | 'following',
526 let firstJoin: string
527 let secondJoin: string
529 if (type === 'followers') {
530 firstJoin = 'targetActorId'
531 secondJoin = 'actorId'
533 firstJoin = 'actorId'
534 secondJoin = 'targetActorId'
537 const selections: string[] = []
538 if (distinct === true) selections.push('DISTINCT("Follows"."' + columnUrl + '") AS "url"')
539 else selections.push('"Follows"."' + columnUrl + '" AS "url"')
541 selections.push('COUNT(*) AS "total"')
543 const tasks: Bluebird<any>[] = []
545 for (let selection of selections) {
546 let query = 'SELECT ' + selection + ' FROM "actor" ' +
547 'INNER JOIN "actorFollow" ON "actorFollow"."' + firstJoin + '" = "actor"."id" ' +
548 'INNER JOIN "actor" AS "Follows" ON "actorFollow"."' + secondJoin + '" = "Follows"."id" ' +
549 'WHERE "actor"."id" = ANY ($actorIds) AND "actorFollow"."state" = \'accepted\' '
551 if (count !== undefined) query += 'LIMIT ' + count
552 if (start !== undefined) query += ' OFFSET ' + start
556 type: QueryTypes.SELECT,
559 tasks.push(ActorFollowModel.sequelize.query(query, options))
562 const [ followers, [ dataTotal ] ] = await Promise.all(tasks)
563 const urls: string[] = followers.map(f => f.url)
567 total: dataTotal ? parseInt(dataTotal.total, 10) : 0
571 private static listBadActorFollows () {
581 return ActorFollowModel.findAll(query)
584 toFormattedJSON (this: MActorFollowFormattable): ActorFollow {
585 const follower = this.ActorFollower.toFormattedJSON()
586 const following = this.ActorFollowing.toFormattedJSON()
594 createdAt: this.createdAt,
595 updatedAt: this.updatedAt