feat(vanity): Only request vanity url when needed

SideProjectGuys · Mar 31, 2020 · b5a7dff · b5a7dff
1 parent 059d373
commit b5a7dff
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/src/invites/cache/VanityUrlCache.ts b/src/invites/cache/VanityUrlCache.ts
@@ -1,15 +1,21 @@
 import { Cache } from '../../framework/cache/Cache';
+import { GuildFeature } from '../../types';
 
 export class VanityUrlCache extends Cache<string> {
 	public async init() {
 		// NO-OP
 	}
 
 	protected async _get(guildId: string): Promise<string> {
+		const guild = this.client.guilds.get(guildId);
+		if (!guild || !guild.features.includes(GuildFeature.VANITY_URL)) {
+			return null;
+		}
+
 		return (
-			this.client.guilds.get(guildId)?.vanityURL ||
-			this.client
-				.getGuildVanity(guildId)
+			guild.vanityURL ||
+			guild
+				.getVanity()
 				.then((r) => r.code as any)
 				.catch(() => null)
 		);

diff --git a/src/types.ts b/src/types.ts
@@ -34,6 +34,22 @@ export enum GuildPermission {
 	READ_MESSAGE_HISTORY = 'readMessageHistory'
 }
 
+export enum GuildFeature {
+	INVITE_SPLASH = 'INVITE_SPLASH',
+	VIP_REGIONS = 'VIP_REGIONS',
+	VANITY_URL = 'VANITY_URL',
+	VERIFIED = 'VERIFIED',
+	PARTNERED = 'PARTNERED',
+	PUBLIC = 'PUBLIC',
+	COMMERCE = 'COMMERCE',
+	NEWS = 'NEWS',
+	DISCOVERABLE = 'DISCOVERABLE',
+	FEATURABLE = 'FEATURABLE',
+	ANIMATED_ICON = 'ANIMATED_ICON',
+	BANNER = 'BANNER',
+	PUBLIC_DISABLED = 'PUBLIC_DISABLED'
+}
+
 export enum PromptResult {
 	SUCCESS,
 	FAILURE,