Merge pull request #1758 from 2anki/feat/core-ki-cleanup2

fix: get the scroll working for really long results

Author: aalemayhu

src/controllers/UsersControllers.ts

@@ -185,6 +185,19 @@ class UsersController {
       );
     }
 
+    const featureFlags = {
+      kiUI: false,
+    };
+
+    const shouldEnableFeatures = user?.id
+      ? parseInt(user.id.toString(), 16) % 10 < 4 // Use user ID for deterministic result (40%)
+      : false;
+
+    console.log('shouldEnableFeatures', shouldEnableFeatures);
+    if (shouldEnableFeatures) {
+      featureFlags.kiUI = user?.patreon || res.locals.subscriber;
+    }
+
     const response = {
       user: {
         id: user?.id,
@@ -195,6 +208,7 @@ class UsersController {
       },
       locals,
       linked_email: linkedEmail,
+      features: featureFlags, // Add feature flags to response
     };
 
     return res.json(response);

src/routes/ki/KiRouter.ts

@@ -160,8 +160,15 @@ const KiRouter = () => {
 
     console.log('[KI] User:', user.id);
     if (!user.patreon) {
-      res.redirect('/pricing?redirect=/ki');
-      return false;
+      const subscriber = await authService.getIsSubscriber(
+        database,
+        user.email
+      );
+      console.log('[KI] Subscriber:', subscriber);
+      if (!subscriber) {
+        res.redirect('/pricing?redirect=/ki');
+        return false;
+      }
     }
 
     return true;
@@ -303,14 +310,25 @@ const KiRouter = () => {
       return false;
     }
 
-    const userId = user.owner; // Assuming token contains userId
-    const userSession = await sessionRepository.getSession(userId); // Ensure session is typed
+    const userId = user.owner;
+    let userSession = await sessionRepository.getSession(userId);
+
+    // Create new session if none exists
+    if (!userSession || !userSession.data) {
+      const w = new Workspace(true, 'fs');
+      await sessionRepository.createSession(userId, {
+        uploadedFiles: [],
+        deckInfo: [],
+        workspace: w,
+      });
+      userSession = await sessionRepository.getSession(userId);
+    }
+
     if (!userSession || !userSession.data) {
-      // Check if userSession.data is defined
       return res.status(403).json({ error: 'Session expired' });
     }
 
-    const files = req.files as UploadedFile[]; // Ensure files are typed
+    const files = req.files as UploadedFile[];
     console.log('[UPLOAD] Request received:', req.body);
     console.log('[UPLOAD] Starting file upload process');
     try {
@@ -1348,6 +1366,98 @@ const KiRouter = () => {
     }
   });
 
+  // Add cache endpoints
+  router.post('/ki/cache', async (req, res) => {
+    const user = await authService.getUserFrom(req.cookies.token);
+    if (!user) {
+      return res.status(403).json({ error: 'Session expired' });
+    }
+
+    const { text, cards } = req.body;
+    if (!text || !cards) {
+      return res.status(400).json({ error: 'Missing required fields' });
+    }
+
+    try {
+      // Store in user session - simple approach
+      const userSession = await sessionRepository.getSession(user.owner);
+      if (!userSession || !userSession.data) {
+        return res.status(403).json({ error: 'Session expired' });
+      }
+
+      // Add cache data to session
+      await sessionRepository.updateSession(
+        user.owner,
+        {
+          cache: { text, cards, timestamp: Date.now() },
+        },
+        userSession.id
+      );
+
+      res.status(200).json({ success: true });
+    } catch (error) {
+      console.error('[CACHE] Error saving cache:', error);
+      res.status(500).json({ error: 'Failed to save cache' });
+    }
+  });
+
+  router.get('/ki/cache', async (req, res) => {
+    const user = await authService.getUserFrom(req.cookies.token);
+    if (!user) {
+      return res.status(403).json({ error: 'Session expired' });
+    }
+
+    const text = req.query.text as string;
+    if (!text) {
+      return res.status(400).json({ error: 'Text parameter is required' });
+    }
+
+    try {
+      const userSession = await sessionRepository.getSession(user.owner);
+      if (!userSession || !userSession.data || !userSession.data.cache) {
+        return res.status(404).json({ error: 'No cache found' });
+      }
+
+      // Simple exact match for now
+      if (userSession.data.cache.text === text) {
+        return res.status(200).json({ cards: userSession.data.cache.cards });
+      }
+
+      res.status(404).json({ error: 'Cache miss' });
+    } catch (error) {
+      console.error('[CACHE] Error retrieving cache:', error);
+      res.status(500).json({ error: 'Failed to retrieve cache' });
+    }
+  });
+
+  router.delete('/ki/cache', async (req, res) => {
+    const user = await authService.getUserFrom(req.cookies.token);
+    if (!user) {
+      return res.status(403).json({ error: 'Session expired' });
+    }
+
+    try {
+      const userSession = await sessionRepository.getSession(user.owner);
+      if (!userSession || !userSession.data) {
+        return res.status(403).json({ error: 'Session expired' });
+      }
+
+      // Remove cache from session
+      await sessionRepository.updateSession(
+        user.owner,
+        {
+          cache: null,
+        },
+        userSession.id
+      );
+
+      res.status(200).json({ success: true });
+    } catch (error) {
+      console.error('[CACHE] Error clearing cache:', error);
+      res.status(500).json({ error: 'Failed to clear cache' });
+    }
+  });
+
   return router;
 };