add: add unit (per second) to lemonsqueezy after run, and sync to db table

web/src/app/(app)/api/update-run/route.ts

@@ -1,6 +1,13 @@
 import { parseDataSafe } from "../../../../lib/parseDataSafe";
 import { db } from "@/db/db";
-import { workflowRunOutputs, workflowRunsTable } from "@/db/schema";
+import {
+  userUsageTable,
+  workflowRunOutputs,
+  workflowRunsTable,
+  workflowTable,
+} from "@/db/schema";
+import { getDuration } from "@/lib/getRelativeTime";
+import { getSubscription, setUsage } from "@/server/linkToPricing";
 import { eq } from "drizzle-orm";
 import { NextResponse } from "next/server";
 import { z } from "zod";
@@ -27,7 +34,6 @@ export async function POST(request: Request) {
       data: output_data,
     });
   } else if (status) {
-    // console.log("status", status);
     const workflow_run = await db
       .update(workflowRunsTable)
       .set({
@@ -35,8 +41,15 @@ export async function POST(request: Request) {
         ended_at:
           status === "success" || status === "failed" ? new Date() : null,
       })
-      .where(eq(workflowRunsTable.id, run_id))
-      .returning();
+      .where(eq(workflowRunsTable.id, run_id));
+
+    // get data from workflowRunsTable
+    const userUsageTime = await importUserUsageData(run_id);
+
+    if (userUsageTime) {
+      // get the usage_time from userUsage
+      await addSubscriptionUnit(userUsageTime);
+    }
   }
 
   // const workflow_version = await db.query.workflowVersionTable.findFirst({
@@ -54,3 +67,47 @@ export async function POST(request: Request) {
     }
   );
 }
+
+async function addSubscriptionUnit(userUsageTime: number) {
+  const subscription = await getSubscription();
+
+  // round up userUsageTime to the nearest integer
+  const roundedUsageTime = Math.ceil(userUsageTime);
+
+  if (subscription) {
+    const usage = await setUsage(
+      subscription.data[0].attributes.first_subscription_item.id,
+      roundedUsageTime
+    );
+  }
+}
+
+async function importUserUsageData(run_id: string) {
+  const workflowRuns = await db.query.workflowRunsTable.findFirst({
+    where: eq(workflowRunsTable.id, run_id),
+  });
+
+  if (!workflowRuns?.workflow_id) return;
+
+  // find if workflowTable id column contains workflowRunsTable workflow_id
+  const workflow = await db.query.workflowTable.findFirst({
+    where: eq(workflowTable.id, workflowRuns.workflow_id),
+  });
+
+  if (workflowRuns?.ended_at === null || workflow == null) return;
+
+  const usageTime = parseFloat(
+    getDuration((workflowRuns?.ended_at - workflowRuns?.started_at) / 1000)
+  );
+
+  // add data to userUsageTable
+  const user_usage = await db.insert(userUsageTable).values({
+    user_id: workflow.user_id,
+    created_at: workflowRuns.ended_at,
+    org_id: workflow.org_id,
+    ended_at: workflowRuns.ended_at,
+    usage_time: usageTime,
+  });
+
+  return usageTime;
+}