P3X GitList

Raw View History

Blames found: 75 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) => {

724156a2

const {uid} = opts

f9a09ec6

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);

724156a2

settings.program = `${path.resolve(settings.generatedDir

a7998c18

)}/p3x-ramdisk.sh`; return settings; }

724156a2

const defaultTerminal = async (opts, command, doesRequireRoot = false) => { if (doesRequireRoot) {

a7998c18

requireRoot(); }

f9a09ec6

const settings = getSettings(opts);

a7998c18

let commandExec = `${settings.program} ${command}`; await utils.childProcess.exec(commandExec, true) }

724156a2

const load = async (opts) => {

f9a09ec6

await defaultTerminal(opts, 'load')

a7998c18

}

724156a2

const link = async (opts) => {

f9a09ec6

await defaultTerminal(opts, 'link')

a7998c18

}

724156a2

const save = async (opts) => {

f9a09ec6

await defaultTerminal(opts, 'save')

a7998c18

}

724156a2

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) }

724156a2

const watch = async (mainOpts, options) => {

f9a09ec6

const settings = getSettings(mainOpts);

600f492e

options.watch = Number(options.watch) || 1000;

724156a2

const watch = ms(options.watch, {long: true}); const timer = ms(settings.timer * 1000 * 60, {long: true});

600f492e

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

}) }

724156a2

const start = async (opts) => {

f9a09ec6

await defaultTerminal(opts, 'start', true)

a7998c18

}

724156a2

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

dab4686c

const userid = require('./userid');

a7998c18

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',

724156a2

persistent: options.persistent || 'ramdisk-persistent', uid: uid,

a7998c18

uidNumber: userid.uid(uid),

cff5d618

gidNumber: userid.gid(uid),

724156a2

gid: options.gid || uid, timer: options.timer || defaults.timer.save, size: options.size || defaults.ramdisk.size, home: homedir,

a7998c18

} 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;

724156a2

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

724156a2

let useredFile = generatedFile for (let type of ['.service', '.timer']) {

f9a09ec6

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;

724156a2

module.exports.requireRoot = requireRoot; module.exports.defaults = defaults;