const express = require("express");
const router = express.Router();
const defineCase = require("../../models/cases");
const defineStep = require("../../models/steps");
const defineAttachment = require("../../models/attachments");
const { DataTypes } = require("sequelize");

module.exports = function (sequelize) {
  const Case = defineCase(sequelize, DataTypes);
  const Step = defineStep(sequelize, DataTypes);
  const Attachment = defineAttachment(sequelize, DataTypes);
  Case.belongsToMany(Step, { through: "caseSteps" });
  Step.belongsToMany(Case, { through: "caseSteps" });
  Case.belongsToMany(Attachment, { through: "caseAttachments" });
  Attachment.belongsToMany(Case, { through: "caseAttachments" });

  router.get("/", async (req, res) => {
    const { caseId, folderId } = req.query;

    if (!caseId && !folderId) {
      return res.status(400).json({ error: "caseId or folderId is required" });
    }

    if (caseId) {
      // Include steps if requested using caseId
      const testcase = await Case.findByPk(caseId, {
        include: [
          {
            model: Step,
            through: { attributes: ["stepNo"] },
          },
          {
            model: Attachment,
          },
        ],
      });
      return res.json(testcase);
    }

    try {
      const cases = await Case.findAll({
        where: {
          folderId: folderId,
        },
      });
      res.json(cases);
    } catch (error) {
      console.error(error);
      res.status(500).send("Internal Server Error");
    }
  });

  return router;
};