Update routes/user.js

routes/user.js

@@ -1,25 +1,24 @@
 const express = require('express');
 const router = express.Router();
-const ApiKey = require('../models/ApiKey');
+const db = require('../database');
 
 router.get('/', (req, res) => {
     if (req.session.user) {
-        res.render('user.html');
+        db.query(`SELECT api_key, credits FROM users WHERE id = ?`, [req.session.user.id], (err, results) => {
+            if (err || results.length === 0) return res.status(500).send('Error loading user data');
+            res.render('user', { apiKey: results[0].api_key, credits: results[0].credits });
+        });
     } else {
-        res.status(403).send('Unauthorized');
+        res.redirect('/auth/login');
     }
 });
 
 router.post('/generate-api-key', (req, res) => {
-    const apiKey = generateApiKey();
-    ApiKey.create(req.session.user.id, apiKey, (err) => {
+    const apiKey = 'api_' + Math.random().toString(36).substr(2, 16);
+    db.query(`UPDATE users SET api_key = ? WHERE id = ?`, [apiKey, req.session.user.id], (err) => {
         if (err) return res.status(500).send('Failed to generate API key');
-        res.send(`API Key: ${apiKey}`);
+        res.redirect('/user');
     });
 });
 
-function generateApiKey() {
-    return 'api_' + Math.random().toString(36).substr(2, 16);
-}
-
 module.exports = router;