54 lines
1.1 KiB
Plaintext
54 lines
1.1 KiB
Plaintext
import { AsyncLocalStorage } from "node:async_hooks";
|
|
//#region src/middleware/context-storage/index.ts
|
|
/**
|
|
* @module
|
|
* Context Storage Middleware for Hono.
|
|
*/
|
|
const asyncLocalStorage = new AsyncLocalStorage();
|
|
/**
|
|
* Context Storage Middleware for Hono.
|
|
*
|
|
* @see {@link https://hono.dev/docs/middleware/builtin/context-storage}
|
|
*
|
|
* @returns {MiddlewareHandler} The middleware handler function.
|
|
*
|
|
* @example
|
|
* ```ts
|
|
* type Env = {
|
|
* Variables: {
|
|
* message: string
|
|
* }
|
|
* }
|
|
*
|
|
* const app = new Hono<Env>()
|
|
*
|
|
* app.use(contextStorage())
|
|
*
|
|
* app.use(async (c, next) => {
|
|
* c.set('message', 'Hono is hot!!')
|
|
* await next()
|
|
* })
|
|
*
|
|
* app.get('/', async (c) => c.text(getMessage()))
|
|
*
|
|
* const getMessage = () => {
|
|
* return getContext<Env>().var.message
|
|
* }
|
|
* ```
|
|
*/
|
|
const contextStorage = () => {
|
|
return async function contextStorage(c, next) {
|
|
await asyncLocalStorage.run(c, next);
|
|
};
|
|
};
|
|
const tryGetContext = () => {
|
|
return asyncLocalStorage.getStore();
|
|
};
|
|
const getContext = () => {
|
|
const context = tryGetContext();
|
|
if (!context) throw new Error("Context is not available");
|
|
return context;
|
|
};
|
|
//#endregion
|
|
export { contextStorage, getContext, tryGetContext };
|