feat: server and client init hook (#13103)

* feat: server and client `init` hook

* fix: provide client_hooks fallback even if universal hooks are present

* chore: add tests

* chore: cache server instance in `vite dev`

* chore: remove only (duh)

* chore: revert single server in dev

* chore: revert weird did_it_run thing

* chore: add additional inited check to prevent multiple init in case of race conditions

* fix: remove `init` from client hooks

* docs: write some documentation, use proper exported types

* chore: regenerate types

* simplify

* chore: docs suggestions from review

Co-authored-by: Rich Harris <[email protected]>

* try this

* chore: fix lint

* chore: fix types

* save a couple bytes

* Apply suggestions from code review

---------

Co-authored-by: Rich Harris <[email protected]>
Co-authored-by: Rich Harris <[email protected]>
Co-authored-by: Simon H <[email protected]>

Author: 3 people

.changeset/eight-poems-learn.md

@@ -0,0 +1,5 @@
+---
+'@sveltejs/kit': minor
+---
+
+feat: server and client `init` hook

documentation/docs/30-advanced/20-hooks.md

@@ -237,6 +237,25 @@ During development, if an error occurs because of a syntax error in your Svelte
 
 > [!NOTE] Make sure that `handleError` _never_ throws an error
 
+### init
+
+This function runs once, when the server is created or the app starts in the browser, and is a useful place to do asynchronous work such as initializing a database connection.
+
+> [!NOTE] If your environment supports top-level await, the `init` function is really no different from writing your initialisation logic at the top level of the module, but some environments — most notably, Safari — don't.
+
+```js
+/// file: src/hooks.server.js
+import * as db from '$lib/server/database';
+
+/** @type {import('@sveltejs/kit').ServerInit} */
+export async function init() {
+	await db.connect();
+}
+```
+
+> [!NOTE]
+> In the browser, asynchronous work in `init` will delay hydration, so be mindful of what you put in there.
+
 ## Universal hooks
 
 The following can be added to `src/hooks.js`. Universal hooks run on both server and client (not to be confused with shared hooks, which are environment-specific).

packages/kit/src/core/sync/write_client_manifest.js

@@ -151,6 +151,7 @@ export function write_client_manifest(kit, manifest_data, output, metadata) {
 				handleError: ${
 					client_hooks_file ? 'client_hooks.handleError || ' : ''
 				}(({ error }) => { console.error(error) }),
+				${client_hooks_file ? 'init: client_hooks.init,' : ''}
 
 				reroute: ${universal_hooks_file ? 'universal_hooks.reroute || ' : ''}(() => {})
 			};

packages/kit/src/exports/public.d.ts

@@ -724,6 +724,16 @@ export type HandleFetch = (input: {
 	fetch: typeof fetch;
 }) => MaybePromise<Response>;
 
+/**
+ * The [`init`](https://svelte.dev/docs/kit/hooks#Shared-hooks-init) will be invoked before the server responds to its first request
+ */
+export type ServerInit = () => MaybePromise<void>;
+
+/**
+ * The [`init`](https://svelte.dev/docs/kit/hooks#Shared-hooks-init) will be invoked once the app starts in the browser
+ */
+export type ClientInit = () => MaybePromise<void>;
+
 /**
  * The [`reroute`](https://svelte.dev/docs/kit/hooks#Universal-hooks-reroute) hook allows you to modify the URL before it is used to determine which route to render.
  * @since 2.3.0

packages/kit/src/runtime/client/client.js

@@ -266,6 +266,9 @@ export async function start(_app, _target, hydrate) {
 	}
 
 	app = _app;
+
+	await _app.hooks.init?.();
+
 	routes = parse(_app);
 	container = __SVELTEKIT_EMBEDDED__ ? _target : document.documentElement;
 	target = _target;

packages/kit/src/runtime/server/index.js

@@ -15,6 +15,9 @@ const prerender_env_handler = {
 	}
 };
 
+/** @type {Promise<any>} */
+let init_promise;
+
 export class Server {
 	/** @type {import('types').SSROptions} */
 	#options;
@@ -63,7 +66,9 @@ export class Server {
 			set_read_implementation(read);
 		}
 
-		if (!this.#options.hooks) {
+		// During DEV and for some adapters this function might be called in quick succession,
+		// so we need to make sure we're not invoking this logic (most notably the init hook) multiple times
+		await (init_promise ??= (async () => {
 			try {
 				const module = await get_hooks();
 
@@ -73,6 +78,10 @@ export class Server {
 					handleFetch: module.handleFetch || (({ request, fetch }) => fetch(request)),
 					reroute: module.reroute || (() => {})
 				};
+
+				if (module.init) {
+					await module.init();
+				}
 			} catch (error) {
 				if (DEV) {
 					this.#options.hooks = {
@@ -87,7 +96,7 @@ export class Server {
 					throw error;
 				}
 			}
-		}
+		})());
 	}
 
 	/**

packages/kit/src/types/internal.d.ts

@@ -17,7 +17,9 @@ import {
 	RequestEvent,
 	SSRManifest,
 	Emulator,
-	Adapter
+	Adapter,
+	ServerInit,
+	ClientInit
 } from '@sveltejs/kit';
 import {
 	HttpMethod,
@@ -109,11 +111,13 @@ export interface ServerHooks {
 	handle: Handle;
 	handleError: HandleServerError;
 	reroute: Reroute;
+	init?: ServerInit;
 }
 
 export interface ClientHooks {
 	handleError: HandleClientError;
 	reroute: Reroute;
+	init?: ClientInit;
 }
 
 export interface Env {

packages/kit/test/apps/basics/src/hooks.client.js

@@ -8,3 +8,7 @@ export function handleError({ error, event, status, message }) {
 		? undefined
 		: { message: `${/** @type {Error} */ (error).message} (${status} ${message})` };
 }
+
+export function init() {
+	console.log('init hooks.client.js');
+}

packages/kit/test/apps/basics/src/hooks.server.js

@@ -1,7 +1,8 @@
-import fs from 'node:fs';
-import { sequence } from '@sveltejs/kit/hooks';
 import { error, isHttpError, redirect } from '@sveltejs/kit';
+import { sequence } from '@sveltejs/kit/hooks';
+import fs from 'node:fs';
 import { COOKIE_NAME } from './routes/cookies/shared';
+import { _set_from_init } from './routes/init-hooks/+page.server';
 
 /**
  * Transform an error into a POJO, by copying its `name`, `message`
@@ -154,3 +155,7 @@ export async function handleFetch({ request, fetch }) {
 
 	return fetch(request);
 }
+
+export function init() {
+	_set_from_init();
+}

packages/kit/test/apps/basics/src/routes/init-hooks/+page.server.ts

@@ -0,0 +1,11 @@
+let did_init_run = 0;
+
+export function _set_from_init() {
+	did_init_run++;
+}
+
+export function load() {
+	return {
+		did_init_run
+	};
+}