@genkit-ai/core/lib/context.mjs

Genkit AI framework core libraries.

import { runInActionRuntimeContext } from "./action.js";
import { getAsyncContext } from "./async-context.js";
import { UserFacingError } from "./error.js";
const contextAlsKey = "core.auth.context";
function runWithContext(context, fn) {
  if (context === void 0) {
    return fn();
  }
  return getAsyncContext().run(
    contextAlsKey,
    context,
    () => runInActionRuntimeContext(fn)
  );
}
function getContext() {
  return getAsyncContext().getStore(contextAlsKey);
}
function apiKey(valueOrPolicy) {
  return async (request) => {
    const context = {
      auth: { apiKey: request.headers["authorization"] }
    };
    if (typeof valueOrPolicy === "string") {
      if (!context.auth?.apiKey) {
        console.error("THROWING UNAUTHENTICATED");
        throw new UserFacingError("UNAUTHENTICATED", "Unauthenticated");
      }
      if (context.auth?.apiKey != valueOrPolicy) {
        console.error("Throwing PERMISSION_DENIED");
        throw new UserFacingError("PERMISSION_DENIED", "Permission Denied");
      }
    } else if (typeof valueOrPolicy === "function") {
      await valueOrPolicy(context);
    } else if (typeof valueOrPolicy !== "undefined") {
      throw new Error(
        `Invalid type ${typeof valueOrPolicy} passed to apiKey()`
      );
    }
    return context;
  };
}
export {
  apiKey,
  getContext,
  runWithContext
};
//# sourceMappingURL=context.mjs.map