mirror of
https://github.com/discordeno/discordeno.git
synced 2026-06-16 19:28:17 +00:00
feat: standalone rest server (#290)
* feat: standalone rest server * desc * fmt * <3 vlad * Update src/rest/README.md Co-authored-by: Ayyan <ayyantee@gmail.com> * Update src/rest/README.md Co-authored-by: Ayyan <ayyantee@gmail.com> * Update README.md * Update src/rest/deps.ts Co-authored-by: Ayyan <ayyantee@gmail.com> * Update src/rest/queue.ts Co-authored-by: Ayyan <ayyantee@gmail.com> * chore: ignore no-explicit-any rule * fix(rest): replace with correct import paths * deno fmt * fixes * fmt * use user agent cons * fix typings * Update src/rest/cache.ts Co-authored-by: Ayyan <ayyantee@gmail.com> * Update src/rest/cache.ts Co-authored-by: Ayyan <ayyantee@gmail.com> Co-authored-by: Ayyan <ayyantee@gmail.com>
This commit is contained in:
@@ -0,0 +1,187 @@
|
||||
import { restCache } from "./cache.ts";
|
||||
import { createRequestBody, processRequestHeaders } from "./request.ts";
|
||||
import { HttpResponseCode } from "./types/mod.ts";
|
||||
|
||||
/** If the queue is not already processing, this will start processing the queue. */
|
||||
export function startQueue() {
|
||||
// IF ALREADY PROCESSING CANCEL
|
||||
if (restCache.processingQueue) return;
|
||||
// MARK AS PROCESSING
|
||||
restCache.processingQueue = true;
|
||||
processQueue();
|
||||
}
|
||||
|
||||
/** Processes the queue by looping over each path separately until the queues are empty. */
|
||||
export function processQueue() {
|
||||
while (restCache.processingQueue) {
|
||||
// FOR EVERY PATH WE WILL START ITS OWN LOOP.
|
||||
restCache.pathQueues.forEach(async (queue) => {
|
||||
// EACH PATH IS UNIQUE LIMITER
|
||||
while (queue.length) {
|
||||
// IF THE BOT IS GLOBALLY RATELIMITED TRY AGAIN
|
||||
if (!restCache.globallyRateLimited) continue;
|
||||
// SELECT THE FIRST ITEM FROM THIS QUEUE
|
||||
const [queuedRequest] = queue;
|
||||
// IF THIS DOESNT HAVE ANY ITEMS JUST CANCEL, THE CLEANER WILL REMOVE IT.
|
||||
if (!queuedRequest) return;
|
||||
|
||||
// IF THIS URL IS STILL RATE LIMITED, TRY AGAIN
|
||||
const urlResetIn = checkRateLimits(queuedRequest.payload.url);
|
||||
if (urlResetIn) continue;
|
||||
|
||||
// IF A BUCKET EXISTS, CHECK THE BUCKET'S RATE LIMITS
|
||||
const bucketResetIn = queuedRequest.payload.bucketID
|
||||
? checkRateLimits(queuedRequest.payload.bucketID)
|
||||
: false;
|
||||
// THIS BUCKET IS STILL RATELIMITED, RE-ADD TO QUEUE
|
||||
if (bucketResetIn) continue;
|
||||
|
||||
// EXECUTE THE REQUEST
|
||||
|
||||
// IF THIS IS A GET REQUEST, CHANGE THE BODY TO QUERY PARAMETERS
|
||||
const query =
|
||||
queuedRequest.payload.method === "get" && queuedRequest.payload.body
|
||||
? Object.entries(queuedRequest.payload.body).map(([key, value]) =>
|
||||
`${encodeURIComponent(key)}=${
|
||||
encodeURIComponent(value as string)
|
||||
}`
|
||||
)
|
||||
.join("&")
|
||||
: "";
|
||||
const urlToUse = queuedRequest.payload.method === "get" && query
|
||||
? `${queuedRequest.payload.url}?${query}`
|
||||
: queuedRequest.payload.url;
|
||||
|
||||
// CUSTOM HANDLER FOR USER TO LOG OR WHATEVER WHENEVER A FETCH IS MADE
|
||||
restCache.eventHandlers.fetching(queuedRequest.payload);
|
||||
|
||||
try {
|
||||
const response = await fetch(
|
||||
urlToUse,
|
||||
createRequestBody(queuedRequest),
|
||||
);
|
||||
restCache.eventHandlers.fetched(queuedRequest.payload);
|
||||
const bucketIDFromHeaders = processRequestHeaders(
|
||||
queuedRequest.payload.url,
|
||||
response.headers,
|
||||
);
|
||||
|
||||
if (response.status < 200 && response.status >= 400) {
|
||||
restCache.eventHandlers.error(
|
||||
"httpError",
|
||||
queuedRequest.payload,
|
||||
response,
|
||||
);
|
||||
|
||||
const error = response.status === HttpResponseCode.BadRequest
|
||||
? "The request was improperly formatted, or the server couldn't understand it."
|
||||
: response.status === HttpResponseCode.Unauthorized
|
||||
? "The Authorization header was missing or invalid."
|
||||
: response.status === HttpResponseCode.Forbidden
|
||||
? "The Authorization token you passed did not have permission to the resource."
|
||||
: response.status === HttpResponseCode.NotFound
|
||||
? "The resource at the location specified doesn't exist."
|
||||
: response.status === HttpResponseCode.MethodNotAllowed
|
||||
? "The HTTP method used is not valid for the location specified."
|
||||
: response.status === HttpResponseCode.GatewayUnavailable
|
||||
? "There was not a gateway available to process your request. Wait a bit and retry."
|
||||
: "REQUEST_UNKNOWN_ERROR";
|
||||
|
||||
queuedRequest.request.respond(
|
||||
{ status: response.status, body: JSON.stringify({ error }) },
|
||||
);
|
||||
queue.shift();
|
||||
continue;
|
||||
}
|
||||
|
||||
// SOMETIMES DISCORD RETURNS AN EMPTY 204 RESPONSE THAT CAN'T BE MADE TO JSON
|
||||
if (response.status === 204) {
|
||||
restCache.eventHandlers.fetchSuccess(queuedRequest.payload);
|
||||
return queuedRequest.request.respond({ status: 204 });
|
||||
}
|
||||
|
||||
// CONVERT THE RESPONSE TO JSON
|
||||
const json = await response.json();
|
||||
|
||||
// IF THE RESPONSE WAS RATE LIMITED, HANDLE ACCORDINGLY
|
||||
if (
|
||||
json.retry_after ||
|
||||
json.message === "You are being rate limited."
|
||||
) {
|
||||
// IF IT HAS MAXED RETRIES SOMETHING SERIOUSLY WRONG. CANCEL OUT.
|
||||
if (
|
||||
queuedRequest.payload.retryCount >=
|
||||
queuedRequest.options.maxRetryCount
|
||||
) {
|
||||
restCache.eventHandlers.retriesMaxed(queuedRequest.payload);
|
||||
queuedRequest.request.respond(
|
||||
{
|
||||
status: 200,
|
||||
body: JSON.stringify(
|
||||
{
|
||||
error:
|
||||
"The request was rate limited and it maxed out the retries limit.",
|
||||
},
|
||||
),
|
||||
},
|
||||
);
|
||||
// REMOVE ITEM FROM QUEUE TO PREVENT RETRY
|
||||
queue.shift();
|
||||
continue;
|
||||
}
|
||||
|
||||
// SET THE BUCKET ID IF IT WAS PRESENT
|
||||
if (bucketIDFromHeaders) {
|
||||
queuedRequest.payload.bucketID = bucketIDFromHeaders;
|
||||
}
|
||||
// SINCE IT WAS RATELIMITE, RETRY AGAIN
|
||||
continue;
|
||||
}
|
||||
|
||||
restCache.eventHandlers.fetchSuccess(queuedRequest.payload);
|
||||
// REMOVE FROM QUEUE
|
||||
queue.shift();
|
||||
queuedRequest.request.respond(
|
||||
{ status: 200, body: JSON.stringify(json) },
|
||||
);
|
||||
} catch (error) {
|
||||
// SOMETHING WENT WRONG, LOG AND RESPOND WITH ERROR
|
||||
restCache.eventHandlers.fetchFailed(queuedRequest.payload, error);
|
||||
queuedRequest.request.respond(
|
||||
{ status: 404, body: JSON.stringify({ error }) },
|
||||
);
|
||||
// REMOVE FROM QUEUE
|
||||
queue.shift();
|
||||
}
|
||||
}
|
||||
|
||||
// ONCE QUEUE IS DONE, WE CAN TRY CLEANING UP
|
||||
cleanupQueues();
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
/** Cleans up the queues by checking if there is nothing left and removing it. */
|
||||
export function cleanupQueues() {
|
||||
restCache.pathQueues.forEach((queue, key) => {
|
||||
if (queue.length) return;
|
||||
// REMOVE IT FROM CACHE
|
||||
restCache.pathQueues.delete(key);
|
||||
});
|
||||
}
|
||||
|
||||
/** Check the rate limits for a url or a bucket. */
|
||||
export function checkRateLimits(url: string) {
|
||||
const ratelimited = restCache.ratelimitedPaths.get(url);
|
||||
const global = restCache.ratelimitedPaths.get("global");
|
||||
const now = Date.now();
|
||||
|
||||
if (ratelimited && now < ratelimited.resetTimestamp) {
|
||||
return ratelimited.resetTimestamp - now;
|
||||
}
|
||||
if (global && now < global.resetTimestamp) {
|
||||
return global.resetTimestamp - now;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
Reference in New Issue
Block a user