scriptoriumadmin/supermemory
1
0
Fork 0
mirror of https://github.com/supermemoryai/supermemory.git synced 2026-08-28 05:25:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update store route

Browse source
This commit is contained in:
yxshv 2024-04-13 06:23:40 +05:30
parent e62cb68ea6
commit af1ef5c958
2 changed files with 18 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
apps/web/drizzle.config.ts
View file

@ -1,9 +1,13 @@
import { type Config } from "drizzle-kit";
const localDb = {
url: process.env.LOCAL_DB_URL!
}
export default {
schema: "./src/server/db/schema.ts",
driver: "d1",
dbCredentials: {
driver: process.env.LOCAL_DB_URL ? "better-sqlite" : "d1",
dbCredentials: process.env.LOCAL_DB_URL ? localDb : {
wranglerConfigPath: "./wrangler.toml",
dbName: "dev-d1-anycontext",
},

40
apps/web/src/app/api/store/route.ts
View file

@ -1,5 +1,5 @@
import { db } from "@/server/db";
import { and, eq } from "drizzle-orm";
import { and, eq, inArray } from "drizzle-orm";
import {
contentToSpace,
sessions,
@ -67,20 +67,18 @@ export async function POST(req: NextRequest) {
const data = (await req.json()) as {
pageContent: string;
url: string;
space?: string;
spaces?: string[];
};
const metadata = await getMetaData(data.url);
let id: number | undefined = undefined;
let storeToSpaces = data.spaces;
let storeToSpace = data.space;
if (!storeToSpace) {
storeToSpace = "none";
if (!storeToSpaces) {
storeToSpaces = [];
}
const storedContentId = await db.insert(storedContent).values({
const { id } = (await db.insert(storedContent).values({
content: data.pageContent,
title: metadata.title,
description: metadata.description,
@ -89,9 +87,7 @@ export async function POST(req: NextRequest) {
image: metadata.image,
savedAt: new Date(),
user: session.user.id,
});
id = storedContentId.meta.last_row_id;
}).returning({ id: storedContent.id }))[0];
if (!id) {
return NextResponse.json(
@ -100,27 +96,15 @@ export async function POST(req: NextRequest) {
);
}
let spaceID = 0;
const spaceData = await db
.select()
.from(space)
.where(and(eq(space.name, storeToSpace), eq(space.user, session.user.id)))
.limit(1);
spaceID = spaceData[0]?.id;
.where(and(inArray(space.name, storeToSpaces), eq(space.user, session.user.id)))
.all()
if (!spaceData || spaceData.length === 0) {
const spaceId = await db.insert(space).values({
name: storeToSpace,
user: session.user.id,
});
spaceID = spaceId.meta.last_row_id;
}
await db.insert(contentToSpace).values({
contentId: id as number,
spaceId: spaceID,
});
await Promise.all([spaceData.forEach(async space => {
await db.insert(contentToSpace).values({ contentId: id, spaceId: space.id })
})])
const res = (await Promise.race([
fetch("https://cf-ai-backend.dhravya.workers.dev/add", {