X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Futils.ts;h=94ceb15e0cbe560b5817431c63b6f9796569e99a;hb=6dd9de95dfa39bd5c1faed00d1dbd52cd112bae0;hp=7ff1556e3c1620f5aec997ff7c138ea90668c622;hpb=c1e791bad0b079af67398f6407221e6dcbb573dd;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/utils.ts b/server/helpers/utils.ts index 7ff1556e3..94ceb15e0 100644 --- a/server/helpers/utils.ts +++ b/server/helpers/utils.ts @@ -1,15 +1,17 @@ -import { Model } from 'sequelize-typescript' -import * as ipaddr from 'ipaddr.js' import { ResultList } from '../../shared' -import { VideoResolution } from '../../shared/models/videos' -import { CONFIG } from '../initializers' -import { UserModel } from '../models/account/user' -import { ActorModel } from '../models/activitypub/actor' import { ApplicationModel } from '../models/application/application' -import { pseudoRandomBytesPromise } from './core-utils' +import { execPromise, execPromise2, pseudoRandomBytesPromise, sha256 } from './core-utils' import { logger } from './logger' - -const isCidr = require('is-cidr') +import { join } from 'path' +import { Instance as ParseTorrent } from 'parse-torrent' +import { remove } from 'fs-extra' +import * as memoizee from 'memoizee' +import { CONFIG } from '../initializers/config' + +function deleteFileAsync (path: string) { + remove(path) + .catch(err => logger.error('Cannot delete the file %s asynchronously.', path, { err })) +} async function generateRandomString (size: number) { const raw = await pseudoRandomBytesPromise(size) @@ -17,10 +19,7 @@ async function generateRandomString (size: number) { return raw.toString('hex') } -interface FormattableToJSON { - toFormattedJSON (args?: any) -} - +interface FormattableToJSON { toFormattedJSON (args?: any) } function getFormattedObjects (objects: T[], objectsTotal: number, formattedArg?: any) { const formattedObjects: U[] = [] @@ -34,139 +33,73 @@ function getFormattedObjects (objects: T[], obje } as ResultList } -async function isSignupAllowed () { - if (CONFIG.SIGNUP.ENABLED === false) { - return false - } +const getServerActor = memoizee(async function () { + const application = await ApplicationModel.load() + if (!application) throw Error('Could not load Application from database.') - // No limit and signup is enabled - if (CONFIG.SIGNUP.LIMIT === -1) { - return true - } + const actor = application.Account.Actor + actor.Account = application.Account - const totalUsers = await UserModel.countTotal() - - return totalUsers < CONFIG.SIGNUP.LIMIT -} - -function isSignupAllowedForCurrentIP (ip: string) { - const addr = ipaddr.parse(ip) - let excludeList = [ 'blacklist' ] - let matched = '' - - // if there is a valid, non-empty whitelist, we exclude all unknown adresses too - if (CONFIG.SIGNUP.FILTERS.CIDR.WHITELIST.filter(cidr => isCidr(cidr)).length > 0) { - excludeList.push('unknown') - } + return actor +}) - if (addr.kind() === 'ipv4') { - const addrV4 = ipaddr.IPv4.parse(ip) - const rangeList = { - whitelist: CONFIG.SIGNUP.FILTERS.CIDR.WHITELIST.filter(cidr => isCidr.v4(cidr)) - .map(cidr => ipaddr.IPv4.parseCIDR(cidr)), - blacklist: CONFIG.SIGNUP.FILTERS.CIDR.BLACKLIST.filter(cidr => isCidr.v4(cidr)) - .map(cidr => ipaddr.IPv4.parseCIDR(cidr)) - } - matched = ipaddr.subnetMatch(addrV4, rangeList, 'unknown') - } else if (addr.kind() === 'ipv6') { - const addrV6 = ipaddr.IPv6.parse(ip) - const rangeList = { - whitelist: CONFIG.SIGNUP.FILTERS.CIDR.WHITELIST.filter(cidr => isCidr.v6(cidr)) - .map(cidr => ipaddr.IPv6.parseCIDR(cidr)), - blacklist: CONFIG.SIGNUP.FILTERS.CIDR.BLACKLIST.filter(cidr => isCidr.v6(cidr)) - .map(cidr => ipaddr.IPv6.parseCIDR(cidr)) - } - matched = ipaddr.subnetMatch(addrV6, rangeList, 'unknown') - } +function generateVideoImportTmpPath (target: string | ParseTorrent) { + const id = typeof target === 'string' ? target : target.infoHash - return !excludeList.includes(matched) + const hash = sha256(id) + return join(CONFIG.STORAGE.TMP_DIR, hash + '-import.mp4') } -function computeResolutionsToTranscode (videoFileHeight: number) { - const resolutionsEnabled: number[] = [] - const configResolutions = CONFIG.TRANSCODING.RESOLUTIONS - - // Put in the order we want to proceed jobs - const resolutions = [ - VideoResolution.H_480P, - VideoResolution.H_360P, - VideoResolution.H_720P, - VideoResolution.H_240P, - VideoResolution.H_1080P - ] - - for (const resolution of resolutions) { - if (configResolutions[ resolution + 'p' ] === true && videoFileHeight > resolution) { - resolutionsEnabled.push(resolution) - } - } - - return resolutionsEnabled +function getSecureTorrentName (originalName: string) { + return sha256(originalName) + '.torrent' } -const timeTable = { - ms: 1, - second: 1000, - minute: 60000, - hour: 3600000, - day: 3600000 * 24, - week: 3600000 * 24 * 7, - month: 3600000 * 24 * 30 -} -export function parseDuration (duration: number | string): number { - if (typeof duration === 'number') return duration +async function getServerCommit () { + try { + const tag = await execPromise2( + '[ ! -d .git ] || git name-rev --name-only --tags --no-undefined HEAD 2>/dev/null || true', + { stdio: [ 0, 1, 2 ] } + ) - if (typeof duration === 'string') { - const split = duration.match(/^([\d\.,]+)\s?(\w+)$/) + if (tag) return tag.replace(/^v/, '') + } catch (err) { + logger.debug('Cannot get version from git tags.', { err }) + } - if (split.length === 3) { - const len = parseFloat(split[1]) - let unit = split[2].replace(/s$/i,'').toLowerCase() - if (unit === 'm') { - unit = 'ms' - } + try { + const version = await execPromise('[ ! -d .git ] || git rev-parse --short HEAD') - return (len || 1) * (timeTable[unit] || 0) - } + if (version) return version.toString().trim() + } catch (err) { + logger.debug('Cannot get version from git HEAD.', { err }) } - throw new Error('Duration could not be properly parsed') + return '' } -function resetSequelizeInstance (instance: Model, savedFields: object) { - Object.keys(savedFields).forEach(key => { - const value = savedFields[key] - instance.set(key, value) - }) -} +/** + * From a filename like "ede4cba5-742b-46fa-a388-9a6eb3a3aeb3.mp4", returns + * only the "ede4cba5-742b-46fa-a388-9a6eb3a3aeb3" part. If the filename does + * not contain a UUID, returns null. + */ +function getUUIDFromFilename (filename: string) { + const regex = /[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}/ + const result = filename.match(regex) -let serverActor: ActorModel -async function getServerActor () { - if (serverActor === undefined) { - const application = await ApplicationModel.load() - if (!application) throw Error('Could not application.') - serverActor = application.Account.Actor - } + if (!result || Array.isArray(result) === false) return null - if (!serverActor) { - logger.error('Cannot load server actor.') - process.exit(0) - } - - return Promise.resolve(serverActor) + return result[0] } -type SortType = { sortModel: any, sortValue: string } - // --------------------------------------------------------------------------- export { + deleteFileAsync, generateRandomString, getFormattedObjects, - isSignupAllowed, - isSignupAllowedForCurrentIP, - computeResolutionsToTranscode, - resetSequelizeInstance, + getSecureTorrentName, getServerActor, - SortType + getServerCommit, + generateVideoImportTmpPath, + getUUIDFromFilename }