Create storylines.js

apps/storylines.js

@@ -0,0 +1,347 @@
+import express from 'express';
+import { generateCompletion, generateImage } from '../ai_engine.js';
+import { supabase } from '../config/supabaseClient.js';
+
+const router = express.Router();
+
+// ─────────────────────────────────────────────
+//  CONSTANTS
+// ─────────────────────────────────────────────
+const DAILY_IMAGE_LIMITS = { quick: 20, premium: 5 };   // per user per day
+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();
+
+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);
+    return fresh;
+  }
+  entry.lastSeen = Date.now();
+  return entry;
+};
+
+const incrementUsage = (device_id, tier) => {
+  const entry = getUsage(device_id);
+  entry[tier] = (entry[tier] || 0) + 1;
+  imageUsage.set(device_id, entry);
+};
+
+// Prune keys not touched in 48 hours
+const pruneUsageCache = () => {
+  const cutoff = Date.now() - 48 * 60 * 60 * 1000;
+  for (const [id, entry] of imageUsage.entries()) {
+    if (entry.lastSeen < cutoff) imageUsage.delete(id);
+  }
+};
+setInterval(pruneUsageCache, 60 * 60 * 1000); // run every hour
+
+// ─────────────────────────────────────────────
+//  MIDDLEWARE: Admin guard
+// ─────────────────────────────────────────────
+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." });
+  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');
+
+  return `You are the narrative engine of a light novel called "${chapter.story_title}".
+
+CURRENT CHAPTER: "${chapter.title}"
+CHAPTER CONTEXT: ${chapter.context || "No additional context."}
+
+ACTIVE CHARACTERS IN SCENE:
+${charList}
+
+THE USER is the main character. They have just taken an action or spoken.
+
+YOUR JOB:
+Respond with a JSON object containing ONE key: "responses"
+"responses" is an array of character reaction objects.
+Each object has exactly one key — the character's name — mapped to their reaction string.
+
+RULES:
+- Only include characters who would realistically react to this specific input
+- Reactions must feel natural, in-character, and emotionally alive
+- Include physical actions using *asterisks* e.g. *slams table* or *avoids eye contact*
+- You may group silent/background characters: e.g. {"The class": "*goes quiet*"}
+- Keep each reaction to 1-3 sentences max
+- Return ONLY the raw JSON. No markdown fences, no extra text.
+
+EXAMPLE OUTPUT:
+{
+  "responses": [
+    {"Andy": "*shouts* What the hell dude, I said we take that to the grave!"},
+    {"Amy": "*covers her mouth, clearly holding back laughter*"},
+    {"Jeff and Amy": "*exchange a look and burst out giggling*"}
+  ]
+}`;
+};
+
+// ─────────────────────────────────────────────
+//  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." });
+
+    // Fetch chapter + its characters from Supabase
+    const { data: chapter, error: chErr } = await supabase
+      .from('storyline_chapters')
+      .select('*, storyline_characters(*)')
+      .eq('id', chapter_id)
+      .single();
+
+    if (chErr || !chapter)
+      return res.status(404).json({ success: false, error: "Chapter not found." });
+
+    const characters = chapter.storyline_characters || [];
+
+    // 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 userPrompt = `${historyBlock}\n\n[PLAYER ACTION]: ${user_input}`;
+
+    const aiResult = await generateCompletion({
+      model: "maverick",
+      prompt: userPrompt,
+      system_prompt: buildStorySystemPrompt(chapter, characters),
+    });
+
+    if (!aiResult.success) throw new Error(aiResult.error);
+
+    const parsed = JSON.parse(aiResult.data);
+
+    res.json({
+      success: true,
+      chapter_id,
+      responses: parsed.responses || [],
+      model_used: aiResult.model_used,
+    });
+
+  } catch (err) {
+    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;
+
+    if (!device_id || !prompt)
+      return res.status(400).json({ success: false, error: "device_id and prompt required." });
+
+    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];
+
+    if (usage[tier] >= limit) {
+      return res.status(429).json({
+        success: false,
+        error: `Daily ${tier} image limit reached (${limit}/day). Resets tomorrow.`,
+        limit,
+        used: usage[tier],
+      });
+    }
+
+    const result = await generateImage({ prompt, tier });
+    if (!result.success) throw new Error(result.error);
+
+    incrementUsage(device_id, tier);
+
+    const updated = getUsage(device_id);
+    res.json({
+      success: true,
+      url: result.url,
+      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);
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  GET /storyline/stories
+//  Fetch all published stories (public)
+// ─────────────────────────────────────────────
+router.get('/stories', async (req, res) => {
+  try {
+    const { data, error } = await supabase
+      .from('storylines')
+      .select('id, title, description, cover_image, genre, created_at')
+      .eq('published', true)
+      .order('created_at', { ascending: false });
+
+    if (error) throw error;
+    res.json({ success: true, data });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  GET /storyline/stories/:story_id/chapters
+//  Fetch chapters for a story (public)
+// ─────────────────────────────────────────────
+router.get('/stories/:story_id/chapters', async (req, res) => {
+  try {
+    const { story_id } = req.params;
+    const { data, error } = await supabase
+      .from('storyline_chapters')
+      .select('id, title, order, context, storyline_characters(name, personality)')
+      .eq('story_id', story_id)
+      .order('order', { ascending: true });
+
+    if (error) throw error;
+    res.json({ success: true, data });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  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." });
+
+    const { data, error } = await supabase
+      .from('storylines')
+      .insert({ title, description, cover_image, genre, published })
+      .select()
+      .single();
+
+    if (error) throw error;
+    res.json({ success: true, data });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  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)
+      .eq('id', id)
+      .select()
+      .single();
+
+    if (error) throw error;
+    res.json({ success: true, data });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  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 {
+    const { id } = req.params;
+    const { error } = await supabase.from('storylines').delete().eq('id', id);
+    if (error) throw error;
+    res.json({ success: true, message: `Story ${id} deleted.` });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  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." });
+
+    const { data: chapter, error: chErr } = await supabase
+      .from('storyline_chapters')
+      .insert({ story_id, title, order: order ?? 0, context })
+      .select()
+      .single();
+
+    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);
+      if (charErr) throw charErr;
+    }
+
+    res.json({ success: true, data: chapter });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+// ─────────────────────────────────────────────
+//  ADMIN: DELETE /storyline/admin/chapters/:id
+// ─────────────────────────────────────────────
+router.delete('/admin/chapters/:id', adminOnly, async (req, res) => {
+  try {
+    const { id } = req.params;
+    const { error } = await supabase.from('storyline_chapters').delete().eq('id', id);
+    if (error) throw error;
+    res.json({ success: true, message: `Chapter ${id} deleted.` });
+  } catch (err) {
+    res.status(500).json({ success: false, error: err.message });
+  }
+});
+
+export default router;