From c880a737712d80b06f83d8914fffc1184b8e70e0 Mon Sep 17 00:00:00 2001
From: Helloyunho <yunho050840@gmail.com>
Date: Mon, 25 Jan 2021 03:49:08 +0900
Subject: [PATCH] Do some tests and fix (or add) some

---
 mod.ts                    |  10 ++-
 src/managers/guilds.ts    |  64 +++++++++---------
 src/managers/roles.ts     |   6 ++
 src/models/rest.ts        |  21 +++++-
 src/models/slashClient.ts |   4 +-
 src/structures/guild.ts   |  24 +++++--
 src/structures/role.ts    |  34 ++++++++++
 src/structures/user.ts    |   5 ++
 src/test/guild.ts         | 137 ++++++++++++++++++++++++++++++++++++++
 src/test/template.ts      |  21 ++++++
 src/test/user.ts          |  14 ++++
 src/types/guild.ts        |   2 +-
 src/utils/snowflake.ts    |   4 ++
 13 files changed, 300 insertions(+), 46 deletions(-)
 create mode 100644 src/test/guild.ts
 create mode 100644 src/test/template.ts
 create mode 100644 src/test/user.ts

diff --git a/mod.ts b/mod.ts
index b279793..52305b0 100644
--- a/mod.ts
+++ b/mod.ts
@@ -102,12 +102,16 @@ export type {
   MessageOptions
 } from './src/types/channel.ts'
 export type { EmojiPayload } from './src/types/emoji.ts'
+export { Verification } from './src/types/guild.ts'
 export type {
-  GuildBanPayload,
-  GuildFeatures,
   GuildIntegrationPayload,
   GuildPayload,
-  GuildChannels
+  GuildBanPayload,
+  GuildFeatures,
+  GuildChannels,
+  GuildCreateOptions,
+  GuildCreateChannelOptions,
+  GuildCreateRolePayload
 } from './src/types/guild.ts'
 export type { InvitePayload, PartialInvitePayload } from './src/types/invite.ts'
 export { PermissionFlags } from './src/types/permissionFlags.ts'
diff --git a/src/managers/guilds.ts b/src/managers/guilds.ts
index 7371267..97d4aad 100644
--- a/src/managers/guilds.ts
+++ b/src/managers/guilds.ts
@@ -70,45 +70,45 @@ export class GuildManager extends BaseManager<GuildPayload, Guild> {
     if (options.icon !== undefined && !options.icon.startsWith('data:')) {
       options.icon = await fetchAuto(options.icon)
     }
+    if (options.roles !== undefined && options.roles[0].name !== '@everyone') {
+      options.roles.unshift({
+        id: Math.floor(Math.random() * 18392375458).toString(),
+        name: '@everyone'
+      })
+    }
 
     const body: GuildCreatePayload = {
       name: options.name,
       region: options.region,
       icon: options.icon,
       verification_level: options.verificationLevel,
-      roles:
-        options.roles !== undefined
-          ? options.roles.map((obj) => {
-              let result: GuildCreateRolePayload
-              if (obj instanceof Role) {
-                result = {
-                  id: obj.id,
-                  name: obj.name,
-                  color: obj.color,
-                  hoist: obj.hoist,
-                  position: obj.position,
-                  permissions: obj.permissions.bitfield.toString(),
-                  managed: obj.managed,
-                  mentionable: obj.mentionable
-                }
-              } else {
-                result = obj
-              }
+      roles: options.roles?.map((obj) => {
+        let result: GuildCreateRolePayload
+        if (obj instanceof Role) {
+          result = {
+            id: obj.id,
+            name: obj.name,
+            color: obj.color,
+            hoist: obj.hoist,
+            position: obj.position,
+            permissions: obj.permissions.bitfield.toString(),
+            managed: obj.managed,
+            mentionable: obj.mentionable
+          }
+        } else {
+          result = obj
+        }
 
-              return result
-            })
-          : undefined,
-      channels:
-        options.channels !== undefined
-          ? options.channels.map(
-              (obj): GuildCreateChannelPayload => ({
-                id: obj.id,
-                name: obj.name,
-                type: obj.type,
-                parent_id: obj.parentID
-              })
-            )
-          : undefined,
+        return result
+      }),
+      channels: options.channels?.map(
+        (obj): GuildCreateChannelPayload => ({
+          id: obj.id,
+          name: obj.name,
+          type: obj.type,
+          parent_id: obj.parentID
+        })
+      ),
       afk_channel_id: options.afkChannelID,
       afk_timeout: options.afkTimeout,
       system_channel_id: options.systemChannelID
diff --git a/src/managers/roles.ts b/src/managers/roles.ts
index c1b371e..3133ae0 100644
--- a/src/managers/roles.ts
+++ b/src/managers/roles.ts
@@ -41,6 +41,12 @@ export class RolesManager extends BaseManager<RolePayload, Role> {
     return new Role(this.client, raw, this.guild)
   }
 
+  async array(): Promise<Role[]> {
+    let arr = await (this.client.cache.array(this.cacheName) as RolePayload[])
+    if (arr === undefined) arr = []
+    return arr.map((e) => new Role(this.client, e, this.guild))
+  }
+
   async fromPayload(roles: RolePayload[]): Promise<boolean> {
     for (const role of roles) {
       await this.set(role.id, role)
diff --git a/src/models/rest.ts b/src/models/rest.ts
index 7d38968..9d00f62 100644
--- a/src/models/rest.ts
+++ b/src/models/rest.ts
@@ -28,8 +28,23 @@ export interface RequestHeaders {
   [name: string]: string
 }
 
+export interface DiscordAPIErrorPayload {
+  url: string
+  status: number
+  method: string
+  code?: number
+  message?: string
+  errors: object
+}
+
 export class DiscordAPIError extends Error {
   name = 'DiscordAPIError'
+  error?: DiscordAPIErrorPayload
+
+  constructor(message?: string, error?: DiscordAPIErrorPayload) {
+    super(message)
+    this.error = error
+  }
 }
 
 export interface QueuedItem {
@@ -399,7 +414,7 @@ export class RESTManager {
       )
 
     // At this point we know it is error
-    const error: { [name: string]: any } = {
+    const error: DiscordAPIErrorPayload = {
       url: response.url,
       status,
       method: data.method,
@@ -443,9 +458,9 @@ export class RESTManager {
         HttpResponseCode.MethodNotAllowed
       ].includes(status)
     ) {
-      reject(new DiscordAPIError(Deno.inspect(error)))
+      reject(new DiscordAPIError(Deno.inspect(error), error))
     } else if (status === HttpResponseCode.GatewayUnavailable) {
-      reject(new DiscordAPIError(Deno.inspect(error)))
+      reject(new DiscordAPIError(Deno.inspect(error), error))
     } else reject(new DiscordAPIError('Request - Unknown Error'))
   }
 
diff --git a/src/models/slashClient.ts b/src/models/slashClient.ts
index 9520025..273d4c4 100644
--- a/src/models/slashClient.ts
+++ b/src/models/slashClient.ts
@@ -381,9 +381,7 @@ export class SlashClient {
     this.commands = new SlashCommandsManager(this)
     this.publicKey = options.publicKey
 
-    if (options !== undefined) {
-      this.enabled = options.enabled ?? true
-    }
+    this.enabled = options.enabled ?? true
 
     if (this.client?._decoratedSlash !== undefined) {
       this.client._decoratedSlash.forEach((e) => {
diff --git a/src/structures/guild.ts b/src/structures/guild.ts
index f438b4d..f7234d5 100644
--- a/src/structures/guild.ts
+++ b/src/structures/guild.ts
@@ -41,6 +41,7 @@ import { RequestMembersOptions } from '../gateway/index.ts'
 import { GuildPresencesManager } from '../managers/presences.ts'
 import { TemplatePayload } from '../types/template.ts'
 import { Template } from './template.ts'
+import { DiscordAPIError } from '../models/rest.ts'
 
 export class GuildBan extends Base {
   guild: Guild
@@ -398,7 +399,22 @@ export class Guild extends Base {
 
   /** Returns a partial invite object for guilds with that feature enabled. */
   async getVanity(): Promise<{ code: string | null; uses: number }> {
-    return this.client.rest.api.guilds[this.id]['vanity-url'].get()
+    try {
+      const value = await this.client.rest.api.guilds[this.id][
+        'vanity-url'
+      ].get()
+      return value
+    } catch (error) {
+      if (error instanceof DiscordAPIError) {
+        if (error.error?.code === 50020) {
+          return {
+            code: null,
+            uses: 0
+          }
+        }
+      }
+      throw error
+    }
   }
 
   /** Returns a PNG (URL) image widget for the guild. */
@@ -441,7 +457,7 @@ export class Guild extends Base {
   async syncTemplate(code: string): Promise<Template> {
     const payload = await this.client.rest.api.guilds[this.id].templates[
       code
-    ].sync.put()
+    ].put()
     return new Template(this.client, payload)
   }
 
@@ -510,7 +526,7 @@ export class Guild extends Base {
 
   async prune(options?: {
     days?: number
-    computePruneCount: true | undefined
+    computePruneCount?: true
     includeRoles?: Array<Role | string>
   }): Promise<number>
   async prune(options?: {
@@ -520,7 +536,7 @@ export class Guild extends Base {
   }): Promise<null>
   async prune(options?: {
     days?: number
-    computePruneCount?: boolean | undefined
+    computePruneCount?: boolean
     includeRoles?: Array<Role | string>
   }): Promise<number | null> {
     const body: GuildBeginPrunePayload = {
diff --git a/src/structures/role.ts b/src/structures/role.ts
index 91050aa..6f435c5 100644
--- a/src/structures/role.ts
+++ b/src/structures/role.ts
@@ -3,6 +3,8 @@ import { Base } from './base.ts'
 import { RoleModifyPayload, RolePayload } from '../types/role.ts'
 import { Permissions } from '../utils/permissions.ts'
 import { Guild } from './guild.ts'
+import { Member } from './member.ts'
+import { User } from './user.ts'
 
 export class Role extends Base {
   id: string
@@ -50,6 +52,38 @@ export class Role extends Base {
     this.mentionable = data.mentionable ?? this.mentionable
   }
 
+  async addTo(member: Member | User | string): Promise<boolean> {
+    if (member instanceof User) {
+      member = member.id
+    }
+    if (typeof member === 'string') {
+      const tempMember = await this.guild.members.get(member)
+      if (tempMember === undefined) {
+        throw new Error(`Couldn't find the member ${member}.`)
+      } else {
+        member = tempMember
+      }
+    }
+
+    return member.roles.add(this.id)
+  }
+
+  async removeFrom(member: Member | User | string): Promise<boolean> {
+    if (member instanceof User) {
+      member = member.id
+    }
+    if (typeof member === 'string') {
+      const tempMember = await this.guild.members.get(member)
+      if (tempMember === undefined) {
+        throw new Error(`Couldn't find the member ${member}.`)
+      } else {
+        member = tempMember
+      }
+    }
+
+    return member.roles.remove(this.id)
+  }
+
   async delete(): Promise<Role | undefined> {
     return this.guild.roles.delete(this)
   }
diff --git a/src/structures/user.ts b/src/structures/user.ts
index a44b6e5..18a4330 100644
--- a/src/structures/user.ts
+++ b/src/structures/user.ts
@@ -5,6 +5,7 @@ import { Base } from './base.ts'
 import { ImageURL } from './cdn.ts'
 import { ImageSize, ImageFormats } from '../types/cdn.ts'
 import { DEFAULT_USER_AVATAR, USER_AVATAR } from '../types/endpoint.ts'
+import { DMChannel } from './dmChannel.ts'
 
 export class User extends Base {
   id: string
@@ -89,4 +90,8 @@ export class User extends Base {
   toString(): string {
     return this.mention
   }
+
+  async getDMchannel(): Promise<DMChannel> {
+    return this.client.createDM(this)
+  }
 }
diff --git a/src/test/guild.ts b/src/test/guild.ts
new file mode 100644
index 0000000..8c39f9b
--- /dev/null
+++ b/src/test/guild.ts
@@ -0,0 +1,137 @@
+import {
+  Client,
+  Intents
+  // Verification
+  // PermissionFlags,
+  // ChannelTypes,
+  // GuildCreateOptions
+} from '../../mod.ts'
+import { TOKEN } from './config.ts'
+
+const client = new Client()
+
+// client.on('guildLoaded', async (guild) => {
+//   if (guild.name === 'OH WOW') {
+//     guild.delete()
+//   }
+// })
+
+// client.on('ready', async () => {
+//   await new Promise((resolve, reject) => setTimeout(resolve, 1000))
+//   const body: GuildCreateOptions = {
+//     name: 'OH WOW',
+//     icon: 'https://helloyunho.xyz/_dist_/images/avatar.png',
+//     verificationLevel: Verification.NONE,
+//     roles: [
+//       {
+//         id: '1',
+//         name: 'a role',
+//         color: 0x103021,
+//         hoist: false,
+//         permissions: PermissionFlags.ADMINISTRATOR.toString(),
+//         mentionable: true
+//       }
+//     ],
+//     channels: [
+//       {
+//         name: 'fucking-awesome',
+//         type: ChannelTypes.GUILD_TEXT,
+//         id: '1'
+//       }
+//     ],
+//     systemChannelID: '1'
+//   }
+//   const guild = await client.guilds.create(body)
+
+//   const channels = await guild.channels.array()
+//   console.log(channels.length)
+//   const invite = await guild.invites.create(channels[0].id)
+//   console.log(invite.link)
+// })
+
+// client.on('guildLoaded', async (guild) => {
+//   if (guild.id === 'GUILD_ID') {
+//     // const roles = await guild.roles.array()
+//     // if (roles.length > 0) {
+//     //   roles.forEach(async (role) => {
+//     //     if (role.name !== '@everyone') {
+//     //       role.addTo('USER_ID')
+//     //     }
+//     //   })
+//     // }
+
+//     // guild.edit({
+//     //   name: 'OH BOI',
+//     //   verificationLevel: Verification.MEDIUM
+//     // })
+
+//     // const role1 = await guild.roles.create({
+//     //   name: 'IDK1'
+//     // })
+//     // const role2 = await guild.roles.create({
+//     //   name: 'IDK2'
+//     // })
+
+//     // alert()
+
+//     // await guild.roles.editPositions(
+//     //   {
+//     //     id: role1.id,
+//     //     position: 1
+//     //   },
+//     //   {
+//     //     id: role2.id,
+//     //     position: 2
+//     //   }
+//     // )
+
+//     // alert()
+
+//     // role1.delete()
+//     // role2.delete()
+
+//     // const role = await guild.roles.create({
+//     //   name: 'IDK'
+//     // })
+
+//     // alert()
+
+//     // await role.edit({
+//     //   name: 'DUMB'
+//     // })
+
+//     // alert()
+
+//     // console.log(
+//     //   await guild.getPruneCount({
+//     //     days: 7,
+//     //     includeRoles: ['ROLE_ID']
+//     //   })
+//     // )
+
+//     // console.log(
+//     //   await guild.prune({
+//     //     days: 7,
+//     //     includeRoles: ['ROLE_ID']
+//     //   })
+//     // )
+//     // console.log(
+//     //   await guild.prune({
+//     //     days: 7,
+//     //     computePruneCount: false,
+//     //     includeRoles: ['ROLE_ID']
+//     //   })
+//     // )
+
+//     // await guild.editWidget({
+//     //   enabled: true,
+//     //   channel: 'CHANNEL_ID'
+//     // })
+
+//     // console.log(await guild.getWidget())
+//     // console.log(await guild.getVanity())
+//     // console.log(await guild.getWidgetImageURL())
+//   }
+// })
+
+client.connect(TOKEN, Intents.All)
diff --git a/src/test/template.ts b/src/test/template.ts
new file mode 100644
index 0000000..c4e0c19
--- /dev/null
+++ b/src/test/template.ts
@@ -0,0 +1,21 @@
+import { Client, Intents } from '../../mod.ts'
+import { TOKEN } from './config.ts'
+
+const client = new Client()
+
+client.on('guildLoaded', async (guild) => {
+  if (guild.id === 'GUILD_ID') {
+    console.log((await guild.syncTemplate('TEMPLATE_ID')).code)
+    console.log(
+      (
+        await guild.editTemplate('TEMPLATE_ID', {
+          name: 'asdf',
+          description: 'asdfasdfasdf'
+        })
+      ).code
+    )
+    console.log((await guild.deleteTemplate('TEMPLATE_ID')).id)
+  }
+})
+
+client.connect(TOKEN, Intents.All)
diff --git a/src/test/user.ts b/src/test/user.ts
new file mode 100644
index 0000000..cd455ee
--- /dev/null
+++ b/src/test/user.ts
@@ -0,0 +1,14 @@
+import { Client, Intents } from '../../mod.ts'
+import { TOKEN } from './config.ts'
+
+const client = new Client()
+
+client.on('ready', async () => {
+  client.editUser({
+    username: 'Learning'
+  })
+  const channel = await client.createDM('USER_ID')
+  channel.send('nice')
+})
+
+client.connect(TOKEN, Intents.All)
diff --git a/src/types/guild.ts b/src/types/guild.ts
index c399de5..3c67657 100644
--- a/src/types/guild.ts
+++ b/src/types/guild.ts
@@ -192,7 +192,7 @@ export interface GuildCreatePayload {
 }
 
 export interface GuildCreateRolePayload {
-  id?: string
+  id: string
   name: string
   color?: number
   hoist?: boolean
diff --git a/src/utils/snowflake.ts b/src/utils/snowflake.ts
index d76e409..2a0e040 100644
--- a/src/utils/snowflake.ts
+++ b/src/utils/snowflake.ts
@@ -24,4 +24,8 @@ export class Snowflake {
   get increment(): string {
     return (this.snowflake & 0xfffn).toString()
   }
+
+  get toString(): string {
+    return this.id
+  }
 }