shared/extra-utils/server/follows.ts

   1 import * as request from 'supertest'
   2 import { ServerInfo } from './servers'
   3 import { waitJobs } from './jobs'
   4 import { makePostBodyRequest } from '../requests/requests'
   5 import { ActivityPubActorType, FollowState } from '@shared/models'
   6 import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes'
   7
   8 function getFollowersListPaginationAndSort (options: {
   9   url: string
  10   start: number
  11   count: number
  12   sort: string
  13   search?: string
  14   actorType?: ActivityPubActorType
  15   state?: FollowState
  16 }) {
  17   const { url, start, count, sort, search, state, actorType } = options
  18   const path = '/api/v1/server/followers'
  19
  20   const query = {
  21     start,
  22     count,
  23     sort,
  24     search,
  25     state,
  26     actorType
  27   }
  28
  29   return request(url)
  30     .get(path)
  31     .query(query)
  32     .set('Accept', 'application/json')
  33     .expect(HttpStatusCode.OK_200)
  34     .expect('Content-Type', /json/)
  35 }
  36
  37 function acceptFollower (url: string, token: string, follower: string, statusCodeExpected = HttpStatusCode.NO_CONTENT_204) {
  38   const path = '/api/v1/server/followers/' + follower + '/accept'
  39
  40   return makePostBodyRequest({
  41     url,
  42     token,
  43     path,
  44     statusCodeExpected
  45   })
  46 }
  47
  48 function rejectFollower (url: string, token: string, follower: string, statusCodeExpected = HttpStatusCode.NO_CONTENT_204) {
  49   const path = '/api/v1/server/followers/' + follower + '/reject'
  50
  51   return makePostBodyRequest({
  52     url,
  53     token,
  54     path,
  55     statusCodeExpected
  56   })
  57 }
  58
  59 function getFollowingListPaginationAndSort (options: {
  60   url: string
  61   start: number
  62   count: number
  63   sort: string
  64   search?: string
  65   actorType?: ActivityPubActorType
  66   state?: FollowState
  67 }) {
  68   const { url, start, count, sort, search, state, actorType } = options
  69   const path = '/api/v1/server/following'
  70
  71   const query = {
  72     start,
  73     count,
  74     sort,
  75     search,
  76     state,
  77     actorType
  78   }
  79
  80   return request(url)
  81     .get(path)
  82     .query(query)
  83     .set('Accept', 'application/json')
  84     .expect(HttpStatusCode.OK_200)
  85     .expect('Content-Type', /json/)
  86 }
  87
  88 function follow (follower: string, following: string[], accessToken: string, expectedStatus = HttpStatusCode.NO_CONTENT_204) {
  89   const path = '/api/v1/server/following'
  90
  91   const followingHosts = following.map(f => f.replace(/^http:\/\//, ''))
  92   return request(follower)
  93     .post(path)
  94     .set('Accept', 'application/json')
  95     .set('Authorization', 'Bearer ' + accessToken)
  96     .send({ hosts: followingHosts })
  97     .expect(expectedStatus)
  98 }
  99
 100 async function unfollow (url: string, accessToken: string, target: ServerInfo, expectedStatus = HttpStatusCode.NO_CONTENT_204) {
 101   const path = '/api/v1/server/following/' + target.host
 102
 103   return request(url)
 104     .delete(path)
 105     .set('Accept', 'application/json')
 106     .set('Authorization', 'Bearer ' + accessToken)
 107     .expect(expectedStatus)
 108 }
 109
 110 function removeFollower (url: string, accessToken: string, follower: ServerInfo, expectedStatus = HttpStatusCode.NO_CONTENT_204) {
 111   const path = '/api/v1/server/followers/peertube@' + follower.host
 112
 113   return request(url)
 114     .delete(path)
 115     .set('Accept', 'application/json')
 116     .set('Authorization', 'Bearer ' + accessToken)
 117     .expect(expectedStatus)
 118 }
 119
 120 async function doubleFollow (server1: ServerInfo, server2: ServerInfo) {
 121   await Promise.all([
 122     follow(server1.url, [ server2.url ], server1.accessToken),
 123     follow(server2.url, [ server1.url ], server2.accessToken)
 124   ])
 125
 126   // Wait request propagation
 127   await waitJobs([ server1, server2 ])
 128
 129   return true
 130 }
 131
 132 // ---------------------------------------------------------------------------
 133
 134 export {
 135   getFollowersListPaginationAndSort,
 136   getFollowingListPaginationAndSort,
 137   unfollow,
 138   removeFollower,
 139   follow,
 140   doubleFollow,
 141   acceptFollower,
 142   rejectFollower
 143 }