new getPrisma function

samselikoff · samselikoff · commit e0d78619898c · 2025-01-21T13:46:09.000-05:00
diff --git a/app/(main)/actions.ts b/app/(main)/actions.ts
@@ -1,25 +1,21 @@
 "use server";
 
-import { PrismaClient } from "@prisma/client";
-import { PrismaNeon } from "@prisma/adapter-neon";
-import { Pool } from "@neondatabase/serverless";
-import { notFound } from "next/navigation";
-import Together from "together-ai";
+import { getPrisma } from "@/lib/prisma";
 import {
   getMainCodingPrompt,
   screenshotToCodePrompt,
   softwareArchitectPrompt,
 } from "@/lib/prompts";
+import { notFound } from "next/navigation";
+import Together from "together-ai";
 
 export async function createChat(
   prompt: string,
   model: string,
   quality: "high" | "low",
   screenshotUrl: string | undefined,
 ) {
-  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
-  const adapter = new PrismaNeon(neon);
-  const prisma = new PrismaClient({ adapter });
+  const prisma = getPrisma();
   let chat;
   try {
     chat = await prisma.chat.create({
@@ -195,9 +191,7 @@ export async function createMessage(
   text: string,
   role: "assistant" | "user",
 ) {
-  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
-  const adapter = new PrismaNeon(neon);
-  const prisma = new PrismaClient({ adapter });
+  const prisma = getPrisma();
   const chat = await prisma.chat.findUnique({
     where: { id: chatId },
     include: { messages: true },
diff --git a/app/(main)/chats/[id]/chat-box.tsx b/app/(main)/chats/[id]/chat-box.tsx
@@ -43,11 +43,8 @@ export default function ChatBox({
     <div className="mx-auto mb-5 flex w-full max-w-prose shrink-0 px-8">
       <form
         className="relative flex w-full"
-        action={async (formData) => {
+        action={async () => {
           startTransition(async () => {
-            const prompt = formData.get("prompt");
-            assert.ok(typeof prompt === "string");
-
             const message = await createMessage(chat.id, prompt, "user");
             const streamPromise = fetch(
               "/api/get-next-completion-stream-promise",
@@ -66,7 +63,10 @@ export default function ChatBox({
             });
 
             onNewStreamPromise(streamPromise);
-            router.refresh();
+            startTransition(() => {
+              router.refresh();
+              setPrompt("");
+            });
           });
         }}
       >
diff --git a/app/(main)/chats/[id]/page.tsx b/app/(main)/chats/[id]/page.tsx
@@ -1,10 +1,8 @@
 // import client from "@/lib/prisma";
-import PageClient from "./page.client";
+import { getPrisma } from "@/lib/prisma";
 import { notFound } from "next/navigation";
 import { cache } from "react";
-import { PrismaClient } from "@prisma/client";
-import { PrismaNeon } from "@prisma/adapter-neon";
-import { Pool } from "@neondatabase/serverless";
+import PageClient from "./page.client";
 
 export default async function Page({
   params,
@@ -20,9 +18,7 @@ export default async function Page({
 }
 
 const getChatById = cache(async (id: string) => {
-  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
-  const adapter = new PrismaNeon(neon);
-  const prisma = new PrismaClient({ adapter });
+  const prisma = getPrisma();
   return await prisma.chat.findFirst({
     where: { id },
     include: { messages: { orderBy: { position: "asc" } } },
diff --git a/app/share/[id]/page.tsx b/app/share/[id]/page.tsx
@@ -1,8 +1,8 @@
 import { notFound } from "next/navigation";
-import client from "@/lib/prisma";
 import type { Metadata } from "next";
 import { cache } from "react";
 import CodeRunner from "@/components/code-runner";
+import { getPrisma } from "@/lib/prisma";
 
 /*
   This is the Share page for v1 apps, before the chat interface was added.
@@ -63,7 +63,8 @@ export default async function Page({
 }
 
 const getGeneratedAppByID = cache(async (id: string) => {
-  return client.generatedApp.findUnique({
+  const prisma = getPrisma();
+  return prisma.generatedApp.findUnique({
     where: {
       id,
     },
diff --git a/app/share/v2/[messageId]/_opengraph-image.tsx b/app/share/v2/[messageId]/_opengraph-image.tsx
@@ -1,8 +1,8 @@
 /* eslint-disable @next/next/no-img-element */
+import { getPrisma } from "@/lib/prisma";
 import { ImageResponse } from "next/og";
 import { readFile } from "node:fs/promises";
 import { join } from "node:path";
-import client from "@/lib/prisma";
 
 export const size = {
   width: 1200,
@@ -17,7 +17,8 @@ export default async function Image({
   params: { messageId: string };
 }) {
   let messageId = params.messageId;
-  let message = await client.message.findUnique({
+  const prisma = getPrisma();
+  let message = await prisma.message.findUnique({
     where: {
       id: messageId,
     },
diff --git a/app/share/v2/[messageId]/page.tsx b/app/share/v2/[messageId]/page.tsx
@@ -1,5 +1,5 @@
 import CodeRunner from "@/components/code-runner";
-import client from "@/lib/prisma";
+import { getPrisma } from "@/lib/prisma";
 import { extractFirstCodeBlock } from "@/lib/utils";
 import { Metadata } from "next";
 import { notFound } from "next/navigation";
@@ -41,7 +41,8 @@ export default async function SharePage({
 }) {
   const { messageId } = await params;
 
-  const message = await client.message.findUnique({ where: { id: messageId } });
+  const prisma = getPrisma();
+  const message = await prisma.message.findUnique({ where: { id: messageId } });
   if (!message) {
     notFound();
   }
@@ -59,7 +60,8 @@ export default async function SharePage({
 }
 
 const getMessage = cache(async (messageId: string) => {
-  return client.message.findUnique({
+  const prisma = getPrisma();
+  return prisma.message.findUnique({
     where: {
       id: messageId,
     },
diff --git a/lib/prisma.ts b/lib/prisma.ts
@@ -1,21 +1,10 @@
 import { PrismaClient } from "@prisma/client";
 import { PrismaNeon } from "@prisma/adapter-neon";
 import { Pool } from "@neondatabase/serverless";
+import { cache } from "react";
 
-declare global {
-  var prisma: PrismaClient | undefined;
-}
-
-const neon = new Pool({ connectionString: process.env.DATABASE_URL });
-const adapter = new PrismaNeon(neon);
-
-let client: PrismaClient;
-
-if (process.env.NODE_ENV !== "production") {
-  client = globalThis.prisma || new PrismaClient({ adapter });
-  globalThis.prisma = client;
-} else {
-  client = new PrismaClient({ adapter });
-}
-
-export default client;
+export const getPrisma = cache(() => {
+  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
+  const adapter = new PrismaNeon(neon);
+  return new PrismaClient({ adapter });
+});
