mirror of
https://github.com/discordjs/discord.js.git
synced 2026-03-09 16:13:31 +01:00
cdaa0a36f5
BREAKING CHANGE: NodeJS v18+ is required when using node due to the use of global `fetch` BREAKING CHANGE: The raw method of REST now returns a web compatible `Respone` object. BREAKING CHANGE: The `parseResponse` utility method has been updated to operate on a web compatible `Response` object. BREAKING CHANGE: Many underlying internals have changed, some of which were exported. BREAKING CHANGE: `DefaultRestOptions` used to contain a default `agent`, which is now set to `null` instead.
85 lines
2.6 KiB
TypeScript
85 lines
2.6 KiB
TypeScript
import type { ServerResponse } from 'node:http';
|
|
import { Readable } from 'node:stream';
|
|
import { pipeline } from 'node:stream/promises';
|
|
import { DiscordAPIError, HTTPError, RateLimitError, type ResponseLike } from '@discordjs/rest';
|
|
|
|
/**
|
|
* Populates a server response with the data from a Discord 2xx REST response
|
|
*
|
|
* @param res - The server response to populate
|
|
* @param data - The data to populate the response with
|
|
*/
|
|
export async function populateSuccessfulResponse(res: ServerResponse, data: ResponseLike): Promise<void> {
|
|
res.statusCode = data.status;
|
|
|
|
for (const [header, value] of data.headers) {
|
|
// Strip ratelimit headers
|
|
if (/^x-ratelimit/i.test(header)) {
|
|
continue;
|
|
}
|
|
|
|
res.setHeader(header, value);
|
|
}
|
|
|
|
if (data.body) {
|
|
await pipeline(data.body instanceof Readable ? data.body : Readable.fromWeb(data.body), res);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Populates a server response with the data from a Discord non-2xx REST response that is NOT a 429
|
|
*
|
|
* @param res - The server response to populate
|
|
* @param error - The error to populate the response with
|
|
*/
|
|
export function populateGeneralErrorResponse(res: ServerResponse, error: DiscordAPIError | HTTPError): void {
|
|
res.statusCode = error.status;
|
|
|
|
if ('rawError' in error) {
|
|
res.setHeader('Content-Type', 'application/json');
|
|
res.write(JSON.stringify(error.rawError));
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Populates a server response with the data from a Discord 429 REST response
|
|
*
|
|
* @param res - The server response to populate
|
|
* @param error - The error to populate the response with
|
|
*/
|
|
export function populateRatelimitErrorResponse(res: ServerResponse, error: RateLimitError): void {
|
|
res.statusCode = 429;
|
|
res.setHeader('Retry-After', error.timeToReset / 1_000);
|
|
}
|
|
|
|
/**
|
|
* Populates a server response with data relevant for a timeout
|
|
*
|
|
* @param res - The sever response to populate
|
|
*/
|
|
export function populateAbortErrorResponse(res: ServerResponse): void {
|
|
res.statusCode = 504;
|
|
res.statusMessage = 'Upstream timed out';
|
|
}
|
|
|
|
/**
|
|
* Tries to populate a server response from an error object
|
|
*
|
|
* @param res - The server response to populate
|
|
* @param error - The error to check and use
|
|
* @returns - True if the error is known and the response object was populated, otherwise false
|
|
*/
|
|
export function populateErrorResponse(res: ServerResponse, error: unknown): boolean {
|
|
if (error instanceof DiscordAPIError || error instanceof HTTPError) {
|
|
populateGeneralErrorResponse(res, error);
|
|
} else if (error instanceof RateLimitError) {
|
|
populateRatelimitErrorResponse(res, error);
|
|
} else if (error instanceof Error && error.name === 'AbortError') {
|
|
populateAbortErrorResponse(res);
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|