Update server.js

server.js

@@ -1,64 +1,47 @@
 const express = require('express');
 const session = require('express-session');
+const path = require('path');
+const multer = require('multer');
 const authRoutes = require('./routes/auth');
-const userRoutes = require('./routes/user');
 const adminRoutes = require('./routes/admin');
+const userRoutes = require('./routes/user');
 const investmentRoutes = require('./routes/investment');
-const cron = require('./cron/roiCron');
-const path = require('path');
-const fs = require('fs');
-const multer = require('multer');
-require('dotenv').config();
 
 const app = express();
 
-// Create upload directory in project root
-const uploadDir = path.join(__dirname, 'kyc_uploads');
-if (!fs.existsSync(uploadDir)) {
-  fs.mkdirSync(uploadDir, { recursive: true });
-}
-
-// Configure multer for file uploads
-const storage = multer.diskStorage({
-  destination: (req, file, cb) => {
-    cb(null, uploadDir);
-  },
-  filename: (req, file, cb) => {
-    cb(null, `${Date.now()}-${file.originalname}`);
-  }
-});
-const upload = multer({ 
-  storage,
-  fileFilter: (req, file, cb) => {
-    const allowedTypes = ['image/jpeg', 'image/png', 'application/pdf'];
-    if (allowedTypes.includes(file.mimetype)) {
-      cb(null, true);
-    } else {
-      cb(new Error('Invalid file type. Only JPG, PNG, PDF allowed.'));
-    }
-  },
-  limits: { fileSize: 5 * 1024 * 1024 } // 5MB limit
-});
+// Set view engine
+app.set('view engine', 'ejs');
+app.set('views', path.join(__dirname, 'public/views'));
 
-app.use(express.json());
+// Middleware
 app.use(express.urlencoded({ extended: true }));
+app.use(express.static(path.join(__dirname, 'public')));
 app.use(session({
-  secret: process.env.SESSION_SECRET || 'your-secret-key',
+  secret: process.env.SESSION_SECRET || 'your-session-secret',
   resave: false,
   saveUninitialized: false
 }));
-app.set('view engine', 'ejs');
-app.set('views', path.join(__dirname, 'public/views'));
-app.use(express.static(path.join(__dirname, 'public')));
-app.use('/kyc_uploads', express.static(uploadDir));
 
-// Make upload middleware available to routes
-app.use((req, res, next) => {
-  req.upload = upload;
-  next();
+// Multer setup for file uploads
+const storage = multer.diskStorage({
+  destination: (req, file, cb) => cb(null, 'kyc_uploads/'),
+  filename: (req, file, cb) => cb(null, `${Date.now()}-${file.originalname}`)
 });
+const upload = multer({ storage });
+app.use('/user/kyc', upload.fields([
+  { name: 'id_doc', maxCount: 1 },
+  { name: 'utility_doc', maxCount: 1 },
+  { name: 'selfie_doc', maxCount: 1 },
+  { name: 'ssn_doc', maxCount: 1 }
+]));
+
+// Routes
+app.use('/', authRoutes);
+app.use('/admin', adminRoutes);
+app.use('/user', userRoutes);
+app.use('/investment', investmentRoutes);
 
-// Serve KYC documents securely
+// Serve KYC documents (admin access only)
 app.get('/admin/kyc-document/:id', async (req, res) => {
   try {
     const [doc] = await require('./config/db').query('SELECT file_path FROM kyc_documents WHERE id = ?', [req.params.id]);
@@ -69,12 +52,11 @@ app.get('/admin/kyc-document/:id', async (req, res) => {
   }
 });
 
-app.use('/', authRoutes);
-app.use('/user', userRoutes);
-app.use('/admin', adminRoutes);
-app.use('/investment', investmentRoutes);
-
-app.get('/', (req, res) => res.render('index'));
+// Error handling
+app.use((err, req, res, next) => {
+  console.error(err.stack);
+  res.status(500).send('Something went wrong!');
+});
 
 const PORT = process.env.PORT || 3000;
 app.listen(PORT, () => console.log(`Server running on port ${PORT}`));