import { omit } from 'lodash' import * as request from 'supertest' import { UserUpdateMe } from '../../models/users' import { UserAdminFlag } from '../../models/users/user-flag.model' import { UserRegister } from '../../models/users/user-register.model' import { UserRole } from '../../models/users/user-role' import { makeGetRequest, makePostBodyRequest, makePutBodyRequest, updateImageRequest } from '../requests/requests' import { ServerInfo } from '../server/servers' import { userLogin } from './login' import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes' type CreateUserArgs = { url: string accessToken: string username: string password: string videoQuota?: number videoQuotaDaily?: number role?: UserRole adminFlags?: UserAdminFlag specialStatus?: number } function createUser (parameters: CreateUserArgs) { const { url, accessToken, username, adminFlags, password = 'password', videoQuota = 1000000, videoQuotaDaily = -1, role = UserRole.USER, specialStatus = HttpStatusCode.OK_200 } = parameters const path = '/api/v1/users' const body = { username, password, role, adminFlags, email: username + '@example.com', videoQuota, videoQuotaDaily } return request(url) .post(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .send(body) .expect(specialStatus) } async function generateUserAccessToken (server: ServerInfo, username: string) { const password = 'my super password' await createUser({ url: server.url, accessToken: server.accessToken, username: username, password: password }) return userLogin(server, { username, password }) } function registerUser (url: string, username: string, password: string, specialStatus = HttpStatusCode.NO_CONTENT_204) { const path = '/api/v1/users/register' const body = { username, password, email: username + '@example.com' } return request(url) .post(path) .set('Accept', 'application/json') .send(body) .expect(specialStatus) } function registerUserWithChannel (options: { url: string user: { username: string, password: string, displayName?: string } channel: { name: string, displayName: string } }) { const path = '/api/v1/users/register' const body: UserRegister = { username: options.user.username, password: options.user.password, email: options.user.username + '@example.com', channel: options.channel } if (options.user.displayName) { Object.assign(body, { displayName: options.user.displayName }) } return makePostBodyRequest({ url: options.url, path, fields: body, statusCodeExpected: HttpStatusCode.NO_CONTENT_204 }) } function getMyUserInformation (url: string, accessToken: string, specialStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/me' return request(url) .get(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(specialStatus) .expect('Content-Type', /json/) } function getUserScopedTokens (url: string, token: string, statusCodeExpected = HttpStatusCode.OK_200) { const path = '/api/v1/users/scoped-tokens' return makeGetRequest({ url, path, token, statusCodeExpected }) } function renewUserScopedTokens (url: string, token: string, statusCodeExpected = HttpStatusCode.OK_200) { const path = '/api/v1/users/scoped-tokens' return makePostBodyRequest({ url, path, token, statusCodeExpected }) } function deleteMe (url: string, accessToken: string, specialStatus = HttpStatusCode.NO_CONTENT_204) { const path = '/api/v1/users/me' return request(url) .delete(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(specialStatus) } function getMyUserVideoQuotaUsed (url: string, accessToken: string, specialStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/me/video-quota-used' return request(url) .get(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(specialStatus) .expect('Content-Type', /json/) } function getUserInformation (url: string, accessToken: string, userId: number, withStats = false) { const path = '/api/v1/users/' + userId return request(url) .get(path) .query({ withStats }) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(HttpStatusCode.OK_200) .expect('Content-Type', /json/) } function getMyUserVideoRating (url: string, accessToken: string, videoId: number | string, specialStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/me/videos/' + videoId + '/rating' return request(url) .get(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(specialStatus) .expect('Content-Type', /json/) } function getUsersList (url: string, accessToken: string) { const path = '/api/v1/users' return request(url) .get(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(HttpStatusCode.OK_200) .expect('Content-Type', /json/) } function getUsersListPaginationAndSort ( url: string, accessToken: string, start: number, count: number, sort: string, search?: string, blocked?: boolean ) { const path = '/api/v1/users' const query = { start, count, sort, search, blocked } return request(url) .get(path) .query(query) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(HttpStatusCode.OK_200) .expect('Content-Type', /json/) } function removeUser (url: string, userId: number | string, accessToken: string, expectedStatus = HttpStatusCode.NO_CONTENT_204) { const path = '/api/v1/users' return request(url) .delete(path + '/' + userId) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(expectedStatus) } function blockUser ( url: string, userId: number | string, accessToken: string, expectedStatus = HttpStatusCode.NO_CONTENT_204, reason?: string ) { const path = '/api/v1/users' let body: any if (reason) body = { reason } return request(url) .post(path + '/' + userId + '/block') .send(body) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(expectedStatus) } function unblockUser (url: string, userId: number | string, accessToken: string, expectedStatus = HttpStatusCode.NO_CONTENT_204) { const path = '/api/v1/users' return request(url) .post(path + '/' + userId + '/unblock') .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(expectedStatus) } function updateMyUser (options: { url: string, accessToken: string, statusCodeExpected?: HttpStatusCode } & UserUpdateMe) { const path = '/api/v1/users/me' const toSend: UserUpdateMe = omit(options, 'url', 'accessToken') return makePutBodyRequest({ url: options.url, path, token: options.accessToken, fields: toSend, statusCodeExpected: options.statusCodeExpected || HttpStatusCode.NO_CONTENT_204 }) } function updateMyAvatar (options: { url: string accessToken: string fixture: string }) { const path = '/api/v1/users/me/avatar/pick' return updateImageRequest({ ...options, path, fieldname: 'avatarfile' }) } function updateUser (options: { url: string userId: number accessToken: string email?: string emailVerified?: boolean videoQuota?: number videoQuotaDaily?: number password?: string adminFlags?: UserAdminFlag pluginAuth?: string role?: UserRole }) { const path = '/api/v1/users/' + options.userId const toSend = {} if (options.password !== undefined && options.password !== null) toSend['password'] = options.password if (options.email !== undefined && options.email !== null) toSend['email'] = options.email if (options.emailVerified !== undefined && options.emailVerified !== null) toSend['emailVerified'] = options.emailVerified if (options.videoQuota !== undefined && options.videoQuota !== null) toSend['videoQuota'] = options.videoQuota if (options.videoQuotaDaily !== undefined && options.videoQuotaDaily !== null) toSend['videoQuotaDaily'] = options.videoQuotaDaily if (options.role !== undefined && options.role !== null) toSend['role'] = options.role if (options.adminFlags !== undefined && options.adminFlags !== null) toSend['adminFlags'] = options.adminFlags if (options.pluginAuth !== undefined) toSend['pluginAuth'] = options.pluginAuth return makePutBodyRequest({ url: options.url, path, token: options.accessToken, fields: toSend, statusCodeExpected: HttpStatusCode.NO_CONTENT_204 }) } function askResetPassword (url: string, email: string) { const path = '/api/v1/users/ask-reset-password' return makePostBodyRequest({ url, path, fields: { email }, statusCodeExpected: HttpStatusCode.NO_CONTENT_204 }) } function resetPassword ( url: string, userId: number, verificationString: string, password: string, statusCodeExpected = HttpStatusCode.NO_CONTENT_204 ) { const path = '/api/v1/users/' + userId + '/reset-password' return makePostBodyRequest({ url, path, fields: { password, verificationString }, statusCodeExpected }) } function askSendVerifyEmail (url: string, email: string) { const path = '/api/v1/users/ask-send-verify-email' return makePostBodyRequest({ url, path, fields: { email }, statusCodeExpected: HttpStatusCode.NO_CONTENT_204 }) } function verifyEmail ( url: string, userId: number, verificationString: string, isPendingEmail = false, statusCodeExpected = HttpStatusCode.NO_CONTENT_204 ) { const path = '/api/v1/users/' + userId + '/verify-email' return makePostBodyRequest({ url, path, fields: { verificationString, isPendingEmail }, statusCodeExpected }) } // --------------------------------------------------------------------------- export { createUser, registerUser, getMyUserInformation, getMyUserVideoRating, deleteMe, registerUserWithChannel, getMyUserVideoQuotaUsed, getUsersList, getUsersListPaginationAndSort, removeUser, updateUser, updateMyUser, getUserInformation, blockUser, unblockUser, askResetPassword, resetPassword, renewUserScopedTokens, updateMyAvatar, askSendVerifyEmail, generateUserAccessToken, verifyEmail, getUserScopedTokens }