import { readFile } from 'fs' import * as request from 'supertest' import { join, isAbsolute } from 'path' import * as parseTorrent from 'parse-torrent' import { makeGetRequest } from './requests' import { readFilePromise } from './miscs' import { ServerInfo } from './servers' import { getMyUserInformation } from './users' import { VideoPrivacy } from '../../../shared' type VideoAttributes = { name?: string category?: number licence?: number language?: number nsfw?: boolean description?: string tags?: string[] channelId?: number privacy?: VideoPrivacy fixture?: string } function getVideoCategories (url: string) { const path = '/api/v1/videos/categories' return makeGetRequest(url, path) } function getVideoLicences (url: string) { const path = '/api/v1/videos/licences' return makeGetRequest(url, path) } function getVideoLanguages (url: string) { const path = '/api/v1/videos/languages' return makeGetRequest(url, path) } function getVideoPrivacies (url: string) { const path = '/api/v1/videos/privacies' return makeGetRequest(url, path) } function getVideo (url: string, id: number | string, expectedStatus = 200) { const path = '/api/v1/videos/' + id return request(url) .get(path) .set('Accept', 'application/json') .expect(expectedStatus) } function viewVideo (url: string, id: number | string, expectedStatus = 204) { const path = '/api/v1/videos/' + id + '/views' return request(url) .post(path) .set('Accept', 'application/json') .expect(expectedStatus) } function getVideoWithToken (url: string, token: string, id: number | string, expectedStatus = 200) { const path = '/api/v1/videos/' + id return request(url) .get(path) .set('Authorization', 'Bearer ' + token) .set('Accept', 'application/json') .expect(expectedStatus) } function getVideoDescription (url: string, descriptionPath: string) { return request(url) .get(descriptionPath) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) } function getVideosList (url: string) { const path = '/api/v1/videos' return request(url) .get(path) .query({ sort: 'name' }) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) } function getMyVideos (url: string, accessToken: string, start: number, count: number, sort?: string) { const path = '/api/v1/users/me/videos' const req = request(url) .get(path) .query({ start: start }) .query({ count: count }) if (sort) req.query({ sort }) return req.set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(200) .expect('Content-Type', /json/) } function getVideosListPagination (url: string, start: number, count: number, sort?: string) { const path = '/api/v1/videos' const req = request(url) .get(path) .query({ start: start }) .query({ count: count }) if (sort) req.query({ sort }) return req.set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) } function getVideosListSort (url: string, sort: string) { const path = '/api/v1/videos' return request(url) .get(path) .query({ sort: sort }) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) } function removeVideo (url: string, token: string, id: number, expectedStatus = 204) { const path = '/api/v1/videos' return request(url) .delete(path + '/' + id) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + token) .expect(expectedStatus) } function searchVideo (url: string, search: string) { const path = '/api/v1/videos' const req = request(url) .get(path + '/search') .query({ search }) .set('Accept', 'application/json') return req.expect(200) .expect('Content-Type', /json/) } function searchVideoWithPagination (url: string, search: string, start: number, count: number, sort?: string) { const path = '/api/v1/videos' const req = request(url) .get(path + '/search') .query({ start }) .query({ search }) .query({ count }) if (sort) req.query({ sort }) return req.set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) } function searchVideoWithSort (url: string, search: string, sort: string) { const path = '/api/v1/videos' return request(url) .get(path + '/search') .query({ search }) .query({ sort }) .set('Accept', 'application/json') .expect(200) .expect('Content-Type', /json/) } async function testVideoImage (url: string, imageName: string, imagePath: string) { // Don't test images if the node env is not set // Because we need a special ffmpeg version for this test if (process.env['NODE_TEST_IMAGE']) { const res = await request(url) .get(imagePath) .expect(200) const data = await readFilePromise(join(__dirname, '..', 'api', 'fixtures', imageName + '.jpg')) return data.equals(res.body) } else { console.log('Do not test images. Enable it by setting NODE_TEST_IMAGE env variable.') return true } } async function uploadVideo (url: string, accessToken: string, videoAttributesArg: VideoAttributes, specialStatus = 204) { const path = '/api/v1/videos/upload' let defaultChannelId = '1' try { const res = await getMyUserInformation(url, accessToken) defaultChannelId = res.body.videoChannels[0].id } catch (e) { /* empty */ } // Default attributes let attributes = { name: 'my super video', category: 5, licence: 4, language: 3, channelId: defaultChannelId, nsfw: true, description: 'my super description', tags: [ 'tag' ], privacy: VideoPrivacy.PUBLIC, fixture: 'video_short.webm' } attributes = Object.assign(attributes, videoAttributesArg) const req = request(url) .post(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .field('name', attributes.name) .field('category', attributes.category.toString()) .field('licence', attributes.licence.toString()) .field('nsfw', JSON.stringify(attributes.nsfw)) .field('description', attributes.description) .field('privacy', attributes.privacy.toString()) .field('channelId', attributes.channelId) if (attributes.language !== undefined) { req.field('language', attributes.language.toString()) } for (let i = 0; i < attributes.tags.length; i++) { req.field('tags[' + i + ']', attributes.tags[i]) } let filePath = '' if (isAbsolute(attributes.fixture)) { filePath = attributes.fixture } else { filePath = join(__dirname, '..', 'api', 'fixtures', attributes.fixture) } return req.attach('videofile', filePath) .expect(specialStatus) } function updateVideo (url: string, accessToken: string, id: number, attributes: VideoAttributes, specialStatus = 204) { const path = '/api/v1/videos/' + id const body = {} if (attributes.name) body['name'] = attributes.name if (attributes.category) body['category'] = attributes.category if (attributes.licence) body['licence'] = attributes.licence if (attributes.language) body['language'] = attributes.language if (attributes.nsfw) body['nsfw'] = attributes.nsfw if (attributes.description) body['description'] = attributes.description if (attributes.tags) body['tags'] = attributes.tags if (attributes.privacy) body['privacy'] = attributes.privacy return request(url) .put(path) .send(body) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .expect(specialStatus) } function rateVideo (url: string, accessToken: string, id: number, rating: string, specialStatus = 204) { const path = '/api/v1/videos/' + id + '/rate' return request(url) .put(path) .set('Accept', 'application/json') .set('Authorization', 'Bearer ' + accessToken) .send({ rating }) .expect(specialStatus) } function parseTorrentVideo (server: ServerInfo, videoUUID: string, resolution: number) { return new Promise((res, rej) => { const torrentName = videoUUID + '-' + resolution + '.torrent' const torrentPath = join(__dirname, '..', '..', '..', 'test' + server.serverNumber, 'torrents', torrentName) readFile(torrentPath, (err, data) => { if (err) return rej(err) return res(parseTorrent(data)) }) }) } // --------------------------------------------------------------------------- export { getVideoDescription, getVideoCategories, getVideoLicences, getVideoPrivacies, getVideoLanguages, getMyVideos, getVideo, getVideoWithToken, getVideosList, getVideosListPagination, getVideosListSort, removeVideo, searchVideo, searchVideoWithPagination, searchVideoWithSort, testVideoImage, uploadVideo, updateVideo, rateVideo, viewVideo, parseTorrentVideo }