tokenmeter 0.9.0

package/dist/integrations/next/index.js

@@ -0,0 +1,170 @@
+/**
+ * Next.js Integration
+ *
+ * Provides helpers for App Router route handlers to automatically
+ * set up cost tracking context.
+ */
+import { withAttributes } from "../../context.js";
+/**
+ * Default context extractor that reads from common headers
+ */
+function defaultContextExtractor(request, options) {
+    const userIdHeader = options.userIdHeader ?? "x-user-id";
+    const orgIdHeader = options.orgIdHeader ?? "x-org-id";
+    const workflowIdHeader = options.workflowIdHeader ?? "x-request-id";
+    return {
+        userId: request.headers.get(userIdHeader) ?? undefined,
+        orgId: request.headers.get(orgIdHeader) ?? undefined,
+        workflowId: request.headers.get(workflowIdHeader) ?? crypto.randomUUID(),
+    };
+}
+/**
+ * Convert TokenMeterContext to attributes
+ */
+function contextToAttributes(ctx) {
+    const attrs = {};
+    if (ctx.userId) {
+        attrs["user.id"] = ctx.userId;
+    }
+    if (ctx.orgId) {
+        attrs["org.id"] = ctx.orgId;
+    }
+    if (ctx.workflowId) {
+        attrs["workflow.id"] = ctx.workflowId;
+    }
+    if (ctx.metadata) {
+        for (const [key, value] of Object.entries(ctx.metadata)) {
+            attrs[key] = value;
+        }
+    }
+    return attrs;
+}
+/**
+ * Wrap a Next.js App Router route handler with tokenmeter context.
+ *
+ * All AI calls made with `monitor()`-wrapped clients within the handler
+ * will automatically be attributed to the user/org extracted from the request.
+ *
+ * @example
+ * ```typescript
+ * // app/api/chat/route.ts
+ * import { withTokenmeter } from 'tokenmeter/next';
+ * import { monitor } from 'tokenmeter';
+ * import OpenAI from 'openai';
+ *
+ * const openai = monitor(new OpenAI());
+ *
+ * export const POST = withTokenmeter(async (req) => {
+ *   const { message } = await req.json();
+ *
+ *   // This call is automatically attributed to the user
+ *   const result = await openai.chat.completions.create({
+ *     model: 'gpt-4o',
+ *     messages: [{ role: 'user', content: message }],
+ *   });
+ *
+ *   return Response.json({ reply: result.choices[0].message.content });
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // With custom context extraction (e.g., from auth)
+ * import { auth } from '@clerk/nextjs/server';
+ *
+ * export const POST = withTokenmeter(
+ *   async (req) => {
+ *     // Handler code
+ *   },
+ *   {
+ *     getContext: async (req) => {
+ *       const { userId, orgId } = await auth();
+ *       return { userId, orgId };
+ *     },
+ *   }
+ * );
+ * ```
+ */
+export function withTokenmeter(handler, options = {}) {
+    return async (request, routeContext) => {
+        // Extract context
+        let ctx;
+        if (options.getContext) {
+            ctx = await options.getContext(request);
+        }
+        else {
+            ctx = defaultContextExtractor(request, options);
+        }
+        // Convert to attributes
+        const attrs = contextToAttributes(ctx);
+        // Execute handler within context
+        return withAttributes(attrs, async () => {
+            return handler(request, routeContext);
+        });
+    };
+}
+/**
+ * Create a reusable tokenmeter wrapper with preset options.
+ *
+ * Useful when you have common context extraction logic across multiple routes.
+ *
+ * @example
+ * ```typescript
+ * // lib/tokenmeter.ts
+ * import { createTokenmeterWrapper } from 'tokenmeter/next';
+ * import { auth } from '@clerk/nextjs/server';
+ *
+ * export const withCostTracking = createTokenmeterWrapper({
+ *   getContext: async (req) => {
+ *     const { userId, orgId } = await auth();
+ *     return { userId, orgId };
+ *   },
+ * });
+ *
+ * // app/api/chat/route.ts
+ * import { withCostTracking } from '@/lib/tokenmeter';
+ *
+ * export const POST = withCostTracking(async (req) => {
+ *   // ...
+ * });
+ * ```
+ */
+export function createTokenmeterWrapper(defaultOptions) {
+    return (handler, options) => {
+        return withTokenmeter(handler, { ...defaultOptions, ...options });
+    };
+}
+/**
+ * Helper to extract common headers from a request
+ */
+export const headerExtractors = {
+    /**
+     * Extract user ID from x-user-id header
+     */
+    userId: (request) => {
+        return request.headers.get("x-user-id") ?? undefined;
+    },
+    /**
+     * Extract org ID from x-org-id header
+     */
+    orgId: (request) => {
+        return request.headers.get("x-org-id") ?? undefined;
+    },
+    /**
+     * Extract API key from Authorization header (Bearer token)
+     */
+    apiKey: (request) => {
+        const auth = request.headers.get("authorization");
+        if (auth?.startsWith("Bearer ")) {
+            return auth.slice(7);
+        }
+        return undefined;
+    },
+    /**
+     * Extract request ID from x-request-id header
+     */
+    requestId: (request) => {
+        return request.headers.get("x-request-id") ?? undefined;
+    },
+};
+//# sourceMappingURL=index.js.map

package/dist/integrations/next/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/integrations/next/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AA8ClD;;GAEG;AACH,SAAS,uBAAuB,CAC9B,OAAgB,EAChB,OAA8B;IAE9B,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,WAAW,CAAC;IACzD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,UAAU,CAAC;IACtD,MAAM,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,cAAc,CAAC;IAEpE,OAAO;QACL,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,SAAS;QACtD,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,SAAS;QACpD,UAAU,EACR,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,MAAM,CAAC,UAAU,EAAE;KAC/D,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,GAAsB;IACjD,MAAM,KAAK,GAAyB,EAAE,CAAC;IAEvC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QACf,KAAK,CAAC,SAAS,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC;IAChC,CAAC;IACD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;QACd,KAAK,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;IAC9B,CAAC;IACD,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;QACnB,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,UAAU,CAAC;IACxC,CAAC;IACD,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QACjB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxD,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACrB,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6CG;AACH,MAAM,UAAU,cAAc,CAC5B,OAAqB,EACrB,UAAiC,EAAE;IAEnC,OAAO,KAAK,EAAE,OAAgB,EAAE,YAA2D,EAAE,EAAE;QAC7F,kBAAkB;QAClB,IAAI,GAAsB,CAAC;QAC3B,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,GAAG,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;aAAM,CAAC;YACN,GAAG,GAAG,uBAAuB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAClD,CAAC;QAED,wBAAwB;QACxB,MAAM,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAEvC,iCAAiC;QACjC,OAAO,cAAc,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;YACtC,OAAO,OAAO,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAM,UAAU,uBAAuB,CACrC,cAAqC;IAErC,OAAO,CAAC,OAAqB,EAAE,OAA+B,EAAE,EAAE;QAChE,OAAO,cAAc,CAAC,OAAO,EAAE,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B;;OAEG;IACH,MAAM,EAAE,CAAC,OAAgB,EAAsB,EAAE;QAC/C,OAAO,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,SAAS,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,KAAK,EAAE,CAAC,OAAgB,EAAsB,EAAE;QAC9C,OAAO,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,SAAS,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,MAAM,EAAE,CAAC,OAAgB,EAAsB,EAAE;QAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAClD,IAAI,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,SAAS,EAAE,CAAC,OAAgB,EAAsB,EAAE;QAClD,OAAO,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,SAAS,CAAC;IAC1D,CAAC;CACF,CAAC"}

package/dist/integrations/nextjs/index.d.ts

@@ -0,0 +1,198 @@
+import type { CostTraceOptions } from "../../types.js";
+/**
+ * Minimal type definitions for Next.js server types.
+ * These are defined inline to avoid requiring Next.js as a hard dependency.
+ */
+/** Minimal NextRequest interface */
+export interface NextRequest {
+    headers: Headers;
+    nextUrl: {
+        pathname: string;
+        searchParams: URLSearchParams;
+    };
+    method: string;
+    url: string;
+}
+/** Minimal NextResponse static methods */
+export interface NextResponseStatic {
+    next: (init?: {
+        headers?: Headers;
+    }) => NextResponse;
+    redirect: (url: string | URL, status?: number) => NextResponse;
+    rewrite: (url: string | URL) => NextResponse;
+    json: (body: unknown, init?: ResponseInit) => NextResponse;
+}
+/** Minimal NextResponse interface */
+export interface NextResponse extends Response {
+    cookies: {
+        set: (name: string, value: string, options?: unknown) => void;
+        delete: (name: string) => void;
+    };
+}
+/** Next.js middleware function type */
+export type NextMiddleware = (request: NextRequest) => Promise<NextResponse | Response | undefined> | NextResponse | Response | undefined;
+/**
+ * Configuration for the Next.js middleware
+ */
+export interface TokenmeterMiddlewareConfig {
+    /**
+     * Extract the identifier from the request.
+     * This is required and should return a user ID, org ID, or other billing identifier.
+     *
+     * @example
+     * ```typescript
+     * getIdentifier: (req) => req.headers.get('x-user-id') || 'anonymous'
+     * ```
+     */
+    getIdentifier: (req: NextRequest) => string | null | undefined | Promise<string | null | undefined>;
+    /**
+     * Optional: Extract a secondary identifier from the request.
+     *
+     * @example
+     * ```typescript
+     * getSecondaryIdentifier: (req) => req.headers.get('x-org-id')
+     * ```
+     */
+    getSecondaryIdentifier?: (req: NextRequest) => string | null | undefined | Promise<string | null | undefined>;
+    /**
+     * Optional: Extract or generate a workflow ID from the request.
+     * Defaults to using the x-request-id header or generating a UUID.
+     */
+    getWorkflowId?: (req: NextRequest) => string | null | undefined | Promise<string | null | undefined>;
+    /**
+     * Optional: Extract or specify a workflow type from the request.
+     */
+    getWorkflowType?: (req: NextRequest) => string | null | undefined | Promise<string | null | undefined>;
+    /**
+     * Optional: Extract additional metadata from the request.
+     */
+    getMetadata?: (req: NextRequest) => Record<string, unknown> | undefined | Promise<Record<string, unknown> | undefined>;
+    /**
+     * Optional: Filter which routes should have cost tracking enabled.
+     * Return true to enable tracking for the request, false to skip.
+     * Defaults to tracking all routes.
+     *
+     * @example
+     * ```typescript
+     * shouldTrack: (req) => req.nextUrl.pathname.startsWith('/api/')
+     * ```
+     */
+    shouldTrack?: (req: NextRequest) => boolean;
+}
+/**
+ * Wrap a Next.js middleware function with cost tracking context.
+ *
+ * This is the recommended way to add tokenmeter to Next.js middleware.
+ * It wraps your existing middleware and sets up cost tracking context
+ * before your middleware runs.
+ *
+ * @param middleware - Your existing Next.js middleware function
+ * @param config - Configuration for extracting identifiers and metadata
+ * @returns A wrapped middleware function
+ *
+ * @example
+ * ```typescript
+ * // middleware.ts
+ * import { NextResponse } from 'next/server';
+ * import type { NextRequest } from 'next/server';
+ * import { wrapMiddleware } from 'tokenmeter/nextjs';
+ *
+ * function middleware(request: NextRequest) {
+ *   // Your middleware logic here
+ *   return NextResponse.next();
+ * }
+ *
+ * export default wrapMiddleware(middleware, {
+ *   getIdentifier: (req) => req.headers.get('x-user-id'),
+ *   shouldTrack: (req) => req.nextUrl.pathname.startsWith('/api/'),
+ * });
+ * ```
+ */
+export declare function wrapMiddleware(middleware: NextMiddleware, config: TokenmeterMiddlewareConfig): NextMiddleware;
+/**
+ * Create a standalone Next.js middleware for cost tracking.
+ *
+ * Use this when you want tokenmeter to be the only middleware,
+ * or when composing middleware manually.
+ *
+ * NOTE: This middleware only sets up the tracking context. The actual
+ * AI calls must be made in API routes that run in the same request context.
+ * Due to Next.js Edge Runtime limitations, the context may not propagate
+ * to API routes. For API routes, use `withTokenmeter` instead.
+ *
+ * @param config - Configuration for extracting identifiers and metadata
+ * @param NextResponse - The NextResponse class from 'next/server'
+ * @returns A Next.js middleware function
+ *
+ * @example
+ * ```typescript
+ * // middleware.ts
+ * import { NextResponse } from 'next/server';
+ * import { createTokenmeterMiddleware } from 'tokenmeter/nextjs';
+ *
+ * export default createTokenmeterMiddleware({
+ *   getIdentifier: (req) => req.headers.get('x-user-id'),
+ *   shouldTrack: (req) => req.nextUrl.pathname.startsWith('/api/'),
+ * }, NextResponse);
+ * ```
+ */
+export declare function createTokenmeterMiddleware(config: TokenmeterMiddlewareConfig, NextResponse: NextResponseStatic): NextMiddleware;
+/**
+ * Higher-order function to wrap a Next.js API route handler with cost tracking.
+ *
+ * Use this for App Router route handlers when you need more control than middleware.
+ * This is the RECOMMENDED approach for tracking AI costs in API routes.
+ *
+ * @param handler - The route handler function
+ * @param getOptions - Function to extract trace options from the request
+ * @returns A wrapped route handler
+ *
+ * @example
+ * ```typescript
+ * // app/api/chat/route.ts
+ * import { withTokenmeter } from 'tokenmeter/nextjs';
+ * import { auth } from '@/auth';
+ *
+ * async function handler(req: NextRequest) {
+ *   // Your AI calls here are automatically tracked
+ *   const result = await openai.chat.completions.create({ ... });
+ *   return Response.json({ result });
+ * }
+ *
+ * export const POST = withTokenmeter(handler, async (req) => {
+ *   const session = await auth();
+ *   return {
+ *     identifier: session?.user?.id || 'anonymous',
+ *     workflowType: 'chat',
+ *   };
+ * });
+ * ```
+ */
+export declare function withTokenmeter<T extends NextRequest>(handler: (req: T, ...args: unknown[]) => Promise<Response>, getOptions: (req: T) => CostTraceOptions | Promise<CostTraceOptions>): (req: T, ...args: unknown[]) => Promise<Response>;
+/**
+ * Helper to extract common identifiers from Next.js request headers.
+ */
+export declare const headerExtractors: {
+    /**
+     * Get identifier from x-user-id header
+     */
+    userId: (req: NextRequest) => string | null;
+    /**
+     * Get identifier from x-org-id header
+     */
+    orgId: (req: NextRequest) => string | null;
+    /**
+     * Get identifier from x-api-key header
+     */
+    apiKey: (req: NextRequest) => string | null;
+    /**
+     * Get identifier from Authorization header (Bearer token)
+     */
+    bearerToken: (req: NextRequest) => string | null;
+    /**
+     * Get request ID from common headers
+     */
+    requestId: (req: NextRequest) => string | null;
+};
+export default wrapMiddleware;
+//# sourceMappingURL=index.d.ts.map

package/dist/integrations/nextjs/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/integrations/nextjs/index.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAEvD;;;GAGG;AAEH,oCAAoC;AACpC,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE;QACP,QAAQ,EAAE,MAAM,CAAC;QACjB,YAAY,EAAE,eAAe,CAAC;KAC/B,CAAC;IACF,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,EAAE,MAAM,CAAC;CACb;AAED,0CAA0C;AAC1C,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,OAAO,CAAA;KAAE,KAAK,YAAY,CAAC;IACrD,QAAQ,EAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,EAAE,MAAM,CAAC,EAAE,MAAM,KAAK,YAAY,CAAC;IAC/D,OAAO,EAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,KAAK,YAAY,CAAC;IAC7C,IAAI,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,YAAY,KAAK,YAAY,CAAC;CAC5D;AAED,qCAAqC;AACrC,MAAM,WAAW,YAAa,SAAQ,QAAQ;IAC5C,OAAO,EAAE;QACP,GAAG,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;QAC9D,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;KAChC,CAAC;CACH;AAED,uCAAuC;AACvC,MAAM,MAAM,cAAc,GAAG,CAC3B,OAAO,EAAE,WAAW,KAElB,OAAO,CAAC,YAAY,GAAG,QAAQ,GAAG,SAAS,CAAC,GAC5C,YAAY,GACZ,QAAQ,GACR,SAAS,CAAC;AAEd;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC;;;;;;;;OAQG;IACH,aAAa,EAAE,CACb,GAAG,EAAE,WAAW,KACb,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC,CAAC;IAEpE;;;;;;;OAOG;IACH,sBAAsB,CAAC,EAAE,CACvB,GAAG,EAAE,WAAW,KACb,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC,CAAC;IAEpE;;;OAGG;IACH,aAAa,CAAC,EAAE,CACd,GAAG,EAAE,WAAW,KACb,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC,CAAC;IAEpE;;OAEG;IACH,eAAe,CAAC,EAAE,CAChB,GAAG,EAAE,WAAW,KACb,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC,CAAC;IAEpE;;OAEG;IACH,WAAW,CAAC,EAAE,CACZ,GAAG,EAAE,WAAW,KAEd,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACvB,SAAS,GACT,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;IAEjD;;;;;;;;;OASG;IACH,WAAW,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,KAAK,OAAO,CAAC;CAC7C;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,wBAAgB,cAAc,CAC5B,UAAU,EAAE,cAAc,EAC1B,MAAM,EAAE,0BAA0B,GACjC,cAAc,CAgDhB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,0BAA0B,EAClC,YAAY,EAAE,kBAAkB,GAC/B,cAAc,CAEhB;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAgB,cAAc,CAAC,CAAC,SAAS,WAAW,EAClD,OAAO,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,QAAQ,CAAC,EAC1D,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC,GACnE,CAAC,GAAG,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,QAAQ,CAAC,CAmBnD;AAED;;GAEG;AACH,eAAO,MAAM,gBAAgB;IAC3B;;OAEG;kBACW,WAAW;IAEzB;;OAEG;iBACU,WAAW;IAExB;;OAEG;kBACW,WAAW;IAEzB;;OAEG;uBACgB,WAAW;IAQ9B;;OAEG;qBACc,WAAW;CAI7B,CAAC;AAEF,eAAe,cAAc,CAAC"}

package/dist/integrations/nextjs/index.js

@@ -0,0 +1,181 @@
+import { withCostTrace } from "../../context.js";
+import { isInitialized } from "../../config.js";
+/**
+ * Wrap a Next.js middleware function with cost tracking context.
+ *
+ * This is the recommended way to add tokenmeter to Next.js middleware.
+ * It wraps your existing middleware and sets up cost tracking context
+ * before your middleware runs.
+ *
+ * @param middleware - Your existing Next.js middleware function
+ * @param config - Configuration for extracting identifiers and metadata
+ * @returns A wrapped middleware function
+ *
+ * @example
+ * ```typescript
+ * // middleware.ts
+ * import { NextResponse } from 'next/server';
+ * import type { NextRequest } from 'next/server';
+ * import { wrapMiddleware } from 'tokenmeter/nextjs';
+ *
+ * function middleware(request: NextRequest) {
+ *   // Your middleware logic here
+ *   return NextResponse.next();
+ * }
+ *
+ * export default wrapMiddleware(middleware, {
+ *   getIdentifier: (req) => req.headers.get('x-user-id'),
+ *   shouldTrack: (req) => req.nextUrl.pathname.startsWith('/api/'),
+ * });
+ * ```
+ */
+export function wrapMiddleware(middleware, config) {
+    return async function tokenmeterMiddleware(request) {
+        // Check if tokenmeter is initialized
+        if (!isInitialized()) {
+            console.warn("[tokenmeter] Not initialized. Call init() before using middleware.");
+            return middleware(request);
+        }
+        // Check if we should track this request
+        if (config.shouldTrack && !config.shouldTrack(request)) {
+            return middleware(request);
+        }
+        // Get identifier (may be async)
+        const identifier = await config.getIdentifier(request);
+        if (!identifier) {
+            // No identifier - skip tracking but continue request
+            return middleware(request);
+        }
+        // Build trace options (await any async getters)
+        const traceOptions = {
+            identifier,
+            secondaryIdentifier: (await config.getSecondaryIdentifier?.(request)) ?? undefined,
+            workflowId: (await config.getWorkflowId?.(request)) ??
+                request.headers.get("x-request-id") ??
+                crypto.randomUUID(),
+            workflowType: (await config.getWorkflowType?.(request)) ?? undefined,
+            requestId: request.headers.get("x-request-id") ?? undefined,
+            metadata: {
+                ...(await config.getMetadata?.(request)),
+                path: request.nextUrl.pathname,
+                method: request.method,
+            },
+        };
+        // Wrap the middleware execution in a cost trace context
+        return withCostTrace(traceOptions, async () => {
+            const result = await middleware(request);
+            return result;
+        });
+    };
+}
+/**
+ * Create a standalone Next.js middleware for cost tracking.
+ *
+ * Use this when you want tokenmeter to be the only middleware,
+ * or when composing middleware manually.
+ *
+ * NOTE: This middleware only sets up the tracking context. The actual
+ * AI calls must be made in API routes that run in the same request context.
+ * Due to Next.js Edge Runtime limitations, the context may not propagate
+ * to API routes. For API routes, use `withTokenmeter` instead.
+ *
+ * @param config - Configuration for extracting identifiers and metadata
+ * @param NextResponse - The NextResponse class from 'next/server'
+ * @returns A Next.js middleware function
+ *
+ * @example
+ * ```typescript
+ * // middleware.ts
+ * import { NextResponse } from 'next/server';
+ * import { createTokenmeterMiddleware } from 'tokenmeter/nextjs';
+ *
+ * export default createTokenmeterMiddleware({
+ *   getIdentifier: (req) => req.headers.get('x-user-id'),
+ *   shouldTrack: (req) => req.nextUrl.pathname.startsWith('/api/'),
+ * }, NextResponse);
+ * ```
+ */
+export function createTokenmeterMiddleware(config, NextResponse) {
+    return wrapMiddleware(() => NextResponse.next(), config);
+}
+/**
+ * Higher-order function to wrap a Next.js API route handler with cost tracking.
+ *
+ * Use this for App Router route handlers when you need more control than middleware.
+ * This is the RECOMMENDED approach for tracking AI costs in API routes.
+ *
+ * @param handler - The route handler function
+ * @param getOptions - Function to extract trace options from the request
+ * @returns A wrapped route handler
+ *
+ * @example
+ * ```typescript
+ * // app/api/chat/route.ts
+ * import { withTokenmeter } from 'tokenmeter/nextjs';
+ * import { auth } from '@/auth';
+ *
+ * async function handler(req: NextRequest) {
+ *   // Your AI calls here are automatically tracked
+ *   const result = await openai.chat.completions.create({ ... });
+ *   return Response.json({ result });
+ * }
+ *
+ * export const POST = withTokenmeter(handler, async (req) => {
+ *   const session = await auth();
+ *   return {
+ *     identifier: session?.user?.id || 'anonymous',
+ *     workflowType: 'chat',
+ *   };
+ * });
+ * ```
+ */
+export function withTokenmeter(handler, getOptions) {
+    return async (req, ...args) => {
+        if (!isInitialized()) {
+            console.warn("[tokenmeter] Not initialized. Call init() before using withTokenmeter.");
+            return handler(req, ...args);
+        }
+        const options = await getOptions(req);
+        if (!options.identifier) {
+            return handler(req, ...args);
+        }
+        return withCostTrace(options, async () => {
+            return handler(req, ...args);
+        });
+    };
+}
+/**
+ * Helper to extract common identifiers from Next.js request headers.
+ */
+export const headerExtractors = {
+    /**
+     * Get identifier from x-user-id header
+     */
+    userId: (req) => req.headers.get("x-user-id"),
+    /**
+     * Get identifier from x-org-id header
+     */
+    orgId: (req) => req.headers.get("x-org-id"),
+    /**
+     * Get identifier from x-api-key header
+     */
+    apiKey: (req) => req.headers.get("x-api-key"),
+    /**
+     * Get identifier from Authorization header (Bearer token)
+     */
+    bearerToken: (req) => {
+        const auth = req.headers.get("authorization");
+        if (auth?.startsWith("Bearer ")) {
+            return auth.slice(7);
+        }
+        return null;
+    },
+    /**
+     * Get request ID from common headers
+     */
+    requestId: (req) => req.headers.get("x-request-id") ||
+        req.headers.get("x-correlation-id") ||
+        req.headers.get("x-trace-id"),
+};
+export default wrapMiddleware;
+//# sourceMappingURL=index.js.map

package/dist/integrations/nextjs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/integrations/nextjs/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AA+GhD;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,UAAU,cAAc,CAC5B,UAA0B,EAC1B,MAAkC;IAElC,OAAO,KAAK,UAAU,oBAAoB,CACxC,OAAoB;QAEpB,qCAAqC;QACrC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,IAAI,CACV,oEAAoE,CACrE,CAAC;YACF,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QAED,wCAAwC;QACxC,IAAI,MAAM,CAAC,WAAW,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC;YACvD,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QAED,gCAAgC;QAChC,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACvD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,qDAAqD;YACrD,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QAED,gDAAgD;QAChD,MAAM,YAAY,GAAqB;YACrC,UAAU;YACV,mBAAmB,EACjB,CAAC,MAAM,MAAM,CAAC,sBAAsB,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,SAAS;YAC/D,UAAU,EACR,CAAC,MAAM,MAAM,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC;gBACvC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;gBACnC,MAAM,CAAC,UAAU,EAAE;YACrB,YAAY,EAAE,CAAC,MAAM,MAAM,CAAC,eAAe,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,SAAS;YACpE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,SAAS;YAC3D,QAAQ,EAAE;gBACR,GAAG,CAAC,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,CAAC;gBACxC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;aACvB;SACF,CAAC;QAEF,wDAAwD;QACxD,OAAO,aAAa,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC;YACzC,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,MAAM,UAAU,0BAA0B,CACxC,MAAkC,EAClC,YAAgC;IAEhC,OAAO,cAAc,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC;AAC3D,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,cAAc,CAC5B,OAA0D,EAC1D,UAAoE;IAEpE,OAAO,KAAK,EAAE,GAAM,EAAE,GAAG,IAAe,EAAqB,EAAE;QAC7D,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,IAAI,CACV,wEAAwE,CACzE,CAAC;YACF,OAAO,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC;QAEtC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,OAAO,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,aAAa,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;YACvC,OAAO,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B;;OAEG;IACH,MAAM,EAAE,CAAC,GAAgB,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAE1D;;OAEG;IACH,KAAK,EAAE,CAAC,GAAgB,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAExD;;OAEG;IACH,MAAM,EAAE,CAAC,GAAgB,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAE1D;;OAEG;IACH,WAAW,EAAE,CAAC,GAAgB,EAAE,EAAE;QAChC,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAC9C,IAAI,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAChC,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,SAAS,EAAE,CAAC,GAAgB,EAAE,EAAE,CAC9B,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAC/B,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QACnC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;CAChC,CAAC;AAEF,eAAe,cAAc,CAAC"}