feat(plugin): PlatformIndicators (#227)

Co-authored-by: Ven <vendicated@riseup.net>

2 files changed, 138 insertions, 0 deletions

diff --git a/src/plugins/platformIndicators.tsx b/src/plugins/platformIndicators.tsx
new file mode 100644
index 0000000..d8f69a6
--- /dev/null
+++ b/src/plugins/platformIndicators.tsx
@@ -0,0 +1,137 @@
+/*
+ * Vencord, a modification for Discord's desktop app
+ * Copyright (c) 2022 Vendicated and contributors
+ *
+ * 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 { User } from "discord-types/general";
+
+import { Settings } from "../api/settings";
+import ErrorBoundary from "../components/ErrorBoundary";
+import { Devs } from "../utils/constants";
+import { lazyWebpack } from "../utils/misc";
+import definePlugin, { OptionType } from "../utils/types";
+import { filters } from "../webpack";
+import { PresenceStore, Tooltip } from "../webpack/common";
+
+function Icon(path: string, viewBox = "0 0 24 24") {
+    return ({ color, tooltip }: { color: string; tooltip: string; }) => (
+        <Tooltip text={tooltip} >
+            {(tooltipProps: any) => (
+                <svg
+                    {...tooltipProps}
+                    height="18"
+                    width="18"
+                    viewBox={viewBox}
+                    fill={color}
+                >
+                    <path d={path} />
+                </svg>
+            )}
+        </Tooltip>
+    );
+}
+
+const Icons = {
+    desktop: Icon("M4 2.5c-1.103 0-2 .897-2 2v11c0 1.104.897 2 2 2h7v2H7v2h10v-2h-4v-2h7c1.103 0 2-.896 2-2v-11c0-1.103-.897-2-2-2H4Zm16 2v9H4v-9h16Z"),
+    web: Icon("M12 2C6.48 2 2 6.48 2 12s4.48 10 10 10 10-4.48 10-10S17.52 2 12 2Zm-1 17.93c-3.95-.49-7-3.85-7-7.93 0-.62.08-1.21.21-1.79L9 15v1c0 1.1.9 2 2 2v1.93Zm6.9-2.54c-.26-.81-1-1.39-1.9-1.39h-1v-3c0-.55-.45-1-1-1H8v-2h2c.55 0 1-.45 1-1V7h2c1.1 0 2-.9 2-2v-.41c2.93 1.19 5 4.06 5 7.41 0 2.08-.8 3.97-2.1 5.39Z"),
+    mobile: Icon("M15.5 1h-8A2.5 2.5 0 0 0 5 3.5v17A2.5 2.5 0 0 0 7.5 23h8a2.5 2.5 0 0 0 2.5-2.5v-17A2.5 2.5 0 0 0 15.5 1zm-4 21c-.83 0-1.5-.67-1.5-1.5s.67-1.5 1.5-1.5 1.5.67 1.5 1.5-.67 1.5-1.5 1.5zm4.5-4H7V4h9v14z"),
+    console: Icon("M14.8 2.7 9 3.1V47h3.3c1.7 0 6.2.3 10 .7l6.7.6V2l-4.2.2c-2.4.1-6.9.3-10 .5zm1.8 6.4c1 1.7-1.3 3.6-2.7 2.2C12.7 10.1 13.5 8 15 8c.5 0 1.2.5 1.6 1.1zM16 33c0 6-.4 10-1 10s-1-4-1-10 .4-10 1-10 1 4 1 10zm15-8v23.3l3.8-.7c2-.3 4.7-.6 6-.6H43V3h-2.2c-1.3 0-4-.3-6-.6L31 1.7V25z", "0 0 50 50"),
+};
+type Platform = keyof typeof Icons;
+
+const getStatusColor = lazyWebpack(filters.byCode("STATUS_YELLOW", "TWITCH", "STATUS_GREY"));
+
+const PlatformIcon = ({ platform, status }: { platform: Platform, status: string; }) => {
+    const tooltip = platform[0].toUpperCase() + platform.slice(1);
+    const Icon = Icons[platform] ?? Icons.desktop;
+
+    return <Icon color={getStatusColor(status)} tooltip={tooltip} />;
+};
+
+const PlatformIndicator = ({ user }: { user: User; }) => {
+    if (!user || user.bot) return null;
+
+    const status = PresenceStore.getState()?.clientStatuses?.[user.id] as Record<Platform, string>;
+    if (!status) return null;
+
+    const icons = Object.entries(status).map(([platform, status]) => (
+        <PlatformIcon
+            key={platform}
+            platform={platform as Platform}
+            status={status}
+        />
+    ));
+
+    return (
+        <div
+            className="vc-platform-indicator"
+            style={{
+                display: "flex", alignItems: "center", marginLeft: "4px"
+            }}
+        >
+            {icons}
+        </div>
+    );
+};
+
+export default definePlugin({
+    name: "PlatformIndicators",
+    description: "Adds platform indicators (Desktop, Mobile, Web...) to users",
+    authors: [Devs.kemo],
+
+    patches: [
+        {
+            // Server member list decorators
+            find: "this.renderPremium()",
+            replacement: {
+                match: /this.renderPremium\(\).+?\]/,
+                replace: "$&.concat(Vencord.Plugins.plugins.PlatformIndicators.renderPlatformIndicators(this.props))"
+            }
+        },
+        {
+            // Dm list decorators
+            find: "PrivateChannel.renderAvatar",
+            replacement: {
+                match: /(subText:(.{1,3})\..+?decorators:)(.+?:null)/,
+                replace: "$1[$3].concat(Vencord.Plugins.plugins.PlatformIndicators.renderPlatformIndicators($2.props))"
+            }
+        },
+        {
+            // User badges
+            find: "Messages.PROFILE_USER_BADGES",
+            predicate: () => Settings.plugins.PlatformIndicators.showAsBadges,
+            replacement: {
+                match: /(Messages\.PROFILE_USER_BADGES,role:"group",children:)(.+?\.key\)\}\)\))/,
+                replace: "$1[Vencord.Plugins.plugins.PlatformIndicators.renderPlatformIndicators(e)].concat($2)"
+            }
+        }
+    ],
+
+    renderPlatformIndicators: ({ user }: { user: User; }) => (
+        <ErrorBoundary noop>
+            <PlatformIndicator user={user} />
+        </ErrorBoundary>
+    ),
+
+    options: {
+        showAsBadges: {
+            description: "Show platform icons in user badges",
+            type: OptionType.BOOLEAN,
+            default: true,
+            restartNeeded: true,
+        }
+    }
+});
diff --git a/src/webpack/common.tsx b/src/webpack/common.tsx
index 5853e56..c649c19 100644
--- a/src/webpack/common.tsx
+++ b/src/webpack/common.tsx
@@ -37,6 +37,7 @@ export const PermissionStore = lazyWebpack(filters.byProps("can", "getGuildPermi
 export const PrivateChannelsStore = lazyWebpack(filters.byProps("openPrivateChannel"));
 export const GuildChannelStore = lazyWebpack(filters.byProps("getChannels"));
 export const ReadStateStore = lazyWebpack(filters.byProps("lastMessageId"));
+export const PresenceStore = lazyWebpack(filters.byProps("setCurrentUserOnConnectionOpen"));
 export let GuildStore: Stores.GuildStore;
 export let UserStore: Stores.UserStore;
 export let SelectedChannelStore: Stores.SelectedChannelStore;