app.js

const express = require('express');
const http = require('http');
const mongoose = require('mongoose');
const session = require('express-session');
const MongoStore = require('connect-mongo');
const flash = require('connect-flash');
const expressLayouts = require('express-ejs-layouts');
const path = require('path');
const methodOverride = require('method-override');
const bodyParser = require("body-parser");
const cookieParser = require("cookie-parser");
const { initializeSocket } = require('./config/socket');
require('dotenv').config();
// const ejsLocals = require("ejs-locals");
require("dotenv").config();

// Import models for admin initialization
const User = require("./models/User");

// Import routes
const indexRoutes = require("./routes/index");
const shipmentRoutes = require("./routes/shipment");
const serviceRoutes = require("./routes/service");
const aboutRoutes = require("./routes/about");
const contactRoutes = require("./routes/contact");
const blogRoutes = require("./routes/blog");
const teamRoutes = require("./routes/team");
const projectRoutes = require("./routes/project");
const adminRoutes = require("./routes/admin");
const authRoutes = require("./routes/auth");
const userRoutes = require("./routes/user");
const legalRoutes = require("./routes/legal");
const dashboardRoutes = require("./routes/dashboard");
const apiRoutes = require("./routes/api");
const trackRoutes = require("./routes/track");
const newsletterRoutes = require("./routes/newsletter");
const ecommerceRoutes = require("./routes/ecommerce");
const journalRoutes = require("./routes/journal");
const shippingRoutes = require("./routes/shipping");
const ratesRoutes = require("./routes/rates");

// Import middleware
const { isAuthenticated } = require("./middleware/auth");

// Initialize Express app
const app = express();

// Function to ensure admin user exists
const initializeAdminUser = async () => {
  try {
    // Check if an admin user already exists
    const adminExists = await User.findOne({ role: "admin" });

    if (!adminExists) {
      console.log("No admin user found, creating default admin...");

      // Get admin email from environment or use default
      const adminEmail = process.env.ADMIN_EMAIL || "admin@dxpress.uk";
      const adminPassword = process.env.ADMIN_PASSWORD || "$IamtheAdmin11";

      // Create default admin user
      const admin = new User({
        email: adminEmail,
        password: adminPassword, // The pre-save hook will hash this
        fullName: "Admin User",
        role: "admin",
        isActive: true,
      });

      await admin.save();
      console.log(
        `Default admin user created successfully with email: ${admin.email}`
      );
    } else {
      console.log(`Admin user already exists with email: ${adminExists.email}`);
    }
  } catch (error) {
    console.error("Error initializing admin user:", error);
  }
};

// Connect to MongoDB
mongoose
  .connect(getDatabaseUri())
  .then(() => {
    console.log(
      "MongoDB connected to",
      process.env.NODE_ENV === "production"
        ? "production database"
        : "development database"
    );
    // Initialize admin user after connecting to the database
    initializeAdminUser();
  })
  .catch((err) => console.error("MongoDB connection error:", err));

// Function to determine which database URI to use based on environment
function getDatabaseUri() {
  // Always use MONGODB_URI for all environments
  console.log("Using database connection from MONGODB_URI");
  return process.env.MONGODB_URI;
}

// Set view engine
//app.engine("ejs", ejsLocals); // Use ejs-locals for layout, partial and block templates
app.set("view engine", "ejs");
app.set("views", path.join(__dirname, "views"));

// Middleware
app.use(express.static("public"));
app.use(express.urlencoded({ extended: true }));
app.use(express.json());

// Debug middleware to log all requests
app.use((req, res, next) => {
  console.log(`🌐 [DEBUG] ${req.method} ${req.path}`);
  if (req.method === 'POST') {
    console.log('📋 [DEBUG] POST body:', req.body);
  }
  next();
});
app.use(cookieParser());
app.use(express.static(path.join(__dirname, "public")));
app.use("/admin", express.static(path.join(__dirname, "public/admin")));

// Session middleware
app.use(
  session({
    secret: process.env.SESSION_SECRET || "your-secret-key",
    resave: false,
    saveUninitialized: false,
    cookie: {
      secure: process.env.NODE_ENV === "production",
      maxAge: 1000 * 60 * 60 * 24, // 1 day
    },
  })
);

// Setup express-ejs-layouts
app.use(expressLayouts);
app.set("layout", "layouts/main");

// Add authentication middleware
app.use((req, res, next) => {
  res.locals.isLoggedIn = req.session.isLoggedIn || false;
  next();
});

// Middleware to disable layouts for admin routes
app.use((req, res, next) => {
  if (req.path.startsWith("/admin")) {
    res.locals.layout = false;
  }
  next();
});

// HTML extension redirect middleware
app.use((req, res, next) => {
  if (req.path.endsWith(".html")) {
    const withoutHtml = req.path.slice(0, -5);
    const query = Object.keys(req.query).length
      ? `?${new URLSearchParams(req.query)}`
      : "";
    return res.redirect(301, `${withoutHtml}${query}`);
  }
  next();
});

// Add debug logging middleware
app.use((req, res, next) => {
  const start = Date.now();
  console.log(`[${new Date().toISOString()}] ${req.method} ${req.url}`);
  console.log("[Debug] Headers:", JSON.stringify(req.headers, null, 2));

  // Log environment variables (safely)
  console.log("[Debug] Environment Check:", {
    NODE_ENV: process.env.NODE_ENV,
    STRIPE_PUBLIC_KEY: process.env.STRIPE_PUBLIC_KEY ? "[SET]" : "[NOT SET]",
    STRIPE_SECRET_KEY: process.env.STRIPE_SECRET_KEY ? "[SET]" : "[NOT SET]",
    PAYPAL_CLIENT_ID: process.env.PAYPAL_CLIENT_ID ? "[SET]" : "[NOT SET]",
    PAYPAL_HOSTED_BUTTON_ID: process.env.PAYPAL_HOSTED_BUTTON_ID
      ? "[SET]"
      : "[NOT SET]",
    API_KEY: process.env.API_KEY ? "[SET]" : "[NOT SET]",
    JWT_SECRET: process.env.JWT_SECRET ? "[SET]" : "[NOT SET]",
    SESSION_SECRET: process.env.SESSION_SECRET ? "[SET]" : "[NOT SET]",
  });

  // Log response
  res.on("finish", () => {
    const duration = Date.now() - start;
    console.log(
      `[${new Date().toISOString()}] ${req.method} ${req.url} ${
        res.statusCode
      } - ${duration}ms`
    );
  });

  next();
});

// Routes
app.use("/api", apiRoutes);
app.use("/", indexRoutes);
app.use("/shipment", shipmentRoutes);
app.use("/services", serviceRoutes);
app.use("/about", aboutRoutes);
app.use("/contact", contactRoutes);
app.use("/blog", blogRoutes);
app.use("/team", teamRoutes);
app.use("/project", projectRoutes);
app.use("/admin", adminRoutes);
app.use("/auth", authRoutes);
app.use("/", userRoutes);
app.use("/", legalRoutes);
app.use("/dashboard", dashboardRoutes);
app.use("/api", apiRoutes);
app.use("/track", trackRoutes);
app.use("/track-shipment", trackRoutes);
app.use("/newsletter", newsletterRoutes);
app.use("/service", serviceRoutes);
app.use("/ecommerce-integration", ecommerceRoutes);
app.use("/dxpress-journal", journalRoutes);
app.use("/journal", journalRoutes);
app.use("/international-shipping", shippingRoutes);
app.use("/admin/rates", ratesRoutes);
app.use("/admin/rates", require("./routes/rates"));
app.use("/admin/carrier-rates", require("./routes/admin/carrierRates"));
app.use("/admin/audit-logs", require("./routes/auditLogs"));
app.use("/api/tracking", require("./routes/tracking"));
app.use("/", require("./routes/notifications"));

// 404 handler
app.use((req, res, next) => {
  if (req.path.startsWith("/admin")) {
    return res.status(404).render("admin/404", {
      title: "404 - Page Not Found",
      path: "/admin/error",
      layout: false,
    });
  }
  res.status(404).render("404", {
    title: "404 - Page Not Found",
    path: req.url,
  });
});

// Error logging middleware
app.use((err, req, res, next) => {
  console.error("[Error]", {
    timestamp: new Date().toISOString(),
    method: req.method,
    url: req.url,
    error: {
      message: err.message,
      stack: err.stack,
      name: err.name,
    },
    headers: req.headers,
    body: req.body,
    query: req.query,
    params: req.params,
  });

  // Render error page
  const isAdminRoute = req.path.startsWith('/admin');
  const layoutToUse = isAdminRoute ? "layouts/admin-dashboard" : "layouts/main";
  
  res.status(err.status || 500).render(isAdminRoute ? "admin/500" : "500", {
    title: "Server Error",
    path: "/500",
    layout: layoutToUse,
    error: process.env.NODE_ENV === "production" ? null : err,
    stylesheets: "",
    scripts: "",
  });
});

// Start server
const PORT = process.env.PORT || 3000;
const server = http.createServer(app);

// Initialize Socket.IO
initializeSocket(server);

server.listen(PORT, () => {
  console.log(`Server running on port ${PORT}`);
  console.log(`WebSocket server initialized for real-time notifications`);
});

module.exports = app;