PeerTube/server/helpers/utils.ts

100 lines
2.8 KiB
TypeScript
Raw Normal View History

import { ResultList } from '../../shared'
2020-01-31 16:56:52 +01:00
import { execPromise, execPromise2, randomBytesPromise, 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'
2019-04-11 11:33:44 +02:00
import { CONFIG } from '../initializers/config'
2020-04-03 15:41:39 +02:00
import { isVideoFileExtnameValid } from './custom-validators/videos'
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) {
2020-01-31 16:56:52 +01:00
const raw = await randomBytesPromise(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
}
2020-04-03 15:41:39 +02:00
function generateVideoImportTmpPath (target: string | ParseTorrent, extensionArg?: string) {
const id = typeof target === 'string'
? target
: target.infoHash
let extension = '.mp4'
if (extensionArg && isVideoFileExtnameValid(extensionArg)) {
extension = extensionArg
}
2018-08-07 09:54:36 +02:00
const hash = sha256(id)
2020-04-03 15:41:39 +02:00
return join(CONFIG.STORAGE.TMP_DIR, `${hash}-import${extension}`)
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,
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
}