1. ramdisk.git
  2. src
  3. index.js
 RSS Git Download  Clone
Raw View History
Blames found: 60 Mode: javascript Binary: false


Hang on, we reloading big blames...
 a7998c18
const mz = require('mz'); const path = require('path'); const utils = require('corifeus-utils'); const fsx = require('fs-extra'); const _ = require('lodash'); const clear = require('clear');
 600f492e
const ms = require('ms');
 a7998c18
 f9a09ec6
const settingsFileFun = (opts) => { const { uid } = opts return `/etc/p3x-ramdisk-${uid}.json`; }
 a7998c18
 1a19e065
const defaults = { ramdisk: {
 c0b64a3f
size: 2048,
 1a19e065
}, timer: { save: 20, } };
 a7998c18
const requireRoot = () => { if (process.getuid && process.getuid() === 0) { return true; } throw new Error('you are not root! please sudo!') } const getGeneratedDir = (homedir = process.env.HOME) => { //return path.resolve(`${__dirname}/../generated`); return path.resolve(`${homedir}/.p3x-ramdisk`) }
 f9a09ec6
const getSettings = (opts) => { const settings = require(settingsFileFun(opts));
 a7998c18
settings.generatedDir = getGeneratedDir(settings.home); settings.program = `${path.resolve(settings.generatedDir )}/p3x-ramdisk.sh`; return settings; }
 f9a09ec6
const defaultTerminal= async (opts, command, doesRequireRoot = false) => {
 a7998c18
if (doesRequireRoot ) { requireRoot(); }
 f9a09ec6
const settings = getSettings(opts);
 a7998c18
let commandExec = `${settings.program} ${command}`; await utils.childProcess.exec(commandExec, true) }
 f9a09ec6
const load = async(opts) => { await defaultTerminal(opts, 'load')
 a7998c18
}
 f9a09ec6
const link = async(opts) => { await defaultTerminal(opts, 'link')
 a7998c18
}
 f9a09ec6
const save = async(opts) => { await defaultTerminal(opts, 'save')
 a7998c18
}
 f9a09ec6
const status = async(opts) => {
 a7998c18
requireRoot(); let commandExec = `
 f9a09ec6
sudo service p3x-ramdisk-${opts.uid} status sudo systemctl status p3x-ramdisk-timer-${opts.uid}.timer
 a7998c18
`; await utils.childProcess.exec(commandExec, true) }
 f9a09ec6
const watch = async(mainOpts, options) => { const settings = getSettings(mainOpts);
 600f492e
options.watch = Number(options.watch) || 1000; const watch = ms(options.watch, { long: true }); const timer = ms(settings.timer * 1000 * 60, { long: true }); return new Promise(async () => {
 a7998c18
const show = async () => { const df = await utils.childProcess.exec(`df -h | grep -e ${settings.home}/${settings.rampath} -e Size`); const free = await utils.childProcess.exec(`free -h`); const loadLogFile = `${settings.home}/${settings.persistent}/update-at-load.log`; let loadLog; if (await mz.fs.exists(loadLogFile)) { loadLog = await mz.fs.readFile(loadLogFile); } else {
 f9a09ec6
loadLog = 'Load is not done.'
 a7998c18
} const saveLogFile = `${settings.home}/${settings.persistent}/update-at-save.log`; let saveLog; if (await mz.fs.exists(saveLogFile)) { saveLog = await mz.fs.readFile(saveLogFile); } else {
 f9a09ec6
saveLog = 'Save is not done.'
 a7998c18
} const logFile = `${settings.home}/${settings.persistent}/ramdisk-persistent.log`; let log; if (await mz.fs.exists(logFile)) {
 f9a09ec6
log = await utils.childProcess.exec(`tail -n 5 ${logFile}`);
 a7998c18
} else {
 f9a09ec6
log = 'Log is not done.'
 a7998c18
} clear() console.log(`${df.stdout.trim()} ${free.stdout} Load: ${loadLog.toString().trim().split('\n').join(' ')} Save: ${saveLog.toString().trim().split('\n').join(' ')} ${log.stdout.trim()}
 600f492e
${new Date().toLocaleString()} | Persistence ${timer} | Watch ${watch}`);
 a7998c18
}; show();
 600f492e
setInterval(show, options.watch)
 a7998c18
}) }
 f9a09ec6
const start = async(opts) => { await defaultTerminal(opts, 'start', true)
 a7998c18
}
 f9a09ec6
const stop = async(opts) => { await defaultTerminal(opts,'stop', true)
 a7998c18
} const install = async (uid, options) => { requireRoot(); // .option('-r, --rampath [type]', 'The path of the ram disk') // .option('-p, --persistent [type]', 'The path of the ram persistent') // .option('-u, --uid [type]', 'The username, it you omit it is the current user') // .option('-g, --gid [type]', 'The gid, it you omit it is the current user') // .option('-t, --timer [type]', 'The timer in minutes, minimum about 10 minutes, 20 is the best') // .option('-s, --size [type]', 'Ramdisk in size of MegaBYTEs, default is 8192') // .option('-h, --home [type]', 'Home path, defaults to /home/uid') // uid - current user default // gid - default uid // timer -- 20 min // rampath - default /home/uid/ramdisk // persistent - default /home/uid/ramdisk-persistent // home - default /home/uid/ // sciripts - is given const userid = require('userid'); const homedir = (await utils.childProcess.exec(`sudo -H -u ${uid} -i eval 'echo $HOME'`)).stdout.trim();
 c0b64a3f
options = options || {};
 a7998c18
const generateOptions = { rampath: options.rampath || 'ramdisk', persistent : options.persistent || 'ramdisk-persistent', uid : uid, uidNumber: userid.uid(uid),
 cff5d618
gidNumber: userid.gid(uid),
 a7998c18
gid : options.gid || uid,
 1a19e065
timer : options.timer || defaults.timer.save,
 c0b64a3f
size : options.size || defaults.ramdisk.size,
 a7998c18
home : homedir, } const generatedDir = getGeneratedDir(homedir) await fsx.emptyDir(generatedDir); generateOptions.script = generatedDir; const origin = `${__dirname}/../templates`; const files = await mz.fs.readdir(origin); _.templateSettings.interpolate = /{{([\s\S]+?)}}/g;
 f9a09ec6
await utils.fs.ensureFile(settingsFileFun(generateOptions), JSON.stringify(generateOptions, null, 4), true )
 a7998c18
await files.forEachAsync(async (file) => { const generatedFile = path.basename(file, '.hbs'); const buffer = await mz.fs.readFile(`${origin}/${file}`); const string = buffer.toString(); const generatedString = _.template(string)(generateOptions);
 f9a09ec6
let useredFile = generatedFile for(let type of ['.service', '.timer']) { if (useredFile.endsWith(type)) { useredFile = useredFile.substring(0, useredFile.length - type.length) useredFile += '-' + generateOptions.uid + type } } await utils.fs.ensureFile(`${generatedDir}/${useredFile}`, generatedString, true);
 a7998c18
}) const program = `${path.resolve(generateOptions.script)}/p3x-ramdisk.sh`;
 cff5d618
let command = `chown -R ${generateOptions.uidNumber}:${generateOptions.gidNumber} ${generatedDir}
 a7998c18
chmod u+x ${program} ${program} install `; //console.log(command) await utils.childProcess.exec(command, true) console.log(`
 cd1adc61
Settings: ${JSON.stringify(generateOptions, null, 4)}
 a7998c18
Final commands: -------------------------- 1) You only have to do it once, if you haven't done it before echo "tmpfs ${generateOptions.home}/${generateOptions.rampath} tmpfs gid=${userid.gid(generateOptions.gid)},uid=${userid.uid(generateOptions.uid)},size=${generateOptions.size}M 0 0" | sudo tee -a /etc/fstab sudo mount -a -------------------------- 2) verify that ramdisk is working, see it here df -h -------------------------- 3) if everything is ok, start the persistent ramdisk sudo p3x-ramdisk start `) } module.exports.install = install; module.exports.load = load; module.exports.save = save; module.exports.start = start; module.exports.stop = stop; module.exports.watch = watch; module.exports.link = link; module.exports.status = status;
 1a19e065
module.exports.requireRoot = requireRoot; module.exports.defaults = defaults;