File size: 4,781 Bytes
a2b2aac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 |
// @ts-check
import yargs from 'yargs'
import os from 'os'
import path from 'path'
import { fileURLToPath, pathToFileURL } from 'url'
import { createRequire } from 'module'
import fs from 'fs'
import Stream, { Readable } from 'stream'
/**
* @param {ImportMeta | string} pathURL
* @param {boolean?} rmPrefix if value is `'true'`, it will remove `'file://'` prefix, if windows it will automatically false
*/
const __filename = function filename(pathURL = import.meta, rmPrefix = os.platform() !== 'win32') {
const path = /** @type {ImportMeta} */ (pathURL).url || /** @type {String} */ (pathURL)
return rmPrefix ?
/file:\/\/\//.test(path) ?
fileURLToPath(path) :
path : /file:\/\/\//.test(path) ?
path : pathToFileURL(path).href
}
/** @param {ImportMeta | string} pathURL */
const __dirname = function dirname(pathURL) {
const dir = __filename(pathURL, true)
const regex = /\/$/
return regex.test(dir) ?
dir : fs.existsSync(dir) &&
fs.statSync(dir).isDirectory() ?
dir.replace(regex, '') :
path.dirname(dir)
}
/** @param {ImportMeta | string} dir */
const __require = function require(dir = import.meta) {
const path = /** @type {ImportMeta} */ (dir).url || /** @type {String} */ (dir)
return createRequire(path)
}
/** @param {string} file */
const checkFileExists = (file) => fs.promises.access(file, fs.constants.F_OK).then(() => true).catch(() => false)
/** @type {(name: string, path: string, query: { [Key: string]: any }, apikeyqueryname: string) => string} */
const API = (name, path = '/', query = {}, apikeyqueryname) => (name in global.APIs ? global.APIs[name] : name) + path + (query || apikeyqueryname ? '?' + new URLSearchParams(Object.entries({ ...query, ...(apikeyqueryname ? { [apikeyqueryname]: global.APIKeys[name in global.APIs ? global.APIs[name] : name] } : {}) })) : '')
/** @type {ReturnType<yargs.Argv['parse']>} */
const opts = new Object(yargs(process.argv.slice(2)).exitProcess(false).parse())
const prefix = new RegExp('^[' + (opts['prefix'] || 'xzXZ/i!#$%+£¢€¥^°=¶∆×÷π√✓©®:;?&.\\-').replace(/[|\\{}()[\]^$+*?.\-\^]/g, '\\$&') + ']')
/**
* @param {Readable} stream
* @param {string} file
* @returns {Promise<void>}
*/
const saveStreamToFile = (stream, file) => new Promise((resolve, reject) => {
const writable = stream.pipe(fs.createWriteStream(file))
writable.once('finish', () => {
resolve()
writable.destroy()
})
writable.once('error', () => {
reject()
writable.destroy()
})
})
const kDestroyed = Symbol('kDestroyed');
const kIsReadable = Symbol('kIsReadable');
const isReadableNodeStream = (obj, strict = false) => {
return !!(
obj &&
typeof obj.pipe === 'function' &&
typeof obj.on === 'function' &&
(
!strict ||
(typeof obj.pause === 'function' && typeof obj.resume === 'function')
) &&
(!obj._writableState || obj._readableState?.readable !== false) && // Duplex
(!obj._writableState || obj._readableState) // Writable has .pipe.
);
}
const isNodeStream = (obj) => {
return (
obj &&
(
obj._readableState ||
obj._writableState ||
(typeof obj.write === 'function' && typeof obj.on === 'function') ||
(typeof obj.pipe === 'function' && typeof obj.on === 'function')
)
);
}
const isDestroyed = (stream) => {
if (!isNodeStream(stream)) return null;
const wState = stream._writableState;
const rState = stream._readableState;
const state = wState || rState;
return !!(stream.destroyed || stream[kDestroyed] || state?.destroyed);
}
const isReadableFinished = (stream, strict) => {
if (!isReadableNodeStream(stream)) return null;
const rState = stream._readableState;
if (rState?.errored) return false;
if (typeof rState?.endEmitted !== 'boolean') return null;
return !!(
rState.endEmitted ||
(strict === false && rState.ended === true && rState.length === 0)
);
}
const isReadableStream = (stream) => {
if (typeof Stream.isReadable === 'function') return Stream.isReadable(stream)
if (stream && stream[kIsReadable] != null) return stream[kIsReadable];
if (typeof stream?.readable !== 'boolean') return null;
if (isDestroyed(stream)) return false;
return (
isReadableNodeStream(stream) &&
!!stream.readable &&
!isReadableFinished(stream)
) || stream instanceof fs.ReadStream || stream instanceof Readable;
}
export default {
__filename,
__dirname,
__require,
checkFileExists,
API,
saveStreamToFile,
isReadableStream,
opts,
prefix,
} |