1
0
Fork 0

fixes to util/exec

refactor/models
Matthew 2020-05-03 23:55:12 -04:00
parent e8cfac48cc
commit 5fffe9f5f2
No known key found for this signature in database
GPG Key ID: 766BE43AE75F7559
1 changed files with 13 additions and 23 deletions

View File

@ -1,5 +1,6 @@
/* eslint-disable no-param-reassign */
import { exec, ExecOptions } from 'child_process';
import { promisify } from 'util';
import childProcess from 'child_process';
import nodemailer from 'nodemailer';
import { Message, PrivateChannel, GroupChannel, Member, User } from 'eris';
import uuid from 'uuid/v4';
@ -30,27 +31,16 @@ export default class Util {
* @param command The command to execute
* @param options childProcess.ExecOptions
*/
public async exec(command: string, options: ExecOptions = {}): Promise<string> {
return new Promise((res, rej) => {
let error = false;
let output = '';
const writeFunction = (data: string|Buffer|Error) => {
if (data instanceof Error) error = true;
output += `${data}`;
};
const cmd = exec(command, options);
cmd.stdout.on('data', writeFunction);
cmd.stderr.on('data', writeFunction);
cmd.on('error', writeFunction);
cmd.once('close', (code, signal) => {
output += `Command exited with code ${code}${signal ? `/${signal}` : ''}`;
cmd.stdout.removeListener('data', writeFunction);
cmd.stderr.removeListener('data', writeFunction);
cmd.removeListener('error', writeFunction);
if (error) rej(output);
res(output);
});
});
public async exec(command: string, options: childProcess.ExecOptions = {}): Promise<string> {
const ex = promisify(childProcess.exec);
let result: string;
try {
const res = await ex(command, options);
result = `${res.stdout}${res.stderr}`;
} catch (err) {
return Promise.reject(new Error(`Command failed: ${err.cmd}\n${err.stderr}${err.stdout}`));
}
return result;
}
/**
@ -163,7 +153,7 @@ export default class Util {
const tier = await this.client.db.Tier.findOne({ id: 1 });
const account = new this.client.db.Account({
username, userID, emailAddress, createdBy: moderatorID, ramLimitNotification: tier.resourceLimits.ram - 50, createdAt: new Date(), locked: false, tier: 1, supportKey: code, ssInit: false, homepath: `/home/${username}`,
username, userID, emailAddress, createdBy: moderatorID, createdAt: new Date(), locked: false, tier: 1, supportKey: code, ssInit: false, ramLimitNotification: tier.resourceLimits.ram - 50, homepath: `/home/${username}`,
});
return account.save();
}