mirror of https://github.com/jkjoy/sunpeiwen.git
57 lines
1.5 KiB
JavaScript
57 lines
1.5 KiB
JavaScript
'use strict';
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const uniqueString = require('unique-string');
|
|
const tempDir = require('temp-dir');
|
|
const isStream = require('is-stream');
|
|
const stream = require('stream');
|
|
const {promisify} = require('util');
|
|
|
|
const pipeline = promisify(stream.pipeline);
|
|
const {writeFile} = fs.promises;
|
|
|
|
const getPath = (prefix = '') => path.join(tempDir, prefix + uniqueString());
|
|
|
|
const writeStream = async (filePath, data) => pipeline(data, fs.createWriteStream(filePath));
|
|
|
|
module.exports.file = options => {
|
|
options = {
|
|
...options
|
|
};
|
|
|
|
if (options.name) {
|
|
if (options.extension !== undefined && options.extension !== null) {
|
|
throw new Error('The `name` and `extension` options are mutually exclusive');
|
|
}
|
|
|
|
return path.join(module.exports.directory(), options.name);
|
|
}
|
|
|
|
return getPath() + (options.extension === undefined || options.extension === null ? '' : '.' + options.extension.replace(/^\./, ''));
|
|
};
|
|
|
|
module.exports.directory = ({prefix = ''} = {}) => {
|
|
const directory = getPath(prefix);
|
|
fs.mkdirSync(directory);
|
|
return directory;
|
|
};
|
|
|
|
module.exports.write = async (data, options) => {
|
|
const filename = module.exports.file(options);
|
|
const write = isStream(data) ? writeStream : writeFile;
|
|
await write(filename, data);
|
|
return filename;
|
|
};
|
|
|
|
module.exports.writeSync = (data, options) => {
|
|
const filename = module.exports.file(options);
|
|
fs.writeFileSync(filename, data);
|
|
return filename;
|
|
};
|
|
|
|
Object.defineProperty(module.exports, 'root', {
|
|
get() {
|
|
return tempDir;
|
|
}
|
|
});
|