@apicity/cost 0.2.0-alpha.1 → 0.2.1

package/dist/src/with-paid-gate.js

@@ -0,0 +1,91 @@
+import { dispatchWithPaidGate, createReplayStore, } from "./paygate.js";
+import { isPaidEndpoint } from "./paid-endpoints.js";
+/**
+ * HTTP-method roots that the walker will descend into. Properties outside
+ * this allowlist (sub-providers like `kie.veo`, schema collections like
+ * `kie.modelInputSchemas`, attached helpers like `kie.examples`) are
+ * returned by reference and never wrapped.
+ */
+const DEFAULT_ROOTS = ["post", "get", "delete", "patch", "put"];
+/**
+ * Walk a provider tree once and route every paid-endpoint leaf through
+ * `dispatchWithPaidGate`. Free leaves are returned untouched. Callable
+ * namespaces (functions with child properties — e.g. `xai.v1.models` is a
+ * function with `.languageModels` children) preserve all children and their
+ * `.schema` attachments.
+ *
+ * @param providerName  Provider identifier matching `PAID_ENDPOINTS`.
+ * @param tree          The provider object returned by the factory.
+ * @param opts          Optional roots allowlist and IO injection.
+ * @returns A new tree of the same shape; paid leaves accept `(req, approval?)`.
+ */
+export function withPaidGate(providerName, tree, opts) {
+    const roots = opts?.roots ?? DEFAULT_ROOTS;
+    // Normalize the config once so every gated dispatch on this provider instance
+    // shares a single replay store (single-use OTPs must be remembered across
+    // calls). A custom store passed by the code client is preserved.
+    const config = opts?.config
+        ? {
+            ...opts.config,
+            replayStore: opts.config.replayStore ?? createReplayStore(),
+        }
+        : undefined;
+    const out = {};
+    for (const [key, value] of Object.entries(tree)) {
+        if (roots.includes(key) && value && typeof value === "object") {
+            out[key] = walk(providerName, [key], value, config);
+        }
+        else {
+            out[key] = value;
+        }
+    }
+    return out;
+}
+function walk(providerName, path, node, config) {
+    if (typeof node === "function") {
+        return wrapCallable(providerName, path, node, config);
+    }
+    if (node && typeof node === "object" && !Array.isArray(node)) {
+        const out = {};
+        for (const [k, v] of Object.entries(node)) {
+            out[k] = walk(providerName, [...path, k], v, config);
+        }
+        return out;
+    }
+    return node;
+}
+/**
+ * Wrap a callable leaf. If the path resolves to a paid endpoint, the returned
+ * function routes through `dispatchWithPaidGate`. Otherwise the original
+ * function is returned. Either way, any properties hanging off the function
+ * (`.schema`, callable-namespace children) are preserved and recursively
+ * walked so nested paid leaves are also gated.
+ */
+function wrapCallable(providerName, path, fn, config) {
+    const method = path[0].toUpperCase();
+    const dotPath = path.slice(1).join(".");
+    const paid = dotPath.length > 0 && isPaidEndpoint(providerName, method, dotPath);
+    const base = paid
+        ? (...args) => {
+            const [req, approval] = args;
+            return dispatchWithPaidGate(providerName, method, dotPath, req, approval, () => fn(req), config);
+        }
+        : fn;
+    // For callable namespaces (functions with own properties), only recurse into
+    // function-typed children — those are sub-endpoints that may themselves be
+    // paid. Non-function props (`.schema`, other metadata) are preserved by
+    // reference so test assertions and runtime callers can compare by identity.
+    const children = {};
+    for (const [k, v] of Object.entries(fn)) {
+        if (typeof v === "function") {
+            children[k] = walk(providerName, [...path, k], v, config);
+        }
+        else {
+            children[k] = v;
+        }
+    }
+    // If `base === fn` we still merge children, but they're already attached;
+    // the assignment is a no-op for own enumerable keys.
+    return Object.assign(base, children);
+}
+//# sourceMappingURL=with-paid-gate.js.map

package/dist/src/with-paid-gate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"with-paid-gate.js","sourceRoot":"","sources":["../../src/with-paid-gate.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,EACpB,iBAAiB,GAGlB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD;;;;;GAKG;AACH,MAAM,aAAa,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAU,CAAC;AAmBzE;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,YAAY,CAC1B,YAAoB,EACpB,IAAO,EACP,IAA0B;IAE1B,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,aAAa,CAAC;IAC3C,8EAA8E;IAC9E,0EAA0E;IAC1E,iEAAiE;IACjE,MAAM,MAAM,GAA8B,IAAI,EAAE,MAAM;QACpD,CAAC,CAAC;YACE,GAAG,IAAI,CAAC,MAAM;YACd,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,iBAAiB,EAAE;SAC5D;QACH,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,GAAG,GAA4B,EAAE,CAAC;IACxC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9D,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACnB,CAAC;IACH,CAAC;IACD,OAAO,GAAQ,CAAC;AAClB,CAAC;AAED,SAAS,IAAI,CACX,YAAoB,EACpB,IAAc,EACd,IAAa,EACb,MAAiC;IAEjC,IAAI,OAAO,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/B,OAAO,YAAY,CAAC,YAAY,EAAE,IAAI,EAAE,IAAkB,EAAE,MAAM,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7D,MAAM,GAAG,GAA4B,EAAE,CAAC;QACxC,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1C,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;;;GAMG;AACH,SAAS,YAAY,CACnB,YAAoB,EACpB,IAAc,EACd,EAAc,EACd,MAAiC;IAEjC,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC,WAAW,EAAE,CAAC;IACtC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,MAAM,IAAI,GACR,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,cAAc,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAEtE,MAAM,IAAI,GAAe,IAAI;QAC3B,CAAC,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;YACrB,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,GAAG,IAA8C,CAAC;YACvE,OAAO,oBAAoB,CACzB,YAAY,EACZ,MAAM,EACN,OAAO,EACP,GAA8B,EAC9B,QAAQ,EACR,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,EACb,MAAM,CACP,CAAC;QACJ,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IAEP,6EAA6E;IAC7E,2EAA2E;IAC3E,wEAAwE;IACxE,4EAA4E;IAC5E,MAAM,QAAQ,GAA4B,EAAE,CAAC;IAC7C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;QACxC,IAAI,OAAO,CAAC,KAAK,UAAU,EAAE,CAAC;YAC5B,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC5D,CAAC;aAAM,CAAC;YACN,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IACD,0EAA0E;IAC1E,qDAAqD;IACrD,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AACvC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@apicity/cost",
-  "version": "0.2.0-alpha.1",
+  "version": "0.2.1",
   "description": "Pure-table cost & token estimation across @apicity providers — zero network calls, all rates bundled.",
   "license": "MIT",
   "repository": {