@serverkgg/bridge 0.1.3 → 0.1.5

package/src/runtime/serve.ts

@@ -98,6 +98,14 @@ export const serve = async (driver: BridgeDriver) => {
 
 	const setup = createSetupHandlers(driver, log, redact);
 
+	const runHook = (hook: string, run: () => Promise<void>) => {
+		void run().catch((error: unknown) => {
+			log.warn(`lifecycle ${hook} hook failed`, {
+				error: errorText(error),
+			});
+		});
+	};
+
 	const supervisor = createSupervisor({
 		cwd: config.cwd,
 		uid: config.gameUid,
@@ -108,6 +116,8 @@ export const serve = async (driver: BridgeDriver) => {
 		subject: driver.lifecycle.subject,
 		ready: driver.lifecycle.ready,
 		failed: driver.lifecycle.failed,
+		graceSeconds: driver.lifecycle.graceSeconds,
+		stableLine: driver.lifecycle.stableLine,
 		redact,
 		onLog: log,
 		async resolveCommand() {
@@ -124,20 +134,31 @@ export const serve = async (driver: BridgeDriver) => {
 			await driver.lifecycle.stop?.(context);
 		},
 		onReady() {
-			void Promise.resolve(driver.lifecycle.onReady?.(context)).catch((error: unknown) => {
-				log.warn("lifecycle ready hook failed", {
-					error: errorText(error),
-				});
+			ring.discard();
+
+			runHook("ready", async () => {
+				await driver.lifecycle.onReady?.(context);
 			});
 		},
 		onStable() {
 			ring.discard();
+
+			runHook("stable", async () => {
+				await driver.lifecycle.onStable?.(context);
+			});
 		},
 		onStopped() {
 			ring.discard();
 		},
 		onExit(record) {
-			ring.crash(record);
+			const cause = ring.crash(record);
+
+			runHook("exit", async () => {
+				await driver.lifecycle.onExit?.(context, {
+					...record,
+					cause,
+				});
+			});
 		},
 	});