import { ProjectType } from '@/types/project';
import { testTypes, priorities, testRunCaseStatus } from '@/config/selection';
import { TestRunCaseStatusMessages } from '@/types/status';

// aggregate folder, case, run mum
function aggregateBasicInfo(project: ProjectType) {
  const folderNum = project.Folders.length;
  const runNum = project.Runs.length;

  let caseNum = 0;
  project.Folders.forEach((folder) => {
    caseNum += folder.Cases.length;
  });

  return { folderNum, runNum, caseNum };
}

// aggregate test types of each case
function aggregateTestType(project: ProjectType) {
  const typesCounts = {};
  project.Folders.forEach((folder) => {
    folder.Cases.forEach((testcase) => {
      const type = testcase.type;
      typesCounts[type] = (typesCounts[type] || 0) + 1;
    });
  });

  const result = [];
  for (let type = 0; type <= testTypes.length; type++) {
    result.push({ type: type, count: typesCounts[type] || 0 });
  }

  return result;
}

function aggregateTestPriority(project: ProjectType) {
  const priorityCounts = {};
  project.Folders.forEach((folder) => {
    folder.Cases.forEach((testcase) => {
      const priority = testcase.priority;
      priorityCounts[priority] = (priorityCounts[priority] || 0) + 1;
    });
  });

  const result = [];
  for (let priority = 0; priority <= priorities.length; priority++) {
    result.push({ priority: priority, count: priorityCounts[priority] || 0 });
  }

  return result;
}

function aggregateProgress(project: ProjectType, testRunCaseStatusMessages: TestRunCaseStatusMessages) {
  let series = testRunCaseStatus.map((status) => {
    return { name: testRunCaseStatusMessages[status.uid], data: [] };
  });
  let categories = [];

  project.Runs.forEach((run) => {
    run.RunCases.forEach((runCase) => {
      const createdAtDate = new Date(runCase.createdAt);
      const dateString = createdAtDate.toISOString().slice(0, 10);

      const alreadyExists = categories.includes(dateString);
      if (!alreadyExists) {
        categories.push(dateString);
        series.forEach((itr) => {
          itr.data.push(0);
        });
      }
    });
  });

  project.Runs.forEach((run) => {
    run.RunCases.forEach((runCase) => {
      const createdAtDate = new Date(runCase.createdAt);
      const dateString = createdAtDate.toISOString().slice(0, 10);
      const index = categories.indexOf(dateString);

      const target = series[runCase.status];
      target.data[index]++;
    });
  });

  return { series, categories };
}

export { aggregateBasicInfo, aggregateTestType, aggregateTestPriority, aggregateProgress };