X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Futils.ts;h=a1ed8e72df461650e7f697c7bebf8cc94b62a5a2;hb=41a676db3989fe3eca91301ac5f5aea30d98654a;hp=2ad87951e7b53dbcad6fdabe8edeabf458551508;hpb=990b6a0b0c4fbebc165e5cf7cec8fbc1cbaa6c66;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/utils.ts b/server/helpers/utils.ts index 2ad87951e..a1ed8e72d 100644 --- a/server/helpers/utils.ts +++ b/server/helpers/utils.ts @@ -1,40 +1,15 @@ -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, sha256, unlinkPromise } from './core-utils' +import { pseudoRandomBytesPromise, sha256 } from './core-utils' import { logger } from './logger' -import { isArray } from './custom-validators/misc' -import * as crypto from "crypto" -import { join } from "path" +import { join } from 'path' import { Instance as ParseTorrent } from 'parse-torrent' - -const isCidr = require('is-cidr') - -function cleanUpReqFiles (req: { files: { [ fieldname: string ]: Express.Multer.File[] } | Express.Multer.File[] }) { - const files = req.files - - if (!files) return - - if (isArray(files)) { - (files as Express.Multer.File[]).forEach(f => deleteFileAsync(f.path)) - return - } - - for (const key of Object.keys(files)) { - const file = files[key] - - if (isArray(file)) file.forEach(f => deleteFileAsync(f.path)) - else deleteFileAsync(file.path) - } -} +import { remove } from 'fs-extra' function deleteFileAsync (path: string) { - unlinkPromise(path) + remove(path) .catch(err => logger.error('Cannot delete the file %s asynchronously.', path, { err })) } @@ -61,127 +36,23 @@ function getFormattedObjects (objects: T[], obje } as ResultList } -async function isSignupAllowed () { - if (CONFIG.SIGNUP.ENABLED === false) { - return false - } - - // No limit and signup is enabled - if (CONFIG.SIGNUP.LIMIT === -1) { - return true - } - - 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') - } - - 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') - } - - return !excludeList.includes(matched) -} - -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 -} - -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 - - if (typeof duration === 'string') { - const split = duration.match(/^([\d\.,]+)\s?(\w+)$/) - - if (split.length === 3) { - const len = parseFloat(split[1]) - let unit = split[2].replace(/s$/i,'').toLowerCase() - if (unit === 'm') { - unit = 'ms' - } - - return (len || 1) * (timeTable[unit] || 0) - } - } - - throw new Error('Duration could not be properly parsed') -} - -function resetSequelizeInstance (instance: Model, savedFields: object) { - Object.keys(savedFields).forEach(key => { - const value = savedFields[key] - instance.set(key, value) - }) -} - -let serverActor: ActorModel async function getServerActor () { - if (serverActor === undefined) { + if (getServerActor.serverActor === undefined) { const application = await ApplicationModel.load() if (!application) throw Error('Could not load Application from database.') - serverActor = application.Account.Actor + getServerActor.serverActor = application.Account.Actor } - if (!serverActor) { + if (!getServerActor.serverActor) { logger.error('Cannot load server actor.') process.exit(0) } - return Promise.resolve(serverActor) + return Promise.resolve(getServerActor.serverActor) +} +namespace getServerActor { + export let serverActor: ActorModel } function generateVideoTmpPath (target: string | ParseTorrent) { @@ -195,21 +66,13 @@ function getSecureTorrentName (originalName: string) { return sha256(originalName) + '.torrent' } -type SortType = { sortModel: any, sortValue: string } - // --------------------------------------------------------------------------- export { - cleanUpReqFiles, deleteFileAsync, generateRandomString, getFormattedObjects, - isSignupAllowed, getSecureTorrentName, - isSignupAllowedForCurrentIP, - computeResolutionsToTranscode, - resetSequelizeInstance, getServerActor, - SortType, generateVideoTmpPath }