@goscribe/server 1.1.7 → 1.3.0

package/src/routers/studyguide.ts

@@ -1,12 +1,9 @@
 import { z } from 'zod';
 import { TRPCError } from '@trpc/server';
 import { router, authedProcedure } from '../trpc.js';
-import { title } from 'node:process';
-
-// Mirror Prisma enum to avoid direct type import
-const ArtifactType = {
-  STUDY_GUIDE: 'STUDY_GUIDE',
-} as const;
+import { ArtifactType } from '../lib/constants.js';
+import { getUserUsage, getUserPlanLimits } from '../lib/usage_service.js';
+import { workspaceAccessFilter } from '../lib/workspace-access.js';
 
 const initializeEditorJsEmptyBlock = () => ({
   time: Date.now(),
@@ -14,7 +11,7 @@ const initializeEditorJsEmptyBlock = () => ({
     {
       id: 'initial',
       type: 'paragraph',
-      data: { text: 'Start writing your study guide...' },
+      data: { text: 'Upload some files to begin creating your revision workspace...' },
     },
   ],
   version: '2.27.0',
@@ -34,15 +31,25 @@ export const studyguide = router({
         where: {
           workspaceId: input.workspaceId!,
           type: ArtifactType.STUDY_GUIDE,
-          workspace: { ownerId: ctx.session.user.id },
+          workspace: workspaceAccessFilter(ctx.session.user.id),
         },
         include: {
           versions: { orderBy: { version: 'desc' }, take: 1 },
         },
       });
 
-      console.log('artifact', artifact);
       if (!artifact) {
+        const [usage, limits] = await Promise.all([
+          getUserUsage(ctx.session.user.id),
+          getUserPlanLimits(ctx.session.user.id)
+        ]);
+        if (limits && usage.studyGuides >= limits.maxStudyGuides) {
+          throw new TRPCError({ 
+            code: 'FORBIDDEN', 
+            message: 'Study Guide limit reached. Please upgrade your plan to create more workspaces with study guides.' 
+          });
+        }
+        
         artifact = await ctx.db.artifact.create({
           data: {
             workspaceId: input.workspaceId,
@@ -67,78 +74,77 @@ export const studyguide = router({
     }),
 
   // Edit study guide content by creating a new version, or create if doesn't exist
-  edit: authedProcedure
-    .input(
-      z.object({
-        workspaceId: z.string(),
-        studyGuideId: z.string().optional(),
-        content: z.string().min(1),
-        data: z.record(z.string(), z.unknown()).optional(),
-        title: z.string().min(1).optional(),
-      })
-    )
-    .mutation(async ({ ctx, input }) => {
-      let artifact;
+  // edit: authedProcedure
+  //   .input(
+  //     z.object({
+  //       workspaceId: z.string(),
+  //       studyGuideId: z.string().optional(),
+  //       content: z.string().min(1),
+  //       data: z.record(z.string(), z.unknown()).optional(),
+  //       title: z.string().min(1).optional(),
+  //     })
+  //   )
+  //   .mutation(async ({ ctx, input }) => {
+  //     let artifact;
       
-      if (input.studyGuideId) {
-        // Try to find existing study guide
-        artifact = await ctx.db.artifact.findFirst({
-          where: {
-            id: input.studyGuideId,
-            type: ArtifactType.STUDY_GUIDE,
-            workspace: { ownerId: ctx.session.user.id },
-          },
-        });
-      } else {
-        // Find by workspace if no specific studyGuideId provided
-        artifact = await ctx.db.artifact.findFirst({
-          where: {
-            workspaceId: input.workspaceId,
-            type: ArtifactType.STUDY_GUIDE,
-            workspace: { ownerId: ctx.session.user.id },
-          },
-        });
-      }
+  //     if (input.studyGuideId) {
+  //       // Try to find existing study guide
+  //       artifact = await ctx.db.artifact.findFirst({
+  //         where: {
+  //           id: input.studyGuideId,
+  //           type: ArtifactType.STUDY_GUIDE,
+  //           workspace: workspaceAccessFilter(ctx.session.user.id),
+  //         },
+  //       });
+  //     } else {
+  //       // Find by workspace if no specific studyGuideId provided
+  //       artifact = await ctx.db.artifact.findFirst({
+  //         where: {
+  //           workspaceId: input.workspaceId,
+  //           type: ArtifactType.STUDY_GUIDE,
+  //           workspace: workspaceAccessFilter(ctx.session.user.id),
+  //         },
+  //       });
+  //     }
       
-      // If no study guide found, create a new one
-      if (!artifact) {
-        artifact = await ctx.db.artifact.create({
-          data: {
-            workspaceId: input.workspaceId,
-            type: ArtifactType.STUDY_GUIDE,
-            title: 'Study Guide',
-            createdById: ctx.session.user.id,
-          },
-        });
-      }
+  //     // If no study guide found, create a new one
+  //     if (!artifact) {
+  //       artifact = await ctx.db.artifact.create({
+  //         data: {
+  //           workspaceId: input.workspaceId,
+  //           type: ArtifactType.STUDY_GUIDE,
+  //           title: 'Study Guide',
+  //           createdById: ctx.session.user.id,
+  //         },
+  //       });
+  //     }
 
-      const last = await ctx.db.artifactVersion.findFirst({
-        where: { artifactId: artifact.id },
-        orderBy: { version: 'desc' },
-      });
+  //     const last = await ctx.db.artifactVersion.findFirst({
+  //       where: { artifactId: artifact.id },
+  //       orderBy: { version: 'desc' },
+  //     });
 
-      if (input.title && input.title !== artifact.title) {
-        console.log('rename')
-        await ctx.db.artifact.update({
-          where: { id: artifact.id },
-          data: { title: input.title },
-        });
-      }
+  //     if (input.title && input.title !== artifact.title) {
+  //       await ctx.db.artifact.update({
+  //         where: { id: artifact.id },
+  //         data: { title: input.title },
+  //       });
+  //     }
       
-      const nextVersion = (last?.version ?? 0) + 1;
+  //     const nextVersion = (last?.version ?? 0) + 1;
 
-      const version = await ctx.db.artifactVersion.create({
-        data: {
-          artifactId: artifact.id,
-          content: input.content,
-          data: input.data as any,
-          version: nextVersion,
-          createdById: ctx.session.user.id,
-        },
-      });
+  //     const version = await ctx.db.artifactVersion.create({
+  //       data: {
+  //         artifactId: artifact.id,
+  //         content: input.content,
+  //         data: input.data ?? undefined,
+  //         version: nextVersion,
+  //         createdById: ctx.session.user.id,
+  //       },
+  //     });
 
-      return { artifactId: artifact.id, version };
-    }),
+  //     return { artifactId: artifact.id, version };
+  //   }),
 });