@voidhash/mimic-effect 0.0.4 → 0.0.6

package/dist/MimicServer.d.cts

@@ -1,7 +1,8 @@
-import { MimicServerConfigOptions } from "./MimicConfig.cjs";
+import { InitialFn, MimicServerConfigOptions } from "./MimicConfig.cjs";
 import { MimicDataStorageTag } from "./MimicDataStorage.cjs";
 import { DocumentManagerTag } from "./DocumentManager.cjs";
 import { MimicAuthServiceTag } from "./MimicAuthService.cjs";
+import * as Effect from "effect/Effect";
 import * as Layer from "effect/Layer";
 import { Presence, Primitive } from "@voidhash/mimic";
 import { HttpLayerRouter } from "@effect/platform";
@@ -9,7 +10,7 @@ import { PathInput } from "@effect/platform/HttpRouter";
 
 //#region src/MimicServer.d.ts
 declare namespace MimicServer_d_exports {
-  export { MimicLayerOptions, documentManagerLayer, layerHttpLayerRouter };
+  export { MimicLayerOptions, MimicLayerRouterOptions, documentManagerLayer, layerHttpLayerRouter };
 }
 /**
  * Options for creating a Mimic server layer.
@@ -36,19 +37,33 @@ interface MimicLayerOptions<TSchema extends Primitive.AnyPrimitive> {
   readonly presence?: Presence.AnyPresence;
   /**
    * Initial state for new documents.
-   * Used when a document is created and no existing state is found in storage.
+   * Can be either:
+   * - A plain object with the initial state values
+   * - A function that receives context (with documentId) and returns an Effect producing the initial state
+   *
+   * When using a function that requires Effect services (has R requirements),
+   * you must also provide `initialLayer` to supply those dependencies.
    *
    * Type-safe: required fields (without defaults) must be provided,
    * while optional fields and fields with defaults can be omitted.
    *
    * @default undefined (documents start empty or use schema defaults)
    */
-  readonly initial?: Primitive.InferSetInput<TSchema>;
+  readonly initial?: Primitive.InferSetInput<TSchema> | InitialFn<TSchema>;
 }
 /**
  * Create the document manager layer.
  */
 declare const documentManagerLayer: <TSchema extends Primitive.AnyPrimitive>(options: MimicServerConfigOptions<TSchema>) => Layer.Layer<DocumentManagerTag>;
+/**
+ * Options for layerHttpLayerRouter including optional custom layers.
+ */
+interface MimicLayerRouterOptions<TSchema extends Primitive.AnyPrimitive> extends MimicLayerOptions<TSchema> {
+  /** Custom auth layer. Defaults to NoAuth (all connections allowed). */
+  readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;
+  /** Custom storage layer. Defaults to InMemoryDataStorage. */
+  readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;
+}
 /**
  * Create a Mimic server layer that integrates with HttpLayerRouter.
  *
@@ -95,12 +110,7 @@ declare const documentManagerLayer: <TSchema extends Primitive.AnyPrimitive>(opt
  * );
  * ```
  */
-declare const layerHttpLayerRouter: <TSchema extends Primitive.AnyPrimitive>(options: MimicLayerOptions<TSchema> & {
-  /** Custom auth layer. Defaults to NoAuth (all connections allowed). */
-  readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;
-  /** Custom storage layer. Defaults to InMemoryDataStorage. */
-  readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;
-}) => Layer.Layer<never, never, HttpLayerRouter.HttpRouter>;
+declare const layerHttpLayerRouter: <TSchema extends Primitive.AnyPrimitive, TError, TRequirements>(optionsEf: Effect.Effect<MimicLayerRouterOptions<TSchema>, TError, TRequirements>) => Layer.Layer<never, TError, TRequirements | HttpLayerRouter.HttpRouter>;
 //#endregion
 export { MimicServer_d_exports };
 //# sourceMappingURL=MimicServer.d.cts.map

package/dist/MimicServer.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"MimicServer.d.cts","names":[],"sources":["../src/MimicServer.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;UA6BiB,kCAAkC,SAAA,CAAU;;;;;EAA5C,SAAA,QAAA,CAAA,EAKK,SALY;EAAiB;;;EAmB7B,SAAS,MAAA,EAVZ,OAUY;EAUc;;;AAO7C;EAAqD,SAAU,qBAAA,CAAA,EAAA,MAAA;EACf;;;;EAClC,SAAA,QAAA,CAAA,EAnBQ,QAAA,CAAS,WAmBjB;EAyGD;;;;;;;;;EAMV,SAAA,OAAA,CAAA,EAxHkB,SAAA,CAAU,aAwH5B,CAxH0C,OAwH1C,CAAA;;;;;cAjHU,uCAAwC,SAAA,CAAU,uBACpD,yBAAqC,aAC7C,KAAA,CAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAyGF,uCAAwC,SAAA,CAAU,uBACpD,kBAAkB;;uBAEJ,KAAA,CAAM,MAAM;;0BAET,KAAA,CAAM,MAAM;MACrC,KAAA,CAAA,oBAAA,eAAA,CAAA"}
+{"version":3,"file":"MimicServer.d.cts","names":[],"sources":["../src/MimicServer.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;;UA6BiB,kCACC,SAAA,CAAU;;AAD5B;;;EAWmB,SAAA,QAAA,CAAA,EAJG,SAIH;EAUG;;;EAewD,SAAA,MAAA,EAzB3D,OAyB2D;EAAtB;;AAOxD;;EACgD,SAAA,qBAAA,CAAA,EAAA,MAAA;EAArC;;;;EA+DM,SAAA,QAAA,CAAA,EAtFK,QAAA,CAAS,WAsFS;EAAiB;;;;;;;;AAsDzD;;;;;;EAKa,SAAO,OAAA,CAAA,EAlIC,SAAA,CAAU,aAkIX,CAlIyB,OAkIzB,CAAA,GAlIoC,SAkIpC,CAlI0D,OAkI1D,CAAA;;;;;AACN,cA5HD,oBA4HC,EAAA,CAAA,gBA5HuC,SAAA,CAAU,YA4HjD,CAAA,CAAA,OAAA,EA3HH,wBA2HG,CA3HkC,OA2HlC,CAAA,EAAA,GA1HX,KAAA,CAAM,KA0HK,CA1HC,kBA0HD,CAAA;;;;UA5DG,wCAAwC,SAAA,CAAU,sBACzD,kBAAkB;;uBAEL,KAAA,CAAM,MAAM;;0BAET,KAAA,CAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAiDzB,uCACK,SAAA,CAAU,gDAIf,MAAA,CAAO,OAAO,wBAAwB,UAAU,QAAQ,mBAClE,KAAA,CAAM,aAAa,QAAQ,gBAAgB,eAAA,CAAgB"}

package/dist/MimicServer.d.mts

@@ -1,7 +1,8 @@
-import { MimicServerConfigOptions } from "./MimicConfig.mjs";
+import { InitialFn, MimicServerConfigOptions } from "./MimicConfig.mjs";
 import { MimicDataStorageTag } from "./MimicDataStorage.mjs";
 import { DocumentManagerTag } from "./DocumentManager.mjs";
 import { MimicAuthServiceTag } from "./MimicAuthService.mjs";
+import * as Effect from "effect/Effect";
 import * as Layer from "effect/Layer";
 import { Presence, Primitive } from "@voidhash/mimic";
 import { HttpLayerRouter } from "@effect/platform";
@@ -9,7 +10,7 @@ import { PathInput } from "@effect/platform/HttpRouter";
 
 //#region src/MimicServer.d.ts
 declare namespace MimicServer_d_exports {
-  export { MimicLayerOptions, documentManagerLayer, layerHttpLayerRouter };
+  export { MimicLayerOptions, MimicLayerRouterOptions, documentManagerLayer, layerHttpLayerRouter };
 }
 /**
  * Options for creating a Mimic server layer.
@@ -36,19 +37,33 @@ interface MimicLayerOptions<TSchema extends Primitive.AnyPrimitive> {
   readonly presence?: Presence.AnyPresence;
   /**
    * Initial state for new documents.
-   * Used when a document is created and no existing state is found in storage.
+   * Can be either:
+   * - A plain object with the initial state values
+   * - A function that receives context (with documentId) and returns an Effect producing the initial state
+   *
+   * When using a function that requires Effect services (has R requirements),
+   * you must also provide `initialLayer` to supply those dependencies.
    *
    * Type-safe: required fields (without defaults) must be provided,
    * while optional fields and fields with defaults can be omitted.
    *
    * @default undefined (documents start empty or use schema defaults)
    */
-  readonly initial?: Primitive.InferSetInput<TSchema>;
+  readonly initial?: Primitive.InferSetInput<TSchema> | InitialFn<TSchema>;
 }
 /**
  * Create the document manager layer.
  */
 declare const documentManagerLayer: <TSchema extends Primitive.AnyPrimitive>(options: MimicServerConfigOptions<TSchema>) => Layer.Layer<DocumentManagerTag>;
+/**
+ * Options for layerHttpLayerRouter including optional custom layers.
+ */
+interface MimicLayerRouterOptions<TSchema extends Primitive.AnyPrimitive> extends MimicLayerOptions<TSchema> {
+  /** Custom auth layer. Defaults to NoAuth (all connections allowed). */
+  readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;
+  /** Custom storage layer. Defaults to InMemoryDataStorage. */
+  readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;
+}
 /**
  * Create a Mimic server layer that integrates with HttpLayerRouter.
  *
@@ -95,12 +110,7 @@ declare const documentManagerLayer: <TSchema extends Primitive.AnyPrimitive>(opt
  * );
  * ```
  */
-declare const layerHttpLayerRouter: <TSchema extends Primitive.AnyPrimitive>(options: MimicLayerOptions<TSchema> & {
-  /** Custom auth layer. Defaults to NoAuth (all connections allowed). */
-  readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;
-  /** Custom storage layer. Defaults to InMemoryDataStorage. */
-  readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;
-}) => Layer.Layer<never, never, HttpLayerRouter.HttpRouter>;
+declare const layerHttpLayerRouter: <TSchema extends Primitive.AnyPrimitive, TError, TRequirements>(optionsEf: Effect.Effect<MimicLayerRouterOptions<TSchema>, TError, TRequirements>) => Layer.Layer<never, TError, TRequirements | HttpLayerRouter.HttpRouter>;
 //#endregion
 export { MimicServer_d_exports };
 //# sourceMappingURL=MimicServer.d.mts.map

package/dist/MimicServer.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"MimicServer.d.mts","names":[],"sources":["../src/MimicServer.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;UA6BiB,kCAAkC,SAAA,CAAU;;;;;EAA5C,SAAA,QAAA,CAAA,EAKK,SALY;EAAiB;;;EAmB7B,SAAS,MAAA,EAVZ,OAUY;EAUc;;;AAO7C;EAAqD,SAAU,qBAAA,CAAA,EAAA,MAAA;EACf;;;;EAClC,SAAA,QAAA,CAAA,EAnBQ,QAAA,CAAS,WAmBjB;EAyGD;;;;;;;;;EAMV,SAAA,OAAA,CAAA,EAxHkB,SAAA,CAAU,aAwH5B,CAxH0C,OAwH1C,CAAA;;;;;cAjHU,uCAAwC,SAAA,CAAU,uBACpD,yBAAqC,aAC7C,KAAA,CAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAyGF,uCAAwC,SAAA,CAAU,uBACpD,kBAAkB;;uBAEJ,KAAA,CAAM,MAAM;;0BAET,KAAA,CAAM,MAAM;MACrC,KAAA,CAAA,oBAAA,eAAA,CAAA"}
+{"version":3,"file":"MimicServer.d.mts","names":[],"sources":["../src/MimicServer.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;;UA6BiB,kCACC,SAAA,CAAU;;AAD5B;;;EAWmB,SAAA,QAAA,CAAA,EAJG,SAIH;EAUG;;;EAewD,SAAA,MAAA,EAzB3D,OAyB2D;EAAtB;;AAOxD;;EACgD,SAAA,qBAAA,CAAA,EAAA,MAAA;EAArC;;;;EA+DM,SAAA,QAAA,CAAA,EAtFK,QAAA,CAAS,WAsFS;EAAiB;;;;;;;;AAsDzD;;;;;;EAKa,SAAO,OAAA,CAAA,EAlIC,SAAA,CAAU,aAkIX,CAlIyB,OAkIzB,CAAA,GAlIoC,SAkIpC,CAlI0D,OAkI1D,CAAA;;;;;AACN,cA5HD,oBA4HC,EAAA,CAAA,gBA5HuC,SAAA,CAAU,YA4HjD,CAAA,CAAA,OAAA,EA3HH,wBA2HG,CA3HkC,OA2HlC,CAAA,EAAA,GA1HX,KAAA,CAAM,KA0HK,CA1HC,kBA0HD,CAAA;;;;UA5DG,wCAAwC,SAAA,CAAU,sBACzD,kBAAkB;;uBAEL,KAAA,CAAM,MAAM;;0BAET,KAAA,CAAM,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAiDzB,uCACK,SAAA,CAAU,gDAIf,MAAA,CAAO,OAAO,wBAAwB,UAAU,QAAQ,mBAClE,KAAA,CAAM,aAAa,QAAQ,gBAAgB,eAAA,CAAgB"}

package/dist/MimicServer.mjs

@@ -92,23 +92,26 @@ const makeMimicHandler = Effect.gen(function* () {
 * );
 * ```
 */
-const layerHttpLayerRouter = (options) => {
-	var _options$basePath, _options$authLayer, _options$storageLayer;
-	const wsPath = `${(_options$basePath = options.basePath) !== null && _options$basePath !== void 0 ? _options$basePath : "/mimic"}/doc/*`;
-	const configLayer = layer({
-		schema: options.schema,
-		maxTransactionHistory: options.maxTransactionHistory,
-		presence: options.presence,
-		initial: options.initial
-	});
-	const authLayer = (_options$authLayer = options.authLayer) !== null && _options$authLayer !== void 0 ? _options$authLayer : layerDefault$1;
-	const storageLayer = (_options$storageLayer = options.storageLayer) !== null && _options$storageLayer !== void 0 ? _options$storageLayer : layerDefault;
-	const registerRoute = Effect.gen(function* () {
-		const router = yield* HttpLayerRouter.HttpRouter;
-		const handler = yield* makeMimicHandler;
-		yield* router.add("GET", wsPath, handler);
-	});
-	return Layer.scopedDiscard(registerRoute).pipe(Layer.provide(layer$1), Layer.provide(layer$2), Layer.provide(configLayer), Layer.provide(storageLayer), Layer.provide(authLayer));
+const layerHttpLayerRouter = (optionsEf) => {
+	return Layer.unwrapScoped(Effect.gen(function* () {
+		var _options$basePath, _options$authLayer, _options$storageLayer;
+		const options = yield* optionsEf;
+		const wsPath = `${(_options$basePath = options.basePath) !== null && _options$basePath !== void 0 ? _options$basePath : "/mimic"}/doc/*`;
+		const configLayer = layer({
+			schema: options.schema,
+			maxTransactionHistory: options.maxTransactionHistory,
+			presence: options.presence,
+			initial: options.initial
+		});
+		const authLayer = (_options$authLayer = options.authLayer) !== null && _options$authLayer !== void 0 ? _options$authLayer : layerDefault$1;
+		const storageLayer = (_options$storageLayer = options.storageLayer) !== null && _options$storageLayer !== void 0 ? _options$storageLayer : layerDefault;
+		const depsLayer = Layer.mergeAll(configLayer, authLayer, storageLayer);
+		return Layer.scopedDiscard(Effect.gen(function* () {
+			const router = yield* HttpLayerRouter.HttpRouter;
+			const handler = yield* makeMimicHandler;
+			yield* router.add("GET", wsPath, handler);
+		})).pipe(Layer.provide(layer$1), Layer.provide(layer$2), Layer.provide(depsLayer));
+	}));
 };
 
 //#endregion

package/dist/MimicServer.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"MimicServer.mjs","names":["MimicConfig.layer","InMemoryDataStorage.layerDefault","NoAuth.layerDefault","MimicConfig.MimicServerConfigTag","DocumentManager.DocumentManagerTag","PresenceManager.PresenceManagerTag","WebSocketHandler.extractDocumentId","WebSocketHandler.handleConnection","wsPath: PathInput","DocumentManager.layer","PresenceManager.layer"],"sources":["../src/MimicServer.ts"],"sourcesContent":["/**\n * @since 0.0.1\n * Mimic server layer composition.\n */\nimport * as Effect from \"effect/Effect\";\nimport * as Layer from \"effect/Layer\";\nimport * as Context from \"effect/Context\";\nimport type * as Socket from \"@effect/platform/Socket\";\nimport { SocketServer } from \"@effect/platform/SocketServer\";\nimport type { Primitive, Presence } from \"@voidhash/mimic\";\n\nimport * as DocumentManager from \"./DocumentManager.js\";\nimport * as WebSocketHandler from \"./WebSocketHandler.js\";\nimport * as MimicConfig from \"./MimicConfig.js\";\nimport { MimicDataStorageTag } from \"./MimicDataStorage.js\";\nimport { MimicAuthServiceTag } from \"./MimicAuthService.js\";\nimport * as PresenceManager from \"./PresenceManager.js\";\nimport * as InMemoryDataStorage from \"./storage/InMemoryDataStorage.js\";\nimport * as NoAuth from \"./auth/NoAuth.js\";\nimport { HttpLayerRouter, HttpServerRequest, HttpServerResponse } from \"@effect/platform\";\nimport { PathInput } from \"@effect/platform/HttpRouter\";\n\n// =============================================================================\n// Layer Composition Options\n// =============================================================================\n\n/**\n * Options for creating a Mimic server layer.\n */\nexport interface MimicLayerOptions<TSchema extends Primitive.AnyPrimitive> {\n  /**\n   * Base path for document routes (used for path matching).\n   * @example \"/mimic/todo\" - documents accessed at \"/mimic/todo/:documentId\"\n   */\n  readonly basePath?: PathInput;\n  /**\n   * The schema defining the document structure.\n   */\n  readonly schema: TSchema;\n  /**\n   * Maximum number of processed transaction IDs to track for deduplication.\n   * @default 1000\n   */\n  readonly maxTransactionHistory?: number;\n  /**\n   * Optional presence schema for ephemeral per-user data.\n   * When provided, enables presence features on WebSocket connections.\n   */\n  readonly presence?: Presence.AnyPresence;\n  /**\n   * Initial state for new documents.\n   * Used when a document is created and no existing state is found in storage.\n   *\n   * Type-safe: required fields (without defaults) must be provided,\n   * while optional fields and fields with defaults can be omitted.\n   *\n   * @default undefined (documents start empty or use schema defaults)\n   */\n  readonly initial?: Primitive.InferSetInput<TSchema>;\n}\n\n\n/**\n * Create the document manager layer.\n */\nexport const documentManagerLayer = <TSchema extends Primitive.AnyPrimitive>(\n  options: MimicConfig.MimicServerConfigOptions<TSchema>\n): Layer.Layer<DocumentManager.DocumentManagerTag> =>\n  DocumentManager.layer.pipe(\n    Layer.provide(MimicConfig.layer(options)),\n    // Provide defaults\n    Layer.provide(InMemoryDataStorage.layerDefault),\n    Layer.provide(NoAuth.layerDefault)\n  );\n\n/**\n * Create the HTTP handler effect for WebSocket upgrade.\n * This handler:\n * 1. Extracts the document ID from the URL path\n * 2. Upgrades the HTTP connection to WebSocket\n * 3. Delegates to the WebSocketHandler for document sync\n */\nconst makeMimicHandler = Effect.gen(function* () {\n  const config = yield* MimicConfig.MimicServerConfigTag;\n  const authService = yield* MimicAuthServiceTag;\n  const documentManager = yield* DocumentManager.DocumentManagerTag;\n  const presenceManager = yield* PresenceManager.PresenceManagerTag;\n\n  return Effect.gen(function* () {\n    const request = yield* HttpServerRequest.HttpServerRequest;\n\n    // Extract document ID from the URL path\n    // Expected format: /basePath/doc/{documentId}\n    const documentId = yield* WebSocketHandler.extractDocumentId(request.url);\n\n    // Upgrade to WebSocket\n    const socket = yield* request.upgrade;\n\n    // Handle the WebSocket connection\n    yield* WebSocketHandler.handleConnection(socket, request.url).pipe(\n      Effect.provideService(MimicConfig.MimicServerConfigTag, config),\n      Effect.provideService(MimicAuthServiceTag, authService),\n      Effect.provideService(DocumentManager.DocumentManagerTag, documentManager),\n      Effect.provideService(PresenceManager.PresenceManagerTag, presenceManager),\n      Effect.scoped,\n      Effect.catchAll((error) =>\n        Effect.logError(\"WebSocket connection error\", error)\n      )\n    );\n\n    // Return empty response - the WebSocket upgrade handles the connection\n    return HttpServerResponse.empty();\n  }).pipe(\n    Effect.catchAll((error) =>\n      Effect.gen(function* () {\n        yield* Effect.logWarning(\"WebSocket upgrade failed\", error);\n        return HttpServerResponse.text(\"WebSocket upgrade failed\", {\n          status: 400,\n        });\n      })\n    )\n  );\n});\n\n\n\n/**\n * Create a Mimic server layer that integrates with HttpLayerRouter.\n *\n * This function creates a layer that:\n * 1. Registers a WebSocket route at the specified base path\n * 2. Handles WebSocket upgrades for document sync\n * 3. Provides all required dependencies (config, auth, storage, document manager)\n *\n * By default, uses in-memory storage and no authentication.\n * To override these defaults, provide custom layers before the defaults:\n *\n * @example\n * ```typescript\n * import { MimicServer, MimicAuthService } from \"@voidhash/mimic-effect\";\n * import { HttpLayerRouter } from \"@effect/platform\";\n * import { Primitive } from \"@voidhash/mimic\";\n *\n * const TodoSchema = Primitive.Struct({\n *   title: Primitive.String(),\n *   completed: Primitive.Boolean(),\n * });\n *\n * // Create the Mimic route layer with defaults\n * const MimicRoute = MimicServer.layerHttpLayerRouter({\n *   basePath: \"/mimic/todo\",\n *   schema: TodoSchema\n * });\n *\n * // Or with custom auth - use Layer.provide to inject before defaults\n * const MimicRouteWithAuth = MimicServer.layerHttpLayerRouter({\n *   basePath: \"/mimic/todo\",\n *   schema: TodoSchema,\n *   authLayer: MimicAuthService.layer({\n *     authHandler: (token) => ({ success: true, userId: token })\n *   })\n * });\n *\n * // Merge with other routes and serve\n * const AllRoutes = Layer.mergeAll(MimicRoute, OtherRoutes);\n * HttpLayerRouter.serve(AllRoutes).pipe(\n *   Layer.provide(BunHttpServer.layer({ port: 3000 })),\n *   Layer.launch,\n *   BunRuntime.runMain\n * );\n * ```\n */\nexport const layerHttpLayerRouter = <TSchema extends Primitive.AnyPrimitive>(\n  options: MimicLayerOptions<TSchema> & {\n    /** Custom auth layer. Defaults to NoAuth (all connections allowed). */\n    readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;\n    /** Custom storage layer. Defaults to InMemoryDataStorage. */\n    readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;\n  }\n) => {\n  // Build the base path pattern for WebSocket routes\n  // Append /doc/* to match /basePath/doc/{documentId}\n  const basePath = options.basePath ?? \"/mimic\";\n  const wsPath: PathInput = `${basePath}/doc/*` as PathInput;\n\n  // Create the config layer\n  const configLayer = MimicConfig.layer({\n    schema: options.schema,\n    maxTransactionHistory: options.maxTransactionHistory,\n    presence: options.presence,\n    initial: options.initial,\n  });\n\n  // Use provided layers or defaults\n  const authLayer = options.authLayer ?? NoAuth.layerDefault;\n  const storageLayer = options.storageLayer ?? InMemoryDataStorage.layerDefault;\n\n  // Create the route registration effect\n  const registerRoute = Effect.gen(function* () {\n    const router = yield* HttpLayerRouter.HttpRouter;\n    const handler = yield* makeMimicHandler;\n    yield* router.add(\"GET\", wsPath, handler);\n  });\n\n  // Build the layer with all dependencies\n  return Layer.scopedDiscard(registerRoute).pipe(\n    Layer.provide(DocumentManager.layer),\n    Layer.provide(PresenceManager.layer),\n    Layer.provide(configLayer),\n    Layer.provide(storageLayer),\n    Layer.provide(authLayer)\n  );\n};"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAiEA,MAAa,wBACX,oBAEsB,KACpB,MAAM,QAAQA,MAAkB,QAAQ,CAAC,EAEzC,MAAM,QAAQC,aAAiC,EAC/C,MAAM,QAAQC,eAAoB,CACnC;;;;;;;;AASH,MAAM,mBAAmB,OAAO,IAAI,aAAa;CAC/C,MAAM,SAAS,OAAOC;CACtB,MAAM,cAAc,OAAO;CAC3B,MAAM,kBAAkB,OAAOC;CAC/B,MAAM,kBAAkB,OAAOC;AAE/B,QAAO,OAAO,IAAI,aAAa;EAC7B,MAAM,UAAU,OAAO,kBAAkB;AAItB,SAAOC,kBAAmC,QAAQ,IAAI;EAGzE,MAAM,SAAS,OAAO,QAAQ;AAG9B,SAAOC,iBAAkC,QAAQ,QAAQ,IAAI,CAAC,KAC5D,OAAO,eAAeJ,sBAAkC,OAAO,EAC/D,OAAO,eAAe,qBAAqB,YAAY,EACvD,OAAO,eAAeC,oBAAoC,gBAAgB,EAC1E,OAAO,eAAeC,oBAAoC,gBAAgB,EAC1E,OAAO,QACP,OAAO,UAAU,UACf,OAAO,SAAS,8BAA8B,MAAM,CACrD,CACF;AAGD,SAAO,mBAAmB,OAAO;GACjC,CAAC,KACD,OAAO,UAAU,UACf,OAAO,IAAI,aAAa;AACtB,SAAO,OAAO,WAAW,4BAA4B,MAAM;AAC3D,SAAO,mBAAmB,KAAK,4BAA4B,EACzD,QAAQ,KACT,CAAC;GACF,CACH,CACF;EACD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDF,MAAa,wBACX,YAMG;;CAIH,MAAMG,SAAoB,wBADT,QAAQ,yEAAY,SACC;CAGtC,MAAM,cAAcR,MAAkB;EACpC,QAAQ,QAAQ;EAChB,uBAAuB,QAAQ;EAC/B,UAAU,QAAQ;EAClB,SAAS,QAAQ;EAClB,CAAC;CAGF,MAAM,kCAAY,QAAQ,4EAAaE;CACvC,MAAM,wCAAe,QAAQ,qFAAgBD;CAG7C,MAAM,gBAAgB,OAAO,IAAI,aAAa;EAC5C,MAAM,SAAS,OAAO,gBAAgB;EACtC,MAAM,UAAU,OAAO;AACvB,SAAO,OAAO,IAAI,OAAO,QAAQ,QAAQ;GACzC;AAGF,QAAO,MAAM,cAAc,cAAc,CAAC,KACxC,MAAM,QAAQQ,QAAsB,EACpC,MAAM,QAAQC,QAAsB,EACpC,MAAM,QAAQ,YAAY,EAC1B,MAAM,QAAQ,aAAa,EAC3B,MAAM,QAAQ,UAAU,CACzB"}
+{"version":3,"file":"MimicServer.mjs","names":["MimicConfig.layer","InMemoryDataStorage.layerDefault","NoAuth.layerDefault","MimicConfig.MimicServerConfigTag","DocumentManager.DocumentManagerTag","PresenceManager.PresenceManagerTag","WebSocketHandler.extractDocumentId","WebSocketHandler.handleConnection","wsPath: PathInput","DocumentManager.layer","PresenceManager.layer"],"sources":["../src/MimicServer.ts"],"sourcesContent":["/**\n * @since 0.0.1\n * Mimic server layer composition.\n */\nimport * as Effect from \"effect/Effect\";\nimport * as Layer from \"effect/Layer\";\nimport * as Context from \"effect/Context\";\nimport type * as Socket from \"@effect/platform/Socket\";\nimport { SocketServer } from \"@effect/platform/SocketServer\";\nimport type { Primitive, Presence } from \"@voidhash/mimic\";\n\nimport * as DocumentManager from \"./DocumentManager.js\";\nimport * as WebSocketHandler from \"./WebSocketHandler.js\";\nimport * as MimicConfig from \"./MimicConfig.js\";\nimport { MimicDataStorageTag } from \"./MimicDataStorage.js\";\nimport { MimicAuthServiceTag } from \"./MimicAuthService.js\";\nimport * as PresenceManager from \"./PresenceManager.js\";\nimport * as InMemoryDataStorage from \"./storage/InMemoryDataStorage.js\";\nimport * as NoAuth from \"./auth/NoAuth.js\";\nimport { HttpLayerRouter, HttpServerRequest, HttpServerResponse } from \"@effect/platform\";\nimport { PathInput } from \"@effect/platform/HttpRouter\";\n\n// =============================================================================\n// Layer Composition Options\n// =============================================================================\n\n/**\n * Options for creating a Mimic server layer.\n */\nexport interface MimicLayerOptions<\n  TSchema extends Primitive.AnyPrimitive,\n> {\n  /**\n   * Base path for document routes (used for path matching).\n   * @example \"/mimic/todo\" - documents accessed at \"/mimic/todo/:documentId\"\n   */\n  readonly basePath?: PathInput;\n  /**\n   * The schema defining the document structure.\n   */\n  readonly schema: TSchema;\n  /**\n   * Maximum number of processed transaction IDs to track for deduplication.\n   * @default 1000\n   */\n  readonly maxTransactionHistory?: number;\n  /**\n   * Optional presence schema for ephemeral per-user data.\n   * When provided, enables presence features on WebSocket connections.\n   */\n  readonly presence?: Presence.AnyPresence;\n  /**\n   * Initial state for new documents.\n   * Can be either:\n   * - A plain object with the initial state values\n   * - A function that receives context (with documentId) and returns an Effect producing the initial state\n   *\n   * When using a function that requires Effect services (has R requirements),\n   * you must also provide `initialLayer` to supply those dependencies.\n   *\n   * Type-safe: required fields (without defaults) must be provided,\n   * while optional fields and fields with defaults can be omitted.\n   *\n   * @default undefined (documents start empty or use schema defaults)\n   */\n  readonly initial?: Primitive.InferSetInput<TSchema> | MimicConfig.InitialFn<TSchema>;\n}\n\n\n/**\n * Create the document manager layer.\n */\nexport const documentManagerLayer = <TSchema extends Primitive.AnyPrimitive>(\n  options: MimicConfig.MimicServerConfigOptions<TSchema>\n): Layer.Layer<DocumentManager.DocumentManagerTag> =>\n  DocumentManager.layer.pipe(\n    Layer.provide(MimicConfig.layer(options)),\n    // Provide defaults\n    Layer.provide(InMemoryDataStorage.layerDefault),\n    Layer.provide(NoAuth.layerDefault)\n  );\n\n/**\n * Create the HTTP handler effect for WebSocket upgrade.\n * This handler:\n * 1. Extracts the document ID from the URL path\n * 2. Upgrades the HTTP connection to WebSocket\n * 3. Delegates to the WebSocketHandler for document sync\n */\nconst makeMimicHandler = Effect.gen(function* () {\n  const config = yield* MimicConfig.MimicServerConfigTag;\n  const authService = yield* MimicAuthServiceTag;\n  const documentManager = yield* DocumentManager.DocumentManagerTag;\n  const presenceManager = yield* PresenceManager.PresenceManagerTag;\n\n  return Effect.gen(function* () {\n    const request = yield* HttpServerRequest.HttpServerRequest;\n\n    // Extract document ID from the URL path\n    // Expected format: /basePath/doc/{documentId}\n    const documentId = yield* WebSocketHandler.extractDocumentId(request.url);\n\n    // Upgrade to WebSocket\n    const socket = yield* request.upgrade;\n\n    // Handle the WebSocket connection\n    yield* WebSocketHandler.handleConnection(socket, request.url).pipe(\n      Effect.provideService(MimicConfig.MimicServerConfigTag, config),\n      Effect.provideService(MimicAuthServiceTag, authService),\n      Effect.provideService(DocumentManager.DocumentManagerTag, documentManager),\n      Effect.provideService(PresenceManager.PresenceManagerTag, presenceManager),\n      Effect.scoped,\n      Effect.catchAll((error) =>\n        Effect.logError(\"WebSocket connection error\", error)\n      )\n    );\n\n    // Return empty response - the WebSocket upgrade handles the connection\n    return HttpServerResponse.empty();\n  }).pipe(\n    Effect.catchAll((error) =>\n      Effect.gen(function* () {\n        yield* Effect.logWarning(\"WebSocket upgrade failed\", error);\n        return HttpServerResponse.text(\"WebSocket upgrade failed\", {\n          status: 400,\n        });\n      })\n    )\n  );\n});\n\n\n\n/**\n * Options for layerHttpLayerRouter including optional custom layers.\n */\nexport interface MimicLayerRouterOptions<TSchema extends Primitive.AnyPrimitive>\n  extends MimicLayerOptions<TSchema> {\n  /** Custom auth layer. Defaults to NoAuth (all connections allowed). */\n  readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;\n  /** Custom storage layer. Defaults to InMemoryDataStorage. */\n  readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;\n}\n\n/**\n * Create a Mimic server layer that integrates with HttpLayerRouter.\n *\n * This function creates a layer that:\n * 1. Registers a WebSocket route at the specified base path\n * 2. Handles WebSocket upgrades for document sync\n * 3. Provides all required dependencies (config, auth, storage, document manager)\n *\n * By default, uses in-memory storage and no authentication.\n * To override these defaults, provide custom layers before the defaults:\n *\n * @example\n * ```typescript\n * import { MimicServer, MimicAuthService } from \"@voidhash/mimic-effect\";\n * import { HttpLayerRouter } from \"@effect/platform\";\n * import { Primitive } from \"@voidhash/mimic\";\n *\n * const TodoSchema = Primitive.Struct({\n *   title: Primitive.String(),\n *   completed: Primitive.Boolean(),\n * });\n *\n * // Create the Mimic route layer with defaults\n * const MimicRoute = MimicServer.layerHttpLayerRouter({\n *   basePath: \"/mimic/todo\",\n *   schema: TodoSchema\n * });\n *\n * // Or with custom auth - use Layer.provide to inject before defaults\n * const MimicRouteWithAuth = MimicServer.layerHttpLayerRouter({\n *   basePath: \"/mimic/todo\",\n *   schema: TodoSchema,\n *   authLayer: MimicAuthService.layer({\n *     authHandler: (token) => ({ success: true, userId: token })\n *   })\n * });\n *\n * // Merge with other routes and serve\n * const AllRoutes = Layer.mergeAll(MimicRoute, OtherRoutes);\n * HttpLayerRouter.serve(AllRoutes).pipe(\n *   Layer.provide(BunHttpServer.layer({ port: 3000 })),\n *   Layer.launch,\n *   BunRuntime.runMain\n * );\n * ```\n */\nexport const layerHttpLayerRouter = <\n  TSchema extends Primitive.AnyPrimitive,\n  TError,\n  TRequirements\n>(\n  optionsEf: Effect.Effect<MimicLayerRouterOptions<TSchema>, TError, TRequirements>\n): Layer.Layer<never, TError, TRequirements | HttpLayerRouter.HttpRouter> => {\n  return Layer.unwrapScoped(\n    Effect.gen(function* () {\n      const options = yield* optionsEf;\n\n      // Build the base path pattern for WebSocket routes\n      // Append /doc/* to match /basePath/doc/{documentId}\n      const basePath = options.basePath ?? \"/mimic\";\n      const wsPath: PathInput = `${basePath}/doc/*` as PathInput;\n\n      // Create the config layer with properly typed initial function\n      const configLayer = MimicConfig.layer<TSchema>({\n        schema: options.schema,\n        maxTransactionHistory: options.maxTransactionHistory,\n        presence: options.presence,\n        initial: options.initial,\n      });\n\n      // Use provided layers or defaults\n      const authLayer = options.authLayer ?? NoAuth.layerDefault;\n      const storageLayer = options.storageLayer ?? InMemoryDataStorage.layerDefault;\n\n      // Combine all dependency layers\n      const depsLayer = Layer.mergeAll(configLayer, authLayer, storageLayer);\n\n      // Create the route registration layer\n      const routeLayer = Layer.scopedDiscard(\n        Effect.gen(function* () {\n          const router = yield* HttpLayerRouter.HttpRouter;\n          const handler = yield* makeMimicHandler;\n          yield* router.add(\"GET\", wsPath, handler);\n        })\n      );\n\n      // Build the complete layer with all dependencies provided\n      return routeLayer.pipe(\n        Layer.provide(DocumentManager.layer),\n        Layer.provide(PresenceManager.layer),\n        Layer.provide(depsLayer),\n      );\n    })\n  );\n};"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAwEA,MAAa,wBACX,oBAEsB,KACpB,MAAM,QAAQA,MAAkB,QAAQ,CAAC,EAEzC,MAAM,QAAQC,aAAiC,EAC/C,MAAM,QAAQC,eAAoB,CACnC;;;;;;;;AASH,MAAM,mBAAmB,OAAO,IAAI,aAAa;CAC/C,MAAM,SAAS,OAAOC;CACtB,MAAM,cAAc,OAAO;CAC3B,MAAM,kBAAkB,OAAOC;CAC/B,MAAM,kBAAkB,OAAOC;AAE/B,QAAO,OAAO,IAAI,aAAa;EAC7B,MAAM,UAAU,OAAO,kBAAkB;AAItB,SAAOC,kBAAmC,QAAQ,IAAI;EAGzE,MAAM,SAAS,OAAO,QAAQ;AAG9B,SAAOC,iBAAkC,QAAQ,QAAQ,IAAI,CAAC,KAC5D,OAAO,eAAeJ,sBAAkC,OAAO,EAC/D,OAAO,eAAe,qBAAqB,YAAY,EACvD,OAAO,eAAeC,oBAAoC,gBAAgB,EAC1E,OAAO,eAAeC,oBAAoC,gBAAgB,EAC1E,OAAO,QACP,OAAO,UAAU,UACf,OAAO,SAAS,8BAA8B,MAAM,CACrD,CACF;AAGD,SAAO,mBAAmB,OAAO;GACjC,CAAC,KACD,OAAO,UAAU,UACf,OAAO,IAAI,aAAa;AACtB,SAAO,OAAO,WAAW,4BAA4B,MAAM;AAC3D,SAAO,mBAAmB,KAAK,4BAA4B,EACzD,QAAQ,KACT,CAAC;GACF,CACH,CACF;EACD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6DF,MAAa,wBAKX,cAC2E;AAC3E,QAAO,MAAM,aACX,OAAO,IAAI,aAAa;;EACtB,MAAM,UAAU,OAAO;EAKvB,MAAMG,SAAoB,wBADT,QAAQ,yEAAY,SACC;EAGtC,MAAM,cAAcR,MAA2B;GAC7C,QAAQ,QAAQ;GAChB,uBAAuB,QAAQ;GAC/B,UAAU,QAAQ;GAClB,SAAS,QAAQ;GAClB,CAAC;EAGF,MAAM,kCAAY,QAAQ,4EAAaE;EACvC,MAAM,wCAAe,QAAQ,qFAAgBD;EAG7C,MAAM,YAAY,MAAM,SAAS,aAAa,WAAW,aAAa;AAYtE,SATmB,MAAM,cACvB,OAAO,IAAI,aAAa;GACtB,MAAM,SAAS,OAAO,gBAAgB;GACtC,MAAM,UAAU,OAAO;AACvB,UAAO,OAAO,IAAI,OAAO,QAAQ,QAAQ;IACzC,CACH,CAGiB,KAChB,MAAM,QAAQQ,QAAsB,EACpC,MAAM,QAAQC,QAAsB,EACpC,MAAM,QAAQ,UAAU,CACzB;GACD,CACH"}

package/dist/PresenceManager.d.cts

@@ -1,5 +1,5 @@
-import * as Layer from "effect/Layer";
 import * as Effect from "effect/Effect";
+import * as Layer from "effect/Layer";
 import * as Context from "effect/Context";
 import * as Scope from "effect/Scope";
 import * as Stream from "effect/Stream";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@voidhash/mimic-effect",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "type": "module",
   "repository": {
     "type": "git",
@@ -24,11 +24,11 @@
     "typescript": "5.8.3",
     "vite-tsconfig-paths": "^5.1.4",
     "vitest": "^3.2.4",
-    "@voidhash/tsconfig": "0.0.4"
+    "@voidhash/tsconfig": "0.0.6"
   },
   "peerDependencies": {
     "effect": "^3.19.12",
-    "@voidhash/mimic": "0.0.4"
+    "@voidhash/mimic": "0.0.6"
   },
   "scripts": {
     "build": "tsdown",

package/src/DocumentManager.ts

@@ -113,10 +113,12 @@ const makeDocumentManager = Effect.gen(function* () {
         () => Effect.succeed(undefined)
       );
 
-      // Transform loaded state with onLoad hook, or use configured initial state for new docs
+      // Transform loaded state with onLoad hook, or compute initial state for new docs
       const initialState = rawState !== undefined
         ? yield* storage.onLoad(rawState)
-        : config.initial;
+        : config.initial !== undefined
+          ? yield* config.initial({ documentId })
+          : undefined;
 
       // Create PubSub for broadcasting
       const pubsub = yield* PubSub.unbounded<Protocol.ServerBroadcast>();
@@ -124,7 +126,7 @@ const makeDocumentManager = Effect.gen(function* () {
       // Create ServerDocument with broadcast callback
       const serverDocument = ServerDocument.make({
         schema: config.schema,
-        initialState: initialState as Primitive.InferState<typeof config.schema> | undefined,
+        initialState: initialState as Primitive.InferSetInput<typeof config.schema> | undefined,
         maxTransactionHistory: config.maxTransactionHistory,
         onBroadcast: (transactionMessage) => {
           // Get current state and save to storage

package/src/MimicConfig.ts

@@ -5,9 +5,32 @@
 import * as Context from "effect/Context";
 import * as Duration from "effect/Duration";
 import type { DurationInput } from "effect/Duration";
+import * as Effect from "effect/Effect";
 import * as Layer from "effect/Layer";
 import { Primitive, Presence } from "@voidhash/mimic";
 
+// =============================================================================
+// Initial State Types
+// =============================================================================
+
+/**
+ * Context available when computing initial state for a document.
+ */
+export interface InitialContext {
+  /**
+   * The document ID being initialized.
+   */
+  readonly documentId: string;
+}
+
+/**
+ * Function that computes initial state for a document.
+ * Receives context with the document ID and returns an Effect that produces the initial state.
+ */
+export type InitialFn<TSchema extends Primitive.AnyPrimitive> = (
+  context: InitialContext
+) => Effect.Effect<Primitive.InferSetInput<TSchema>>;
+
 // =============================================================================
 // Mimic Server Configuration
 // =============================================================================
@@ -18,7 +41,9 @@ import { Primitive, Presence } from "@voidhash/mimic";
  * Note: Authentication and persistence are now handled by injectable services
  * (MimicAuthService and MimicDataStorage) rather than config options.
  */
-export interface MimicServerConfig<TSchema extends Primitive.AnyPrimitive = Primitive.AnyPrimitive> {
+export interface MimicServerConfig<
+  TSchema extends Primitive.AnyPrimitive = Primitive.AnyPrimitive,
+> {
   /**
    * The schema defining the document structure.
    */
@@ -56,17 +81,20 @@ export interface MimicServerConfig<TSchema extends Primitive.AnyPrimitive = Prim
   readonly presence: Presence.AnyPresence | undefined;
 
   /**
-   * Initial state for new documents.
-   * Used when a document is created and no existing state is found in storage.
-   * @default undefined (documents start empty)
+   * Initial state function for new documents.
+   * Called when a document is created and no existing state is found in storage.
+   * Receives the document ID and returns an Effect that produces the initial state.
+   * @default undefined (documents start empty or use schema defaults)
    */
-  readonly initial: Primitive.InferState<TSchema> | undefined;
+  readonly initial: InitialFn<TSchema> | undefined;
 }
 
 /**
  * Options for creating a MimicServerConfig.
  */
-export interface MimicServerConfigOptions<TSchema extends Primitive.AnyPrimitive = Primitive.AnyPrimitive> {
+export interface MimicServerConfigOptions<
+  TSchema extends Primitive.AnyPrimitive = Primitive.AnyPrimitive,
+> {
   /**
    * The schema defining the document structure.
    */
@@ -105,32 +133,63 @@ export interface MimicServerConfigOptions<TSchema extends Primitive.AnyPrimitive
 
   /**
    * Initial state for new documents.
-   * Used when a document is created and no existing state is found in storage.
+   * Can be either:
+   * - A plain object with the initial state values
+   * - A function that receives context (with documentId) and returns an Effect producing the initial state
    *
    * Type-safe: required fields (without defaults) must be provided,
    * while optional fields and fields with defaults can be omitted.
    *
+   * @example
+   * // Plain object
+   * initial: { title: "New Document", count: 0 }
+   *
+   * @example
+   * // Function returning Effect
+   * initial: ({ documentId }) => Effect.succeed({ title: `Doc ${documentId}`, count: 0 })
+   *
    * @default undefined (documents start empty or use schema defaults)
    */
-  readonly initial?: Primitive.InferSetInput<TSchema>;
+  readonly initial?: Primitive.InferSetInput<TSchema> | InitialFn<TSchema>;
 }
 
+/**
+ * Check if a value is an InitialFn (function) rather than a plain object.
+ */
+const isInitialFn = <TSchema extends Primitive.AnyPrimitive>(
+  value: Primitive.InferSetInput<TSchema> | InitialFn<TSchema> | undefined
+): value is InitialFn<TSchema> => typeof value === "function";
+
 /**
  * Create a MimicServerConfig from options.
  */
 export const make = <TSchema extends Primitive.AnyPrimitive>(
   options: MimicServerConfigOptions<TSchema>
-): MimicServerConfig<TSchema> => ({
-  schema: options.schema,
-  maxIdleTime: Duration.decode(options.maxIdleTime ?? "5 minutes"),
-  maxTransactionHistory: options.maxTransactionHistory ?? 1000,
-  heartbeatInterval: Duration.decode(options.heartbeatInterval ?? "30 seconds"),
-  heartbeatTimeout: Duration.decode(options.heartbeatTimeout ?? "10 seconds"),
-  presence: options.presence,
-  initial: options.initial !== undefined
-    ? Primitive.applyDefaults(options.schema, options.initial as Partial<Primitive.InferState<TSchema>>)
-    : undefined,
-});
+): MimicServerConfig<TSchema> => {
+  const { initial, schema } = options;
+
+  // Convert initial to a function that applies defaults
+  const initialFn: InitialFn<TSchema> | undefined = initial === undefined
+    ? undefined
+    : isInitialFn<TSchema>(initial)
+      ? (context) => Effect.map(
+          initial(context),
+          (state) => Primitive.applyDefaults(schema, state as Partial<Primitive.InferState<TSchema>>)
+        ) as Effect.Effect<Primitive.InferSetInput<TSchema>>
+      : () => Effect.succeed(
+          Primitive.applyDefaults(schema, initial as Partial<Primitive.InferState<TSchema>>)
+        ) as Effect.Effect<Primitive.InferSetInput<TSchema>>;
+
+  return {
+    schema,
+    maxIdleTime: Duration.decode(options.maxIdleTime ?? "5 minutes"),
+    maxTransactionHistory: options.maxTransactionHistory ?? 1000,
+    heartbeatInterval: Duration.decode(options.heartbeatInterval ?? "30 seconds"),
+    heartbeatTimeout: Duration.decode(options.heartbeatTimeout ?? "10 seconds"),
+    presence: options.presence,
+    initial: initialFn,
+  };
+};
 
 // =============================================================================
 // Context Tag
@@ -149,4 +208,4 @@ export class MimicServerConfigTag extends Context.Tag(
 export const layer = <TSchema extends Primitive.AnyPrimitive>(
   options: MimicServerConfigOptions<TSchema>
 ): Layer.Layer<MimicServerConfigTag> =>
-  Layer.succeed(MimicServerConfigTag, make(options));
+  Layer.succeed(MimicServerConfigTag, make(options) as unknown as MimicServerConfig);

package/src/MimicServer.ts

@@ -27,7 +27,9 @@ import { PathInput } from "@effect/platform/HttpRouter";
 /**
  * Options for creating a Mimic server layer.
  */
-export interface MimicLayerOptions<TSchema extends Primitive.AnyPrimitive> {
+export interface MimicLayerOptions<
+  TSchema extends Primitive.AnyPrimitive,
+> {
   /**
    * Base path for document routes (used for path matching).
    * @example "/mimic/todo" - documents accessed at "/mimic/todo/:documentId"
@@ -49,14 +51,19 @@ export interface MimicLayerOptions<TSchema extends Primitive.AnyPrimitive> {
   readonly presence?: Presence.AnyPresence;
   /**
    * Initial state for new documents.
-   * Used when a document is created and no existing state is found in storage.
+   * Can be either:
+   * - A plain object with the initial state values
+   * - A function that receives context (with documentId) and returns an Effect producing the initial state
+   *
+   * When using a function that requires Effect services (has R requirements),
+   * you must also provide `initialLayer` to supply those dependencies.
    *
    * Type-safe: required fields (without defaults) must be provided,
    * while optional fields and fields with defaults can be omitted.
    *
    * @default undefined (documents start empty or use schema defaults)
    */
-  readonly initial?: Primitive.InferSetInput<TSchema>;
+  readonly initial?: Primitive.InferSetInput<TSchema> | MimicConfig.InitialFn<TSchema>;
 }
 
 
@@ -124,6 +131,17 @@ const makeMimicHandler = Effect.gen(function* () {
 
 
 
+/**
+ * Options for layerHttpLayerRouter including optional custom layers.
+ */
+export interface MimicLayerRouterOptions<TSchema extends Primitive.AnyPrimitive>
+  extends MimicLayerOptions<TSchema> {
+  /** Custom auth layer. Defaults to NoAuth (all connections allowed). */
+  readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;
+  /** Custom storage layer. Defaults to InMemoryDataStorage. */
+  readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;
+}
+
 /**
  * Create a Mimic server layer that integrates with HttpLayerRouter.
  *
@@ -170,44 +188,52 @@ const makeMimicHandler = Effect.gen(function* () {
  * );
  * ```
  */
-export const layerHttpLayerRouter = <TSchema extends Primitive.AnyPrimitive>(
-  options: MimicLayerOptions<TSchema> & {
-    /** Custom auth layer. Defaults to NoAuth (all connections allowed). */
-    readonly authLayer?: Layer.Layer<MimicAuthServiceTag>;
-    /** Custom storage layer. Defaults to InMemoryDataStorage. */
-    readonly storageLayer?: Layer.Layer<MimicDataStorageTag>;
-  }
-) => {
-  // Build the base path pattern for WebSocket routes
-  // Append /doc/* to match /basePath/doc/{documentId}
-  const basePath = options.basePath ?? "/mimic";
-  const wsPath: PathInput = `${basePath}/doc/*` as PathInput;
-
-  // Create the config layer
-  const configLayer = MimicConfig.layer({
-    schema: options.schema,
-    maxTransactionHistory: options.maxTransactionHistory,
-    presence: options.presence,
-    initial: options.initial,
-  });
-
-  // Use provided layers or defaults
-  const authLayer = options.authLayer ?? NoAuth.layerDefault;
-  const storageLayer = options.storageLayer ?? InMemoryDataStorage.layerDefault;
-
-  // Create the route registration effect
-  const registerRoute = Effect.gen(function* () {
-    const router = yield* HttpLayerRouter.HttpRouter;
-    const handler = yield* makeMimicHandler;
-    yield* router.add("GET", wsPath, handler);
-  });
-
-  // Build the layer with all dependencies
-  return Layer.scopedDiscard(registerRoute).pipe(
-    Layer.provide(DocumentManager.layer),
-    Layer.provide(PresenceManager.layer),
-    Layer.provide(configLayer),
-    Layer.provide(storageLayer),
-    Layer.provide(authLayer)
+export const layerHttpLayerRouter = <
+  TSchema extends Primitive.AnyPrimitive,
+  TError,
+  TRequirements
+>(
+  optionsEf: Effect.Effect<MimicLayerRouterOptions<TSchema>, TError, TRequirements>
+): Layer.Layer<never, TError, TRequirements | HttpLayerRouter.HttpRouter> => {
+  return Layer.unwrapScoped(
+    Effect.gen(function* () {
+      const options = yield* optionsEf;
+
+      // Build the base path pattern for WebSocket routes
+      // Append /doc/* to match /basePath/doc/{documentId}
+      const basePath = options.basePath ?? "/mimic";
+      const wsPath: PathInput = `${basePath}/doc/*` as PathInput;
+
+      // Create the config layer with properly typed initial function
+      const configLayer = MimicConfig.layer<TSchema>({
+        schema: options.schema,
+        maxTransactionHistory: options.maxTransactionHistory,
+        presence: options.presence,
+        initial: options.initial,
+      });
+
+      // Use provided layers or defaults
+      const authLayer = options.authLayer ?? NoAuth.layerDefault;
+      const storageLayer = options.storageLayer ?? InMemoryDataStorage.layerDefault;
+
+      // Combine all dependency layers
+      const depsLayer = Layer.mergeAll(configLayer, authLayer, storageLayer);
+
+      // Create the route registration layer
+      const routeLayer = Layer.scopedDiscard(
+        Effect.gen(function* () {
+          const router = yield* HttpLayerRouter.HttpRouter;
+          const handler = yield* makeMimicHandler;
+          yield* router.add("GET", wsPath, handler);
+        })
+      );
+
+      // Build the complete layer with all dependencies provided
+      return routeLayer.pipe(
+        Layer.provide(DocumentManager.layer),
+        Layer.provide(PresenceManager.layer),
+        Layer.provide(depsLayer),
+      );
+    })
   );
 };