feat: add run origin options, fix client code templates, buttons and badge styles

2024-01-19 17:08:05 +08:00
parent 81cb44c5cf
commit e220921440
14 changed files with 1152 additions and 181 deletions
@@ -0,0 +1,113 @@
+"use client";
+
+import { plainInputsToZod } from "../lib/workflowVersionInputsToZod";
+import { publicRunStore } from "./VersionSelect";
+import { callServerPromise } from "./callServerPromise";
+import { LoadingIcon } from "@/components/LoadingIcon";
+import AutoForm, { AutoFormSubmit } from "@/components/ui/auto-form";
+import { Button } from "@/components/ui/button";
+import type { getInputsFromWorkflow } from "@/lib/getInputsFromWorkflow";
+import { createRun } from "@/server/createRun";
+import { useAuth, useClerk } from "@clerk/nextjs";
+import { Play } from "lucide-react";
+import { useMemo, useState } from "react";
+
+// For share page
+export function RunWorkflowInline({
+  inputs,
+  workflow_version_id,
+  machine_id,
+}: {
+  inputs: ReturnType<typeof getInputsFromWorkflow>;
+  workflow_version_id: string;
+  machine_id: string;
+}) {
+  const [values, setValues] = useState<Record<string, string>>({});
+  const [isLoading, setIsLoading] = useState(false);
+
+  const user = useAuth();
+  const clerk = useClerk();
+
+  const schema = useMemo(() => {
+    return plainInputsToZod(inputs);
+  }, [inputs]);
+
+  const {
+    setRunId,
+    loading,
+    setLoading: setLoading2,
+    setStatus,
+  } = publicRunStore();
+
+  const runWorkflow = async () => {
+    console.log();
+
+    if (!user.isSignedIn) {
+      clerk.openSignIn({
+        redirectUrl: window.location.href,
+      });
+      console.log("hi");
+      return;
+    }
+    console.log(values);
+
+    const val = Object.keys(values).length > 0 ? values : undefined;
+    setLoading2(true);
+    setIsLoading(true);
+    setStatus("preparing");
+    try {
+      const origin = window.location.origin;
+      const a = await callServerPromise(
+        createRun({
+          origin,
+          workflow_version_id: workflow_version_id,
+          machine_id: machine_id,
+          inputs: val,
+          runOrigin: "public-share",
+        })
+      );
+      if (a && !("error" in a)) {
+        setRunId(a.workflow_run_id);
+      } else {
+        setLoading2(false);
+      }
+      console.log(a);
+      setIsLoading(false);
+    } catch (error) {
+      setIsLoading(false);
+      setLoading2(false);
+    }
+  };
+
+  return (
+    <>
+      {schema && (
+        <AutoForm
+          formSchema={schema}
+          values={values}
+          onValuesChange={setValues}
+          onSubmit={runWorkflow}
+          className="px-1"
+        >
+          <div className="flex justify-end">
+            <AutoFormSubmit disabled={isLoading || loading}>
+              Run
+              <span className="ml-2">
+                {isLoading || loading ? <LoadingIcon /> : <Play size={14} />}
+              </span>
+            </AutoFormSubmit>
+          </div>
+        </AutoForm>
+      )}
+      {!schema && (
+        <Button
+          className="gap-2"
+          disabled={isLoading || loading}
+          onClick={runWorkflow}
+        >
+          Confirm {isLoading || loading ? <LoadingIcon /> : <Play size={14} />}
+        </Button>
+      )}
+    </>
+  );
+}