From 91f1d68e29d61820b778d59f33645e59abb24247 Mon Sep 17 00:00:00 2001 From: Nuckyz <61953774+Nuckyz@users.noreply.github.com> Date: Tue, 22 Nov 2022 22:51:45 -0300 Subject: [PATCH] feat(plugins): Keep Current Channel plugin (#248) --- src/plugins/keepCurrentChannel.ts | 99 +++++++++++++++++++++++++++++++ src/webpack/common.tsx | 9 +++ 2 files changed, 108 insertions(+) create mode 100644 src/plugins/keepCurrentChannel.ts diff --git a/src/plugins/keepCurrentChannel.ts b/src/plugins/keepCurrentChannel.ts new file mode 100644 index 00000000..395fd768 --- /dev/null +++ b/src/plugins/keepCurrentChannel.ts @@ -0,0 +1,99 @@ +/* + * 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 . +*/ + +import { DataStore } from "../api"; +import { Devs } from "../utils/constants"; +import definePlugin from "../utils/types"; +import { ChannelStore, FluxDispatcher, NavigationRouter, SelectedChannelStore, SelectedGuildStore } from "../webpack/common"; + +export interface LogoutEvent { + type: "LOGOUT"; + isSwitchingAccount: boolean; +} + +interface ChannelSelectEvent { + type: "CHANNEL_SELECT"; + channelId: string | null; + guildId: string | null; +} + +interface PreviousChannel { + guildId: string | null; + channelId: string | null; +} + + +export default definePlugin({ + name: "KeepCurrentChannel", + description: "Attempt to navigate the channel you were in before switching accounts or loading Discord.", + authors: [Devs.Nuckyz], + + isSwitchingAccount: false, + previousCache: {} as PreviousChannel, + + attemptToNavigateToChannel(guildId: string | null, channelId: string) { + if (!ChannelStore.hasChannel(channelId)) return; + NavigationRouter.transitionTo(`/channels/${guildId ?? "@me"}/${channelId}`); + }, + + onLogout(e: LogoutEvent) { + this.isSwitchingAccount = e.isSwitchingAccount; + }, + + onConnectionOpen() { + if (!this.isSwitchingAccount) return; + this.isSwitchingAccount = false; + + if (this.previousCache.channelId) this.attemptToNavigateToChannel(this.previousCache.guildId, this.previousCache.channelId); + }, + + async onChannelSelect({ guildId, channelId }: ChannelSelectEvent) { + if (this.isSwitchingAccount) return; + + this.previousCache = { + guildId, + channelId + }; + await DataStore.set("KeepCurrentChannel_previousData", this.previousCache); + }, + + async start() { + const previousData = await DataStore.get("KeepCurrentChannel_previousData"); + if (previousData) { + this.previousCache = previousData; + + if (this.previousCache.channelId) this.attemptToNavigateToChannel(this.previousCache.guildId, this.previousCache.channelId); + } else { + this.previousCache = { + guildId: SelectedGuildStore.getGuildId(), + channelId: SelectedChannelStore.getChannelId() ?? null + }; + await DataStore.set("KeepCurrentChannel_previousData", this.previousCache); + } + + FluxDispatcher.subscribe("LOGOUT", this.onLogout.bind(this)); + FluxDispatcher.subscribe("CONNECTION_OPEN", this.onConnectionOpen.bind(this)); + FluxDispatcher.subscribe("CHANNEL_SELECT", this.onChannelSelect.bind(this)); + }, + + stop() { + FluxDispatcher.unsubscribe("LOGOUT", this.onLogout); + FluxDispatcher.unsubscribe("CONNECTION_OPEN", this.onConnectionOpen); + FluxDispatcher.unsubscribe("CHANNEL_SELECT", this.onChannelSelect); + } +}); diff --git a/src/webpack/common.tsx b/src/webpack/common.tsx index c649c19e..fa6bb0e8 100644 --- a/src/webpack/common.tsx +++ b/src/webpack/common.tsx @@ -41,6 +41,7 @@ export const PresenceStore = lazyWebpack(filters.byProps("setCurrentUserOnConnec export let GuildStore: Stores.GuildStore; export let UserStore: Stores.UserStore; export let SelectedChannelStore: Stores.SelectedChannelStore; +export let SelectedGuildStore: any; export let ChannelStore: Stores.ChannelStore; export const Forms = {} as { @@ -125,6 +126,13 @@ export const Clipboard = mapMangledModuleLazy('document.queryCommandEnabled("cop SUPPORTS_COPY: x => typeof x === "boolean", }); +export const NavigationRouter = mapMangledModuleLazy("Transitioning to external path", { + transitionTo: filters.byCode("Transitioning to external path"), + transitionToGuild: filters.byCode("transitionToGuild"), + goBack: filters.byCode("goBack()"), + goForward: filters.byCode("goForward()"), +}); + waitFor("useState", m => React = m); waitFor(["dispatch", "subscribe"], m => { @@ -139,6 +147,7 @@ waitFor(["dispatch", "subscribe"], m => { waitFor(["getCurrentUser", "initialize"], m => UserStore = m); waitFor("getSortedPrivateChannels", m => ChannelStore = m); waitFor("getCurrentlySelectedChannelId", m => SelectedChannelStore = m); +waitFor("getLastSelectedGuildId", m => SelectedGuildStore = m); waitFor("getGuildCount", m => GuildStore = m); waitFor(["Hovers", "Looks", "Sizes"], m => Button = m);