const express = require("express");
const router = express.Router();
const defineProject = require("../../models/projects");
const defineFolder = require("../../models/folders");
const defineCase = require("../../models/cases");
const defineRun = require("../../models/runs");
const defineRunCase = require("../../models/runCases");
const { DataTypes } = require("sequelize");

module.exports = function (sequelize) {
  const Project = defineProject(sequelize, DataTypes);
  const Folder = defineFolder(sequelize, DataTypes);
  const Case = defineCase(sequelize, DataTypes);
  const Run = defineRun(sequelize, DataTypes);
  const RunCase = defineRunCase(sequelize, DataTypes);
  Project.hasMany(Folder, { foreignKey: "projectId" });
  Folder.hasMany(Case, { foreignKey: "folderId" });
  Project.hasMany(Run, { foreignKey: "projectId" });
  Run.hasMany(RunCase, { foreignKey: "runId" });

  router.get("/:projectId", async (req, res) => {
    const projectId = req.params.projectId;

    if (!projectId) {
      return res.status(400).json({ error: "projectId is required" });
    }

    try {
      const project = await Project.findByPk(projectId, {
        include: [
          {
            model: Folder,
            include: [{ model: Case }],
          },
          { model: Run, include: [{ model: RunCase }] },
        ],
      });
      if (!project) {
        return res.status(404).send("Project not found");
      }
      res.json(project);
    } catch (error) {
      console.error(error);
      res.status(500).send("Internal Server Error");
    }
  });

  return router;
};