sendou.ink/hooks/builds.ts

import { Ability } from "@prisma/client";
import { abilities, isMainAbility } from "lib/lists/abilities";
import { weaponToCode } from "lib/lists/weaponCodes";
import { weapons } from "lib/lists/weapons";
import { useRouter } from "next/router";
import { GetBuildsByWeaponData } from "prisma/queries/getBuildsByWeapon";
import { Dispatch, useReducer } from "react";
import useSWR from "swr";

export type BuildFilterType = "AT_LEAST" | "AT_MOST" | "HAS" | "DOES_NOT_HAVE";

interface BuildFilter {
  ability: Ability;
  abilityPoints?: { min: number; max: number };
  hasAbility?: boolean;
}

export interface UseBuildsByWeaponState {
  weapon: string;
  filters: BuildFilter[];
  expandedUsers: Set<number>;
}

type Action =
  | {
      type: "SET_WEAPON";
      weapon: string;
    }
  | {
      type: "EXPAND_USER";
      id: number;
    }
  | {
      type: "ADD_FILTER";
      ability: Ability;
    }
  | {
      type: "REMOVE_FILTER";
      index: number;
    }
  | {
      type: "SET_FILTER_HAS_ABILITY";
      index: number;
      hasAbility: boolean;
    }
  | {
      type: "SET_FILTER_ABILITY_POINTS";
      index: number;
      abilityPoints: { min: number; max: number };
    };

export type UseBuildsByWeaponDispatch = Dispatch<Action>;

export function useBuildsByWeapon() {
  const router = useRouter();
  const [state, dispatch] = useReducer(
    (oldState: UseBuildsByWeaponState, action: Action) => {
      switch (action.type) {
        case "SET_WEAPON":
          router.replace({
            pathname: "/builds",
            query: { weapon: action.weapon },
          });

          return { ...oldState, weapon: action.weapon };
        case "EXPAND_USER":
          return {
            ...oldState,
            expandedUsers: new Set([...oldState.expandedUsers, action.id]),
          };
        case "ADD_FILTER":
          const newFilter = isMainAbility(action.ability)
            ? {
                ability: action.ability,
                hasAbility: true,
              }
            : { ability: action.ability, abilityPoints: { min: 3, max: 57 } };

          return {
            ...oldState,
            filters: [...oldState.filters, newFilter].sort(
              (a, b) =>
                abilities.findIndex((ability) => a.ability === ability.code) -
                abilities.findIndex((ability) => b.ability === ability.code)
            ),
          };
        case "REMOVE_FILTER":
          return {
            ...oldState,
            filters: oldState.filters.filter((_, i) => i !== action.index),
          };
        case "SET_FILTER_HAS_ABILITY":
          const filtersTypeCopy = [...oldState.filters];
          filtersTypeCopy[action.index] = {
            ...filtersTypeCopy[action.index],
            hasAbility: action.hasAbility,
          };

          return {
            ...oldState,
            filters: filtersTypeCopy,
          };
        case "SET_FILTER_ABILITY_POINTS":
          const filtersAbilityPointsCopy = [...oldState.filters];
          filtersAbilityPointsCopy[action.index] = {
            ...filtersAbilityPointsCopy[action.index],
            abilityPoints: action.abilityPoints,
          };

          return {
            ...oldState,
            filters: filtersAbilityPointsCopy,
          };
        default:
          return oldState;
      }
    },
    {
      weapon: getInitialWeapon(),
      filters: [],
      expandedUsers: new Set() as Set<number>,
    }
  );

  function getInitialWeapon() {
    if (
      typeof router.query.weapon !== "string" ||
      !weapons.includes(router.query.weapon as any)
    ) {
      return "";
    }

    return router.query.weapon;
  }

  const { data } = useSWR<GetBuildsByWeaponData>(() => {
    if (!state.weapon) return null;

    const key = state.weapon as keyof typeof weaponToCode;
    return `/api/builds/${weaponToCode[key]}`;
  });

  const buildArrays = data ?? [];

  const buildsToShow = state.filters.length
    ? buildArrays.reduce((acc: GetBuildsByWeaponData, buildArray) => {
        const filteredArray = buildArray.filter((build) => {
          return state.filters.every((filter) => {
            // @ts-ignore
            const apCount = build.abilityPoints[filter.ability] ?? 0;

            if (typeof filter.hasAbility === "boolean")
              return (
                (filter.hasAbility && apCount > 0) ||
                (!filter.hasAbility && apCount === 0)
              );

            if (filter.abilityPoints) {
              return (
                apCount >= filter.abilityPoints.min &&
                apCount <= filter.abilityPoints.max
              );
            }

            console.error("filter had no 'show' or 'abilityPoints' attributes");
            return true;
          });
        });

        if (!filteredArray.length) return acc;

        return [...acc, filteredArray];
      }, [])
    : buildArrays;

  return {
    data: buildsToShow,
    isLoading: state.weapon && !data,
    state,
    dispatch,
    hiddenBuildCount: buildArrays.length - buildsToShow.length,
  };
}