mirror of
https://github.com/HeyPuter/puter.git
synced 2026-08-08 23:31:24 +00:00
KernelDeimos
5433dde6d7
This extensions brings back the dev-socket functionality, which is really important when testing things like broadcast, alarms, events, etc ; it saves a lot of time if you can invoke a command directly to the backend. This is an optional extension that will not be included in production deployments. This is for development purposes only.
105 lines
3.3 KiB
JavaScript
105 lines
3.3 KiB
JavaScript
/*
|
||
* Copyright (C) 2024-present Puter Technologies Inc.
|
||
*
|
||
* This file is part of Puter.
|
||
*
|
||
* Puter is free software: you can redistribute it and/or modify
|
||
* it under the terms of the GNU Affero General Public License as published
|
||
* by the Free Software Foundation, either version 3 of the License, or
|
||
* (at your option) any later version.
|
||
*
|
||
* This program is distributed in the hope that it will be useful,
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
* GNU Affero General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU Affero General Public License
|
||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
*/
|
||
|
||
import fs from 'node:fs';
|
||
import net from 'node:net';
|
||
import path from 'node:path';
|
||
|
||
const SOCKET_NAME = 'dev.sock';
|
||
const WELCOME = [
|
||
'Puter dev socket – enter a command (e.g. help) and press Enter.',
|
||
'Close the connection with Ctrl+C or by typing exit.',
|
||
'',
|
||
].join('\n');
|
||
|
||
function getSocketDir () {
|
||
if ( process.env.PUTER_DEV_SOCKET_DIR ) {
|
||
return process.env.PUTER_DEV_SOCKET_DIR;
|
||
}
|
||
const volatileRuntime = path.join(process.cwd(), 'volatile', 'runtime');
|
||
if ( fs.existsSync(volatileRuntime) ) {
|
||
return volatileRuntime;
|
||
}
|
||
return process.cwd();
|
||
}
|
||
|
||
extension.on('init', async () => {
|
||
if ( process.env.DEVCONSOLE !== '1' ) {
|
||
return;
|
||
}
|
||
|
||
const commands = extension.import('service:commands');
|
||
const socketDir = getSocketDir();
|
||
const socketPath = path.join(socketDir, SOCKET_NAME);
|
||
|
||
try {
|
||
if ( fs.existsSync(socketPath) ) {
|
||
fs.unlinkSync(socketPath);
|
||
}
|
||
fs.mkdirSync(socketDir, { recursive: true });
|
||
} catch ( err ) {
|
||
console.warn('dev-socket: could not prepare socket path', socketPath, err.message);
|
||
return;
|
||
}
|
||
|
||
const server = net.createServer((socket) => {
|
||
socket.setEncoding('utf8');
|
||
socket.write(`${WELCOME }\n> `);
|
||
let buffer = '';
|
||
socket.on('data', (chunk) => {
|
||
buffer += chunk;
|
||
const lines = buffer.split(/\r?\n/);
|
||
buffer = lines.pop() ?? '';
|
||
for ( const line of lines ) {
|
||
const trimmed = line.trim();
|
||
if ( trimmed === '' ) continue;
|
||
if ( trimmed.toLowerCase() === 'exit' ) {
|
||
socket.end();
|
||
return;
|
||
}
|
||
const log = {
|
||
log: (msg) => {
|
||
socket.write(`${String(msg) }\n`);
|
||
},
|
||
error: (msg) => {
|
||
socket.write(`${String(msg) }\n`);
|
||
},
|
||
};
|
||
commands.executeRawCommand(trimmed, log).then(() => {
|
||
socket.write('> ');
|
||
}).catch((err) => {
|
||
log.error(err?.message ?? err);
|
||
socket.write('> ');
|
||
});
|
||
}
|
||
});
|
||
socket.on('end', () => {
|
||
});
|
||
socket.on('error', () => {
|
||
});
|
||
});
|
||
|
||
server.listen(socketPath, () => {
|
||
console.log('dev-socket: socket listening at', socketPath);
|
||
});
|
||
server.on('error', (err) => {
|
||
console.warn('dev-socket: socket error', err.message);
|
||
});
|
||
});
|