client.js

import { io } from "socket.io-client";
import chalk from "chalk";
import CryptoJS from 'crypto-js';

const { AES } = CryptoJS;

function devLog(...args) {
	// if (process.env.ENV === "dev") 
	console.log(chalk.bold(...args));
	// else return
}

function errorLog(...args) {
	console.error(chalk.redBright("Ignition error:", ...args))
}

export class Ignition {

	#socket;#encryptionKey;

	constructor(config) {
		this.url = config.url;
		this.#encryptionKey = config.encryptionKey;
		this.apiKey = config.key;
		this.groupId = undefined;
		this.groupCount = 0;

		// this.#socket = io(this.url, { // public shared ignition websocket server URL - Elastic Ip
		// 	auth: {
		// 		token: "abc12387878787",
		// 	}
		// });

		this.#socket = io(this.url + "/" + this.apiKey);

		this.#socket.on("ERROR", (message) => { errorLog(message) });
		this.#socket.on("LOG", (message) => { devLog(chalk.cyanBright(message)) });
		console.log(chalk.cyanBright("Ignition client connecting ........", this.url, this.apiKey));
		
	}

	ecrypt(message) {
		return AES.encrypt(message, this.#encryptionKey).toString();
	}

	decrypt(message) {
		return AES.decrypt(message, this.#encryptionKey).toString(CryptoJS.enc.Utf8);
	}

	async subscribe(groupId) {
		this.groupId = groupId;
		devLog("Attempting to subscribe to room !!");
		// pub struct JoinLeaveRequestData {
		//     pub key: String,
		//     pub group_id: String,  
		// }
		this.#socket.emit("JOIN", {
			key: this.apiKey,
			group_id: `${this.apiKey}_${groupId}`,
		},
		(data) => { // this would be recived by the server & the server will join this client int that room
			console.log(chalk.cyanBright(data));
			// this.#socket.disconnect()
		});
		// this.#socket.emit("GROUP", groupId);
	}

	async unsubscribe(groupId) {
		this.groupId = groupId; // set groupId as global class state
		devLog("Attempting to unsubscribe from room !!");
		this.#socket.emit("LEAVE", {
			key: this.apiKey,
			group_id: `${this.apiKey}_${groupId}`,
		}, 
		(data) => {
			console.log(chalk.cyanBright(data));
		});
	}

	// this method should only be used by dedictaed, dedictaed+ & enterprize clients
	// emit directly send message to the websocket server instaed appending it to the message queue.
	async emit(eventName, groupId, message) {
		if (this.url == undefined) {
			errorLog("You must have `URL` of a server to emit a direct message, try using `publish()` method for Shared users.")
		}

		if(typeof(message) == "object") {
			message = JSON.stringify(message)
		}

		console.log("EMITITING: -> ",this.#encryptionKey ? this.ecrypt(message) : message)

		// ^ FOR SHARED WSS:
		// pub struct ClientMessage {
		//     pub group_id: String,
		//     pub event_name: String,
		//     pub message: String,
		//     pub key: String,
		// }

		// ^ FOR DEDICATED WSS:
		// struct ClientPayload {
		// 	event: String,
		// 	room: String,
		// 	message: String,
		// }

		devLog("EMITTING EVENT !!")

		// this.#socket.emit("MESSAGE", {
		// 	group_id: this.apiKey + "_" + groupId,
		// 	event_name: eventName,
		// 	message: this.#encryptionKey ? this.ecrypt(message) : message,
		// 	key: this.apiKey
		// })

		this.#socket.emit("MESSAGE", {
			event: eventName,
			room: this.apiKey + "_" + groupId,
			message: this.#encryptionKey ? this.ecrypt(message) : message,
		})
	}

	async on(eventName, callback) {
		if (eventName != "connect" && eventName != "disconnect" && this.groupId == undefined) {
			errorLog("Missing `groupId`. Did you forgot to `subsribe` to a group ?");
		};
		this.#socket.on(eventName, callback);
	}

	async off(eventName, callback=undefined) {
		if (eventName != "connect" && eventName != "disconnect" && this.groupId == undefined) {
			errorLog("Missing `groupId`. Did you forgot to `subsribe` to a group ?");
		};
		this.#socket.off(eventName, callback);
	}

	// this method adds the message to the `message queue` from which shared websocket server 
	// keeps pulling and braodcasting the message to all the clients subscribed to the group.
	async publish(groupId, eventName, message) {
		try {
			const res = await fetch("ignition_application_server_url", {
				method: "POST",
				mode: "cors",
				cache: "no-cache",
				credentials: "same-origin",
				headers: {
					"Content-Type": "application/json",
				},
				body: JSON.stringify({
					"group_id": groupId,
					"event_name": eventName,
					"message": this.#encryptionKey ? this.ecrypt(message) : message,
					"key": this.apiKey
				})
			})
			if (res.status != 200) {
				throw Error(`Failed to publish message, status code: ${res.status}`)
			}
		} catch (error) {
			errorLog(error)
		}
	}

}

// let hj = io("http://localhost:3000", {
// 	auth: {
// 		token: "abc123"
// 	}
// })

// hj.on("connect", () => {
// 	console.log("connected")
// })

// hj.on("server to client event", (data) => {
// 	console.log(data)
// })

// hj.emit("client to server event", "1111111111111111111122222222222222222222333333333333333333333333333334444444444444444")

// let a = new Ignition({
// 	url: "http://localhost:4000",
// 	key:"abc123",
// 	// encryptionKey:"RADHA"
// })

// a.subscribe("test")

// a.on("connect", () => {
// 	console.log("CONNECTED ")
// })






// b.on("test", (data) => {
// 	console.log("message recived by `b`:",data)
// })

// for(let i = 0; i < 100; i++) {
	let b = new Ignition({
		url: "http://127.0.0.1:4000",
		key:"abc123",
		encryptionKey:"RADHA"
	})
	
	b.subscribe("test")
	b.emit("test", "test", "hello world")
// }