Add Web/Desktop specific plugin capabilities; misc fixes

1 files changed, 109 insertions, 0 deletions

diff --git a/src/plugins/arRPC.web.tsx b/src/plugins/arRPC.web.tsx
new file mode 100644
index 0000000..9a16c64
--- /dev/null
+++ b/src/plugins/arRPC.web.tsx
@@ -0,0 +1,109 @@
+/*
+ * Vencord, a modification for Discord's desktop app
+ * Copyright (c) 2022 OpenAsar
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+*/
+
+import { popNotice, showNotice } from "@api/Notices";
+import { Link } from "@components/Link";
+import { Devs } from "@utils/constants";
+import definePlugin from "@utils/types";
+import { filters, findByCodeLazy, mapMangledModuleLazy } from "@webpack";
+import { FluxDispatcher, Forms, Toasts } from "@webpack/common";
+
+const assetManager = mapMangledModuleLazy(
+    "getAssetImage: size must === [number, number] for Twitch",
+    {
+        getAsset: filters.byCode("apply("),
+    }
+);
+
+const lookupRpcApp = findByCodeLazy(".APPLICATION_RPC(");
+
+async function lookupAsset(applicationId: string, key: string): Promise<string> {
+    return (await assetManager.getAsset(applicationId, [key, undefined]))[0];
+}
+
+const apps: any = {};
+async function lookupApp(applicationId: string): Promise<string> {
+    const socket: any = {};
+    await lookupRpcApp(socket, applicationId);
+    return socket.application;
+}
+
+let ws: WebSocket;
+export default definePlugin({
+    name: "WebRichPresence (arRPC)",
+    description: "Client plugin for arRPC to enable RPC on Discord Web (experimental)",
+    authors: [Devs.Ducko],
+
+    settingsAboutComponent: () => (
+        <>
+            <Forms.FormTitle tag="h3">How to use arRPC</Forms.FormTitle>
+            <Forms.FormText>
+                <Link href="https://github.com/OpenAsar/arrpc/tree/main#server">Follow the instructions in the GitHub repo</Link> to get the server running, and then enable the plugin.
+            </Forms.FormText>
+        </>
+    ),
+
+    async start() {
+        // ArmCord comes with its own arRPC implementation, so this plugin just confuses users
+        if ("armcord" in window) return;
+
+        if (ws) ws.close();
+        ws = new WebSocket("ws://127.0.0.1:1337"); // try to open WebSocket
+
+        ws.onmessage = async e => { // on message, set status to data
+            const data = JSON.parse(e.data);
+
+            if (data.activity?.assets?.large_image) data.activity.assets.large_image = await lookupAsset(data.activity.application_id, data.activity.assets.large_image);
+            if (data.activity?.assets?.small_image) data.activity.assets.small_image = await lookupAsset(data.activity.application_id, data.activity.assets.small_image);
+
+            if (data.activity) {
+                const appId = data.activity.application_id;
+                apps[appId] ||= await lookupApp(appId);
+
+                const app = apps[appId];
+                data.activity.name ||= app.name;
+            }
+
+            FluxDispatcher.dispatch({ type: "LOCAL_ACTIVITY_UPDATE", ...data });
+        };
+
+        const connectionSuccessful = await new Promise(res => setTimeout(() => res(ws.readyState === WebSocket.OPEN), 1000)); // check if open after 1s
+        if (!connectionSuccessful) {
+            showNotice("Failed to connect to arRPC, is it running?", "Retry", () => { // show notice about failure to connect, with retry/ignore
+                popNotice();
+                this.start();
+            });
+            return;
+        }
+
+        Toasts.show({ // show toast on success
+            message: "Connected to arRPC",
+            type: Toasts.Type.SUCCESS,
+            id: Toasts.genId(),
+            options: {
+                duration: 1000,
+                position: Toasts.Position.BOTTOM
+            }
+        });
+    },
+
+    stop() {
+        FluxDispatcher.dispatch({ type: "LOCAL_ACTIVITY_UPDATE", activity: null }); // clear status
+        ws.close(); // close WebSocket
+    }
+});