const express = require('express');
const router = express.Router();
const multer = require('multer');
const path = require('path');
const fs = require('fs');
const { protect } = require('../middleware/authMiddleware');
const User = require('../models/User');

// ==========================================
// 📁 Multer Storage Config
// ==========================================
const uploadDir = path.join(__dirname, '..', 'uploads');

// Ensure upload directories exist
const dirs = ['profiles', 'documents', 'parcels', 'places', 'proofs', 'products'];

dirs.forEach(dir => {
    const fullPath = path.join(uploadDir, dir);
    if (!fs.existsSync(fullPath)) {
        fs.mkdirSync(fullPath, { recursive: true });
    }
});

const storage = multer.diskStorage({
    destination: (req, file, cb) => {
        const type = req.params.type || 'profiles';
        cb(null, path.join(uploadDir, type));
    },
    filename: (req, file, cb) => {
        const uniqueName = `${req.user._id}_${Date.now()}${path.extname(file.originalname)}`;
        cb(null, uniqueName);
    }
});

const fileFilter = (req, file, cb) => {
    const allowed = ['image/jpeg', 'image/png', 'image/webp'];
    if (allowed.includes(file.mimetype)) {
        cb(null, true);
    } else {
        cb(new Error('نوع الملف غير مدعوم — يرجى رفع JPG أو PNG أو WebP'), false);
    }
};

const upload = multer({
    storage,
    fileFilter,
    limits: { fileSize: 5 * 1024 * 1024 } // 5MB max
});

// ==========================================
// 📷 1. رفع صورة البروفايل (Profile Photo)
// ==========================================
router.post('/profile-photo', protect, (req, res) => {
    req.params.type = 'profiles';
    upload.single('photo')(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع الصورة' });
        }
        if (!req.file) {
            return res.status(400).json({ message: 'لم يتم اختيار صورة' });
        }

        const fileUrl = `/uploads/profiles/${req.file.filename}`;

        // Update user profile photo
        await User.findByIdAndUpdate(req.user._id, {
            'documents.profilePhoto': fileUrl
        });

        res.json({
            success: true,
            message: 'تم رفع الصورة بنجاح',
            url: fileUrl
        });
    });
});

// ==========================================
// 📄 2. رفع وثائق الكابتن (Captain Documents)
// ==========================================
router.post('/captain-docs', protect, (req, res) => {
    req.params.type = 'documents';
    const uploadFields = upload.fields([
        { name: 'driverLicense', maxCount: 1 },
        { name: 'profilePhoto', maxCount: 1 },
        { name: 'vehiclePhoto', maxCount: 1 }
    ]);

    uploadFields(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع الملفات' });
        }

        const updates = {};
        if (req.files.driverLicense) {
            updates['documents.driverLicense'] = `/uploads/documents/${req.files.driverLicense[0].filename}`;
        }
        if (req.files.profilePhoto) {
            updates['documents.profilePhoto'] = `/uploads/documents/${req.files.profilePhoto[0].filename}`;
        }
        if (req.files.vehiclePhoto) {
            updates['documents.vehiclePhoto'] = `/uploads/documents/${req.files.vehiclePhoto[0].filename}`;
        }

        if (Object.keys(updates).length > 0) {
            await User.findByIdAndUpdate(req.user._id, updates);
        }

        res.json({
            success: true,
            message: 'تم رفع الوثائق بنجاح',
            files: updates
        });
    });
});

// ==========================================
// 📦 3. رفع صورة الشحنة (Parcel Image)
// ==========================================
router.post('/parcel-image', protect, (req, res) => {
    req.params.type = 'parcels';
    upload.single('parcelImage')(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع الصورة' });
        }
        if (!req.file) {
            return res.status(400).json({ message: 'لم يتم اختيار صورة' });
        }

        const fileUrl = `/uploads/parcels/${req.file.filename}`;

        res.json({
            success: true,
            message: 'تم رفع صورة الشحنة',
            url: fileUrl
        });
    });
});

// ==========================================
// 🏪 4. رفع صورة غلاف المحل (Place Image) - Admin Only
// ==========================================
router.post('/place-image', protect, (req, res) => {
    req.params.type = 'places';
    upload.single('placeImage')(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع الصورة' });
        }
        if (!req.file) {
            return res.status(400).json({ message: 'لم يتم اختيار صورة' });
        }
        if (req.user.role !== 'admin') {
            return res.status(403).json({ message: 'Admins only' });
        }

        const fileUrl = `/uploads/places/${req.file.filename}`;
        res.json({ success: true, url: fileUrl });
    });
});

// ==========================================
// 📸 5. رفع صورة إثبات الاستلام (Proof of Pickup) — Captain Only
// ==========================================
router.post('/proof-image', protect, (req, res) => {
    req.params.type = 'proofs';
    upload.single('proofImage')(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع صورة الإثبات' });
        }
        if (!req.file) {
            return res.status(400).json({ message: 'لم يتم اختيار صورة' });
        }
        if (req.user.role !== 'captain') {
            return res.status(403).json({ message: 'مخصص للكباتن فقط' });
        }

        const fileUrl = `/uploads/proofs/${req.file.filename}`;
        res.json({
            success: true,
            message: 'تم رفع صورة الإثبات بنجاح',
            url: fileUrl
        });
    });
});

// ==========================================
// 🏪 6. رفع صور التاجر (Merchant Registration Proofs)
// ==========================================
router.post('/merchant-proof', protect, (req, res) => {
    req.params.type = 'proofs';
    upload.single('image')(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع الصورة' });
        }
        if (!req.file) {
            return res.status(400).json({ message: 'لم يتم اختيار صورة' });
        }

        const fileUrl = `/uploads/proofs/${req.file.filename}`;
        res.json({
            success: true,
            message: 'تم رفع الصورة بنجاح',
            url: fileUrl
        });
    });
});

// ==========================================
// 🍔 7. رفع صورة المنتج (Product Image)
// ==========================================
router.post('/product-image', protect, (req, res) => {
    req.params.type = 'products';
    upload.single('image')(req, res, async (err) => {
        if (err) {
            return res.status(400).json({ message: err.message || 'خطأ في رفع الصورة' });
        }
        if (!req.file) {
            return res.status(400).json({ message: 'لم يتم اختيار صورة' });
        }

        const fileUrl = `/uploads/products/${req.file.filename}`;
        res.json({
            success: true,
            message: 'تم رفع صورة المنتج بنجاح',
            url: fileUrl
        });
    });
});

module.exports = router;
