import compact from 'lodash/compact';
import orderBy from 'lodash/orderBy';
import {
  useTeamQuery,
  type TeamFieldsFragment,
  type TeamStatsFieldsFragment,
  type TeamRefereeFieldsFragment,
  type TeamEntityFragment,
} from './__generated__/Team';
import { DEFAULT_AGGREGATION_EPOCHS } from './use-teams';

export type Team = TeamFieldsFragment;
export type TeamStats = TeamStatsFieldsFragment;
export type Member = TeamRefereeFieldsFragment;
export type TeamEntity = TeamEntityFragment;
export type TeamGame = ReturnType<typeof useTeam>['games'][number];

export const useTeam = (teamId?: string, partyId?: string) => {
  const { data, loading, error, refetch } = useTeamQuery({
    variables: {
      teamId: teamId || '',
      partyId,
      aggregationEpochs: DEFAULT_AGGREGATION_EPOCHS,
    },
    skip: !teamId,
    fetchPolicy: 'cache-and-network',
  });

  const teamEdge = data?.teams?.edges.find((e) => e.node.teamId === teamId);
  const partyTeam = data?.partyTeams?.edges?.length
    ? data.partyTeams.edges[0].node
    : undefined;

  const teamStatsEdge = data?.teamsStatistics?.edges.find(
    (e) => e.node.teamId === teamId
  );
  const members = data?.teamReferees?.edges
    .filter((e) => e.node.teamId === teamId)
    .map((e) => e.node);

  // Find games where the current team participated in
  const gamesWithTeam = compact(data?.games.edges).map((edge) => {
    const team = edge.node.entities.find((e) => {
      if (e.__typename !== 'TeamGameEntity') return false;
      if (e.team.teamId !== teamId) return false;
      return true;
    });

    if (!team) return null;

    return {
      id: edge.node.id,
      epoch: edge.node.epoch,
      numberOfParticipants: edge.node.numberOfParticipants,
      entities: edge.node.entities,
      team: team as TeamEntity, // TS can't infer that all the game entities are teams
    };
  });

  const games = orderBy(compact(gamesWithTeam), 'epoch', 'desc');

  return {
    data,
    loading,
    error,
    refetch,
    stats: teamStatsEdge?.node,
    team: teamEdge?.node,
    members,
    games,
    partyTeam,
  };
};