import type { NextApiRequest, NextApiResponse } from "next";
import { storage } from "@/firebase";
import client from "@/lib/mongodb";
import { withIronSessionApiRoute } from "iron-session/next";
import { sessionOptions } from "@/lib/session";
import ReactPDF from "@react-pdf/renderer";
import TestReport from "@/exams/pdf/test.report";
import LevelTestReport from "@/exams/pdf/level.test.report";
import { ref, uploadBytes, getDownloadURL } from "firebase/storage";
import {
  DemographicInformation,
  Stat,
  StudentUser,
  User,
} from "@/interfaces/user";
import { Module } from "@/interfaces";
import { ModuleScore } from "@/interfaces/module.scores";
import { SkillExamDetails } from "@/exams/pdf/details/skill.exam";
import { calculateBandScore } from "@/utils/score";
import axios from "axios";
import { moduleLabels } from "@/utils/moduleUtils";
import {
  generateQRCode,
  getRadialProgressPNG,
  streamToBuffer,
} from "@/utils/pdf";
import moment from "moment-timezone";
import { getCorporateNameForStudent } from "@/utils/groups.be";

const db = client.db(process.env.MONGODB_DB);

export default withIronSessionApiRoute(handler, sessionOptions);

async function handler(req: NextApiRequest, res: NextApiResponse) {
  if (req.method === "GET") return get(req, res);
  if (req.method === "POST") return post(req, res);
}

const getExamSummary = (score: number) => {
  if (score > 0.8) {
    return "Scoring between 81% and 100% on the English exam demonstrates an outstanding level of proficiency in writing, speaking, listening, and reading. Mastery of key concepts is evident across all language domains, showcasing not only a high level of skill but also a dedication to excellence. Continuing to challenge oneself with advanced material in writing, speaking, listening, and reading will further refine the already impressive command of the English language.";
  }

  if (score > 0.6) {
    return "Scoring between 61% and 80% on the English exam, encompassing writing, speaking, listening, and reading, reflects a commendable level of proficiency in each domain. There's evidence of a solid grasp of key concepts, and effective application of skills. Room for refinement and deeper exploration in writing, speaking, listening, and reading remains, presenting an opportunity for further mastery.";
  }

  if (score > 0.4) {
    return "Scoring between 41% and 60% on the English exam across writing, speaking, listening, and reading demonstrates a moderate level of understanding in each domain. While there's a commendable grasp of key concepts, refining fundamental skills in writing, speaking, listening, and reading can lead to notable improvement. Consistent effort and targeted focus on weaker areas are recommended.";
  }

  if (score > 0.2) {
    return "Scoring between 21% and 40% on the English exam, spanning writing, speaking, listening, and reading, indicates some understanding of key concepts in each domain. However, there's room for improvement in fundamental skills. Strengthening writing, speaking, listening, and reading abilities through consistent effort and focused study will contribute to overall proficiency.";
  }

  return "This student's performance on the English exam, encompassing writing, speaking, listening, and reading, reflects a significant need for improvement, scoring between 0% and 20%. There's a notable gap in understanding key concepts across all language domains. Strengthening fundamental skills in writing, speaking, listening, and reading is crucial. Developing a consistent study routine and seeking additional support in each area can contribute to substantial progress.";
};

const getLevelSummary = (score: number) => {
  if (score > 0.8) {
    return "Scoring between 81% and 100% on the English exam showcases an outstanding level of understanding and proficiency. Your performance reflects a mastery of key concepts, including grammar, vocabulary, and comprehension. You exhibit a high level of skill in applying these elements effectively. Your dedication to excellence is evident, and your consistent, stellar performance is commendable. Continue to challenge yourself with advanced material to further refine your already impressive command of the English language. Your commitment to excellence positions you as a standout student in English studies, and your achievements are a testament to your hard work and capability.";
  }

  if (score > 0.6) {
    return "Scoring between 61% and 80% on the English exam reflects a commendable level of understanding and proficiency. You have demonstrated a solid grasp of key concepts, including grammar, vocabulary, and comprehension. There's evidence of effective application of skills, but room for refinement and deeper exploration remains. Consistent effort in honing nuanced aspects of language will contribute to even greater mastery. Continue engaging with challenging material and seeking opportunities for advanced comprehension. With sustained dedication, you have the potential to elevate your performance to an exceptional level and further excel in your English studies.";
  }

  if (score > 0.4) {
    return "Scoring between 41% and 60% on the English exam reflects a moderate level of understanding. You demonstrate a grasp of some key concepts, but there's room for refinement in areas like grammar, vocabulary, and comprehension. Consistent effort and a strategic focus on weaker areas can lead to notable improvement. Engaging with supplementary resources and seeking feedback will further enhance your skills. With continued dedication, there's a solid foundation to build upon, and achieving a higher level of proficiency is within reach. Keep up the good work and aim for sustained progress in your English studies.";
  }

  if (score > 0.2) {
    return "Scoring between 21% and 40% on the English exam shows some understanding of key concepts, but there's still ample room for improvement. Strengthening foundational skills, such as grammar, vocabulary, and comprehension, is essential. Consistent effort and focused study can help bridge gaps in knowledge and elevate your performance. Consider seeking additional guidance or resources to refine your understanding of the material. With commitment and targeted improvements, you have the potential to make significant strides in your English proficiency.";
  }

  return "Your performance on the English exam falls within the 0% to 20% range, indicating a need for improvement. There's room to enhance your grasp of fundamental concepts like grammar, vocabulary, and comprehension. Establishing a consistent study routine and seeking extra support can be beneficial. With dedication and targeted efforts, you have the potential to significantly boost your performance in upcoming assessments.";
};

const getPerformanceSummary = (module: Module, score: number) => {
  if (module === "level") return getLevelSummary(score);
  return getExamSummary(score);
};
interface SkillsFeedbackRequest {
  code: Module;
  name: string;
  grade: number;
}

interface SkillsFeedbackResponse extends SkillsFeedbackRequest {
  evaluation: string;
  suggestions: string;
  bullet_points?: string[];
}

const getSkillsFeedback = async (sections: SkillsFeedbackRequest[]) => {
  const backendRequest = await axios.post(
    `${process.env.BACKEND_URL}/grading_summary`,
    { sections },
    {
      headers: {
        Authorization: `Bearer ${process.env.BACKEND_JWT}`,
      },
    }
  );

  return backendRequest.data?.sections;
};

// perform the request with several retries if needed
const handleSkillsFeedbackRequest = async (
  sections: SkillsFeedbackRequest[]
): Promise<SkillsFeedbackResponse[] | null> => {
  let i = 0;
  try {
    const data = await getSkillsFeedback(sections);
    return data;
  } catch (err) {
    if (i < 3) {
      i++;
      return handleSkillsFeedbackRequest(sections);
    }

    return null;
  }
};

async function getDefaultPDFStream(
  stats: Stat[],
  user: User,
  qrcodeUrl: string
) {
  const [stat] = stats;
  // generate the QR code for the report
  const qrcode = await generateQRCode(qrcodeUrl);

  if (!qrcode) {
    throw new Error("Failed to generate QR code");
  }

  // stats may contain multiple exams of the same type so we need to aggregate them
  const results = stats
    .reduce((accm: ModuleScore[], stat: Stat) => {
      const { module, score } = stat;

      const fixedModuleStr = module[0].toUpperCase() + module.substring(1);
      if (accm.find((e: ModuleScore) => e.module === fixedModuleStr)) {
        return accm.map((e: ModuleScore) => {
          if (e.module === fixedModuleStr) {
            return {
              ...e,
              score: e.score + score.correct,
              total: e.total + score.total,
            };
          }

          return e;
        });
      }

      const value = {
        module: fixedModuleStr,
        score: score.correct,
        total: score.total,
        code: module,
      } as ModuleScore;

      return [...accm, value];
    }, [])
    .map((moduleScore: ModuleScore) => {
      const { score, total } = moduleScore;
      // with all the scores aggreated we can calculate the band score for each module
      const bandScore = calculateBandScore(
        score,
        total,
        moduleScore.code as Module,
        user.focus
      );

      return {
        ...moduleScore,
        // generate the closest radial progress png for the score
        png: getRadialProgressPNG("azul", score, total),
        bandScore,
      };
    });

  // get the skills feedback from the backend based on the module grade
  const skillsFeedback = (await handleSkillsFeedbackRequest(
    results.map(({ code, bandScore }) => ({
      code,
      name: moduleLabels[code],
      grade: bandScore,
    }))
  )) as SkillsFeedbackResponse[];

  if (!skillsFeedback) {
    throw new Error("Failed to get skills feedback");
  }

  // assign the feedback to the results
  const finalResults = results.map((result) => {
    const feedback = skillsFeedback.find(
      (f: SkillsFeedbackResponse) => f.code === result.code
    );

    if (feedback) {
      return {
        ...result,
        evaluation: feedback?.evaluation,
        suggestions: feedback?.suggestions,
        bullet_points: feedback?.bullet_points,
      };
    }

    return result;
  });

  // calculate the overall score out of all the aggregated results
  const overallScore = results.reduce((accm, { score }) => accm + score, 0);
  const overallTotal = results.reduce((accm, { total }) => accm + total, 0);
  const overallResult = overallScore / overallTotal;

  const overallPNG = getRadialProgressPNG(
    "laranja",
    overallScore,
    overallTotal
  );

  // generate the overall detail report
  const overallDetail = {
    module: "Overall",
    score: overallScore,
    total: overallTotal,
    png: overallPNG,
  } as ModuleScore;
  const testDetails = [overallDetail, ...finalResults];

  // generate the performance summary based on the overall result
  const performanceSummary = getPerformanceSummary(stat.module, overallResult);

  const title = "ENGLISH SKILLS TEST RESULT REPORT";
  const details = <SkillExamDetails testDetails={testDetails} />;

  const demographicInformation =
    user.demographicInformation as DemographicInformation;
  return ReactPDF.renderToStream(
    <TestReport
      title={title}
      date={moment(stat.date)
        .tz(user.demographicInformation?.timezone || "UTC")
        .format("ll HH:mm:ss")}
      name={user.name}
      email={user.email}
      id={user.id}
      gender={demographicInformation?.gender}
      summary={performanceSummary}
      testDetails={testDetails}
      renderDetails={details}
      logo={"public/logo_title.png"}
      qrcode={qrcode}
      summaryPNG={overallPNG}
      summaryScore={`${Math.floor(overallResult * 100)}%`}
      passportId={demographicInformation?.passport_id || ""}
    />
  );
}

async function getPdfUrl(pdfStream: any, docsSnap: any) {
  // generate the file ref for storage
  const fileName = `${Date.now().toString()}.pdf`;
  const refName = `exam_report/${fileName}`;
  const fileRef = ref(storage, refName);

  // upload the pdf to storage
  const pdfBuffer = await streamToBuffer(pdfStream);
  await uploadBytes(fileRef, pdfBuffer, {
    contentType: "application/pdf",
  });

  // update the stats entries with the pdf url to prevent duplication
  await db.collection("stats").updateOne(
    { id: docsSnap.id },
    {
      $set: {
        pdf: {
          path: refName,
          version: process.env.PDF_VERSION,
        }
      }
    }
  );
  return getDownloadURL(fileRef);
}

async function post(req: NextApiRequest, res: NextApiResponse) {
  // verify if it's a logged user that is trying to export
  if (req.session.user) {
    const { id } = req.query as { id: string };
    // fetch stats entries for this particular user with the requested exam session
    const stats = await db.collection("stats").find<Stat>({ session: id }).toArray();

    if (stats.length == 0) {
      res.status(400).end();
      return;
    }

    // verify if the stats already have a pdf generated
    const hasPDF = stats.find(
      (s) => s.pdf?.path && s.pdf?.version === process.env.PDF_VERSION
    );
    // find the user that generated the stats
    const statIndex = stats.findIndex((s) => s.user);

    if (statIndex === -1) {
      res.status(401).json({ ok: false });
      return;
    }
    const userId = stats[statIndex].user;

    if (hasPDF) {
      // if it does, return the pdf url
      const fileRef = ref(storage, hasPDF.pdf!.path);
      const url = await getDownloadURL(fileRef);

      res.status(200).end(url);
      return;
    }

    try {
      // generate the pdf report
      const docUser = await db.collection("users").findOne<User>({ id: userId});

      if (docUser) {
        // we'll need the user in order to get the user data (name, email, focus, etc);

        const [stat] = stats;

        if (stat.module === "level") {
          const user = docUser as StudentUser;

          const uniqueExercises = stats.map((s) => ({
            name: "Gramar & Vocabulary",
            result: `${s.score.correct}/${s.score.total}`,
          }));
          const dates = stats.map((s) => moment(s.date));
          const timeSpent = `${stats.reduce((accm, s: Stat) => accm + (s.timeSpent || 0), 0) / 60
            } minutes`;
          const score = stats.reduce((accm, s) => accm + s.score.correct, 0);
          const corporateName = await getCorporateNameForStudent(userId);
          const pdfStream = await ReactPDF.renderToStream(
            <LevelTestReport
              date={moment.max(dates).format("DD/MM/YYYY")}
              name={user.name}
              email={user.email}
              id={stat.exam}
              gender={user.demographicInformation?.gender || ""}
              passportId={user.demographicInformation?.passport_id || ""}
              corporateName={corporateName}
              downloadDate={moment().format("DD/MM/YYYY")}
              userId={userId}
              uniqueExercises={uniqueExercises}
              timeSpent={timeSpent}
              score={score.toString()}
            />
          );

          const url = await getPdfUrl(pdfStream, docUser);
          res.status(200).end(url);
          return;
        }
        const user = docUser as User;

        try {
          const pdfStream = await getDefaultPDFStream(
            stats,
            user,
            `${req.headers.origin || ""}${req.url}`
          );

          const url = await getPdfUrl(pdfStream, docUser);
          res.status(200).end(url);
          return;
        } catch (err) {
          console.error(err);
          res.status(500).json({ ok: false });
          return;
        }
      }

      res.status(401).json({ ok: false });
      return;
    } catch (err) {
      console.error(err);
      res.status(500).json({ ok: false });
      return;
    }
  }

  res.status(401).json({ ok: false });
  return;
}

async function get(req: NextApiRequest, res: NextApiResponse) {
  const { id } = req.query as { id: string };
  const stats = await db.collection("stats").find<Stat>({ session: id }).toArray();

  if (stats.length == 0) {
    res.status(404).end();
    return;
  }

  const hasPDF = stats.find((s) => s.pdf?.path);

  if (hasPDF) {
    const fileRef = ref(storage, hasPDF.pdf!.path);
    const url = await getDownloadURL(fileRef);
    return res.redirect(url);
  }

  res.status(500).end();
}
