app.js

const express = require('express');
const cors = require('cors');
const ArangoDBManager = require('./database');
require('dotenv').config();

const app = express();
const port = process.env.PORT || 3000;

// Middleware
app.use(cors());
app.use(express.json());
app.use(express.static('public'));

// Initialize database
const dbManager = new ArangoDBManager();

// Routes

// User Management (Document operations)
app.post('/api/users', async (req, res) => {
  try {
    const { username, displayName, email, bio } = req.body;
    const result = await dbManager.createUser({
      username,
      displayName,
      email,
      bio
    });
    res.status(201).json(result);
  } catch (error) {
    res.status(400).json({ error: error.message });
  }
});

app.get('/api/users', async (req, res) => {
  try {
    const users = await dbManager.getAllUsers();
    res.json(users);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

// Popular users endpoint - must be before the generic :id route
app.get('/api/users/popular', async (req, res) => {
  try {
    const limit = parseInt(req.query.limit) || 5;
    console.log('Getting popular users with limit:', limit);
    const popularUsers = await dbManager.getPopularUsers(limit);
    console.log('Popular users result:', popularUsers);
    res.json(popularUsers);
  } catch (error) {
    console.error('Error in /api/users/popular:', error);
    res.status(500).json({ error: error.message });
  }
});

app.get('/api/users/:id', async (req, res) => {
  try {
    const user = await dbManager.getUser(req.params.id);
    res.json(user);
  } catch (error) {
    res.status(404).json({ error: 'User not found' });
  }
});

// Post Management (Document operations)
app.post('/api/posts', async (req, res) => {
  try {
    const { authorId, content, title, tags } = req.body;
    const result = await dbManager.createPost({
      authorId,
      content,
      title,
      tags: tags || []
    });
    res.status(201).json(result);
  } catch (error) {
    res.status(400).json({ error: error.message });
  }
});

app.get('/api/posts', async (req, res) => {
  try {
    const posts = await dbManager.getAllPosts();
    res.json(posts);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

// Graph Operations
app.post('/api/users/:followerId/follow/:followeeId', async (req, res) => {
  try {
    const { followerId, followeeId } = req.params;
    
    // Validation
    if (followerId === followeeId) {
      return res.status(400).json({ error: 'Cannot follow yourself' });
    }
    
    const result = await dbManager.followUser(followerId, followeeId);
    res.status(201).json(result);
  } catch (error) {
    console.error('Follow user error:', error);
    res.status(400).json({ error: error.message });
  }
});

app.post('/api/users/:userId/like/:postId', async (req, res) => {
  try {
    const { userId, postId } = req.params;
    console.log('Like post request:', { userId, postId });
    const result = await dbManager.likePost(userId, postId);
    console.log('Like post result:', result);
    res.status(201).json(result);
  } catch (error) {
    console.error('Like post error:', error);
    res.status(400).json({ error: error.message });
  }
});

// Debug endpoints for graph operations
app.get('/api/debug/follows/:userId?', async (req, res) => {
  try {
    const { userId } = req.params;
    const follows = await dbManager.getFollowRelationships(userId);
    res.json(follows);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

app.get('/api/debug/likes/:userId?', async (req, res) => {
  try {
    const { userId } = req.params;
    const likes = await dbManager.getLikeRelationships(userId);
    res.json(likes);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

// Simplified debug endpoints for frontend
app.get('/api/debug/graph', async (req, res) => {
  try {
    const follows = await dbManager.getFollowRelationships();
    const likes = await dbManager.getLikeRelationships();
    res.json({
      relationships: {
        follows: follows.length,
        likes: likes.length
      },
      details: {
        follows: follows,
        likes: likes
      }
    });
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

app.get('/api/users/:userId/stats', async (req, res) => {
  try {
    const { userId } = req.params;
    const stats = await dbManager.getUserStats(userId);
    res.json(stats);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

// Key-Value Operations
app.post('/api/users/:userId/settings/:key', async (req, res) => {
  try {
    const { userId, key } = req.params;
    const { value } = req.body;
    const result = await dbManager.setUserSetting(userId, key, value);
    res.status(201).json(result);
  } catch (error) {
    res.status(400).json({ error: error.message });
  }
});

app.get('/api/users/:userId/settings/:key', async (req, res) => {
  try {
    const { userId, key } = req.params;
    const setting = await dbManager.getUserSetting(userId, key);
    if (setting) {
      res.json(setting);
    } else {
      res.status(404).json({ error: 'Setting not found' });
    }
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

// Complex Multimodal Queries
app.get('/api/users/:userId/feed', async (req, res) => {
  try {
    const { userId } = req.params;
    const limit = parseInt(req.query.limit) || 10;
    const feed = await dbManager.getUserFeed(userId, limit);
    res.json(feed);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

app.get('/api/users/:userId1/mutual/:userId2', async (req, res) => {
  try {
    const { userId1, userId2 } = req.params;
    const mutualFollows = await dbManager.getMutualFollows(userId1, userId2);
    res.json(mutualFollows);
  } catch (error) {
    res.status(500).json({ error: error.message });
  }
});

// Health check
app.get('/health', (req, res) => {
  res.json({ status: 'OK', timestamp: new Date().toISOString() });
});

// Serve the demo page
app.get('/', (req, res) => {
  res.sendFile(__dirname + '/public/index.html');
});

// Start server
async function startServer() {
  try {
    await dbManager.initializeDatabase();
    app.listen(port, () => {
      console.log(`🚀 ArangoDB Multimodal App running on http://localhost:${port}`);
      console.log(`📊 Database: ${process.env.ARANGO_DATABASE || 'multimodal_demo'}`);
      console.log(`🔗 ArangoDB: ${process.env.ARANGO_URL || 'http://localhost:8529'}`);
    });
  } catch (error) {
    console.error('Failed to start server:', error);
    process.exit(1);
  }
}

startServer();