123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
import type { Embed } from '../structures/embed.ts'import type { MessageAttachment } from '../structures/message.ts'import type { RESTManager } from './manager.ts'import type { RequestMethods } from './types.ts'
export interface RequestOptions {  headers?: { [name: string]: string }  query?: { [name: string]: string }  files?: MessageAttachment[]  // Untyped JSON  data?: any  reason?: string  rawResponse?: boolean  route?: string}
export class APIRequest {  retries = 0  route: string
  constructor(    public rest: RESTManager,    public method: RequestMethods,    public path: string,    public options: RequestOptions  ) {    this.route = options.route ?? path    if (      (method === 'get' || method === 'delete' || method === 'head') &&      typeof options.data === 'object'    ) {      if (options.query !== undefined) {        Object.assign(options.query, options.data)      } else options.query = options.data      options.data = undefined    }    if (typeof options.query === 'object') {      const entries = Object.entries(options.query).filter(        (e) => e[1] !== undefined && e[1] !== null      )      if (entries.length > 0) {        this.path += '?'        entries.forEach((entry, i) => {          this.path += `${i === 0 ? '' : '&'}${encodeURIComponent(            entry[0]          )}=${encodeURIComponent(entry[1])}`        })      }    }
    let _files: undefined | MessageAttachment[]    if (      options.data?.embed?.files !== undefined &&      Array.isArray(options.data?.embed?.files)    ) {      _files = [...options.data?.embed?.files]    }    if (      options.data?.embeds !== undefined &&      Array.isArray(options.data?.embeds)    ) {      const files1 = options.data?.embeds        .map((e: Embed) => e.files)        .filter((e: MessageAttachment[]) => e !== undefined)      for (const files of files1) {        for (const file of files) {          if (_files === undefined) _files = []          _files?.push(file)        }      }    }
    if (options.data?.file !== undefined) {      if (_files === undefined) _files = []      _files.push(options.data?.file)    }
    if (      options.data?.files !== undefined &&      Array.isArray(options.data?.files)    ) {      if (_files === undefined) _files = []      options.data?.files.forEach((file: MessageAttachment) => {        _files!.push(file)      })    }
    // Interaction response    if (      options.data?.data?.files !== undefined &&      Array.isArray(options.data?.data?.files)    ) {      if (_files === undefined) _files = []      options.data?.data?.files.forEach((file: MessageAttachment) => {        _files!.push(file)      })    }    if (      options.data?.data?.embeds !== undefined &&      Array.isArray(options.data?.data?.embeds)    ) {      const files1 = options.data?.data?.embeds        .map((e: Embed) => e.files)        .filter((e: MessageAttachment[]) => e !== undefined)      for (const files of files1) {        for (const file of files) {          if (_files === undefined) _files = []          _files?.push(file)        }      }    }
    if (_files !== undefined && _files.length > 0) {      if (options.files === undefined) options.files = _files      else options.files = [...options.files, ..._files]    }  }
  async execute(): Promise<Response> {    let contentType: string | undefined    let body: BodyInit | undefined    if (      this.method === 'post' ||      this.method === 'put' ||      this.method === 'patch'    ) {      // Use empty JSON object as body by default      // this should not be required, but seems like CloudFlare is not      // letting these requests without a body through.      body = this.options.data ?? {}      if (this.options.files !== undefined && this.options.files.length > 0) {        contentType = undefined        const form = new FormData()        this.options.files.forEach((file, i) =>          form.append(`files[${i}]`, file.blob, file.name)        )        if (typeof body === 'object' && body !== null) {          // Differentiate between interaction response          // and other endpoints.          const target =            'data' in body && 'type' in body              ? (body as unknown as { data: Record<string, unknown> }).data              : (body as unknown as Record<string, unknown>)          target.attachments = this.options.files.map((e, i) => ({            id: i,            filename: e.name,            description: e.description          }))        }        form.append('payload_json', JSON.stringify(body))        body = form      } else if (body instanceof FormData) {        contentType = 'multipart/form-data'      } else {        contentType = 'application/json'        body = JSON.stringify(body)      }    }
    const controller = new AbortController()    const timer = setTimeout(() => {      controller.abort()    }, this.rest.requestTimeout)    this.rest.timers.add(timer)
    const url = this.path.startsWith('http')      ? this.path      : `${this.rest.apiURL}/v${this.rest.version}${this.path}`
    const headers: Record<string, string | undefined> = {      'User-Agent':        this.rest.userAgent ??        `DiscordBot (harmony, https://github.com/harmonyland/harmony)`,      Authorization:        this.rest.token === undefined          ? undefined          : `${this.rest.tokenType} ${              typeof this.rest.token === 'string'                ? this.rest.token                : this.rest.token()            }`.trim()    }
    if (contentType !== undefined) headers['Content-Type'] = contentType
    const init: RequestInit = {      method: this.method.toUpperCase(),      signal: controller.signal,      headers: Object.assign(headers, this.rest.headers, this.options.headers),      body    }
    if (this.options.reason !== undefined) {      ;(init.headers as { [name: string]: string })['X-Audit-Log-Reason'] =        encodeURIComponent(this.options.reason)    }
    return fetch(url, init).finally(() => {      clearTimeout(timer)      this.rest.timers.delete(timer)    })  }}