From 6e01bf5302822022d0a6da89ad341b5fb89d1cc5 Mon Sep 17 00:00:00 2001
From: "wont.stream" <drns.pw@icloud.com>
Date: Fri, 27 Sep 2024 00:44:57 -0400
Subject: [PATCH] add arRPC plugin

---
 plugins/arRPC/index.ts    | 113 ++++++++++++++++++++++++++++++++++++++
 plugins/arRPC/plugin.json |   5 ++
 2 files changed, 118 insertions(+)
 create mode 100644 plugins/arRPC/index.ts
 create mode 100644 plugins/arRPC/plugin.json

diff --git a/plugins/arRPC/index.ts b/plugins/arRPC/index.ts
new file mode 100644
index 0000000..f352015
--- /dev/null
+++ b/plugins/arRPC/index.ts
@@ -0,0 +1,113 @@
+const {
+	flux: {
+		dispatcher: FluxDispatcher,
+		stores: { ApplicationStore },
+	},
+	http: { get },
+} = shelter;
+const apps = {};
+const assetCache = {};
+
+const fetchAssetId = async (applicationId, assetName) => {
+	// TO-DO Use APPLICATION_ASSETS_UPDATE and APPLICATION_ASSETS_FETCH
+	if (!assetCache[applicationId] && get) {
+		try {
+			const response = await get(
+				`/oauth2/applications/${applicationId}/assets`,
+			);
+
+			if (response.status !== 200) {
+				console.error("Error fetching resources");
+				return null;
+			}
+
+			assetCache[applicationId] = response.body;
+		} catch (error) {
+			console.error("Request failed", error);
+			return null;
+		}
+	}
+
+	// Find the resource ID by its name in the cached resources
+	const resource = assetCache[applicationId].find(
+		(item) => item.name === assetName,
+	);
+	return resource ? resource.id : null;
+};
+
+export class Socket extends WebSocket {
+	private _reconnectDelay: number;
+	private _url: string;
+
+	constructor(url: string) {
+		super(url);
+		this._url = url;
+		this._reconnectDelay = 1000;
+
+		/*
+		this.addEventListener("open", () => {
+			console.log(`Socket > Connected to ${new URL(url).hostname}`);
+		});
+		*/
+
+		this.addEventListener("close", () => {
+			setTimeout(() => {
+				this._reconnect();
+			}, this._reconnectDelay);
+		});
+
+		this.addEventListener("error", () => {
+			setTimeout(() => {
+				this._reconnect();
+			}, this._reconnectDelay);
+		});
+	}
+
+	private _reconnect() {
+		this.close();
+		const newSocket = new Socket(this._url);
+		Object.setPrototypeOf(this, Object.getPrototypeOf(newSocket));
+		Object.assign(this, newSocket);
+	}
+}
+
+// https://images-ext-1.discordapp.net/external/1u9jSonO6pasZ41RA1LGqSbTysKHHF0MzoL0YDeTJg0/https/lh3.googleusercontent.com/fhDgRO0LrPPo9CDqsLQlxR3CVZc8xPmSi9Ja8DKAS5zhoWsZKdj2scyWqBUU2t4DHxK1xcbWKY2Q7cpj%3Dw544-h544-l90-rj?format=webp&width=300&height=300
+export function onLoad() {
+	const ws = new Socket("ws://localhost:1337");
+
+	ws.onmessage = async ({ data }) => {
+		const msg = JSON.parse(data);
+
+		if (msg.activity?.assets?.large_image.startsWith("https://")) {
+			msg.activity.assets.large_image = `https://images-ext-1.discordapp.net/external/${msg.activity.assets.large_image.replace("https://", "https/")}?format=webp&width=300&height=300`;
+		} else {
+			if (msg.activity?.assets?.large_image)
+				msg.activity.assets.large_image = await fetchAssetId(
+					msg.activity.application_id,
+					msg.activity.assets.large_image,
+				);
+			if (msg.activity?.assets?.small_image)
+				msg.activity.assets.small_image = await fetchAssetId(
+					msg.activity.application_id,
+					msg.activity.assets.small_image,
+				);
+		}
+		if (msg.activity) {
+			// TODO - Support games from DB too lool
+			const appId = msg.activity.application_id;
+			if (!apps[appId])
+				apps[appId] =
+					(await (Array.isArray(ApplicationStore)
+						? ApplicationStore[0].getApplication(appId)
+						: ApplicationStore.getApplication(appId))) || "??";
+
+			const app = apps[appId];
+			if (!msg.activity.name) msg.activity.name = app.name;
+		}
+		//console.warn(msg); ??
+		FluxDispatcher.dispatch({
+			type: "LOCAL_ACTIVITY_UPDATE",
+			...msg,
+		}); // set RPC status
+	};
+}
diff --git a/plugins/arRPC/plugin.json b/plugins/arRPC/plugin.json
new file mode 100644
index 0000000..a0faa87
--- /dev/null
+++ b/plugins/arRPC/plugin.json
@@ -0,0 +1,5 @@
+{
+	"name": "ArmCord RPC (modded)",
+	"author": "smartfrigde",
+	"description": "Links arRPC to ArmCord"
+}