Latest
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267import { ServerRequest } from "https://deno.land/std@0.62.0/http/server.ts";import { JsonRpcRequest, JsonRpcSuccess, JsonRpcFailure, JsonRpcId, JsonValue, JsonArray, JsonObject,} from "./jsonRpc2Types.ts";type ResultOrError = | { id: JsonRpcId | undefined; result: unknown } | ServerError;type NotNotification = | { id: JsonRpcId; result: unknown } | (Omit<ServerError, "rpcRequestID"> & { rpcRequestID: JsonRpcId; });type ResponseTypes = JsonRpcSuccess | JsonRpcFailure;type Methods = { [method: string]: (...args: any[]) => unknown };
class ServerError extends Error { rpcErrorID: number; rpcRequestID: JsonRpcId | undefined; constructor( rpcErrorID: number, message: string, rpcRequestID: JsonRpcId | undefined ) { super(message); this.rpcErrorID = rpcErrorID; this.rpcRequestID = rpcRequestID; }}
function isJsonRpcVersion(input: unknown): input is "2.0" { return input === "2.0";}function isJsonRpcMethod(input: unknown): input is string { return typeof input === "string" && !input.startsWith("rpc.");}function isJsonRpcParams(input: unknown): input is JsonArray | JsonObject { return typeof input === "object";}function isJsonRpcId(input: unknown): input is JsonRpcId { switch (typeof input) { case "string": return true; case "number": return input % 1 === 0; case "object": let isNull = input === null; if (isNull) { console.warn("Use of null ID in JSONRPC 2.0 is discouraged."); return true; } else { return false; } default: return false; }}
function validateRpcObj( decodedBody: JsonValue | ServerError, methods: Methods): JsonRpcRequest | ServerError { if (decodedBody instanceof ServerError) return decodedBody; if ( typeof decodedBody === "object" && !Array.isArray(decodedBody) && decodedBody !== null ) { if ( !isJsonRpcVersion(decodedBody.jsonrpc) || !isJsonRpcMethod(decodedBody.method) ) return new ServerError( -32600, "Invalid Request", isJsonRpcId(decodedBody.id) ? decodedBody.id : null ); else if (typeof methods[decodedBody.method] !== "function") return new ServerError( -32601, "Method not found", "id" in decodedBody ? isJsonRpcId(decodedBody.id) ? decodedBody.id : null : undefined ); else if ("params" in decodedBody && typeof decodedBody.params !== "object") return new ServerError( -32602, "Invalid parameters", "id" in decodedBody ? isJsonRpcId(decodedBody.id) ? decodedBody.id : null : undefined ); else { return "id" in decodedBody ? isJsonRpcId(decodedBody.id) ? ((decodedBody as unknown) as JsonRpcRequest) : new ServerError(-32600, "Invalid Request", null) : ({ ...decodedBody, id: undefined } as JsonRpcRequest); } } else return new ServerError(-32600, "Invalid Request", null);}
async function executeMethods( rpcReqObj: JsonRpcRequest | ServerError, methods: Methods, req?: ServerRequest): Promise<ResultOrError> { try { if (rpcReqObj instanceof ServerError) return rpcReqObj; else { if (req) return Array.isArray(rpcReqObj.params) ? { result: await methods[rpcReqObj.method](req, ...rpcReqObj.params), id: rpcReqObj.id, } : { result: await methods[rpcReqObj.method]({ ...rpcReqObj.params, req, }), id: rpcReqObj.id, }; else { return Array.isArray(rpcReqObj.params) ? { result: await methods[rpcReqObj.method](...rpcReqObj.params), id: rpcReqObj.id, } : { result: await methods[rpcReqObj.method](rpcReqObj.params), id: rpcReqObj.id, }; } } } catch (err) { return new ServerError( -32000, "Server error", "id" in rpcReqObj ? isJsonRpcId(rpcReqObj.id) ? rpcReqObj.id : null : undefined ); }}
async function respondRpc( req: ServerRequest, methods: Methods, { includeServerErrorStack = false, callMethodsWithRequestObj = false } = {}) { const decodedBody = new TextDecoder().decode(await Deno.readAll(req.body)); const resObject = callMethodsWithRequestObj ? await handleData(decodedBody, methods, includeServerErrorStack, req) : await handleData(decodedBody, methods, includeServerErrorStack); const headers = new Headers(); headers.set("content-type", "application/json"); req.respond( resObject ? { body: new TextEncoder().encode(JSON.stringify(resObject)), headers, status: 200, } : { status: 204 } ); return resObject;}
function parseJson(json: string): JsonValue | ServerError { try { return JSON.parse(json); } catch (err) { return new ServerError(-32700, "Parse error", null); }}
async function handleData( decodedBody: string, methods: Methods, includeServerErrorStack = false, req?: ServerRequest): Promise<ResponseTypes | ResponseTypes[] | null> { const data = parseJson(decodedBody); const result: ResultOrError | ResultOrError[] = Array.isArray(data) && data.length > 0 ? await Promise.all( data .map((body: JsonValue) => validateRpcObj(body, methods)) .map(validatedRpcReq => req ? executeMethods(validatedRpcReq, methods, req) : executeMethods(validatedRpcReq, methods) ) ) : req ? await executeMethods(validateRpcObj(data, methods), methods, req) : await executeMethods(validateRpcObj(data, methods), methods); return createRPCResponseObject(result, includeServerErrorStack);}
function createRPCResponseObject( result: ResultOrError | ResultOrError[], includeServerErrorStack: boolean): ResponseTypes | ResponseTypes[] | null { if (Array.isArray(result)) { const responseBatchObj = result .map(result => createObject(result, includeServerErrorStack)) .filter( (result: ResponseTypes | null): result is ResponseTypes => result != null ); return responseBatchObj.length > 0 ? responseBatchObj : null; } else { return createObject(result, includeServerErrorStack); }}
function createObject( data: ResultOrError, includeServerErrorStack: boolean): ResponseTypes | null { function isNotNotification(result: ResultOrError): result is NotNotification { return ( ("id" in result && result.id !== undefined) || (result instanceof ServerError && result.rpcRequestID !== undefined) ); } if (isNotNotification(data)) { if ("id" in data) { return { jsonrpc: "2.0", result: data.result === undefined ? null : data.result, // can't return undefined id: data.id, }; } else { const rpcResObj: JsonRpcFailure = { jsonrpc: "2.0", error: { code: data.rpcErrorID || -32603, message: data.message, }, id: data.rpcRequestID, }; if (data.stack && includeServerErrorStack) rpcResObj.error.data = { stack: data.stack }; return rpcResObj; } } else { return null; }}
export { respondRpc, handleData };