import { Buffer, net, path } from "../../deps.ts";import { RPCError } from "../utils/RPCError.ts";import { CUSTOM_RPC_ERROR_CODE, Transport, TransportOptions,} from "../structures/Transport.ts";
export enum IPC_OPCODE { HANDSHAKE, FRAME, CLOSE, PING, PONG,}
export type FormatFunction = (id: number) => string | undefined;
export type IPCTransportOptions = { pathList?: FormatFunction[];} & TransportOptions;
const defaultPathList: FormatFunction[] = [ (id: number): string | undefined => {
const isWindows = Deno.build.os === "windows";
return isWindows ? `\\\\?\\pipe\\discord-ipc-${id}` : undefined; }, (id: number): string | undefined => {
if (Deno.build.os === "windows") return;
const { XDG_RUNTIME_DIR, TMPDIR, TMP, TEMP } = Deno.env.toObject();
const prefix = Deno.realPathSync( XDG_RUNTIME_DIR ?? TMPDIR ?? TMP ?? TEMP ?? `${path.sep}tmp`, ); return path.join(prefix, `discord-ipc-${id}`); }, (id: number): string | undefined => {
if (Deno.build.os === "windows") return;
const { XDG_RUNTIME_DIR, TMPDIR, TMP, TEMP } = Deno.env.toObject();
const prefix = Deno.realPathSync( XDG_RUNTIME_DIR ?? TMPDIR ?? TMP ?? TEMP ?? `${path.sep}tmp`, ); return path.join(prefix, "snap.discord", `discord-ipc-${id}`); }, (id: number): string | undefined => {
if (Deno.build.os === "windows") return;
const { XDG_RUNTIME_DIR, TMPDIR, TMP, TEMP } = Deno.env.toObject();
const prefix = Deno.realPathSync( XDG_RUNTIME_DIR ?? TMPDIR ?? TMP ?? TEMP ?? `${path.sep}tmp`, ); return path.join( prefix, "app", "com.discordapp.Discord", `discord-ipc-${id}`, ); },];
const createSocket = async (path: string): Promise<net.Socket> => { return new Promise((resolve, reject) => { const onError = () => { socket.removeListener("conect", onConnect); reject(); };
const onConnect = () => { socket.removeListener("error", onError); resolve(socket); };
const socket = net.createConnection(path);
socket.once("connect", onConnect); socket.once("error", onError); });};
const exists = async (filename: string): Promise<boolean> => { return new Promise((resolve, reject) => { Deno.stat(filename) .then(() => resolve(true)) .catch((err) => { if (err && err.kind === Deno.errors.NotFound) return resolve(false); reject(err); }); });};
export class IPCTransport extends Transport { pathList: FormatFunction[] = defaultPathList;
private socket?: net.Socket;
get isConnected() { return this.socket !== undefined && this.socket.readyState === "open"; }
constructor(options: IPCTransportOptions) { super(options);
this.pathList = options.pathList ?? this.pathList; }
private async getSocket(): Promise<net.Socket> { if (this.socket) return this.socket;
const pathList = this.pathList; return new Promise(async (resolve, reject) => { for (const formatFunc of pathList) { const tryCreateSocket = async (path: string) => { const socket = await createSocket(path).catch(() => undefined); return socket; };
const handleSocketId = async ( id: number, ): Promise<net.Socket | undefined> => { const socketPath = formatFunc(id);
if (!socketPath || socketPath.trim() === "") return;
if ( Deno.build.os !== "windows" && !(await exists(path.dirname(socketPath))) ) { return; }
const socket = await tryCreateSocket(socketPath); return socket; };
if (this.client.pipeId) { const socket = await handleSocketId(this.client.pipeId); if (socket) { return resolve(socket); } } else { for (let i = 0; i < 10; i++) { const socket = await handleSocketId(i); if (socket) { return resolve(socket); } } } }
reject( new RPCError( CUSTOM_RPC_ERROR_CODE.RPC_COULD_NOT_CONNECT, "Could not connect", ), ); }); }
async connect(): Promise<void> { if (!this.socket) { this.socket = await this.getSocket(); }
this.emit("open");
this.send( { v: 1, client_id: this.client.clientId, }, IPC_OPCODE.HANDSHAKE, );
this.socket.on("readable", () => { let data = this.socket?.read() as Buffer | undefined; if (!data) return; this.client.emit( "debug", `SERVER => CLIENT | ${ data .toString("hex") .match(/.{1,2}/g) ?.join(" ") .toUpperCase() }`, );
do { const chunk = this.socket?.read() as Buffer | undefined; if (!chunk) break; this.client.emit( "debug", `SERVER => CLIENT | ${ chunk .toString("hex") .match(/.{1,2}/g) ?.join(" ") .toUpperCase() }`, ); data = Buffer.concat([data, chunk]); } while (true);
const op = data.readUInt32LE(0); const length = data.readUInt32LE(4); const parsedData = JSON.parse(data.subarray(8, length + 8).toString());
this.client.emit( "debug", `SERVER => CLIENT | OPCODE.${IPC_OPCODE[op]} |`, parsedData, );
switch (op) { case IPC_OPCODE.FRAME: { if (!data) break;
this.emit("message", parsedData); break; } case IPC_OPCODE.CLOSE: { this.emit("close", parsedData); break; } case IPC_OPCODE.PING: { this.send(parsedData, IPC_OPCODE.PONG); this.emit("ping"); break; } } });
this.socket.on("close", () => { this.socket = undefined; this.emit("close", "Closed by Discord"); }); }
send(message?: any, op: IPC_OPCODE = IPC_OPCODE.FRAME): void { this.client.emit( "debug", `| [CLIENT] => [SERVER] | OPCODE.${IPC_OPCODE[op]} | ${ JSON.stringify( message, null, 2, ) }`, );
const dataBuffer = message ? Buffer.from(JSON.stringify(message)) : Buffer.alloc(0);
const packet = Buffer.alloc(8); packet.writeUInt32LE(op, 0); packet.writeUInt32LE(dataBuffer.length, 4);
this.socket?.write(Buffer.concat([packet, dataBuffer])); }
ping(): void { this.send(crypto.randomUUID(), IPC_OPCODE.PING); }
close(): Promise<void> { if (!this.socket) return new Promise((resolve) => void resolve());
return new Promise((resolve) => { this.socket!.once("close", () => { this.emit("close", "Closed by client"); this.socket = undefined; resolve(); }); this.socket!.end(); }); }}