112 lines
3.6 KiB
TypeScript
112 lines
3.6 KiB
TypeScript
/* eslint-disable */
|
|
|
|
import { Readable } from "stream";
|
|
import { us_listen_socket_close, TemplatedApp, HttpResponse, HttpRequest } from "uWebSockets.js";
|
|
|
|
import formData from "./formdata";
|
|
import { stob } from "./utils";
|
|
import { Handler } from "./types";
|
|
import { join } from "path";
|
|
|
|
const contTypes = ["application/x-www-form-urlencoded", "multipart/form-data"];
|
|
const noOp = () => true;
|
|
|
|
const handleBody = (res: HttpResponse, req: HttpRequest) => {
|
|
const contType = req.getHeader("content-type");
|
|
|
|
res.bodyStream = function () {
|
|
const stream = new Readable();
|
|
stream._read = noOp; // eslint-disable-line @typescript-eslint/unbound-method
|
|
|
|
this.onData((ab: ArrayBuffer, isLast: boolean) => {
|
|
// uint and then slicing is bit faster than slice and then uint
|
|
stream.push(new Uint8Array(ab.slice((ab as any).byteOffset, ab.byteLength))); // eslint-disable-line @typescript-eslint/no-explicit-any
|
|
if (isLast) {
|
|
stream.push(null);
|
|
}
|
|
});
|
|
|
|
return stream;
|
|
};
|
|
|
|
res.body = () => stob(res.bodyStream());
|
|
|
|
if (contType.includes("application/json")) res.json = async () => JSON.parse(await res.body());
|
|
if (contTypes.map((t) => contType.includes(t)).includes(true)) res.formData = formData.bind(res, contType);
|
|
};
|
|
|
|
class BaseApp {
|
|
_sockets = new Map();
|
|
ws!: TemplatedApp["ws"];
|
|
get!: TemplatedApp["get"];
|
|
_post!: TemplatedApp["post"];
|
|
_put!: TemplatedApp["put"];
|
|
_patch!: TemplatedApp["patch"];
|
|
_listen!: TemplatedApp["listen"];
|
|
|
|
post(pattern: string, handler: Handler) {
|
|
if (typeof handler !== "function") throw Error(`handler should be a function, given ${typeof handler}.`);
|
|
this._post(pattern, (res, req) => {
|
|
handleBody(res, req);
|
|
handler(res, req);
|
|
});
|
|
return this;
|
|
}
|
|
|
|
put(pattern: string, handler: Handler) {
|
|
if (typeof handler !== "function") throw Error(`handler should be a function, given ${typeof handler}.`);
|
|
this._put(pattern, (res, req) => {
|
|
handleBody(res, req);
|
|
|
|
handler(res, req);
|
|
});
|
|
return this;
|
|
}
|
|
|
|
patch(pattern: string, handler: Handler) {
|
|
if (typeof handler !== "function") throw Error(`handler should be a function, given ${typeof handler}.`);
|
|
this._patch(pattern, (res, req) => {
|
|
handleBody(res, req);
|
|
|
|
handler(res, req);
|
|
});
|
|
return this;
|
|
}
|
|
|
|
listen(h: string | number, p: Function | number = noOp, cb?: Function) {
|
|
if (typeof p === "number" && typeof h === "string") {
|
|
this._listen(h, p, (socket) => {
|
|
this._sockets.set(p, socket);
|
|
if (cb === undefined) {
|
|
throw new Error("cb undefined");
|
|
}
|
|
cb(socket);
|
|
});
|
|
} else if (typeof h === "number" && typeof p === "function") {
|
|
this._listen(h, (socket) => {
|
|
this._sockets.set(h, socket);
|
|
p(socket);
|
|
});
|
|
} else {
|
|
throw Error("Argument types: (host: string, port: number, cb?: Function) | (port: number, cb?: Function)");
|
|
}
|
|
|
|
return this;
|
|
}
|
|
|
|
close(port: null | number = null) {
|
|
if (port) {
|
|
this._sockets.has(port) && us_listen_socket_close(this._sockets.get(port));
|
|
this._sockets.delete(port);
|
|
} else {
|
|
this._sockets.forEach((app) => {
|
|
us_listen_socket_close(app);
|
|
});
|
|
this._sockets.clear();
|
|
}
|
|
return this;
|
|
}
|
|
}
|
|
|
|
export default BaseApp;
|