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');
const ms = require('ms');
const settingsFile = `/etc/p3x-ramdisk.json`;
const defaults = {
ramdisk: {
size: 4096,
},
timer: {
save: 20,
}
};
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`)
}
const getSettings = () => {
const settings = require(settingsFile);
settings.generatedDir = getGeneratedDir(settings.home);
settings.program = `${path.resolve(settings.generatedDir
)}/p3x-ramdisk.sh`;
return settings;
}
const defaultTerminal= async (command, doesRequireRoot = false) => {
if (doesRequireRoot ) {
requireRoot();
}
const settings = getSettings();
let commandExec = `${settings.program} ${command}`;
await utils.childProcess.exec(commandExec, true)
}
const load = async() => {
await defaultTerminal('load')
}
const link = async() => {
await defaultTerminal('link')
}
const save = async() => {
await defaultTerminal('save')
}
const status = async() => {
requireRoot();
let commandExec = `
sudo service p3x-ramdisk status
sudo systemctl status p3x-ramdisk-timer.timer
`;
await utils.childProcess.exec(commandExec, true)
}
const watch = async(options) => {
const settings = getSettings();
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 () => {
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 {
loadLog = Promise.resolve('Load is not done.')
}
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 {
saveLog = Promise.resolve('Save is not done.')
}
const logFile = `${settings.home}/${settings.persistent}/ramdisk-persistent.log`;
let log;
if (await mz.fs.exists(logFile)) {
log = await utils.childProcess.exec(`tail -n 5 ${logFile}`);
} else {
log = Promise.resolve({
stdout: 'Log is not done.'
})
}
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()}
${new Date().toLocaleString()} | Persistence ${timer} | Watch ${watch}`);
};
show();
setInterval(show, options.watch)
})
}
const start = async() => {
await defaultTerminal('start', true)
}
const stop = async() => {
await defaultTerminal('stop', true)
}
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();
const generateOptions = {
rampath: options.rampath || 'ramdisk',
persistent : options.persistent || 'ramdisk-persistent',
uid : uid,
uidNumber: userid.uid(uid),
gid : options.gid || uid,
timer : options.timer || defaults.timer.save,
size : options.size || defaults.size.ramdisk,
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;
await utils.fs.ensureFile(settingsFile, JSON.stringify(generateOptions, null, 4))
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);
await utils.fs.ensureFile(`${generatedDir}/${generatedFile}`, generatedString);
})
const program = `${path.resolve(generateOptions.script)}/p3x-ramdisk.sh`;
let command = `chown -R ${generateOptions.uid}:${generateOptions.gid} ${generatedDir}
chmod u+x ${program}
${program} install
`;
//console.log(command)
await utils.childProcess.exec(command, true)
console.log(`
Settings: ${JSON.stringify(generateOptions, null, 2)}
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;
module.exports.requireRoot = requireRoot;
module.exports.defaults = defaults;
