vibezcheck 0.4.0 → 0.4.2

package/dist/index.js

@@ -58,6 +58,7 @@ __export(src_exports, {
   getModelPricing: () => getModelPricing,
   inspectOpenAIStreamChunk: () => inspectOpenAIStreamChunk,
   meteredModel: () => meteredModel,
+  normalizeCustomer: () => normalizeCustomer,
   normalizeModelKey: () => normalizeModelKey,
   registerModelPricing: () => registerModelPricing,
   trackTokens: () => trackTokens,
@@ -682,6 +683,58 @@ function calculateUsageCost(model, usage, optionsOrMarkup) {
   });
 }
 
+// src/customers/helpers.ts
+function normalizeCustomer(customer, extraCustomerId) {
+  if (!customer && !extraCustomerId) {
+    return {
+      customerId: void 0,
+      customerEmail: void 0,
+      customerObj: void 0,
+      customerMetadata: {}
+    };
+  }
+  if (typeof customer === "string") {
+    const isEmail = customer.includes("@");
+    const customerObj2 = {
+      id: customer,
+      email: isEmail ? customer : void 0
+    };
+    return {
+      customerId: customer,
+      customerEmail: isEmail ? customer : void 0,
+      customerObj: customerObj2,
+      customerMetadata: {}
+    };
+  }
+  const customerObj = customer || (extraCustomerId ? { id: extraCustomerId } : void 0);
+  const customerId = customerObj?.id || customerObj?.userId || extraCustomerId;
+  const customerEmail = customerObj?.email;
+  const customerMetadata = {};
+  if (customerObj) {
+    const orgId = customerObj.orgId || customerObj.organizationId;
+    if (orgId) customerMetadata.org_id = String(orgId);
+    const teamId = customerObj.teamId || customerObj.workspaceId;
+    if (teamId) customerMetadata.team_id = String(teamId);
+    if (customerObj.plan) customerMetadata.plan = String(customerObj.plan);
+    if (customerObj.tier) customerMetadata.tier = String(customerObj.tier);
+    if (customerObj.role) customerMetadata.role = String(customerObj.role);
+    if (customerObj.orgName) customerMetadata.org_name = String(customerObj.orgName);
+    if (customerObj.metadata) {
+      for (const [key, val] of Object.entries(customerObj.metadata)) {
+        if (val !== null && val !== void 0) {
+          customerMetadata[key] = val;
+        }
+      }
+    }
+  }
+  return {
+    customerId,
+    customerEmail,
+    customerObj,
+    customerMetadata
+  };
+}
+
 // src/meter/stream.ts
 function wrapOpenAIStream(stream, options, onComplete) {
   let detectedModel = options.model || "gpt-4o";
@@ -702,15 +755,20 @@ function wrapOpenAIStream(stream, options, onComplete) {
       } finally {
         if (finalUsage) {
           const cost = calculateUsageCost(detectedModel, finalUsage);
-          const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+          const normalized = normalizeCustomer(options.customer, options.customerId);
           const event = {
             timestamp: (/* @__PURE__ */ new Date()).toISOString(),
             model: detectedModel,
             provider: "openai",
             usage: finalUsage,
             cost,
-            customerId,
-            metadata: options.metadata
+            customerId: normalized.customerId,
+            customerEmail: normalized.customerEmail,
+            customer: normalized.customerObj,
+            metadata: {
+              ...normalized.customerMetadata,
+              ...options.metadata
+            }
           };
           onComplete(event);
           if (options.onUsage) {
@@ -735,15 +793,20 @@ function wrapAnthropicStream(stream, options, onComplete) {
         const extracted = accumulator.getUsage();
         const model = options.model || extracted.model;
         const cost = calculateUsageCost(model, extracted.usage);
-        const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+        const normalized = normalizeCustomer(options.customer, options.customerId);
         const usageEvent = {
           timestamp: (/* @__PURE__ */ new Date()).toISOString(),
           model,
           provider: "anthropic",
           usage: extracted.usage,
           cost,
-          customerId,
-          metadata: options.metadata
+          customerId: normalized.customerId,
+          customerEmail: normalized.customerEmail,
+          customer: normalized.customerObj,
+          metadata: {
+            ...normalized.customerMetadata,
+            ...options.metadata
+          }
         };
         onComplete(usageEvent);
         if (options.onUsage) {
@@ -772,15 +835,20 @@ function wrapGeminiStream(result, options, onComplete) {
         const extracted = extractGeminiResponseUsage(lastChunkWithUsage, model);
         if (extracted) {
           const cost = calculateUsageCost(extracted.model, extracted.usage);
-          const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+          const normalized = normalizeCustomer(options.customer, options.customerId);
           const event = {
             timestamp: (/* @__PURE__ */ new Date()).toISOString(),
             model: extracted.model,
             provider: "google",
             usage: extracted.usage,
             cost,
-            customerId,
-            metadata: options.metadata
+            customerId: normalized.customerId,
+            customerEmail: normalized.customerEmail,
+            customer: normalized.customerObj,
+            metadata: {
+              ...normalized.customerMetadata,
+              ...options.metadata
+            }
           };
           onComplete(event);
           if (options.onUsage) {
@@ -823,7 +891,7 @@ var VibezMeter = class {
       this.stripeClient = new import_stripe.default(key, {
         appInfo: {
           name: "vibezcheck",
-          version: "0.1.0",
+          version: "0.4.2",
           url: "https://vibezcheck.xyz"
         }
       });
@@ -843,15 +911,21 @@ var VibezMeter = class {
     }
     const model = options.model || extracted.model;
     const cost = calculateUsageCost(model, extracted.usage, this.markupMultiplier);
-    const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+    const normalized = normalizeCustomer(options.customer, options.customerId);
+    const mergedMeta = {
+      ...normalized.customerMetadata,
+      ...options.metadata
+    };
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model,
       provider: extracted.provider,
       usage: extracted.usage,
       cost,
-      customerId,
-      metadata: options.metadata
+      customerId: normalized.customerId,
+      customerEmail: normalized.customerEmail,
+      customer: normalized.customerObj,
+      metadata: mergedMeta
     };
     this.batcher.enqueue(event);
     return event;
@@ -881,15 +955,21 @@ var VibezMeter = class {
       cachedTokens
     };
     const cost = calculateUsageCost(options.model, usage, this.markupMultiplier);
-    const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+    const normalized = normalizeCustomer(options.customer, options.customerId);
+    const mergedMeta = {
+      ...normalized.customerMetadata,
+      ...options.metadata
+    };
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model: options.model,
       provider: options.provider || "custom",
       usage,
       cost,
-      customerId,
-      metadata: options.metadata
+      customerId: normalized.customerId,
+      customerEmail: normalized.customerEmail,
+      customer: normalized.customerObj,
+      metadata: mergedMeta
     };
     this.batcher.enqueue(event);
     return event;
@@ -936,7 +1016,11 @@ function withBilling(model, options = {}) {
     apiKey: options.stripeApiKey,
     eventName: options.eventName
   });
-  const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+  const normalized = normalizeCustomer(options.customer, options.customerId);
+  const customerId = normalized.customerId;
+  const customerEmail = normalized.customerEmail;
+  const customerObj = normalized.customerObj;
+  const customerMetadata = normalized.customerMetadata;
   const modelId = model.modelId || "unknown-model";
   const provider = model.provider?.replace(/^@ai-sdk\//, "") || "ai-sdk";
   const maxCostPerCall = options.maxCostPerCallUSD !== void 0 ? options.maxCostPerCallUSD : 0.5;
@@ -1005,6 +1089,12 @@ function withBilling(model, options = {}) {
         throw new VibezCircuitBreakerError(budgetEvent);
       }
     }
+    const mergedMetadata = {
+      ...customerMetadata,
+      ...options.metadata,
+      ...extraMeta,
+      billingMode: options.billing?.mode || "postpaid"
+    };
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model: modelId,
@@ -1012,11 +1102,9 @@ function withBilling(model, options = {}) {
       usage,
       cost,
       customerId,
-      metadata: {
-        ...options.metadata,
-        ...extraMeta,
-        billingMode: options.billing?.mode || "postpaid"
-      }
+      customerEmail,
+      customer: customerObj,
+      metadata: mergedMetadata
     };
     meter.recordUsage({
       model: modelId,
@@ -1026,11 +1114,84 @@ function withBilling(model, options = {}) {
       reasoningTokens: usage.reasoningTokens,
       cachedTokens: usage.cachedTokens,
       customerId,
+      customer: customerObj,
       metadata: event.metadata
     });
     if (options.onUsage) {
       options.onUsage(event);
     }
+    if (options.database) {
+      const dbTarget = options.database;
+      const executeDbWrite = async () => {
+        try {
+          const supabaseClient = typeof dbTarget === "object" && dbTarget !== null && typeof dbTarget.from === "function" ? dbTarget : typeof dbTarget === "object" && dbTarget !== null && typeof dbTarget.client?.from === "function" ? dbTarget.client : null;
+          if (supabaseClient) {
+            const tableName = dbTarget?.table || "vibez_usage";
+            await supabaseClient.from(tableName).insert({
+              customer_id: customerId,
+              user_id: customerObj?.userId || customerId,
+              model: modelId,
+              input_tokens: usage.inputTokens,
+              output_tokens: usage.outputTokens,
+              total_tokens: usage.totalTokens,
+              cached_tokens: usage.cachedTokens,
+              reasoning_tokens: usage.reasoningTokens,
+              cost_usd: cost.totalUSD,
+              created_at: event.timestamp,
+              metadata: event.metadata
+            });
+            return;
+          }
+          if (typeof dbTarget === "object" && dbTarget !== null && typeof dbTarget.save === "function") {
+            await dbTarget.save(event);
+            return;
+          }
+          if (typeof dbTarget === "function") {
+            await dbTarget(event);
+            return;
+          }
+        } catch (err) {
+          console.warn(`[vibezcheck] Database record failed safely:`, err?.message || err);
+        }
+      };
+      if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
+        globalThis.after(() => executeDbWrite());
+      } else {
+        executeDbWrite().catch(() => {
+        });
+      }
+    }
+    if (options.billing?.charge && typeof options.billing.charge === "function") {
+      const chargeFn = options.billing.charge;
+      const executeCharge = async () => {
+        try {
+          await chargeFn(cost.totalUSD, event);
+        } catch (err) {
+          console.warn(`[vibezcheck] Custom charge handler failed safely:`, err?.message || err);
+        }
+      };
+      if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
+        globalThis.after(() => executeCharge());
+      } else {
+        executeCharge().catch(() => {
+        });
+      }
+    } else if (options.billing?.provider && typeof options.billing.provider === "object" && typeof options.billing.provider.charge === "function") {
+      const providerCharge = options.billing.provider.charge;
+      const executeCharge = async () => {
+        try {
+          await providerCharge(cost.totalUSD, event);
+        } catch (err) {
+          console.warn(`[vibezcheck] Payment provider charge failed safely:`, err?.message || err);
+        }
+      };
+      if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
+        globalThis.after(() => executeCharge());
+      } else {
+        executeCharge().catch(() => {
+        });
+      }
+    }
     scheduleFlush();
   };
   const handler = {
@@ -1078,6 +1239,8 @@ function withBilling(model, options = {}) {
               controller.enqueue(chunk);
               if (chunk.type === "text-delta" && chunk.textDelta) {
                 accumulatedChars += chunk.textDelta.length;
+              } else if (chunk.type === "reasoning" && (chunk.textDelta || chunk.reasoning)) {
+                accumulatedChars += (chunk.textDelta || chunk.reasoning || "").length;
               }
               if (chunk.type === "finish" && chunk.usage) {
                 streamCompleted = true;
@@ -1276,7 +1439,7 @@ var CustomerManager = class {
     this.cache = new CustomerCache(options.cacheTtlMs);
   }
   /**
-   * Retrieves existing Stripe Customer or automatically provisions a new one
+   * Retrieves existing Stripe Customer or automatically provisions a new one with full metadata
    */
   async getOrCreate(params) {
     const cacheKey = params.userId || params.email;
@@ -1317,20 +1480,66 @@ var CustomerManager = class {
         return { id: customer.id, isNew: false, customer };
       }
     }
+    const orgId = params.orgId || params.organizationId;
+    const teamId = params.teamId || params.workspaceId;
+    const stripeMetadata = {
+      vibez_user_id: params.userId || "",
+      created_by: "vibezcheck"
+    };
+    if (orgId) stripeMetadata.org_id = String(orgId);
+    if (params.orgName) stripeMetadata.org_name = String(params.orgName);
+    if (teamId) stripeMetadata.team_id = String(teamId);
+    if (params.plan) stripeMetadata.plan = String(params.plan);
+    if (params.tier) stripeMetadata.tier = String(params.tier);
+    if (params.role) stripeMetadata.role = String(params.role);
+    if (params.metadata) {
+      for (const [k, v] of Object.entries(params.metadata)) {
+        if (v !== void 0 && v !== null) {
+          stripeMetadata[k] = String(v);
+        }
+      }
+    }
     const newCustomer = await this.stripe.customers.create({
       email: params.email,
       name: params.name,
-      metadata: {
-        vibez_user_id: params.userId,
-        created_by: "vibezcheck",
-        ...params.metadata || {}
-      }
+      phone: params.phone,
+      metadata: stripeMetadata
     });
     if (cacheKey) this.cache.set(cacheKey, newCustomer.id);
     if (params.userId) this.cache.set(params.userId, newCustomer.id);
     if (params.email) this.cache.set(params.email, newCustomer.id);
     return { id: newCustomer.id, isNew: true, customer: newCustomer };
   }
+  /**
+   * Updates an existing Stripe Customer's information and metadata
+   */
+  async updateCustomer(customerId, params) {
+    const updatePayload = {};
+    if (params.email) updatePayload.email = params.email;
+    if (params.name) updatePayload.name = params.name;
+    if (params.phone) updatePayload.phone = params.phone;
+    const metadataToUpdate = {};
+    if (params.userId) metadataToUpdate.vibez_user_id = params.userId;
+    const orgId = params.orgId || params.organizationId;
+    if (orgId) metadataToUpdate.org_id = String(orgId);
+    if (params.orgName) metadataToUpdate.org_name = String(params.orgName);
+    const teamId = params.teamId || params.workspaceId;
+    if (teamId) metadataToUpdate.team_id = String(teamId);
+    if (params.plan) metadataToUpdate.plan = String(params.plan);
+    if (params.tier) metadataToUpdate.tier = String(params.tier);
+    if (params.role) metadataToUpdate.role = String(params.role);
+    if (params.metadata) {
+      for (const [k, v] of Object.entries(params.metadata)) {
+        if (v !== void 0 && v !== null) {
+          metadataToUpdate[k] = String(v);
+        }
+      }
+    }
+    if (Object.keys(metadataToUpdate).length > 0) {
+      updatePayload.metadata = metadataToUpdate;
+    }
+    return await this.stripe.customers.update(customerId, updatePayload);
+  }
   /**
    * Clears in-memory resolution cache
    */
@@ -1636,6 +1845,7 @@ var vibes = vibez;
   getModelPricing,
   inspectOpenAIStreamChunk,
   meteredModel,
+  normalizeCustomer,
   normalizeModelKey,
   registerModelPricing,
   trackTokens,