PeerTube/server/helpers/utils.ts

105 lines
3.0 KiB
TypeScript
Raw Normal View History

import { ResultList } from '../../shared'
2017-12-14 17:38:41 +01:00
import { ApplicationModel } from '../models/application/application'
2018-10-03 14:35:35 +02:00
import { execPromise, execPromise2, pseudoRandomBytesPromise, sha256 } from './core-utils'
import { logger } from './logger'
2018-08-07 15:17:17 +02:00
import { join } from 'path'
2018-08-07 09:54:36 +02:00
import { Instance as ParseTorrent } from 'parse-torrent'
2018-08-27 16:23:34 +02:00
import { remove } from 'fs-extra'
2018-09-14 11:52:23 +02:00
import * as memoizee from 'memoizee'
2019-04-11 11:33:44 +02:00
import { CONFIG } from '../initializers/config'
2018-07-31 15:09:34 +02:00
function deleteFileAsync (path: string) {
2018-08-27 16:23:34 +02:00
remove(path)
2018-07-31 15:09:34 +02:00
.catch(err => logger.error('Cannot delete the file %s asynchronously.', path, { err }))
}
async function generateRandomString (size: number) {
const raw = await pseudoRandomBytesPromise(size)
return raw.toString('hex')
2017-02-26 18:57:33 +01:00
}
2019-08-21 14:31:57 +02:00
interface FormattableToJSON<U, V> {
toFormattedJSON (args?: U): V
}
2019-07-23 09:48:48 +02:00
function getFormattedObjects<U, V, T extends FormattableToJSON<U, V>> (objects: T[], objectsTotal: number, formattedArg?: U) {
const formattedObjects = objects.map(o => o.toFormattedJSON(formattedArg))
2017-01-04 20:59:23 +01:00
return {
2017-01-04 20:59:23 +01:00
total: objectsTotal,
2017-08-25 11:45:31 +02:00
data: formattedObjects
2019-07-23 09:48:48 +02:00
} as ResultList<V>
2017-01-04 20:59:23 +01:00
}
2018-09-14 11:52:23 +02:00
const getServerActor = memoizee(async function () {
const application = await ApplicationModel.load()
if (!application) throw Error('Could not load Application from database.')
2018-07-26 10:45:10 +02:00
const actor = application.Account.Actor
actor.Account = application.Account
return actor
2020-01-03 13:47:45 +01:00
}, { promise: true })
2017-11-13 17:39:41 +01:00
2018-12-04 16:02:49 +01:00
function generateVideoImportTmpPath (target: string | ParseTorrent) {
2018-08-07 09:54:36 +02:00
const id = typeof target === 'string' ? target : target.infoHash
const hash = sha256(id)
2018-12-04 16:02:49 +01:00
return join(CONFIG.STORAGE.TMP_DIR, hash + '-import.mp4')
2018-08-06 17:13:39 +02:00
}
2018-08-07 09:54:36 +02:00
function getSecureTorrentName (originalName: string) {
return sha256(originalName) + '.torrent'
}
Handle blacklist (#84) * Client: Add list blacklist feature * Server: Add list blacklist feature * Client: Add videoId column * Server: Add some video infos in the REST api * Client: Add video information in the blacklist list * Fix sortable columns :) * Client: Add removeFromBlacklist feature * Server: Add removeFromBlacklist feature * Move to TypeScript * Move to TypeScript and Promises * Server: Fix blacklist list sort * Server: Fetch videos informations * Use common shared interface for client and server * Add check-params remove blacklisted video tests * Add check-params list blacklisted videos tests * Add list blacklist tests * Add remove from blacklist tests * Add video blacklist management tests * Fix rebase onto develop issues * Server: Add sort on blacklist id column * Server: Add blacklists library * Add blacklist id sort test * Add check-params tests for blacklist list pagination, count and sort * Fix coding style * Increase Remote API tests timeout * Increase Request scheduler API tests timeout * Fix typo * Increase video transcoding API tests timeout * Move tests to Typescript * Use lodash orderBy method * Fix typos * Client: Remove optional tests in blacklist model attributes * Move blacklist routes from 'blacklists' to 'blacklist' * CLient: Remove blacklist-list.component.scss * Rename 'blacklists' files to 'blacklist' * Use only BlacklistedVideo interface * Server: Use getFormattedObjects method in listBlacklist method * Client: Use new coding style * Server: Use new sort validator methods * Server: Use new checkParams methods * Client: Fix sortable columns
2017-09-22 09:13:43 +02:00
2018-11-23 11:06:10 +01:00
async function getServerCommit () {
2018-10-03 14:35:35 +02:00
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 (tag) return tag.replace(/^v/, '')
} catch (err) {
logger.debug('Cannot get version from git tags.', { err })
}
try {
const version = await execPromise('[ ! -d .git ] || git rev-parse --short HEAD')
if (version) return version.toString().trim()
} catch (err) {
logger.debug('Cannot get version from git HEAD.', { err })
}
2018-11-23 11:06:10 +01:00
return ''
}
/**
* 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)
if (!result || Array.isArray(result) === false) return null
return result[0]
}
// ---------------------------------------------------------------------------
2016-01-31 11:23:52 +01:00
2017-05-15 22:22:03 +02:00
export {
2018-07-31 15:09:34 +02:00
deleteFileAsync,
2017-05-15 22:22:03 +02:00
generateRandomString,
2017-08-25 11:45:31 +02:00
getFormattedObjects,
2018-08-07 09:54:36 +02:00
getSecureTorrentName,
2017-12-14 17:38:41 +01:00
getServerActor,
2018-11-23 11:06:10 +01:00
getServerCommit,
2018-12-04 16:02:49 +01:00
generateVideoImportTmpPath,
getUUIDFromFilename
2017-05-15 22:22:03 +02:00
}