Add support for consuming one-time entitlements

OceanicJS · Jun 20, 2024 · f66367a · f66367a
1 parent 05899ff
commit f66367a
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 0 deletions.
diff --git a/lib/routes/Applications.ts b/lib/routes/Applications.ts
@@ -91,6 +91,18 @@ export default class Applications {
         }).then(data => data.map(d => new ApplicationCommand(d, this._manager.client)));
     }
 
+    /**
+     * Mark an entitlement as consumed.
+     * @param applicationID The ID of the application to the entitlement is for.
+     * @param entitlementID The ID of the entitlement to consume.
+     */
+    async consumeEntitlement(applicationID: string, entitlementID: string): Promise<void> {
+        await this._manager.authRequest<void>({
+            method: "POST",
+            path:   Routes.CONSUME_ENTITLEMENT(applicationID, entitlementID)
+        });
+    }
+
     /**
      * Create a global application command.
      * @param applicationID The ID of the application.

diff --git a/lib/structures/BaseEntitlement.ts b/lib/structures/BaseEntitlement.ts
@@ -28,6 +28,11 @@ export default class BaseEntitlement extends Base {
         this.userID = data.user_id;
     }
 
+    /** Mark this entitlement as consumed. */
+    async consume(): Promise<void> {
+        return this.client.rest.applications.consumeEntitlement(this.applicationID, this.id);
+    }
+
     override toJSON(): JSONBaseEntitlement {
         return {
             ...super.toJSON(),

diff --git a/lib/structures/ClientApplication.ts b/lib/structures/ClientApplication.ts
@@ -73,6 +73,14 @@ export default class ClientApplication extends Base {
         return this.client.rest.applications.bulkEditGuildCommands(this.id, guildID, options);
     }
 
+    /**
+     * Mark an entitlement as consumed.
+     * @param entitlementID The ID of the entitlement to consume.
+     */
+    async consumeEntitlement(entitlementID: string): Promise<void> {
+        return this.client.rest.applications.consumeEntitlement(this.id, entitlementID);
+    }
+
     /**
      * Create a global application command.
      * @param options The options for creating the command.

diff --git a/lib/util/Routes.ts b/lib/util/Routes.ts
@@ -147,6 +147,7 @@ export const INTERACTION_CALLBACK                  = (interactionID: string, int
 export const APPLICATION                           = "/applications/@me";
 export const ENTITLEMENTS                          = (applicationID: string) => encode`/applications/${applicationID}/entitlements`;
 export const ENTITLEMENT                           = (applicationID: string, entitlementID: string) => encode`/applications/${applicationID}/entitlements/${entitlementID}`;
+export const CONSUME_ENTITLEMENT                   = (applicationID: string, entitlementID: string) => encode`/applications/${applicationID}/entitlements/${entitlementID}/consume`;
 export const SKUS                                  = (applicationID: string) => encode`/applications/${applicationID}/skus`;
 
 // Misc