server.js

if (process.env.NODE_ENV != "production") {
  require("dotenv").config();
}

const express = require("express");
const mongoose = require("mongoose");
const passport = require("passport");
const LocalStrategy = require("passport-local");
const session = require("express-session");
const MongoStore = require("connect-mongo");
const flash = require("express-flash");
const User = require("./models/user.js");
const path = require("path");
const helmet = require("helmet");
const { errorHandler, logUserInfo } = require('./middleware.js');
const methodOverride = require('method-override');

const dbUrl = process.env.ATLAS_URL;

// const dbUrl = 'mongodb://localhost:27017/music'

const authRoutes = require("./routes/auth.js");
const songRoutes = require("./routes/music.js");

const app = express();

app.set("views", "./views");
app.set("view engine", "ejs");

app.use(express.static(path.join(__dirname, "public")));
app.use(express.urlencoded({ extended: true }));
app.use(express.json());
app.use(methodOverride('_method'));


app.use(helmet
  ({ contentSecurityPolicy: false }));

// Mongo Session store
const store = MongoStore.create({
  mongoUrl: dbUrl,
  touchAfter: 24 * 3600,
  crypto: {
    secret: process.env.SESSION_SECRET,
  },
});

store.on("error", (err) => {
  console.error("Error in Mongo Session Store:", err);
});

const sessionOptions = {
  store,
  secret: process.env.SESSION_SECRET,
  resave: false,
  saveUninitialized: false,
  cookie: {
    expires: Date.now() + 7 * 24 * 60 * 60 * 1000,
    httpOnly: true,
    maxAge: 7 * 24 * 60 * 60 * 1000,
  },
};

// Middleware
app.use(session(sessionOptions));
app.use(flash());
app.use(passport.initialize());
app.use(passport.session());

// Passport Strategy
passport.use(User.createStrategy());
passport.serializeUser(User.serializeUser());
passport.deserializeUser(User.deserializeUser());

// Prevent caching
app.use((req, res, next) => {
  res.set("Cache-Control", "no-store, no-cache, must-revalidate, private");
  res.set("Pragma", "no-cache");
  res.set("Expires", "0");
  next();
});

// Connect to MongoDB
async function main() {
  await mongoose.connect(dbUrl);
}
main()
  .then(() => console.log("Connected to MongoDB"))
  .catch((err) => console.error("Error connecting to MongoDB:", err));

// Routes
app.use(authRoutes);
app.use(songRoutes);

app.get('/health', async (req, res) => {
  const mongo = mongoose.connection.readyState; 
  res.status(200).json({
    status: 'ok',
    mongo: mongo === 1 ? 'connected' : 'not-connected'
  });
});

// 404 Error Page
app.use((req, res, next) => {
  res.status(404).render("error", {
    errorTitle: "Page Not Found",
    errorMessage: "The page you're looking for doesn't exist.",
  });
});

app.use(errorHandler);

const PORT = process.env.PORT || 3000;
app.listen(PORT, '0.0.0.0', () => {
  console.log(`🚀 WaveTune running on port ${PORT}`);
});

app.get('/health', (req, res) => {
  res.json({ status: 'ok' });
});