From 8b564a4b49c6a047b6aab6b0740403275e28bf68 Mon Sep 17 00:00:00 2001
From: Helloyunho <yunho050840@gmail.com>
Date: Thu, 21 Jan 2021 14:37:10 +0900
Subject: [PATCH] Add guild prune function

---
 src/structures/guild.ts | 56 +++++++++++++++++++++++++++++++++--------
 src/types/guild.ts      |  5 ++--
 2 files changed, 49 insertions(+), 12 deletions(-)

diff --git a/src/structures/guild.ts b/src/structures/guild.ts
index a1ee116..59393ad 100644
--- a/src/structures/guild.ts
+++ b/src/structures/guild.ts
@@ -12,9 +12,9 @@ import {
   MessageNotification,
   ContentFilter,
   GuildModifyOptions,
-  GuildGetPruneCountOptions,
   GuildGetPruneCountPayload,
-  GuildPruneCountPayload
+  GuildPruneCountPayload,
+  GuildBeginPrunePayload
 } from '../types/guild.ts'
 import { Base } from './base.ts'
 import { CreateGuildRoleOptions, RolesManager } from '../managers/roles.ts'
@@ -363,15 +363,15 @@ export class Guild extends Base {
     return result === undefined ? this : result
   }
 
-  async getPruneCount(options: GuildGetPruneCountOptions): Promise<number> {
+  async getPruneCount(
+    days?: number,
+    includeRoles?: Array<Role | string>
+  ): Promise<number> {
     const query: GuildGetPruneCountPayload = {
-      days: options.days,
-      include_roles:
-        options.includeRoles !== undefined
-          ? options.includeRoles
-              .map((role) => (role instanceof Role ? role.id : role))
-              .join(',')
-          : undefined
+      days: days,
+      include_roles: includeRoles
+        ?.map((role) => (role instanceof Role ? role.id : role))
+        .join(',')
     }
 
     const result: GuildPruneCountPayload = await this.client.rest.get(
@@ -385,6 +385,42 @@ export class Guild extends Base {
 
     return result.pruned as number
   }
+
+  async prune(
+    days?: number,
+    computePruneCount?: false,
+    includeRoles?: Array<Role | string>
+  ): Promise<null>
+  async prune(
+    days?: number,
+    computePruneCount?: true,
+    includeRoles?: Array<Role | string>
+  ): Promise<number>
+  async prune(
+    days?: number,
+    computePruneCount?: undefined,
+    includeRoles?: Array<Role | string>
+  ): Promise<number>
+  async prune(
+    days?: number,
+    computePruneCount?: boolean,
+    includeRoles?: Array<Role | string>
+  ): Promise<number | null> {
+    const body: GuildBeginPrunePayload = {
+      days: days,
+      compute_prune_count: computePruneCount,
+      include_roles: includeRoles?.map((role) =>
+        role instanceof Role ? role.id : role
+      )
+    }
+
+    const result: GuildPruneCountPayload = await this.client.rest.post(
+      GUILD_PRUNE(this.id),
+      body
+    )
+
+    return result.pruned
+  }
 }
 
 export class GuildIntegration extends Base {
diff --git a/src/types/guild.ts b/src/types/guild.ts
index 938b6bd..6c4c882 100644
--- a/src/types/guild.ts
+++ b/src/types/guild.ts
@@ -290,7 +290,8 @@ export interface GuildGetPruneCountPayload {
   include_roles?: string
 }
 
-export interface GuildGetPruneCountOptions {
+export interface GuildBeginPrunePayload {
   days?: number
-  includeRoles?: Array<Role | string>
+  compute_prune_count?: boolean
+  include_roles?: string[]
 }