import * as request from 'supertest' import { ServerInfo } from '../server/servers' import { getClient } from '../server/clients' import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes' type Client = { id: string, secret: string } type User = { username: string, password: string } type Server = { url: string, client: Client, user: User } function login (url: string, client: Client, user: User, expectedStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/token' const body = { client_id: client.id, client_secret: client.secret, username: user.username, password: user.password, response_type: 'code', grant_type: 'password', scope: 'upload' } return request(url) .post(path) .type('form') .send(body) .expect(expectedStatus) } function logout (url: string, token: string, expectedStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/revoke-token' return request(url) .post(path) .set('Authorization', 'Bearer ' + token) .type('form') .expect(expectedStatus) } async function serverLogin (server: Server) { const res = await login(server.url, server.client, server.user, HttpStatusCode.OK_200) return res.body.access_token as string } function refreshToken (server: ServerInfo, refreshToken: string, expectedStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/token' const body = { client_id: server.client.id, client_secret: server.client.secret, refresh_token: refreshToken, response_type: 'code', grant_type: 'refresh_token' } return request(server.url) .post(path) .type('form') .send(body) .expect(expectedStatus) } async function userLogin (server: Server, user: User, expectedStatus = HttpStatusCode.OK_200) { const res = await login(server.url, server.client, user, expectedStatus) return res.body.access_token as string } async function getAccessToken (url: string, username: string, password: string) { const resClient = await getClient(url) const client = { id: resClient.body.client_id, secret: resClient.body.client_secret } const user = { username, password } try { const res = await login(url, client, user) return res.body.access_token } catch (err) { throw new Error('Cannot authenticate. Please check your username/password.') } } function setAccessTokensToServers (servers: ServerInfo[]) { const tasks: Promise[] = [] for (const server of servers) { const p = serverLogin(server).then(t => { server.accessToken = t }) tasks.push(p) } return Promise.all(tasks) } function loginUsingExternalToken (server: Server, username: string, externalAuthToken: string, expectedStatus = HttpStatusCode.OK_200) { const path = '/api/v1/users/token' const body = { client_id: server.client.id, client_secret: server.client.secret, username: username, response_type: 'code', grant_type: 'password', scope: 'upload', externalAuthToken } return request(server.url) .post(path) .type('form') .send(body) .expect(expectedStatus) } // --------------------------------------------------------------------------- export { login, logout, serverLogin, refreshToken, userLogin, getAccessToken, setAccessTokensToServers, Server, Client, User, loginUsingExternalToken }