import { db } from "@/db/db";
import type { WorkflowVersionType } from "@/db/schema";
import { workflowTable, workflowVersionTable } from "@/db/schema";

export async function createNewWorkflow({
  workflow_name,
  user_id,
  org_id,
  workflowData,
}: {
  workflow_name: string;
  user_id: string;
  org_id?: string;
  workflowData: Pick<
    WorkflowVersionType,
    "workflow" | "workflow_api" | "snapshot"
  >;
}) {
  // Create a new parent workflow
  const workflow_parent = await db
    .insert(workflowTable)
    .values({
      user_id,
      name: workflow_name,
      org_id: org_id,
    })
    .returning();

  const workflow_id = workflow_parent[0].id;

  // Create a new version
  const data = await db
    .insert(workflowVersionTable)
    .values({
      workflow_id: workflow_id,
      version: 1,
      ...workflowData,
    })
    .returning();
  const version = data[0].version;

  return {
    workflow_id,
    version,
  };
}