import { remove } from 'fs-extra' import { Instance as ParseTorrent } from 'parse-torrent' import { join } from 'path' import { sha256 } from '@shared/extra-utils' import { ResultList } from '@shared/models' import { CONFIG } from '../initializers/config' import { randomBytesPromise } from './core-utils' import { logger } from './logger' function deleteFileAndCatch (path: string) { remove(path) .catch(err => logger.error('Cannot delete the file %s asynchronously.', path, { err })) } async function generateRandomString (size: number) { const raw = await randomBytesPromise(size) return raw.toString('hex') } interface FormattableToJSON { toFormattedJSON (args?: U): V } function getFormattedObjects> (objects: T[], objectsTotal: number, formattedArg?: U) { const formattedObjects = objects.map(o => o.toFormattedJSON(formattedArg)) return { total: objectsTotal, data: formattedObjects } as ResultList } function generateVideoImportTmpPath (target: string | ParseTorrent, extension = '.mp4') { const id = typeof target === 'string' ? target : target.infoHash const hash = sha256(id) return join(CONFIG.STORAGE.TMP_DIR, `${hash}-import${extension}`) } function getSecureTorrentName (originalName: string) { return sha256(originalName) + '.torrent' } /** * 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] } // --------------------------------------------------------------------------- export { deleteFileAndCatch, generateRandomString, getFormattedObjects, getSecureTorrentName, generateVideoImportTmpPath, getUUIDFromFilename }