working version!

Nutlope · Nutlope · commit dedc9b525016 · 2024-08-08T10:03:35.000-07:00
diff --git a/app/(main)/actions.ts b/app/(main)/actions.ts
@@ -0,0 +1,24 @@
+"use server";
+
+import client from "@/lib/prisma";
+import { redirect } from "next/navigation";
+
+export async function shareApp({
+  generatedCode,
+  prompt,
+  model,
+}: {
+  generatedCode: string;
+  prompt: string;
+  model: string;
+}) {
+  let newApp = await client.generatedApp.create({
+    data: {
+      code: generatedCode,
+      model: model,
+      prompt: prompt,
+    },
+  });
+
+  return redirect(`/share/${newApp.id}`);
+}
diff --git a/app/(main)/page.tsx b/app/(main)/page.tsx
@@ -6,11 +6,8 @@ import { useScrollTo } from "@/hooks/use-scroll-to";
 import { Sandpack } from "@codesandbox/sandpack-react";
 import { dracula as draculaTheme } from "@codesandbox/sandpack-themes";
 import { CheckIcon } from "@heroicons/react/16/solid";
-import {
-  ArrowLongRightIcon,
-  ChevronDownIcon,
-  PlusIcon,
-} from "@heroicons/react/20/solid";
+import { ArrowLongRightIcon, ChevronDownIcon } from "@heroicons/react/20/solid";
+import { ArrowUpOnSquareIcon } from "@heroicons/react/24/outline";
 import * as Select from "@radix-ui/react-select";
 import * as Tooltip from "@radix-ui/react-tooltip";
 import {
@@ -21,6 +18,7 @@ import {
 import { AnimatePresence, motion } from "framer-motion";
 import { FormEvent, useEffect, useState } from "react";
 import LoadingDots from "../../components/loading-dots";
+import { shareApp } from "./actions";
 
 export default function Home() {
   let [status, setStatus] = useState<
@@ -332,24 +330,30 @@ export default function Home() {
                   <Tooltip.Root delayDuration={0}>
                     <Tooltip.Trigger asChild>
                       <button
-                        onClick={() => {
-                          location.reload();
+                        onClick={async () => {
+                          let userMessages = messages.filter(
+                            (message) => message.role === "user",
+                          );
+                          let prompt =
+                            userMessages[userMessages.length - 1].content;
 
-                          // TODO: Cancel stream and reset this state
-                          // setMessages([]);
-                          // setStatus("initial");
+                          await shareApp({
+                            generatedCode,
+                            prompt,
+                            model: modelUsedForInitialCode,
+                          });
                         }}
                         className="inline-flex size-[68px] items-center justify-center rounded-3xl bg-blue-500"
                       >
-                        <PlusIcon className="size-10 text-white" />
+                        <ArrowUpOnSquareIcon className="size-6 text-white" />
                       </button>
                     </Tooltip.Trigger>
                     <Tooltip.Portal>
                       <Tooltip.Content
                         className="select-none rounded bg-white px-4 py-2.5 text-sm leading-none shadow-md shadow-black/20"
                         sideOffset={5}
                       >
-                        Create a new app
+                        Share your app
                         <Tooltip.Arrow className="fill-white" />
                       </Tooltip.Content>
                     </Tooltip.Portal>
diff --git a/app/share/[id]/page.tsx b/app/share/[id]/page.tsx
@@ -1,5 +1,6 @@
 import { notFound } from "next/navigation";
 import CodeViewer from "./code-viewer";
+import client from "@/lib/prisma";
 
 async function getCode(id: string) {
   await new Promise((resolve) => setTimeout(resolve, 1000));
@@ -13,9 +14,13 @@ export default async function Page({ params }: { params: { id: string } }) {
     notFound();
   }
 
-  let code = await getCode(params.id);
+  const generatedApp = await client.generatedApp.findUnique({
+    where: {
+      id: params.id,
+    },
+  });
 
-  return <CodeViewer code={code} />;
+  return <CodeViewer code={generatedApp?.code} />;
 }
 
 let sampleCode = `
