You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
3.1 KiB
108 lines
3.1 KiB
// Internal
|
|
const { spawn } = require('child_process');
|
|
const { EOL } = require('os');
|
|
const os = require('os');
|
|
|
|
// Modules
|
|
const Telegraf = require('telegraf');
|
|
|
|
//Config
|
|
const config = require('./config.js');
|
|
|
|
// Lib
|
|
const validator = require('./lib/validator.js');
|
|
const responder = require('./lib/responseHandler.js');
|
|
const sessionFinder = require('./lib/sessionFinder.js');
|
|
const listeners = require('./lib/listeners.js');
|
|
|
|
const dateOptions = {
|
|
weekday: 'long',
|
|
year: 'numeric',
|
|
month: 'long',
|
|
day: 'numeric',
|
|
hour: 'numeric',
|
|
minute: 'numeric',
|
|
};
|
|
|
|
const bot = new Telegraf(config.botApiKey);
|
|
const sessions = [];
|
|
let identifierState = 0;
|
|
sessions.history = [];
|
|
|
|
const getSession = sessionFinder(sessions);
|
|
|
|
// get os info
|
|
const home = os.homedir();
|
|
const hostname = os.hostname();
|
|
const username = os.userInfo().username;
|
|
const defaultShell = os.platform() === 'win32' ? 'cmd.exe' : 'bash';
|
|
|
|
// Validate bot's master
|
|
bot.use(validator);
|
|
|
|
bot.command('start',
|
|
ctx => {
|
|
const newProc = spawn(defaultShell, {
|
|
cwd: home
|
|
});
|
|
newProc.identifier = identifierState;
|
|
sessions[identifierState] = newProc;
|
|
identifierState++;
|
|
sessions.currentSession = newProc;
|
|
listeners.add(sessions.currentSession, responder, ctx);
|
|
return ctx.replyWithHTML(`Welcome to tsh -- <code>Telegram Shell!</code>\n\n`
|
|
+ `You are now connected to <code>${hostname}</code>`
|
|
+ ` as <strong>${username}</strong>`);
|
|
});
|
|
|
|
bot.command('ls',
|
|
ctx => ctx.reply(
|
|
sessions.reduce((acc, _, index) =>
|
|
acc ? `${acc}\n${index}` : `${index}`, '')
|
|
|| `No sessions found. Start one with /start.`
|
|
));
|
|
|
|
bot.command('attach',
|
|
ctx => {
|
|
const session = getSession(ctx)('attach');
|
|
if(!session)
|
|
return responder.fail('Session not found. /ls for list of sessions')(ctx);
|
|
sessions.currentSession = session;
|
|
listeners.add(sessions.currentSession, responder, ctx);
|
|
return responder.success(`Reattached to shell ${session.identifier}`)(ctx);
|
|
});
|
|
|
|
bot.command('detach',
|
|
ctx => {
|
|
const session = getSession(ctx)('detach') || sessions.currentSession;
|
|
if(!session)
|
|
return responder.fail('Session not found. /ls for list of sessions.')(ctx);
|
|
listeners.remove(session);
|
|
sessions.currentSession = undefined;
|
|
return responder.success(`Detached from shell ${session.identifier}`)(ctx);
|
|
});
|
|
|
|
bot.command('kill',
|
|
ctx => {
|
|
const session = getSession(ctx)('kill') || sessions.currentSession;
|
|
if(!session)
|
|
return responder.fail('Session not found. /ls for list of sessions.')(ctx);
|
|
delete sessions[session.identifier];
|
|
if(session === sessions.currentSession) sessions.currentSession = undefined;
|
|
session.kill();
|
|
ctx.reply('Session killed. /ls for list of sessions.')
|
|
})
|
|
|
|
bot.use(ctx => {
|
|
if(!sessions.currentSession)
|
|
return responder.fail(`No active session. `
|
|
+ `Start one with /start or view list of sessions by sending /ls.`)(ctx);
|
|
const cmd = ctx.update.message.text;
|
|
const history = `${new Date().toLocaleDateString('en-IN', dateOptions)}: ${cmd}`;
|
|
sessions.history.push(history);
|
|
console.log(history);
|
|
sessions.currentSession.stdin.write(cmd + EOL);
|
|
});
|
|
|
|
bot.startPolling();
|
|
console.log(`Polling for updates.`);
|
|
|