@fragno-dev/core 0.1.0 → 0.1.2

package/src/api/request-middleware.test.ts

@@ -500,8 +500,7 @@ describe("Request Middleware", () => {
     });
   });
 
-  // TODO: This is not currently supported
-  test.todo("middleware can modify query parameters", async () => {
+  test("middleware can modify query parameters", async () => {
     const fragment = defineFragment("test-lib");
 
     const routes = [
@@ -542,4 +541,91 @@ describe("Request Middleware", () => {
       role: "some-other-role-defined-in-middleware",
     });
   });
+
+  test("middleware can modify request body", async () => {
+    const fragment = defineFragment("test-lib");
+
+    const routes = [
+      defineRoute({
+        method: "POST",
+        path: "/users",
+        inputSchema: z.object({ name: z.string(), role: z.string().optional() }),
+        outputSchema: z.object({ name: z.string(), role: z.string() }),
+        handler: async ({ input }, { json }) => {
+          const body = await input.valid();
+          return json({
+            name: body.name,
+            role: body.role ?? "user",
+          });
+        },
+      }),
+    ] as const;
+
+    const instance = createFragment(fragment, {}, routes, {
+      mountRoute: "/api",
+    }).withMiddleware(async ({ ifMatchesRoute, requestState }) => {
+      // Middleware modifies the request body
+      const result = await ifMatchesRoute("POST", "/users", async ({ input }) => {
+        const body = await input.valid();
+        // Modify the body by adding a role field
+        requestState.setBody({
+          ...body,
+          role: "admin-from-middleware",
+        });
+      });
+
+      return result;
+    });
+
+    const req = new Request("http://localhost/api/users", {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ name: "John Doe" }),
+    });
+
+    const res = await instance.handler(req);
+    expect(res.status).toBe(200);
+    expect(await res.json()).toEqual({
+      name: "John Doe",
+      role: "admin-from-middleware",
+    });
+  });
+
+  test("middleware can modify request headers", async () => {
+    const fragment = defineFragment("test-lib");
+
+    const routes = [
+      defineRoute({
+        method: "GET",
+        path: "/data",
+        outputSchema: z.object({ auth: z.string(), custom: z.string() }),
+        handler: async ({ headers }, { json }) => {
+          return json({
+            auth: headers.get("Authorization") ?? "none",
+            custom: headers.get("X-Custom-Header") ?? "none",
+          });
+        },
+      }),
+    ] as const;
+
+    const instance = createFragment(fragment, {}, routes, {
+      mountRoute: "/api",
+    }).withMiddleware(async ({ headers }) => {
+      // Middleware modifies headers
+      headers.set("Authorization", "Bearer middleware-token");
+      headers.set("X-Custom-Header", "middleware-value");
+      return undefined;
+    });
+
+    const req = new Request("http://localhost/api/data", {
+      method: "GET",
+    });
+
+    const res = await instance.handler(req);
+    expect(res.status).toBe(200);
+    expect(await res.json()).toEqual({
+      auth: "Bearer middleware-token",
+      custom: "middleware-value",
+    });
+  });
 });

package/src/api/request-middleware.ts

@@ -2,10 +2,10 @@ import type { StandardSchemaV1 } from "@standard-schema/spec";
 import type { ExtractRouteByPath, ExtractRoutePath } from "../client/client";
 import type { HTTPMethod } from "./api";
 import type { ExtractPathParams } from "./internal/path";
-import type { RequestBodyType } from "./request-input-context";
 import type { AnyFragnoRouteConfig } from "./route";
 import { RequestInputContext } from "./request-input-context";
 import { OutputContext, RequestOutputContext } from "./request-output-context";
+import { MutableRequestState } from "./mutable-request-state";
 
 export type FragnoMiddlewareCallback<
   TRoutes extends readonly AnyFragnoRouteConfig[],
@@ -19,10 +19,8 @@ export type FragnoMiddlewareCallback<
 export interface RequestMiddlewareOptions {
   path: string;
   method: HTTPMethod;
-  pathParams?: Record<string, string>;
-  searchParams: URLSearchParams;
-  body: RequestBodyType;
   request: Request;
+  state: MutableRequestState;
 }
 
 export class RequestMiddlewareOutputContext<
@@ -50,9 +48,11 @@ export class RequestMiddlewareOutputContext<
 export class RequestMiddlewareInputContext<const TRoutes extends readonly AnyFragnoRouteConfig[]> {
   readonly #options: RequestMiddlewareOptions;
   readonly #route: TRoutes[number];
+  readonly #state: MutableRequestState;
 
   constructor(routes: TRoutes, options: RequestMiddlewareOptions) {
     this.#options = options;
+    this.#state = options.state;
 
     const route = routes.find(
       (route) => route.path === options.path && route.method === options.method,
@@ -74,11 +74,15 @@ export class RequestMiddlewareInputContext<const TRoutes extends readonly AnyFra
   }
 
   get pathParams(): Record<string, string> {
-    return this.#options.pathParams ?? {};
+    return this.#state.pathParams;
   }
 
   get queryParams(): URLSearchParams {
-    return this.#options.searchParams;
+    return this.#state.searchParams;
+  }
+
+  get headers(): Headers {
+    return this.#state.headers;
   }
 
   get inputSchema(): StandardSchemaV1 | undefined {
@@ -89,6 +93,23 @@ export class RequestMiddlewareInputContext<const TRoutes extends readonly AnyFra
     return this.#route.outputSchema;
   }
 
+  /**
+   * Access to the mutable request state.
+   * Use this to modify query parameters, path parameters, or request body.
+   *
+   * @example
+   * ```typescript
+   * // Modify body
+   * requestState.setBody({ modified: true });
+   *
+   * // Query params are already accessible via queryParams getter
+   * // Path params are already accessible via pathParams getter
+   * ```
+   */
+  get requestState(): MutableRequestState {
+    return this.#state;
+  }
+
   // Defined as a field so that `this` reference stays in tact when destructuring
   ifMatchesRoute = async <
     const TMethod extends HTTPMethod,
@@ -116,6 +137,7 @@ export class RequestMiddlewareInputContext<const TRoutes extends readonly AnyFra
       path: path,
       pathParams: this.pathParams as ExtractPathParams<TPath>,
       inputSchema: this.#route.inputSchema,
+      state: this.#state,
     });
 
     const outputContext = new RequestOutputContext(this.#route.outputSchema);

package/src/api/request-output-context.test.ts

@@ -327,7 +327,9 @@ describe("RequestOutputContext", () => {
       try {
         while (true) {
           const { done, value } = await reader.read();
-          if (done) break;
+          if (done) {
+            break;
+          }
           chunks.push(decoder.decode(value));
         }
       } catch {
@@ -354,7 +356,9 @@ describe("RequestOutputContext", () => {
       try {
         while (true) {
           const { done, value } = await reader.read();
-          if (done) break;
+          if (done) {
+            break;
+          }
           chunks.push(decoder.decode(value));
         }
       } catch {

package/src/api/request-output-context.ts

@@ -19,7 +19,9 @@ function mergeHeaders(...headerSources: (HeadersInit | undefined)[]): HeadersIni
   const mergedHeaders = new Headers();
 
   for (const headerSource of headerSources) {
-    if (!headerSource) continue;
+    if (!headerSource) {
+      continue;
+    }
 
     if (headerSource instanceof Headers) {
       for (const [key, value] of headerSource.entries()) {
@@ -45,11 +47,11 @@ export abstract class OutputContext<const TOutput, const TErrorCode extends stri
    *
    * Shortcut for `throw new FragnoApiError(...)`
    */
-  error(
+  error = (
     { message, code }: { message: string; code: TErrorCode },
     initOrStatus?: ResponseInit | StatusCode,
     headers?: HeadersInit,
-  ): Response {
+  ): Response => {
     if (typeof initOrStatus === "undefined") {
       return Response.json({ message: message, code }, { status: 500, headers });
     }
@@ -63,12 +65,12 @@ export abstract class OutputContext<const TOutput, const TErrorCode extends stri
       { message: message, code },
       { status: initOrStatus.status, headers: mergedHeaders },
     );
-  }
+  };
 
-  empty(
+  empty = (
     initOrStatus?: ResponseInit<ContentlessStatusCode> | ContentlessStatusCode,
     headers?: HeadersInit,
-  ): Response {
+  ): Response => {
     const defaultHeaders = {};
 
     if (typeof initOrStatus === "undefined") {
@@ -92,9 +94,13 @@ export abstract class OutputContext<const TOutput, const TErrorCode extends stri
       status: initOrStatus.status,
       headers: mergedHeaders,
     });
-  }
+  };
 
-  json(object: TOutput, initOrStatus?: ResponseInit | StatusCode, headers?: HeadersInit): Response {
+  json = (
+    object: TOutput,
+    initOrStatus?: ResponseInit | StatusCode,
+    headers?: HeadersInit,
+  ): Response => {
     if (typeof initOrStatus === "undefined") {
       return Response.json(object, {
         status: 200,
@@ -114,7 +120,7 @@ export abstract class OutputContext<const TOutput, const TErrorCode extends stri
       status: initOrStatus.status,
       headers: mergedHeaders,
     });
-  }
+  };
 
   jsonStream = (
     cb: (stream: ResponseStream<TOutput>) => void | Promise<void>,

package/src/client/client.ts

@@ -25,7 +25,7 @@ import {
 } from "./internal/ndjson-streaming";
 import { addStore, getInitialData, SSR_ENABLED } from "../util/ssr";
 import { unwrapObject } from "../util/nanostores";
-import type { FragmentBuilder } from "../api/fragment-builder";
+import type { FragmentDefinition } from "../api/fragment-builder";
 import {
   type AnyRouteOrFactory,
   type FlattenRouteFactories,
@@ -1001,15 +1001,18 @@ export function createClientBuilder<
   TDeps,
   TServices extends Record<string, unknown>,
   const TRoutesOrFactories extends readonly AnyRouteOrFactory[],
+  const TAdditionalContext extends Record<string, unknown>,
 >(
-  fragmentDefinition: FragmentBuilder<TConfig, TDeps, TServices>,
+  fragmentBuilder: {
+    definition: FragmentDefinition<TConfig, TDeps, TServices, TAdditionalContext>;
+  },
   publicConfig: FragnoPublicClientConfig,
   routesOrFactories: TRoutesOrFactories,
 ): ClientBuilder<
   FlattenRouteFactories<TRoutesOrFactories>,
   FragnoFragmentSharedConfig<FlattenRouteFactories<TRoutesOrFactories>>
 > {
-  const definition = fragmentDefinition.definition;
+  const definition = fragmentBuilder.definition;
 
   // For client-side, we resolve route factories with dummy context
   // This will be removed by the bundle plugin anyway

package/src/client/component.test.svelte

@@ -2,7 +2,9 @@
   import type { Readable } from "svelte/store";
   import { useFragno } from "./client.svelte";
 
+  // oxlint-disable-next-line no-unassigned-vars
   export let clientObj: Record<string, unknown>;
+  // oxlint-disable-next-line no-unassigned-vars
   export let hookName: string;
   export let args: Record<string, unknown> = {};
 

package/src/client/internal/ndjson-streaming.ts

@@ -191,7 +191,9 @@ async function continueStreaming<TOutputSchema extends StandardSchemaV1, TErrorC
         if (buffer.trim()) {
           const lines = buffer.split("\n");
           for (const line of lines) {
-            if (!line.trim()) continue;
+            if (!line.trim()) {
+              continue;
+            }
 
             try {
               const jsonObject = JSON.parse(line) as StandardSchemaV1.InferOutput<TOutputSchema>;
@@ -215,7 +217,9 @@ async function continueStreaming<TOutputSchema extends StandardSchemaV1, TErrorC
       buffer = lines.pop() || ""; // Keep incomplete line in buffer
 
       for (const line of lines) {
-        if (!line.trim()) continue;
+        if (!line.trim()) {
+          continue;
+        }
 
         try {
           const jsonObject = JSON.parse(line) as StandardSchemaV1.InferOutput<TOutputSchema>;

package/src/client/react.ts

@@ -210,7 +210,9 @@ export function useStore<SomeStore extends Store>(
 
   const subscribe = useCallback((onChange: () => void) => {
     const emitChange = (value: StoreValue<SomeStore>) => {
-      if (snapshotRef.current === value) return;
+      if (snapshotRef.current === value) {
+        return;
+      }
       snapshotRef.current = value;
       onChange();
     };

package/src/util/async.test.ts

@@ -20,7 +20,9 @@ describe("createAsyncIteratorFromCallback", () => {
       const values: string[] = [];
       for await (const value of iterator) {
         values.push(value);
-        if (values.length === 3) break;
+        if (values.length === 3) {
+          break;
+        }
       }
       return values;
     })();
@@ -53,7 +55,9 @@ describe("createAsyncIteratorFromCallback", () => {
       const values: string[] = [];
       for await (const value of iterator) {
         values.push(value);
-        if (values.length === 2) break; // Break after 2 values
+        if (values.length === 2) {
+          break;
+        } // Break after 2 values
       }
       return values;
     })();