1 files changed, 73 insertions, 17 deletions

diff --git a/src/plugins/openInApp.ts b/src/plugins/openInApp.ts
index 8249ed7..52b418f 100644
--- a/src/plugins/openInApp.ts
+++ b/src/plugins/openInApp.ts
@@ -16,22 +16,41 @@
  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
+import { definePluginSettings } from "@api/Settings";
 import { Devs } from "@utils/constants";
-import definePlugin from "@utils/types";
+import definePlugin, { OptionType } from "@utils/types";
+import { showToast, Toasts } from "@webpack/common";
+import { MouseEvent } from "react";
 
-const SpotifyMatcher = /https:\/\/open\.spotify\.com\/(track|album|artist|playlist|user)\/([^S]+)/;
+const ShortUrlMatcher = /^https:\/\/(spotify\.link|s\.team)\/.+$/;
+const SpotifyMatcher = /^https:\/\/open\.spotify\.com\/(track|album|artist|playlist|user)\/(.+)(?:\?.+?)?$/;
+const SteamMatcher = /^https:\/\/(steamcommunity\.com|(?:help|store)\.steampowered\.com)\/.+$/;
+
+const settings = definePluginSettings({
+    spotify: {
+        type: OptionType.BOOLEAN,
+        description: "Open Spotify links in the Spotify app",
+        default: true,
+    },
+    steam: {
+        type: OptionType.BOOLEAN,
+        description: "Open Steam links in the Steam app",
+        default: true,
+    }
+});
 
 export default definePlugin({
     name: "OpenInApp",
-    description: "Open spotify URLs in app",
+    description: "Open Spotify and Steam URLs in the Spotify and Steam app instead of your Browser",
     authors: [Devs.Ven],
+    settings,
 
     patches: [
         {
             find: '"MaskedLinkStore"',
             replacement: {
                 match: /return ((\i)\.apply\(this,arguments\))(?=\}function \i.{0,200}\.trusted)/,
-                replace: "return $self.handleLink(...arguments)||$1"
+                replace: "return $self.handleLink(...arguments).then(handled => handled || $1)"
             }
         },
         // Make Spotify profile activity links open in app on web
@@ -52,23 +71,60 @@ export default definePlugin({
         }
     ],
 
-    handleLink(data: { href: string; }, event: MouseEvent) {
-        if (!data) return;
+    async handleLink(data: { href: string; }, event: MouseEvent) {
+        if (!data) return false;
+
+        let url = data.href;
+        if (!IS_WEB && ShortUrlMatcher.test(url)) {
+            event.preventDefault();
+            // CORS jumpscare
+            url = await VencordNative.pluginHelpers.OpenInApp.resolveRedirect(url);
+        }
 
-        const match = SpotifyMatcher.exec(data.href);
-        if (!match) return;
+        spotify: {
+            if (!settings.store.spotify) break spotify;
 
-        const [, type, id] = match;
-        VencordNative.native.openExternal(`spotify:${type}:${id}`);
-        event.preventDefault();
+            const match = SpotifyMatcher.exec(url);
+            if (!match) break spotify;
 
-        return Promise.resolve();
-    },
+            const [, type, id] = match;
+            VencordNative.native.openExternal(`spotify:${type}:${id}`);
+
+            event.preventDefault();
+            return true;
+        }
+
+        steam: {
+            if (!settings.store.steam) break steam;
 
-    handleAccountView(event: MouseEvent, platformType: string, otherUserId: string) {
-        if (platformType !== "spotify") return;
+            if (!SteamMatcher.test(url)) break steam;
 
-        VencordNative.native.openExternal(`spotify:user:${otherUserId}`);
-        event.preventDefault();
+            VencordNative.native.openExternal(`steam://openurl/${url}`);
+
+            event.preventDefault();
+
+            // Steam does not focus itself so show a toast so it's slightly less confusing
+            showToast("Opened link in Steam", Toasts.Type.SUCCESS);
+            return true;
+        }
+
+        // in case short url didn't end up being something we can handle
+        if (event.defaultPrevented) {
+            window.open(url, "_blank");
+            return true;
+        }
+
+        return false;
+    },
+
+    handleAccountView(event: { preventDefault(): void; }, platformType: string, userId: string) {
+        if (platformType === "spotify") {
+            VencordNative.native.openExternal(`spotify:user:${userId}`);
+            event.preventDefault();
+        } else if (platformType === "steam") {
+            VencordNative.native.openExternal(`steam://openurl/https://steamcommunity.com/profiles/${userId}`);
+            showToast("Opened link in Steam", Toasts.Type.SUCCESS);
+            event.preventDefault();
+        }
     }
 });