-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
a0e138d
commit d9e1017
Showing
24 changed files
with
496 additions
and
92 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| import type { Context } from "hono"; | ||
|
|
||
| // inversion of control container to store parked requests | ||
| export type ParkingLot = Map< | ||
| string, | ||
| [Context, (value: Response) => void, (reason: unknown) => void] | ||
| >; | ||
|
|
||
| export const parkingLot: ParkingLot = new Map(); | ||
|
|
||
| export { deflectorMiddleware } from "./middleware.js"; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,130 @@ | ||
| import { headersToObject, resolveBody } from "@fiberplane/fpx-utils"; | ||
| import type { MiddlewareHandler } from "hono"; | ||
|
|
||
| import { eq } from "drizzle-orm"; | ||
| import * as schema from "../../db/schema.js"; | ||
| import logger from "../../logger.js"; | ||
| import { | ||
| handleFailedRequest, | ||
| handleSuccessfulRequest, | ||
| } from "../proxy-request/index.js"; | ||
| import type { Bindings, Variables } from "../types.js"; | ||
| import { parkingLot } from "./index.js"; | ||
|
|
||
| let isDeflectorEnabled = false; | ||
|
|
||
| export const setDeflectorStatus = (status: boolean) => { | ||
| isDeflectorEnabled = status; | ||
| }; | ||
|
|
||
| export const deflectorMiddleware: MiddlewareHandler<{ | ||
| Bindings: Bindings; | ||
| Variables: Variables; | ||
| }> = async (c, next) => { | ||
| const deflectTo = c.req.header("x-fpx-deflect-to"); | ||
| if (!isDeflectorEnabled || !deflectTo) { | ||
| return next(); | ||
| } | ||
|
|
||
| const db = c.get("db"); | ||
| const traceId = crypto.randomUUID(); | ||
| const [requestUrl, deflectionType] = getTargetUrlAndDeflectionType( | ||
| deflectTo, | ||
| c.req.url, | ||
| ); | ||
| logger.info(`Deflecting request to ${requestUrl}`); | ||
| const newHeaders = new Headers(c.req.raw.headers); | ||
| newHeaders.append("x-fpx-trace-id", traceId); | ||
|
|
||
| const [{ id: requestId }] = await db | ||
| .insert(schema.appRequests) | ||
| .values({ | ||
| requestMethod: c.req.method as schema.NewAppRequest["requestMethod"], | ||
| requestUrl: requestUrl.toString(), | ||
| requestHeaders: headersToObject(newHeaders), | ||
| requestPathParams: {}, | ||
| requestQueryParams: Object.fromEntries(requestUrl.searchParams), | ||
| requestBody: await resolveBody(c.req), | ||
| requestRoute: requestUrl.pathname, | ||
| }) | ||
| .returning({ id: schema.appRequests.id }); | ||
|
|
||
| const startTime = Date.now(); | ||
| newHeaders.delete("x-fpx-deflect-to"); | ||
|
|
||
| try { | ||
| let response: Response; | ||
| if (deflectionType === "proxy") { | ||
| response = await fetch(requestUrl, { | ||
| method: c.req.method, | ||
| headers: newHeaders, | ||
| body: c.req.raw.body, | ||
| }); | ||
| } else if (deflectionType === "serverSimulator") { | ||
| response = await new Promise((resolve, reject) => { | ||
| parkingLot.set(traceId, [c, resolve, reject]); | ||
| }); | ||
| } else if (deflectionType === "mock") { | ||
| const [r1] = await db | ||
| .select() | ||
| .from(schema.appRequests) | ||
| .then((requests) => | ||
| requests.filter((request) => { | ||
| return request.requestHeaders?.["x-fpx-deflect-to"] !== undefined; | ||
| }), | ||
| ); | ||
|
|
||
| if (r1?.id) { | ||
| const [matchingResponse] = await db | ||
| .select() | ||
| .from(schema.appResponses) | ||
| .where(eq(schema.appResponses.requestId, r1.id)); | ||
| response = new Response(matchingResponse.responseBody, { | ||
| status: matchingResponse.responseStatusCode ?? 200, | ||
| headers: matchingResponse.responseHeaders ?? {}, | ||
| }); | ||
| } else { | ||
| throw new Error(); | ||
| } | ||
| } else { | ||
| throw new Error(); | ||
| } | ||
| const duration = Date.now() - startTime; | ||
| await handleSuccessfulRequest( | ||
| db, | ||
| requestId, | ||
| duration, | ||
| response.clone(), | ||
| traceId, | ||
| ); | ||
|
|
||
| return response; | ||
| } catch (error) { | ||
| logger.error("Error making request", error); | ||
| const duration = Date.now() - startTime; | ||
| await handleFailedRequest(db, requestId, traceId, duration, error); | ||
|
|
||
| return c.json({ error: "Internal server error" }, 500); | ||
| } | ||
| }; | ||
|
|
||
| type DeflectionType = "proxy" | "serverSimulator" | "mock"; | ||
|
|
||
| function getTargetUrlAndDeflectionType( | ||
| targetString: string, | ||
| requestString: string, | ||
| ): [finalUrl: URL, deflectionType: DeflectionType] { | ||
| try { | ||
| const [targetUrl, requestUrl] = [targetString, requestString].map( | ||
| (url) => new URL(url), | ||
| ); | ||
| for (const prop of ["hostname", "port", "protocol"] as const) { | ||
| requestUrl[prop] = targetUrl[prop]; | ||
| } | ||
| return [requestUrl, "proxy"]; | ||
| } catch { | ||
| const url = new URL(requestString); | ||
| url.hostname = targetString; | ||
| return [url, "serverSimulator"]; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,31 @@ | ||
| import { zValidator } from "@hono/zod-validator"; | ||
| import { Hono } from "hono"; | ||
| import { z } from "zod"; | ||
| import { parkingLot } from "../lib/deflector/index.js"; | ||
| import type { Bindings, Variables } from "../lib/types.js"; | ||
|
|
||
| const app = new Hono<{ Bindings: Bindings; Variables: Variables }>(); | ||
|
|
||
| app.post( | ||
| "/v0/deflector", | ||
| zValidator( | ||
| "json", | ||
| z.object({ | ||
| key: z.string(), | ||
| value: z.string(), | ||
| }), | ||
| ), | ||
| async (ctx) => { | ||
| const { key, value } = ctx.req.valid("json"); | ||
| const fromCache = parkingLot.get(key); | ||
| if (fromCache) { | ||
| parkingLot.delete(key); | ||
| const [parkedContext, resolve] = fromCache; | ||
| resolve(parkedContext.json(JSON.parse(value))); | ||
| return ctx.json({ result: "success" }); | ||
| } | ||
| return ctx.json({ error: `Unknown key: ${key}` }, 404); | ||
| }, | ||
| ); | ||
|
|
||
| export default app; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.