tmux-iocompiler
Version:
Realtime i/o compiler bot
213 lines (212 loc) • 8.29 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const helpers_1 = __importDefault(require("../helpers"));
const config_1 = __importDefault(require("../config"));
const child_process_1 = require("child_process");
const fs_1 = __importDefault(require("fs"));
const path_1 = __importDefault(require("path"));
let h = new helpers_1.default();
let cyoyoc = async (ctx, obj = {}) => {
// obj = obj || {}
const edit = async (messageId, messageText) => {
return await ctx.telegram.editMessageText(ctx.chat.id, messageId, undefined, messageText + " ```", { parse_mode: "MarkdownV2" });
};
let newObj = obj[ctx.from.id];
newObj.fromId = ctx.from.id;
try {
if (newObj.status == "leave") {
reply(ctx, 'Session terminated');
ctx.scene.leave();
return await terminate(ctx, obj);
}
let previous = Date.now();
let repeats = 0;
let looperr = false;
let cout = async (tempdata) => {
if (tempdata != '-1a\n') {
let current = Date.now();
if (previous + 1000 > current)
repeats++;
if (repeats > 5 && !looperr) {
looperr = true;
await terminate(ctx, obj);
reply(ctx, 'It seems you are created infinite loop');
ctx.scene.leave();
return;
}
newObj.editedMes += tempdata.toString();
if (repeats > 5 || looperr)
return;
if (newObj.mid == 0) {
newObj.mid = await ctx.reply("" + newObj.editedMes + " ```", { parse_mode: "MarkdownV2" })
.catch((err) => {
if (err.message.includes('too long')) {
looperr = true;
reply(ctx, 'message is too long');
terminate(ctx, obj);
ctx.scene.leave();
}
});
}
else {
// newObj.editedMes += data
try {
await edit(newObj.mid.message_id, newObj.editedMes)
.catch((err) => { console.error(err); });
}
catch (err) { }
}
}
// return
if (!newObj.firstlistener)
return;
newObj.firstlistener = false;
newObj.ctxemitter.on('ctx', async (ctxx) => {
ctxx.deleteMessage().catch(() => { });
try {
newObj.editedMes += ctxx.message.text + "\n";
if (newObj.mid == 0)
newObj.mid = await ctx.reply("" + newObj.editedMes + " ```", { parse_mode: "MarkdownV2" });
else
await edit(newObj.mid.message_id, newObj.editedMes);
await newObj.ccode.stdin.write(ctxx.message.text + "\n");
}
catch (err) {
console.error(err);
}
});
};
if (newObj.status == "input") {
return await newObj.ctxemitter.emit('ctx', await (ctx));
}
obj[ctx.from.id].status = "input";
cout('-1a\n');
newObj.code = newObj.code.replace(/\u00A0/mg, ' ');
let ttl = ctx.scene.options.ttl;
newObj.fromId = ctx.message.from.id;
newObj.code = newObj.code.replace(/^(\s*)(pt)(.*)/gim, '$1printf($3);');
newObj.code = newObj.code.replace(/"start"/gi, "#include <stdio.h>\nint main(){\n")
.replace(/"end"/gi, "\nreturn 0;\n}")
.replace(/(^\s*pt)(.*)/gim, "printf($2);")
.replace(/#include\s*\<conio\.h\>/, `#include "conio.h"`);
newObj.code = newObj.code.replace(/.*\n.*printf\s{0,4}\(.+\);/g, (match) => {
if (match.includes("if")) {
return match;
}
else
return match + " fflush(stdout);";
});
newObj.timeid = setTimeout(() => {
newObj.code = false;
if (newObj && newObj.ccode) {
ctx.reply("Timout: " + ttl + " Seconds");
terminate(ctx, obj);
}
}, ttl * 1000);
newObj.cCodeDir = path_1.default.join('.', 'files', 'ccode', `c${newObj.fromId}c`);
newObj.cCodeFilePath = path_1.default.join(newObj.cCodeDir, `main.c`);
newObj.cCodeExecutablePath = path_1.default.join(newObj.cCodeDir, `main`);
try {
fs_1.default.mkdirSync(newObj.cCodeDir);
}
catch (err) {
// Handle file writing error
}
try {
fs_1.default.writeFileSync(newObj.cCodeFilePath, newObj.code);
}
catch (err) {
// Handle file writing error
}
const gccArgs = [
'-I', path_1.default.join(__dirname, "..", 'lib'),
'-o', newObj.cCodeExecutablePath,
newObj.cCodeFilePath,
path_1.default.join(__dirname, "..", 'lib', 'conio.c'),
'-lm'
];
const { status, stderr } = (0, child_process_1.spawnSync)('gcc', gccArgs);
if (status !== 0) {
terminate(ctx, obj);
reply(ctx, stderr.toString().replace(new RegExp(newObj.cCodeFilePath, 'g'), 'See'));
return ctx.scene.leave();
}
newObj.ccode = (0, child_process_1.spawn)(newObj.cCodeExecutablePath, [], config_1.default.spawnOptions || { env: {} });
newObj.ccode.setMaxListeners(0);
newObj.ccode.stdout.on('data', cout);
let m = true;
newObj.ccode.stderr.on('data', async (data) => {
if (newObj.mid == 0 && m) {
m = false;
newObj.ErrorMes = newObj.ErrorMes + data;
reply(ctx, "" + newObj.ErrorMes, 30);
}
else {
newObj.ErrorMes = newObj.ErrorMes + data;
ctx.telegram.editMessageText(ctx.chat.id, newObj.mid.message_id, undefined, newObj.ErrorMes)
.then(async (mmm) => {
await h.sleep(30000);
ctx.deleteMessage(mmm).catch(() => { });
}).catch(() => { });
}
terminate(ctx, obj);
});
newObj.code = "";
newObj.ccode.on("error", (err) => { console.log(err); terminate(ctx, obj); });
newObj.ccode.on('close', function (statusCode) {
if (statusCode == 0) {
reply(ctx, 'Program terminated successfully');
}
else {
reply(ctx, 'Program terminated unsuccessfully');
}
terminate(ctx, obj);
});
newObj.ccode.on('exit', (ex) => {
if (ex != 0) {
console.error("Something went wrong");
}
});
return newObj.ccode;
}
catch (errr) {
console.error(errr);
reply(ctx, errr.message);
terminate(ctx, obj);
}
};
module.exports = cyoyoc;
async function reply(ctx, mss, tim = 10) {
return await ctx.reply(mss).then(async (mi) => {
await h.sleep(tim * 1000);
return await ctx.deleteMessage(mi.message_id).catch((err) => { });
})
.catch((err) => { });
}
let terminate = async (ctx, options = {}) => {
let newObj = options[ctx.from.id];
if (ctx.scene)
ctx.scene.leave();
await h.sleep(100);
try {
clearTimeout(newObj.timeid);
if (newObj && newObj.ccode)
newObj.ccode.removeAllListeners();
}
catch (error) {
}
try {
if (fs_1.default.existsSync(newObj.cCodeDir)) {
fs_1.default.rmSync(newObj.cCodeDir, { recursive: true });
}
}
catch (err) { }
console.log('terminating...');
if (newObj && newObj.ctxemitter)
newObj.ctxemitter.removeAllListeners();
await h.sleep(10);
options[ctx.from.id] = undefined;
};