Update server.js

server.js

@@ -9,6 +9,7 @@ import jwt from 'jsonwebtoken';
 import FormData from 'form-data';
 import pg from 'pg';
 import fetch from 'node-fetch';
+import NodeCache from 'node-cache';
 
 dotenv.config();
 
@@ -99,6 +100,9 @@ const pool = new Pool({
   }
 });
 
+// Initialize cache (default TTL: 60 seconds)
+const cache = new NodeCache({ stdTTL: 60, checkperiod: 120 });
+
 // Create tables if they don't exist
 const initializeDatabase = async () => {
   try {
@@ -132,6 +136,11 @@ const initializeDatabase = async () => {
     `);
     // Add censored column if it doesn't exist
     await pool.query(`ALTER TABLE images ADD COLUMN IF NOT EXISTS censored BOOLEAN DEFAULT false;`);
+    // Add indexes for performance
+    await pool.query(`CREATE INDEX IF NOT EXISTS idx_images_user_id ON images(user_id);`);
+    await pool.query(`CREATE INDEX IF NOT EXISTS idx_images_created_at ON images(created_at);`);
+    await pool.query(`CREATE INDEX IF NOT EXISTS idx_user_generations_user_id ON user_generations(user_id);`);
+    await pool.query(`CREATE INDEX IF NOT EXISTS idx_user_generations_created_at ON user_generations(created_at);`);
     console.log('Database initialized successfully');
   } catch (error) {
     console.error('Error initializing database:', error);
@@ -322,14 +331,25 @@ app.post('/api/upload-image', authenticateToken, async (req, res) => {
   }
 });
 
+// Community images with cache and pagination
 app.get('/api/community-images', async (req, res) => {
   try {
+    const page = parseInt(req.query.page) || 1;
+    const limit = parseInt(req.query.limit) || 50;
+    const offset = (page - 1) * limit;
+    const cacheKey = `community-images:${page}:${limit}`;
+    const cached = cache.get(cacheKey);
+    if (cached) {
+      return res.json(cached);
+    }
     const result = await pool.query(`
       SELECT i.id, i.image_url, i.uploaded_url, i.prompt, i.created_at, i.censored, u.username 
       FROM images i 
       JOIN users u ON i.user_id = u.id 
       ORDER BY i.created_at DESC
-    `);
+      LIMIT $1 OFFSET $2
+    `, [limit, offset]);
+    cache.set(cacheKey, result.rows);
     res.json(result.rows);
   } catch (error) {
     console.error('Error fetching community images:', error);
@@ -796,18 +816,22 @@ app.post('/api/add-community-image', authenticateToken, async (req, res) => {
     }
 });
 
-// Auth status endpoint
+// Auth status endpoint with cache
 app.get('/api/auth/status', authenticateToken, async (req, res) => {
     try {
+        const cacheKey = `auth-status:${req.user.id}`;
+        const cached = cache.get(cacheKey);
+        if (cached) {
+            return res.json({ user: cached });
+        }
         const result = await pool.query(
             'SELECT id, username, email FROM users WHERE id = $1',
             [req.user.id]
         );
-        
         if (result.rows.length === 0) {
             return res.status(401).json({ error: 'User not found' });
         }
-        
+        cache.set(cacheKey, result.rows[0]);
         res.json({ user: result.rows[0] });
     } catch (error) {
         console.error('Auth status error:', error);
@@ -817,21 +841,27 @@ app.get('/api/auth/status', authenticateToken, async (req, res) => {
 
 // Admin endpoints (no authentication required for admin functions)
 
-// Get image statistics
+// Get image statistics with cache
 app.get('/api/admin/stats', async (req, res) => {
   try {
+    const cacheKey = 'admin-stats';
+    const cached = cache.get(cacheKey);
+    if (cached) {
+      return res.json(cached);
+    }
     const totalImages = await pool.query('SELECT COUNT(*) as count FROM images');
     const uniqueUsers = await pool.query('SELECT COUNT(DISTINCT user_id) as count FROM images');
     const last24Hours = await pool.query(`
       SELECT COUNT(*) as count FROM images 
       WHERE created_at >= NOW() - INTERVAL '24 hours'
     `);
-    
-    res.json({
+    const stats = {
       totalImages: parseInt(totalImages.rows[0].count),
       uniqueUsers: parseInt(uniqueUsers.rows[0].count),
       last24Hours: parseInt(last24Hours.rows[0].count)
-    });
+    };
+    cache.set(cacheKey, stats);
+    res.json(stats);
   } catch (error) {
     console.error('Error getting admin stats:', error);
     res.status(500).json({ error: 'Error getting statistics' });