@goscribe/server 1.2.0 → 1.3.1

package/src/routers/podcast.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
 import { TRPCError } from '@trpc/server';
-import { router, authedProcedure } from '../trpc.js';
+import { router, authedProcedure, verifiedProcedure,limitedProcedure } from '../trpc.js';
 import { v4 as uuidv4 } from 'uuid';
 import inference from '../lib/inference.js';
 import { uploadToSupabase, generateSignedUrl, deleteFromSupabase } from '../lib/storage.js';
@@ -9,6 +9,7 @@ import { aiSessionService } from '../lib/ai-session.js';
 import { ArtifactType } from '../lib/constants.js';
 import { workspaceAccessFilter } from '../lib/workspace-access.js';
 import { logger } from '../lib/logger.js';
+import { notifyArtifactFailed, notifyArtifactReady } from '../lib/notification-service.js';
 
 // Podcast segment schema
 const podcastSegmentSchema = z.object({
@@ -72,11 +73,11 @@ export const podcast = router({
       // Check if workspace exists
 
       if (!workspace) throw new TRPCError({ code: 'NOT_FOUND' });
-      
+
       const artifacts = await ctx.db.artifact.findMany({
-        where: { 
-          workspaceId: input.workspaceId, 
-          type: ArtifactType.PODCAST_EPISODE 
+        where: {
+          workspaceId: input.workspaceId,
+          type: ArtifactType.PODCAST_EPISODE
         },
         include: {
           versions: {
@@ -89,7 +90,7 @@ export const podcast = router({
         },
         orderBy: { updatedAt: 'desc' },
       });
-      
+
       logger.debug(`Found ${artifacts.length} podcast artifacts`);
       artifacts.forEach((artifact, i) => {
         logger.debug(`  Podcast ${i + 1}: "${artifact.title}" - ${artifact.podcastSegments.length} segments`);
@@ -105,7 +106,7 @@ export const podcast = router({
           if (artifact.imageObjectKey) {
             objectUrl = await generateSignedUrl(artifact.imageObjectKey, 24);
           }
-          
+
           // Generate fresh signed URLs for all segments
           const segmentsWithUrls = await Promise.all(
             artifact.podcastSegments.map(async (segment) => {
@@ -150,7 +151,7 @@ export const podcast = router({
           let metadata = null;
           if (latestVersion) {
             try {
-              logger.debug(latestVersion.data)
+              logger.debug(JSON.stringify(latestVersion.data))
               metadata = podcastMetadataSchema.parse(latestVersion.data);
             } catch (error) {
               logger.error('Failed to parse podcast metadata:', error);
@@ -184,7 +185,7 @@ export const podcast = router({
     .input(z.object({ episodeId: z.string() }))
     .query(async ({ ctx, input }) => {
       const episode = await ctx.db.artifact.findFirst({
-        where: { 
+        where: {
           id: input.episodeId,
           type: ArtifactType.PODCAST_EPISODE,
           workspace: workspaceAccessFilter(ctx.session.user.id)
@@ -200,14 +201,14 @@ export const podcast = router({
         },
       });
 
-      logger.debug(episode)
+      logger.debug(JSON.stringify(episode))
 
       if (!episode) throw new TRPCError({ code: 'NOT_FOUND' });
-      
+
       const latestVersion = episode.versions[0];
       if (!latestVersion) throw new TRPCError({ code: 'NOT_FOUND', message: 'No version found' });
-      
-      logger.debug(latestVersion)
+
+      logger.debug(JSON.stringify(latestVersion))
       try {
         const metadata = podcastMetadataSchema.parse(latestVersion.data);
       } catch (error) {
@@ -216,7 +217,7 @@ export const podcast = router({
       const metadata = podcastMetadataSchema.parse(latestVersion.data);
 
       const imageUrl = episode.imageObjectKey ? await generateSignedUrl(episode.imageObjectKey, 24) : null;
-  
+
 
       // Generate fresh signed URLs for all segments
       const segmentsWithUrls = await Promise.all(
@@ -263,7 +264,7 @@ export const podcast = router({
           };
         })
       );
-      
+
       return {
         id: episode.id,
         title: metadata.title, // Use title from version metadata
@@ -278,7 +279,7 @@ export const podcast = router({
     }),
 
   // Generate podcast episode from text input
-  generateEpisode: authedProcedure
+  generateEpisode: limitedProcedure
     .input(z.object({
       workspaceId: z.string(),
       podcastData: podcastInputSchema,
@@ -288,33 +289,33 @@ export const podcast = router({
         where: { id: input.workspaceId, ownerId: ctx.session.user.id },
       });
       if (!workspace) throw new TRPCError({ code: 'NOT_FOUND' });
-      
-        // Emit podcast generation start notification
-        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_generation_start', { 
-          title: input.podcastData.title 
-        });
 
-        const BEGIN_PODCAST_GENERATION_MESSAGE = 'Structuring podcast contents...';
+      // Emit podcast generation start notification
+      await PusherService.emitTaskComplete(input.workspaceId, 'podcast_generation_start', {
+        title: input.podcastData.title
+      });
 
-        const newArtifact = await ctx.db.artifact.create({
-          data: {
-            title: '----',
-            type: ArtifactType.PODCAST_EPISODE,
-            generating: true,
-            generatingMetadata: {
-              message: BEGIN_PODCAST_GENERATION_MESSAGE,
-            },
-            workspace: {
-              connect: {
-                id: input.workspaceId,
-              }
+      const BEGIN_PODCAST_GENERATION_MESSAGE = 'Structuring podcast contents...';
+
+      const newArtifact = await ctx.db.artifact.create({
+        data: {
+          title: '----',
+          type: ArtifactType.PODCAST_EPISODE,
+          generating: true,
+          generatingMetadata: {
+            message: BEGIN_PODCAST_GENERATION_MESSAGE,
+          },
+          workspace: {
+            connect: {
+              id: input.workspaceId,
             }
           }
-        });
+        }
+      });
 
-        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', { 
-          message: BEGIN_PODCAST_GENERATION_MESSAGE,
-        });
+      await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', {
+        message: BEGIN_PODCAST_GENERATION_MESSAGE,
+      });
 
       try {
 
@@ -328,9 +329,9 @@ export const podcast = router({
         );
 
         if (!structureResult.success || !structureResult.structure) {
-          throw new TRPCError({ 
-            code: 'INTERNAL_SERVER_ERROR', 
-            message: 'Failed to generate podcast structure' 
+          throw new TRPCError({
+            code: 'INTERNAL_SERVER_ERROR',
+            message: 'Failed to generate podcast structure'
           });
         }
 
@@ -362,7 +363,7 @@ export const podcast = router({
           }
         });
 
-        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', { 
+        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', {
           message: `Generating podcast image...`,
         });
 
@@ -405,7 +406,7 @@ export const podcast = router({
               }
             });
 
-            await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', { 
+            await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', {
               message: `Generating audio for segment ${i + 1} of ${structure.segments.length}...`,
             });
 
@@ -445,42 +446,51 @@ export const podcast = router({
               error: errorMessage,
               stack: audioError instanceof Error ? audioError.stack : undefined,
             });
-            
+
             // Track failed segment
             failedSegments.push({
               index: i + 1,
               title: segment.title || `Segment ${i + 1}`,
               error: errorMessage,
             });
-            
-            await PusherService.emitTaskComplete(input.workspaceId, 'podcast_segment_error', { 
+
+            await PusherService.emitTaskComplete(input.workspaceId, 'podcast_segment_error', {
               segmentIndex: i + 1,
               segmentTitle: segment.title || `Segment ${i + 1}`,
               error: errorMessage,
               successfulSegments: segments.length,
               failedSegments: failedSegments.length,
             });
-            
+
             // Continue with other segments even if one fails
           }
         }
-        
+
         // Check if any segments were successfully generated
         if (segments.length === 0) {
           logger.error('No segments were successfully generated');
-          await PusherService.emitError(input.workspaceId, 
-            `Failed to generate any segments. ${failedSegments.length} segment(s) failed.`, 
+          await PusherService.emitError(input.workspaceId,
+            `Failed to generate any segments. ${failedSegments.length} segment(s) failed.`,
             'podcast'
           );
-          
+
           // Cleanup the artifact
+          await notifyArtifactFailed(ctx.db, {
+            userId: ctx.session.user.id,
+            workspaceId: input.workspaceId,
+            artifactType: ArtifactType.PODCAST_EPISODE,
+            artifactId: newArtifact.id,
+            title: input.podcastData.title,
+            message: `Failed to generate any audio segments. All ${failedSegments.length} attempts failed.`,
+          }).catch(() => {});
+
           await ctx.db.artifact.delete({
             where: { id: newArtifact.id },
           });
-          
-          throw new TRPCError({ 
-            code: 'INTERNAL_SERVER_ERROR', 
-            message: `Failed to generate any audio segments. All ${failedSegments.length} attempts failed.` 
+
+          throw new TRPCError({
+            code: 'INTERNAL_SERVER_ERROR',
+            message: `Failed to generate any audio segments. All ${failedSegments.length} attempts failed.`
           });
         }
 
@@ -496,7 +506,7 @@ export const podcast = router({
           }
         });
 
-        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', { 
+        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', {
           message: `Preparing podcast summary...`,
         });
 
@@ -523,7 +533,7 @@ export const podcast = router({
         Podcast Title: ${structure.episodeTitle}
         Segments: ${JSON.stringify(segments.map(s => ({ title: s.title, keyPoints: s.keyPoints })))}`;
 
-        const summaryResponse = await inference(summaryPrompt);
+        const summaryResponse = await inference([{ role: "user", content: summaryPrompt }]);
         const summaryContent: string = summaryResponse.choices[0].message.content || '';
 
         let episodeSummary;
@@ -536,7 +546,7 @@ export const podcast = router({
           episodeSummary = JSON.parse(jsonMatch[0]);
         } catch (parseError) {
           logger.error('Failed to parse summary response:', summaryContent);
-          await PusherService.emitTaskComplete(input.workspaceId, 'podcast_summary_error', { 
+          await PusherService.emitTaskComplete(input.workspaceId, 'podcast_summary_error', {
             error: 'Failed to parse summary response'
           });
           episodeSummary = {
@@ -551,13 +561,13 @@ export const podcast = router({
         }
 
         // Emit summary generation completion notification
-        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', { 
+        await PusherService.emitTaskComplete(input.workspaceId, 'podcast_info', {
           message: `Podcast summary generated.`,
         });
 
         // Step 4: Create artifact and initial version
         const episodeTitle = structure.episodeTitle || input.podcastData.title;
-        
+
         await ctx.db.artifact.update({
           where: {
             id: newArtifact.id,
@@ -570,7 +580,7 @@ export const podcast = router({
             createdById: ctx.session.user.id,
           },
         });
-        
+
         const createdSegments = await ctx.db.podcastSegment.createMany({
           data: segments.map(segment => ({
             artifactId: newArtifact.id,
@@ -587,7 +597,7 @@ export const podcast = router({
             },
           })),
         });
-        
+
         const metadata = {
           title: episodeTitle,
           description: input.podcastData.description,
@@ -617,7 +627,14 @@ export const podcast = router({
         });
 
         // Emit podcast generation completion notification
-        await PusherService.emitPodcastComplete(input.workspaceId, {});
+        await PusherService.emitPodcastComplete(input.workspaceId, newArtifact);
+        await notifyArtifactReady(ctx.db, {
+          userId: ctx.session.user.id,
+          workspaceId: input.workspaceId,
+          artifactId: newArtifact.id,
+          artifactType: ArtifactType.PODCAST_EPISODE,
+          title: metadata.title,
+        }).catch(() => {});
 
         return {
           id: newArtifact.id,
@@ -631,15 +648,27 @@ export const podcast = router({
 
         logger.error('Error generating podcast episode:', error);
 
+        await notifyArtifactFailed(ctx.db, {
+          userId: ctx.session.user.id,
+          workspaceId: input.workspaceId,
+          artifactType: ArtifactType.PODCAST_EPISODE,
+          artifactId: newArtifact.id,
+          title: input.podcastData.title,
+          message:
+            error instanceof Error
+              ? error.message
+              : 'Podcast generation failed.',
+        }).catch(() => {});
+
         await ctx.db.artifact.delete({
           where: {
             id: newArtifact.id,
           },
         });
         await PusherService.emitError(input.workspaceId, `Failed to generate podcast episode: ${error instanceof Error ? error.message : 'Unknown error'}`, 'podcast');
-        throw new TRPCError({ 
-          code: 'INTERNAL_SERVER_ERROR', 
-          message: `Failed to generate podcast episode: ${error instanceof Error ? error.message : 'Unknown error'}` 
+        throw new TRPCError({
+          code: 'INTERNAL_SERVER_ERROR',
+          message: `Failed to generate podcast episode: ${error instanceof Error ? error.message : 'Unknown error'}`
         });
       }
     }),
@@ -656,7 +685,7 @@ export const podcast = router({
     .input(z.object({ episodeId: z.string() }))
     .query(async ({ ctx, input }) => {
       const episode = await ctx.db.artifact.findFirst({
-        where: { 
+        where: {
           id: input.episodeId,
           type: ArtifactType.PODCAST_EPISODE,
           workspace: workspaceAccessFilter(ctx.session.user.id)
@@ -671,14 +700,14 @@ export const podcast = router({
           },
         },
       });
-      
+
       if (!episode) throw new TRPCError({ code: 'NOT_FOUND' });
-      
+
       const latestVersion = episode.versions[0];
       if (!latestVersion) throw new TRPCError({ code: 'NOT_FOUND', message: 'No version found' });
 
       const metadata = podcastMetadataSchema.parse(latestVersion.data);
-      
+
       return {
         segments: episode.podcastSegments.map(s => ({
           id: s.id,
@@ -707,7 +736,7 @@ export const podcast = router({
     }))
     .mutation(async ({ ctx, input }) => {
       const episode = await ctx.db.artifact.findFirst({
-        where: { 
+        where: {
           id: input.episodeId,
           type: ArtifactType.PODCAST_EPISODE,
           workspace: workspaceAccessFilter(ctx.session.user.id)
@@ -719,14 +748,14 @@ export const podcast = router({
           },
         },
       });
-      
+
       if (!episode) throw new TRPCError({ code: 'NOT_FOUND' });
-      
+
       const latestVersion = episode.versions[0];
       if (!latestVersion) throw new TRPCError({ code: 'NOT_FOUND', message: 'No version found' });
 
       const metadata = podcastMetadataSchema.parse(latestVersion.data);
-      
+
       // Update metadata
       if (input.title) metadata.title = input.title;
       if (input.description) metadata.description = input.description;
@@ -742,7 +771,7 @@ export const podcast = router({
           createdById: ctx.session.user.id,
         },
       });
-      
+
       // Update the artifact with basic info for listing/searching
       return ctx.db.artifact.update({
         where: { id: input.episodeId },
@@ -759,7 +788,7 @@ export const podcast = router({
     .input(z.object({ episodeId: z.string() }))
     .mutation(async ({ ctx, input }) => {
       const episode = await ctx.db.artifact.findFirst({
-        where: { 
+        where: {
           id: input.episodeId,
           type: ArtifactType.PODCAST_EPISODE,
           workspace: workspaceAccessFilter(ctx.session.user.id)
@@ -771,12 +800,12 @@ export const podcast = router({
           },
         },
       });
-      
+
       if (!episode) throw new TRPCError({ code: 'NOT_FOUND' });
 
       try {
         // Emit episode deletion start notification
-        await PusherService.emitTaskComplete(episode.workspaceId, 'podcast_deletion_start', { 
+        await PusherService.emitTaskComplete(episode.workspaceId, 'podcast_deletion_start', {
           episodeId: input.episodeId,
           episodeTitle: episode.title || 'Untitled Episode'
         });
@@ -813,7 +842,7 @@ export const podcast = router({
         });
 
         // Emit episode deletion completion notification
-        await PusherService.emitTaskComplete(episode.workspaceId, 'podcast_deletion_complete', { 
+        await PusherService.emitTaskComplete(episode.workspaceId, 'podcast_deletion_complete', {
           episodeId: input.episodeId,
           episodeTitle: episode.title || 'Untitled Episode'
         });
@@ -823,9 +852,9 @@ export const podcast = router({
       } catch (error) {
         logger.error('Error deleting episode:', error);
         await PusherService.emitError(episode.workspaceId, `Failed to delete episode: ${error instanceof Error ? error.message : 'Unknown error'}`, 'podcast');
-        throw new TRPCError({ 
-          code: 'INTERNAL_SERVER_ERROR', 
-          message: 'Failed to delete episode' 
+        throw new TRPCError({
+          code: 'INTERNAL_SERVER_ERROR',
+          message: 'Failed to delete episode'
         });
       }
     }),
@@ -835,7 +864,7 @@ export const podcast = router({
     .input(z.object({ segmentId: z.string() }))
     .query(async ({ ctx, input }) => {
       const segment = await ctx.db.podcastSegment.findFirst({
-        where: { 
+        where: {
           id: input.segmentId,
           artifact: {
             workspace: workspaceAccessFilter(ctx.session.user.id)

package/src/routers/studyguide.ts

@@ -2,6 +2,7 @@ import { z } from 'zod';
 import { TRPCError } from '@trpc/server';
 import { router, authedProcedure } from '../trpc.js';
 import { ArtifactType } from '../lib/constants.js';
+import { getUserUsage, getUserPlanLimits } from '../lib/usage_service.js';
 import { workspaceAccessFilter } from '../lib/workspace-access.js';
 
 const initializeEditorJsEmptyBlock = () => ({
@@ -38,6 +39,17 @@ export const studyguide = router({
       });
 
       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,