Update apps/storylines.js

apps/storylines.js

@@ -7,13 +7,12 @@ const router = express.Router();
 // ─────────────────────────────────────────────
 //  CONSTANTS
 // ─────────────────────────────────────────────
-const DAILY_IMAGE_LIMITS = { quick: 20, premium: 5 };   // per user per day
+const DAILY_IMAGE_LIMITS = { quick: 20, premium: 5 };
 const ADMIN_KEY = process.env.STORYLINE_ADMIN_KEY;
 
 // ─────────────────────────────────────────────
 //  IN-MEMORY IMAGE USAGE TRACKER
 //  { [device_id]: { date: "YYYY-MM-DD", quick: n, premium: n } }
-//  Keys untouched for 2 days are pruned automatically
 // ─────────────────────────────────────────────
 const imageUsage = new Map();
 
@@ -22,8 +21,6 @@ const todayStr = () => new Date().toISOString().slice(0, 10);
 const getUsage = (device_id) => {
   const today = todayStr();
   const entry = imageUsage.get(device_id);
-
-  // New day or new user — reset
   if (!entry || entry.date !== today) {
     const fresh = { date: today, quick: 0, premium: 0, lastSeen: Date.now() };
     imageUsage.set(device_id, fresh);
@@ -46,7 +43,7 @@ const pruneUsageCache = () => {
     if (entry.lastSeen < cutoff) imageUsage.delete(id);
   }
 };
-setInterval(pruneUsageCache, 60 * 60 * 1000); // run every hour
+setInterval(pruneUsageCache, 60 * 60 * 1000);
 
 // ─────────────────────────────────────────────
 //  MIDDLEWARE: Admin guard
@@ -54,24 +51,22 @@ setInterval(pruneUsageCache, 60 * 60 * 1000); // run every hour
 const adminOnly = (req, res, next) => {
   const key = req.headers['x-admin-key'];
   if (!ADMIN_KEY || key !== ADMIN_KEY)
-    return res.status(403).json({ success: false, error: "Forbidden." });
+    return res.status(403).json({ success: false, error: 'Forbidden.' });
   next();
 };
 
 // ─────────────────────────────────────────────
 //  STORY SYSTEM PROMPT BUILDER
-//  Pulls active characters from the chapter and
-//  instructs the model to respond as each of them
 // ─────────────────────────────────────────────
-const buildStorySystemPrompt = (chapter, characters) => {
-  const charList = characters.map(c =>
-    `- ${c.name} (${c.personality || "no description"})`
-  ).join('\n');
+const buildStorySystemPrompt = (chapter, characters, storyTitle) => {
+  const charList = characters
+    .map(c => `- ${c.name} (${c.personality || 'no description'})`)
+    .join('\n');
 
-  return `You are the narrative engine of a light novel called "${chapter.story_title}".
+  return `You are the narrative engine of a light novel called "${storyTitle}".
 
 CURRENT CHAPTER: "${chapter.title}"
-CHAPTER CONTEXT: ${chapter.context || "No additional context."}
+CHAPTER CONTEXT: ${chapter.context || 'No additional context.'}
 
 ACTIVE CHARACTERS IN SCENE:
 ${charList}
@@ -103,40 +98,45 @@ EXAMPLE OUTPUT:
 
 // ─────────────────────────────────────────────
 //  POST /storyline/interact
-//  Main story interaction — user sends input,
-//  gets back array of character responses
 // ─────────────────────────────────────────────
 router.post('/interact', async (req, res) => {
   try {
     const { device_id, chapter_id, user_input, history = [] } = req.body;
 
     if (!device_id || !chapter_id || !user_input)
-      return res.status(400).json({ success: false, error: "device_id, chapter_id and user_input are required." });
+      return res.status(400).json({
+        success: false,
+        error: 'device_id, chapter_id and user_input are required.',
+      });
+
+    // Cap history server-side regardless of what the client sends
+    const cappedHistory = history.slice(-10);
 
-    // Fetch chapter + its characters from Supabase
+    // Fetch chapter + parent story title + characters
     const { data: chapter, error: chErr } = await supabase
       .from('storyline_chapters')
-      .select('*, storyline_characters(*)')
+      .select('*, storylines(title), storyline_characters(*)')
       .eq('id', chapter_id)
       .single();
 
     if (chErr || !chapter)
-      return res.status(404).json({ success: false, error: "Chapter not found." });
+      return res.status(404).json({ success: false, error: 'Chapter not found.' });
 
     const characters = chapter.storyline_characters || [];
+    const storyTitle = chapter.storylines?.title ?? 'Unknown Story';
 
-    // Build conversation history into the prompt
-    const historyBlock = history.length > 0
-      ? `\nRECENT HISTORY (last ${history.length} exchanges):\n` +
-        history.map(h => `[${h.role}]: ${h.content}`).join('\n')
-      : '';
+    const historyBlock =
+      cappedHistory.length > 0
+        ? `\nRECENT HISTORY (last ${cappedHistory.length} exchanges):\n` +
+          cappedHistory.map(h => `[${h.role}]: ${h.content}`).join('\n')
+        : '';
 
     const userPrompt = `${historyBlock}\n\n[PLAYER ACTION]: ${user_input}`;
 
     const aiResult = await generateCompletion({
-      model: "maverick",
+      model: 'maverick',
       prompt: userPrompt,
-      system_prompt: buildStorySystemPrompt(chapter, characters),
+      system_prompt: buildStorySystemPrompt(chapter, characters, storyTitle),
     });
 
     if (!aiResult.success) throw new Error(aiResult.error);
@@ -149,28 +149,25 @@ router.post('/interact', async (req, res) => {
       responses: parsed.responses || [],
       model_used: aiResult.model_used,
     });
-
   } catch (err) {
-    console.error("[STORY INTERACT ERROR]", err.message);
+    console.error('[STORY INTERACT ERROR]', err.message);
     res.status(500).json({ success: false, error: err.message });
   }
 });
 
 // ─────────────────────────────────────────────
 //  POST /storyline/generate-image
-//  Generates a scene image, enforces daily limits
 // ─────────────────────────────────────────────
 router.post('/generate-image', async (req, res) => {
   try {
-    const { device_id, prompt, tier = "quick" } = req.body;
+    const { device_id, prompt, tier = 'quick' } = req.body;
 
     if (!device_id || !prompt)
-      return res.status(400).json({ success: false, error: "device_id and prompt required." });
+      return res.status(400).json({ success: false, error: 'device_id and prompt required.' });
 
-    if (!["quick", "premium"].includes(tier))
+    if (!['quick', 'premium'].includes(tier))
       return res.status(400).json({ success: false, error: "tier must be 'quick' or 'premium'." });
 
-    // Check daily limit
     const usage = getUsage(device_id);
     const limit = DAILY_IMAGE_LIMITS[tier];
 
@@ -195,16 +192,14 @@ router.post('/generate-image', async (req, res) => {
       model_used: result.model_used,
       usage: { quick: updated.quick, premium: updated.premium, limits: DAILY_IMAGE_LIMITS },
     });
-
   } catch (err) {
-    console.error("[IMAGE GEN ERROR]", err.message);
+    console.error('[IMAGE GEN ERROR]', err.message);
     res.status(500).json({ success: false, error: err.message });
   }
 });
 
 // ─────────────────────────────────────────────
 //  GET /storyline/stories
-//  Fetch all published stories (public)
 // ─────────────────────────────────────────────
 router.get('/stories', async (req, res) => {
   try {
@@ -221,10 +216,8 @@ router.get('/stories', async (req, res) => {
   }
 });
 
-
 // ─────────────────────────────────────────────
 //  ADMIN: GET /storyline/admin/stories
-//  Fetch ALL stories including drafts
 // ─────────────────────────────────────────────
 router.get('/admin/stories', adminOnly, async (req, res) => {
   try {
@@ -242,7 +235,6 @@ router.get('/admin/stories', adminOnly, async (req, res) => {
 
 // ─────────────────────────────────────────────
 //  GET /storyline/stories/:story_id/chapters
-//  Fetch chapters for a story (public)
 // ─────────────────────────────────────────────
 router.get('/stories/:story_id/chapters', async (req, res) => {
   try {
@@ -262,12 +254,11 @@ router.get('/stories/:story_id/chapters', async (req, res) => {
 
 // ─────────────────────────────────────────────
 //  ADMIN: POST /storyline/admin/stories
-//  Create a new story
 // ─────────────────────────────────────────────
 router.post('/admin/stories', adminOnly, async (req, res) => {
   try {
     const { title, description, cover_image, genre, published = false } = req.body;
-    if (!title) return res.status(400).json({ success: false, error: "title required." });
+    if (!title) return res.status(400).json({ success: false, error: 'title required.' });
 
     const { data, error } = await supabase
       .from('storylines')
@@ -284,15 +275,13 @@ router.post('/admin/stories', adminOnly, async (req, res) => {
 
 // ─────────────���───────────────────────────────
 //  ADMIN: PATCH /storyline/admin/stories/:id
-//  Update a story
 // ─────────────────────────────────────────────
 router.patch('/admin/stories/:id', adminOnly, async (req, res) => {
   try {
     const { id } = req.params;
-    const updates = req.body;
     const { data, error } = await supabase
       .from('storylines')
-      .update(updates)
+      .update(req.body)
       .eq('id', id)
       .select()
       .single();
@@ -306,7 +295,6 @@ router.patch('/admin/stories/:id', adminOnly, async (req, res) => {
 
 // ─────────────────────────────────────────────
 //  ADMIN: DELETE /storyline/admin/stories/:id
-//  Delete a story and its chapters (cascade expected in DB)
 // ─────────────────────────────────────────────
 router.delete('/admin/stories/:id', adminOnly, async (req, res) => {
   try {
@@ -321,12 +309,12 @@ router.delete('/admin/stories/:id', adminOnly, async (req, res) => {
 
 // ─────────────────────────────────────────────
 //  ADMIN: POST /storyline/admin/chapters
-//  Add a chapter (with characters) to a story
 // ─────────────────────────────────────────────
 router.post('/admin/chapters', adminOnly, async (req, res) => {
   try {
     const { story_id, title, order, context, characters = [] } = req.body;
-    if (!story_id || !title) return res.status(400).json({ success: false, error: "story_id and title required." });
+    if (!story_id || !title)
+      return res.status(400).json({ success: false, error: 'story_id and title required.' });
 
     const { data: chapter, error: chErr } = await supabase
       .from('storyline_chapters')
@@ -336,7 +324,6 @@ router.post('/admin/chapters', adminOnly, async (req, res) => {
 
     if (chErr) throw chErr;
 
-    // Insert characters if provided
     if (characters.length > 0) {
       const charRows = characters.map(c => ({ ...c, chapter_id: chapter.id, story_id }));
       const { error: charErr } = await supabase.from('storyline_characters').insert(charRows);
@@ -349,33 +336,41 @@ router.post('/admin/chapters', adminOnly, async (req, res) => {
   }
 });
 
-
+// ─────────────────────────────────────────────
+//  ADMIN: PATCH /storyline/admin/chapters/:id
+// ─────────────────────────────────────────────
 router.patch('/admin/chapters/:id', adminOnly, async (req, res) => {
   try {
     const { id } = req.params;
     const { title, order, content, context, characters } = req.body;
+
     const { data: chapter, error } = await supabase
       .from('storyline_chapters')
       .update({ title, order, content, context })
       .eq('id', id)
       .select()
       .single();
+
     if (error) throw error;
 
     if (characters) {
       await supabase.from('storyline_characters').delete().eq('chapter_id', id);
       if (characters.length > 0) {
-        const rows = characters.map(c => ({ ...c, chapter_id: id, story_id: chapter.story_id }));
+        const rows = characters.map(c => ({
+          ...c,
+          chapter_id: id,
+          story_id: chapter.story_id,
+        }));
         await supabase.from('storyline_characters').insert(rows);
       }
     }
+
     res.json({ success: true, data: chapter });
   } catch (err) {
     res.status(500).json({ success: false, error: err.message });
   }
 });
 
-
 // ─────────────────────────────────────────────
 //  ADMIN: DELETE /storyline/admin/chapters/:id
 // ─────────────────────────────────────────────