NewGuildSettings: Add "Apply NewGuildSettings" button in menu (#2556)

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

src/plugins/newGuildSettings/index.tsx

@@ -16,10 +16,17 @@
  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
+import {
+    findGroupChildrenByChildId,
+    NavContextMenuPatchCallback
+} from "@api/ContextMenu";
 import { definePluginSettings, migratePluginSettings } from "@api/Settings";
+import { CogWheel } from "@components/Icons";
 import { Devs } from "@utils/constants";
 import definePlugin, { OptionType } from "@utils/types";
 import { findByCodeLazy, findByPropsLazy, mapMangledModuleLazy } from "@webpack";
+import { Menu } from "@webpack/common";
+import { Guild } from "discord-types/general";
 
 const { updateGuildNotificationSettings } = findByPropsLazy("updateGuildNotificationSettings");
 const { toggleShowAllChannels } = mapMangledModuleLazy(".onboardExistingMember(", {
@@ -73,31 +80,21 @@ const settings = definePluginSettings({
     }
 });
 
-migratePluginSettings("NewGuildSettings", "MuteNewGuild");
+const makeContextMenuPatch: (shouldAddIcon: boolean) => NavContextMenuPatchCallback = (shouldAddIcon: boolean) => (children, { guild }: { guild: Guild, onClose(): void; }) => {
-export default definePlugin({
+    if (!guild) return;
-    name: "NewGuildSettings",
-    description: "Automatically mute new servers and change various other settings upon joining",
-    tags: ["MuteNewGuild", "mute", "server"],
-    authors: [Devs.Glitch, Devs.Nuckyz, Devs.carince, Devs.Mopi, Devs.GabiRP],
-    patches: [
-        {
-            find: ",acceptInvite(",
-            replacement: {
-                match: /INVITE_ACCEPT_SUCCESS.+?,(\i)=null!==.+?;/,
-                replace: (m, guildId) => `${m}$self.handleMute(${guildId});`
-            }
-        },
-        {
-            find: "{joinGuild:",
-            replacement: {
-                match: /guildId:(\i),lurker:(\i).{0,20}}\)\);/,
-                replace: (m, guildId, lurker) => `${m}if(!${lurker})$self.handleMute(${guildId});`
-            }
-        }
-    ],
-    settings,
 
-    handleMute(guildId: string | null) {
+    const group = findGroupChildrenByChildId("privacy", children);
+    group?.push(
+        <Menu.MenuItem
+            label="Apply NewGuildSettings"
+            id="vc-newguildsettings-apply"
+            icon={shouldAddIcon ? CogWheel : void 0}
+            action={() => applyDefaultSettings(guild.id)}
+        />
+    );
+};
+
+function applyDefaultSettings(guildId: string | null) {
     if (guildId === "@me" || guildId === "null" || guildId == null) return;
     updateGuildNotificationSettings(guildId,
         {
@@ -116,5 +113,35 @@ export default definePlugin({
     if (settings.store.showAllChannels && isOptInEnabledForGuild(guildId)) {
         toggleShowAllChannels(guildId);
     }
+}
+
+
+migratePluginSettings("NewGuildSettings", "MuteNewGuild");
+export default definePlugin({
+    name: "NewGuildSettings",
+    description: "Automatically mute new servers and change various other settings upon joining",
+    tags: ["MuteNewGuild", "mute", "server"],
+    authors: [Devs.Glitch, Devs.Nuckyz, Devs.carince, Devs.Mopi, Devs.GabiRP],
+    contextMenus: {
+        "guild-context": makeContextMenuPatch(false),
+        "guild-header-popout": makeContextMenuPatch(true)
+    },
+    patches: [
+        {
+            find: ",acceptInvite(",
+            replacement: {
+                match: /INVITE_ACCEPT_SUCCESS.+?,(\i)=null!==.+?;/,
+                replace: (m, guildId) => `${m}$self.applyDefaultSettings(${guildId});`
             }
+        },
+        {
+            find: "{joinGuild:",
+            replacement: {
+                match: /guildId:(\i),lurker:(\i).{0,20}}\)\);/,
+                replace: (m, guildId, lurker) => `${m}if(!${lurker})$self.applyDefaultSettings(${guildId});`
+            }
+        }
+    ],
+    settings,
+    applyDefaultSettings
 });