cli.js

const { Command } = require("commander");
const { Client } = require("@stomp/stompjs");
const readline = require("readline");
const { WebSocket } = require("ws");

Object.assign(global, { WebSocket });

const program = new Command();

program
 .version("0.0.1")
 .description("Interactive CLI for STOMP server")
 .argument("<brokerURL>", "STOMP server host (ws:// or wss://)");

program.parse(process.argv);

const brokerURL = program.args[0];

const client = new Client({
 brokerURL: brokerURL,
 onConnect: () => {
 console.log("Connected to the STOMP server.", brokerURL);
 showPrompt();
 },
 onStompError: (frame) => {
 console.error("Broker error: ", frame.headers["message"]);
 console.error("Additional details: ", frame.body);
 },
 onWebSocketClose: () => {
 console.log("WebSocket is closed.");
 },
});

client.activate();

const rl = readline.createInterface({
 input: process.stdin,
 output: process.stdout,
 prompt: "stompjs-cli> ",
});

let subscriptions = {};

function showPrompt() {
 console.log(`
Available commands:
 sub <destination> - Subscribe to a destination, example 'sub /topic/bar'
 pub <destination> <message> - Publish a message to a destination, example 'pub /topic/bar hello'
 list - List current subscriptions
 exit - Exit the CLI
 `);
 rl.prompt();
}

rl.on("line", (line) => {
 const input = line.trim().split(" ");
 const command = input[0];

 switch (command) {
 case "sub": {
 const destination = input[1];
 if (!destination) {
 console.log("Error: Missing destination.");
 break;
 }
 if (subscriptions[destination]) {
 console.log(`Already subscribed to ${destination}.`);
 break;
 }
 subscriptions[destination] = client.subscribe(destination, (message) => {
 console.log(`Received on ${destination}:`, message.body);
 rl.prompt();
 });
 console.log(`Subscribed to ${destination}.`);
 break;
 }

 case "pub": {
 const destination = input[1];
 const message = input.slice(2).join(" ");
 if (!destination ) {
 console.log("Error: Missing destination");
 break;
 }
 if (!message){
 console.log("Error: Missing message.");
 break;
 }
 try {
 let messageBody;
 try {
 messageBody = JSON.stringify(JSON.parse(message));
 } catch (error) {
 messageBody = message;
 }
 client.publish({ destination: destination, body: messageBody });
 console.log(`Published to ${destination}: ${messageBody}`);
 } catch (error) {
 console.log("Error: Invalid JSON format.");
 }
 break;
 }

 case "list": {
 console.log("Current subscriptions:");
 Object.keys(subscriptions).forEach((destination) => {
 console.log(` - ${destination}`);
 });
 break;
 }

 case "exit": {
 console.log("Exiting...");
 Object.values(subscriptions).forEach((sub) => sub.unsubscribe());
 client.deactivate();
 rl.close();
 break;
 }

 default:
 console.log("Unknown command:", command);
 break;
 }

 rl.prompt();
}).on("close", () => {
 console.log("CLI closed.");
 process.exit(0);
});