feat: move logic to EntitlementManager

discordjs · May 1, 2024 · bb623d3 · bb623d3
1 parent 0de148d
commit bb623d3
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 2 deletions.
diff --git a/packages/discord.js/src/managers/EntitlementManager.js b/packages/discord.js/src/managers/EntitlementManager.js
@@ -124,6 +124,19 @@ class EntitlementManager extends CachedManager {
 
     await this.client.rest.delete(Routes.entitlement(this.client.application.id, resolved));
   }
+
+  /**
+   * Marks an entitlement as consumed
+   * <info>Only available for One-Time Purchase consumable SKUs.</info>
+   * @param {EntitlementResolvable} entitlement The entitlement to consume
+   * @returns {Promise<void>}
+   */
+  async consume(entitlement) {
+    const resolved = this.resolveId(entitlement);
+    if (!resolved) throw new DiscordjsTypeError(ErrorCodes.InvalidType, 'entitlement', 'EntitlementResolvable');
+
+    await this.client.rest.post(Routes.consumeEntitlement(this.client.application.id, resolved));
+  }
 }
 
 exports.EntitlementManager = EntitlementManager;
diff --git a/packages/discord.js/src/structures/Entitlement.js b/packages/discord.js/src/structures/Entitlement.js
@@ -1,6 +1,5 @@
 'use strict';
 
-const { Routes } = require('discord-api-types/v10');
 const Base = require('./Base');
 
 /**
@@ -177,7 +176,7 @@ class Entitlement extends Base {
    * @returns {Promise<void>}
    */
   async consume() {
-    await this.client.rest.post(Routes.consumeEntitlement(this.applicationId, this.id));
+    await this.client.application.entitlements.consume(this);
   }
 }
 

diff --git a/packages/discord.js/typings/index.d.ts b/packages/discord.js/typings/index.d.ts
@@ -4143,6 +4143,7 @@ export class EntitlementManager extends CachedManager<Snowflake, Entitlement, En
   public fetch(options?: FetchEntitlementsOptions): Promise<Collection<Snowflake, Entitlement>>;
   public createTest(options: GuildEntitlementCreateOptions | UserEntitlementCreateOptions): Promise<Entitlement>;
   public deleteTest(entitlement: EntitlementResolvable): Promise<void>;
+  public consume(entitlement: EntitlementResolvable): Promise<void>;
 }
 
 export interface FetchGuildApplicationCommandFetchOptions extends Omit<FetchApplicationCommandOptions, 'guildId'> {}

diff --git a/packages/discord.js/typings/index.test-d.ts b/packages/discord.js/typings/index.test-d.ts
@@ -2494,6 +2494,8 @@ declare const sku: SKU;
 
   await application.entitlements.deleteTest(entitlement);
 
+  await application.entitlements.consume(entitlement);
+
   expectType<boolean>(entitlement.isActive());
 
   if (entitlement.isUserSubscription()) {