83 lines
3.0 KiB
JavaScript
83 lines
3.0 KiB
JavaScript
const rgb = (r, g, b, msg) => `\x1b[38;2;${r};${g};${b}m${msg}\x1b[0m`;
|
|
const log = (...args) => console.log(`[${rgb(88, 101, 242, "arRPC")} > ${rgb(87, 242, 135, "bridge")}]`, ...args);
|
|
|
|
const IPCServer = require("./transports/ipc.js");
|
|
|
|
const WSServer = require("./transports/websocket.js");
|
|
const Bridge = require("./bridge.js");
|
|
|
|
const lookupAsset = (name, assets) => {
|
|
return assets.find((x) => x.name === name)?.id;
|
|
};
|
|
class RPCServer {
|
|
constructor() {
|
|
return (async () => {
|
|
this.onConnection = this.onConnection.bind(this);
|
|
this.onMessage = this.onMessage.bind(this);
|
|
|
|
this.ipc = await new IPCServer.IPCServer(this.onMessage, this.onConnection);
|
|
this.ws = await new WSServer.WSServer(this.onMessage, this.onConnection);
|
|
})();
|
|
}
|
|
|
|
onConnection(socket) {
|
|
socket.send({
|
|
cmd: "DISPATCH",
|
|
evt: "READY",
|
|
data: {}
|
|
});
|
|
}
|
|
|
|
async onMessage(socket, {cmd, args}) {
|
|
switch (cmd) {
|
|
case "SET_ACTIVITY":
|
|
if (!socket.application) {
|
|
socket.application = await (
|
|
await fetch(`https://discord.com/api/v9/oauth2/applications/${socket.clientId}/rpc`)
|
|
).json();
|
|
socket.application.assets = await (
|
|
await fetch(`https://discord.com/api/v9/oauth2/applications/${socket.clientId}/assets`)
|
|
).json();
|
|
log("fetched app info for", socket.clientId, socket.application);
|
|
}
|
|
|
|
const {activity, pid} = args;
|
|
const {buttons, timestamps, instance} = activity;
|
|
|
|
const metadata = {};
|
|
const extra = {};
|
|
if (buttons) {
|
|
metadata.button_urls = buttons.map((x) => x.url);
|
|
extra.buttons = buttons.map((x) => x.label);
|
|
}
|
|
|
|
if (timestamps)
|
|
for (const x in timestamps) {
|
|
if (Date.now().toString().length - timestamps[x].toString().length > 2)
|
|
timestamps[x] = Math.floor(1e3 * timestamps[x]);
|
|
}
|
|
|
|
if (activity.assets?.large_image)
|
|
activity.assets.large_image = lookupAsset(activity.assets.large_image, socket.application.assets);
|
|
if (activity.assets?.small_image)
|
|
activity.assets.small_image = lookupAsset(activity.assets.small_image, socket.application.assets);
|
|
|
|
Bridge.send({
|
|
activity: {
|
|
name: socket.application.name,
|
|
application_id: socket.application.id,
|
|
type: 0,
|
|
metadata,
|
|
flags: instance ? 1 << 0 : 0,
|
|
...activity,
|
|
...extra
|
|
},
|
|
pid
|
|
});
|
|
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
module.exports = {RPCServer};
|