import type { Bug } from './bug';
import type { DevTask } from './dev-task';
import type { TestCase } from './test-case';
import type { V22WorkspaceData } from './v22-api';
import type { VersionPlan } from './version-plan';

export interface WorkspaceCollections {
  versionPlans: VersionPlan[];
  devTasks: DevTask[];
  testCases: TestCase[];
  bugs: Bug[];
}

export interface SelectWorkspaceCollectionsInput {
  v22Loaded: boolean;
  v22Failed: boolean;
  v22Data: V22WorkspaceData | null;
  storeData: {
    plans: VersionPlan[];
    devTasks: DevTask[];
    testCases: TestCase[];
    bugs: Bug[];
  };
}

export function selectWorkspaceCollections({
  v22Loaded,
  v22Failed,
  v22Data,
  storeData,
}: SelectWorkspaceCollectionsInput): WorkspaceCollections {
  if (v22Loaded && !v22Failed && v22Data) {
    const hasV22Rows = v22Data.versionPlans.length > 0
      || v22Data.devTasks.length > 0
      || v22Data.testCases.length > 0
      || v22Data.bugs.length > 0;
    if (!hasV22Rows) {
      return {
        versionPlans: v22Data.versionPlans,
        devTasks: v22Data.devTasks,
        testCases: v22Data.testCases,
        bugs: v22Data.bugs,
      };
    }
    return {
      versionPlans: overlayRowsById(v22Data.versionPlans, storeData.plans),
      devTasks: overlayRowsById(v22Data.devTasks, storeData.devTasks),
      testCases: overlayRowsById(v22Data.testCases, storeData.testCases),
      bugs: overlayRowsById(v22Data.bugs, storeData.bugs),
    };
  }

  return {
    versionPlans: storeData.plans,
    devTasks: storeData.devTasks,
    testCases: storeData.testCases,
    bugs: storeData.bugs,
  };
}

function overlayRowsById<T extends { id: string }>(baseRows: T[], overlayRows: T[]): T[] {
  const overlayById = new Map(overlayRows.map((item) => [item.id, item]));
  const seen = new Set<string>();
  const merged = baseRows.map((item) => {
    seen.add(item.id);
    return overlayById.get(item.id) ?? item;
  });
  for (const item of overlayRows) {
    if (!seen.has(item.id)) merged.push(item);
  }
  return merged;
}