require("dotenv").config();
const mongoose = require("mongoose");
const connectDB = require("../config/database");

const COLLECTION_NAME = "newsletter_subscriptions";

const validator = {
  $jsonSchema: {
    bsonType: "object",
    required: ["email", "status", "createdAt", "updatedAt"],
    properties: {
      email: {
        bsonType: "string",
        pattern: "^\\S+@\\S+\\.\\S+$",
      },
      status: {
        enum: ["subscribed", "unsubscribed"],
      },
      pageUrl: {
        bsonType: "string",
      },
      ipAddress: {
        bsonType: "string",
      },
      userAgent: {
        bsonType: "string",
      },
      internalNote: {
        bsonType: "string",
        maxLength: 2000,
      },
      createdAt: {
        bsonType: "date",
      },
      updatedAt: {
        bsonType: "date",
      },
    },
  },
};

async function ensureCollection(db) {
  const collections = await db
    .listCollections({ name: COLLECTION_NAME }, { nameOnly: true })
    .toArray();

  if (collections.length === 0) {
    await db.createCollection(COLLECTION_NAME, {
      validator,
      validationLevel: "moderate",
      validationAction: "error",
    });
    return;
  }

  await db.command({
    collMod: COLLECTION_NAME,
    validator,
    validationLevel: "moderate",
    validationAction: "error",
  });
}

async function ensureIndexes(collection) {
  await collection.createIndex({ email: 1 }, { name: "email_1", unique: true });
  await collection.createIndex(
    { status: 1, createdAt: -1 },
    { name: "status_1_createdAt_-1" },
  );
  await collection.createIndex({ createdAt: -1 }, { name: "createdAt_-1" });
}

async function migrate() {
  try {
    await connectDB();
    const db = mongoose.connection.db;
    await ensureCollection(db);
    await ensureIndexes(db.collection(COLLECTION_NAME));
    console.log("Newsletter subscriptions migration completed successfully");
    await mongoose.disconnect();
    process.exit(0);
  } catch (error) {
    console.error("Newsletter subscriptions migration error:", error);
    process.exit(1);
  }
}

if (require.main === module) {
  migrate();
}

module.exports = { migrate };