From 1ea2a81363d7b76c1ef51487c42e3aaf2e2088da Mon Sep 17 00:00:00 2001
From: LTS20050703 <lts20050703@gmail.com>
Date: Thu, 25 Aug 2022 22:43:33 +0700
Subject: [PATCH] Fix cache plugin doesn't update voice state update (#2388)

* Fix cache plugin doesn't update voice state update

* fix putting fix in wrong file

* deno fmt

* delete voice state

* deno fmt

Co-authored-by: Skillz4Killz <23035000+Skillz4Killz@users.noreply.github.com>
---
 plugins/cache/src/setupCacheEdits.ts    | 16 ++++++++++++++++
 plugins/cache/src/setupCacheRemovals.ts | 20 ++++++++++++++++++--
 2 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/plugins/cache/src/setupCacheEdits.ts b/plugins/cache/src/setupCacheEdits.ts
index 1e4e34a00..d0c370a64 100644
--- a/plugins/cache/src/setupCacheEdits.ts
+++ b/plugins/cache/src/setupCacheEdits.ts
@@ -5,6 +5,7 @@ import type {
   DiscordMessageReactionAdd,
   DiscordMessageReactionRemove,
   DiscordMessageReactionRemoveAll,
+  DiscordVoiceState,
 } from "../deps.ts";
 import type { BotWithCache } from "./addCacheCollections.ts";
 
@@ -15,6 +16,7 @@ export function setupCacheEdits<B extends Bot>(bot: BotWithCache<B>) {
     MESSAGE_REACTION_ADD,
     MESSAGE_REACTION_REMOVE,
     MESSAGE_REACTION_REMOVE_ALL,
+    VOICE_STATE_UPDATE,
   } = bot.handlers;
 
   bot.handlers.GUILD_MEMBER_ADD = function (_, data, shardId) {
@@ -116,4 +118,18 @@ export function setupCacheEdits<B extends Bot>(bot: BotWithCache<B>) {
 
     MESSAGE_REACTION_REMOVE_ALL(bot, data, shardId);
   };
+
+  bot.handlers.VOICE_STATE_UPDATE = (_, data, shardId) => {
+    const payload = data.d as DiscordVoiceState;
+    if (!payload.guild_id) return;
+
+    const vs = bot.transformers.voiceState(bot, {
+      voiceState: payload,
+      guildId: bot.transformers.snowflake(payload.guild_id),
+    });
+
+    bot.guilds.get(vs.guildId)?.voiceStates.set(vs.userId, vs);
+
+    VOICE_STATE_UPDATE(bot, data, shardId);
+  };
 }
diff --git a/plugins/cache/src/setupCacheRemovals.ts b/plugins/cache/src/setupCacheRemovals.ts
index 9ce971749..d8d826722 100644
--- a/plugins/cache/src/setupCacheRemovals.ts
+++ b/plugins/cache/src/setupCacheRemovals.ts
@@ -1,5 +1,4 @@
-import type { Emoji, Guild } from "../deps.ts";
-import type { BotWithCache } from "./addCacheCollections.ts";
+import type { Emoji } from "../deps.ts";
 import {
   Bot,
   Collection,
@@ -11,7 +10,9 @@ import {
   DiscordMessageDelete,
   DiscordMessageDeleteBulk,
   DiscordUnavailableGuild,
+  DiscordVoiceState,
 } from "../deps.ts";
+import type { BotWithCache } from "./addCacheCollections.ts";
 
 export function setupCacheRemovals<B extends Bot>(bot: BotWithCache<B>) {
   const {
@@ -22,6 +23,7 @@ export function setupCacheRemovals<B extends Bot>(bot: BotWithCache<B>) {
     GUILD_MEMBER_REMOVE,
     GUILD_ROLE_DELETE,
     MESSAGE_DELETE_BULK,
+    VOICE_STATE_UPDATE,
   } = bot.handlers;
 
   bot.handlers.GUILD_DELETE = function (_, data, shardId) {
@@ -118,4 +120,18 @@ export function setupCacheRemovals<B extends Bot>(bot: BotWithCache<B>) {
 
     GUILD_ROLE_DELETE(bot, data, shardId);
   };
+
+  bot.handlers.VOICE_STATE_UPDATE = (_, data, shardId) => {
+    const payload = data.d as DiscordVoiceState;
+    if (!payload.guild_id) return;
+
+    const vs = bot.transformers.voiceState(bot, {
+      voiceState: payload,
+      guildId: bot.transformers.snowflake(payload.guild_id),
+    });
+
+    if (!vs.channelId) bot.guilds.get(vs.guildId)?.voiceStates.delete(vs.userId);
+
+    VOICE_STATE_UPDATE(bot, data, shardId);
+  };
 }