@goscribe/server 1.3.0 → 1.3.1

package/dist/routers/podcast.js

@@ -1,17 +1,15 @@
 import { z } from 'zod';
 import { TRPCError } from '@trpc/server';
-import { router, authedProcedure } from '../trpc.js';
+import { router, authedProcedure, limitedProcedure } from '../trpc.js';
 import { v4 as uuidv4 } from 'uuid';
 import inference from '../lib/inference.js';
 import { generateSignedUrl, deleteFromSupabase } from '../lib/storage.js';
 import PusherService from '../lib/pusher.js';
 import { aiSessionService } from '../lib/ai-session.js';
-// Prisma enum values mapped manually to avoid type import issues in ESM
-const ArtifactType = {
-    PODCAST_EPISODE: 'PODCAST_EPISODE',
-    STUDY_GUIDE: 'STUDY_GUIDE',
-    FLASHCARD_SET: 'FLASHCARD_SET',
-};
+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({
     id: z.string(),
@@ -85,9 +83,9 @@ export const podcast = router({
             },
             orderBy: { updatedAt: 'desc' },
         });
-        console.log(`📻 Found ${artifacts.length} podcast artifacts`);
+        logger.debug(`Found ${artifacts.length} podcast artifacts`);
         artifacts.forEach((artifact, i) => {
-            console.log(`  Podcast ${i + 1}: "${artifact.title}" - ${artifact.podcastSegments.length} segments`);
+            logger.debug(`  Podcast ${i + 1}: "${artifact.title}" - ${artifact.podcastSegments.length} segments`);
         });
         // Transform to include segments with fresh signed URLs
         const episodesWithUrls = await Promise.all(artifacts.map(async (artifact) => {
@@ -112,7 +110,7 @@ export const podcast = router({
                         };
                     }
                     catch (error) {
-                        console.error(`Failed to generate signed URL for segment ${segment.id}:`, error);
+                        logger.error(`Failed to generate signed URL for segment ${segment.id}:`, error);
                         return {
                             id: segment.id,
                             title: segment.title,
@@ -138,11 +136,11 @@ export const podcast = router({
             let metadata = null;
             if (latestVersion) {
                 try {
-                    console.log(latestVersion.data);
+                    logger.debug(JSON.stringify(latestVersion.data));
                     metadata = podcastMetadataSchema.parse(latestVersion.data);
                 }
                 catch (error) {
-                    console.error('Failed to parse podcast metadata:', error);
+                    logger.error('Failed to parse podcast metadata:', error);
                 }
             }
             return {
@@ -172,7 +170,7 @@ export const podcast = router({
             where: {
                 id: input.episodeId,
                 type: ArtifactType.PODCAST_EPISODE,
-                workspace: { ownerId: ctx.session.user.id }
+                workspace: workspaceAccessFilter(ctx.session.user.id)
             },
             include: {
                 versions: {
@@ -184,18 +182,18 @@ export const podcast = router({
                 },
             },
         });
-        console.log(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' });
-        console.log(latestVersion);
+        logger.debug(JSON.stringify(latestVersion));
         try {
             const metadata = podcastMetadataSchema.parse(latestVersion.data);
         }
         catch (error) {
-            console.error('Failed to parse podcast metadata:', error);
+            logger.error('Failed to parse podcast metadata:', error);
         }
         const metadata = podcastMetadataSchema.parse(latestVersion.data);
         const imageUrl = episode.imageObjectKey ? await generateSignedUrl(episode.imageObjectKey, 24) : null;
@@ -217,7 +215,7 @@ export const podcast = router({
                     };
                 }
                 catch (error) {
-                    console.error(`Failed to generate signed URL for segment ${segment.id}:`, error);
+                    logger.error(`Failed to generate signed URL for segment ${segment.id}:`, error);
                     return {
                         id: segment.id,
                         title: segment.title,
@@ -256,7 +254,7 @@ export const podcast = router({
         };
     }),
     // Generate podcast episode from text input
-    generateEpisode: authedProcedure
+    generateEpisode: limitedProcedure
         .input(z.object({
         workspaceId: z.string(),
         podcastData: podcastInputSchema,
@@ -378,7 +376,7 @@ export const podcast = router({
                 }
                 catch (audioError) {
                     const errorMessage = audioError instanceof Error ? audioError.message : 'Unknown error';
-                    console.error(`❌ Error generating audio for segment ${i + 1}:`, {
+                    logger.error(`❌ Error generating audio for segment ${i + 1}:`, {
                         title: segment.title,
                         error: errorMessage,
                         stack: audioError instanceof Error ? audioError.stack : undefined,
@@ -401,9 +399,17 @@ export const podcast = router({
             }
             // Check if any segments were successfully generated
             if (segments.length === 0) {
-                console.error('No segments were successfully generated');
+                logger.error('No segments were successfully generated');
                 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 },
                 });
@@ -447,7 +453,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 = summaryResponse.choices[0].message.content || '';
             let episodeSummary;
             try {
@@ -459,7 +465,7 @@ export const podcast = router({
                 episodeSummary = JSON.parse(jsonMatch[0]);
             }
             catch (parseError) {
-                console.error('Failed to parse summary response:', summaryContent);
+                logger.error('Failed to parse summary response:', summaryContent);
                 await PusherService.emitTaskComplete(input.workspaceId, 'podcast_summary_error', {
                     error: 'Failed to parse summary response'
                 });
@@ -533,7 +539,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,
                 title: metadata.title,
@@ -543,7 +556,17 @@ export const podcast = router({
             };
         }
         catch (error) {
-            console.error('Error generating podcast episode:', error);
+            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,
@@ -570,7 +593,7 @@ export const podcast = router({
             where: {
                 id: input.episodeId,
                 type: ArtifactType.PODCAST_EPISODE,
-                workspace: { ownerId: ctx.session.user.id }
+                workspace: workspaceAccessFilter(ctx.session.user.id)
             },
             include: {
                 versions: {
@@ -618,7 +641,7 @@ export const podcast = router({
             where: {
                 id: input.episodeId,
                 type: ArtifactType.PODCAST_EPISODE,
-                workspace: { ownerId: ctx.session.user.id }
+                workspace: workspaceAccessFilter(ctx.session.user.id)
             },
             include: {
                 versions: {
@@ -667,7 +690,7 @@ export const podcast = router({
             where: {
                 id: input.episodeId,
                 type: ArtifactType.PODCAST_EPISODE,
-                workspace: { ownerId: ctx.session.user.id }
+                workspace: workspaceAccessFilter(ctx.session.user.id)
             },
             include: {
                 versions: {
@@ -695,7 +718,7 @@ export const podcast = router({
                         await deleteFromSupabase(segment.objectKey);
                     }
                     catch (error) {
-                        console.error(`Failed to delete audio file ${segment.objectKey}:`, error);
+                        logger.error(`Failed to delete audio file ${segment.objectKey}:`, error);
                     }
                 }
             }
@@ -719,7 +742,7 @@ export const podcast = router({
             return true;
         }
         catch (error) {
-            console.error('Error deleting episode:', 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',
@@ -735,7 +758,7 @@ export const podcast = router({
             where: {
                 id: input.segmentId,
                 artifact: {
-                    workspace: { ownerId: ctx.session.user.id }
+                    workspace: workspaceAccessFilter(ctx.session.user.id)
                 }
             },
             include: {
@@ -751,7 +774,7 @@ export const podcast = router({
                 audioUrl = await generateSignedUrl(segment.objectKey, 24); // 24 hours
             }
             catch (error) {
-                console.error(`Failed to generate signed URL for segment ${segment.id}:`, error);
+                logger.error(`Failed to generate signed URL for segment ${segment.id}:`, error);
             }
         }
         return {

package/dist/routers/studyguide.d.ts

@@ -1,11 +1,5 @@
 export declare const studyguide: import("@trpc/server").TRPCBuiltRouter<{
-    ctx: {
-        db: import("@prisma/client").PrismaClient<import("@prisma/client").Prisma.PrismaClientOptions, never, import("@prisma/client/runtime/library").DefaultArgs>;
-        session: any;
-        req: import("express").Request<import("express-serve-static-core").ParamsDictionary, any, any, import("qs").ParsedQs, Record<string, any>>;
-        res: import("express").Response<any, Record<string, any>>;
-        cookies: Record<string, string | undefined>;
-    };
+    ctx: import("../context.js").Context;
     meta: object;
     errorShape: {
         data: {
@@ -39,26 +33,4 @@ export declare const studyguide: import("@trpc/server").TRPCBuiltRouter<{
         };
         meta: object;
     }>;
-    edit: import("@trpc/server").TRPCMutationProcedure<{
-        input: {
-            workspaceId: string;
-            content: string;
-            studyGuideId?: string | undefined;
-            data?: Record<string, unknown> | undefined;
-            title?: string | undefined;
-        };
-        output: {
-            artifactId: string;
-            version: {
-                id: string;
-                createdAt: Date;
-                createdById: string | null;
-                artifactId: string;
-                content: string;
-                data: import("@prisma/client/runtime/library").JsonValue | null;
-                version: number;
-            };
-        };
-        meta: object;
-    }>;
 }>>;

package/dist/routers/studyguide.js

@@ -1,16 +1,16 @@
 import { z } from 'zod';
+import { TRPCError } from '@trpc/server';
 import { router, authedProcedure } from '../trpc.js';
-// Mirror Prisma enum to avoid direct type import
-const ArtifactType = {
-    STUDY_GUIDE: 'STUDY_GUIDE',
-};
+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(),
     blocks: [
         {
             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',
@@ -27,14 +27,23 @@ 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,
@@ -58,68 +67,68 @@ export const studyguide = router({
         return { artifactId: artifact.id, title: artifact.title, latestVersion };
     }),
     // 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;
-        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 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' },
-        });
-        if (input.title && input.title !== artifact.title) {
-            console.log('rename');
-            await ctx.db.artifact.update({
-                where: { id: artifact.id },
-                data: { title: input.title },
-            });
-        }
-        const nextVersion = (last?.version ?? 0) + 1;
-        const version = await ctx.db.artifactVersion.create({
-            data: {
-                artifactId: artifact.id,
-                content: input.content,
-                data: input.data,
-                version: nextVersion,
-                createdById: ctx.session.user.id,
-            },
-        });
-        return { artifactId: artifact.id, version };
-    }),
+    // 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: 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,
+    //         },
+    //       });
+    //     }
+    //     const last = await ctx.db.artifactVersion.findFirst({
+    //       where: { artifactId: artifact.id },
+    //       orderBy: { version: 'desc' },
+    //     });
+    //     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 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 };
+    //   }),
 });