sendou.ink/hooks/freeagents.ts

import { Playstyle } from "@prisma/client";
import { useUser } from "hooks/common";
import { useRouter } from "next/router";
import { FreeAgentLikeInfo } from "pages/api/freeagents/like";
import { GetAllFreeAgentPostsData } from "prisma/queries/getAllFreeAgentPosts";
import { useReducer } from "react";
import useSWR from "swr";
import { setSearchParams } from "utils/setSearchParams";

interface UseFreeAgentsState {
  playstyle?: Playstyle;
  weapon?: string;
}

type Action = {
  type: "SET_PLAYSTYLE";
  playstyle?: Playstyle;
};

export function useFreeAgents() {
  const router = useRouter();
  const [user] = useUser();

  const { data: postsData } = useSWR<GetAllFreeAgentPostsData>(
    "/api/freeagents"
  );

  const usersPost = postsData?.find(
    (post) => post.user.discordId === user?.discordId
  );

  const { data: likesData } = useSWR<FreeAgentLikeInfo>(
    usersPost ? "/api/freeagents/like" : null
  );

  const [state, dispatch] = useReducer(
    (oldState: UseFreeAgentsState, action: Action) => {
      switch (action.type) {
        case "SET_PLAYSTYLE":
          setSearchParams("playstyle", action.playstyle);

          return { ...oldState, playstyle: action.playstyle };
        default:
          return oldState;
      }
    },
    getInitialState()
  );

  function getInitialState() {
    if (
      typeof router.query.playstyle !== "string" ||
      !["FRONTLINE", "MIDLINE", "BACKLINE"].includes(
        router.query.playstyle as any
      )
    ) {
      return {};
    }

    return { playstyle: router.query.playstyle as Playstyle };
  }

  const filteredPostsData = (postsData ?? []).filter(
    (post) => !state.playstyle || post.playstyles.includes(state.playstyle)
  );

  return {
    postsData: filteredPostsData,
    likesData,
    isLoading: !postsData,
    usersPost,
    matchedPosts: (likesData?.matchedPostIds ?? []).map((id) =>
      (filteredPostsData ?? []).find((post) => post.id === id)
    ),
    playstyleCounts: (postsData ?? []).reduce(
      (acc, cur) => {
        cur.playstyles.forEach((playstyle) => acc[playstyle]++);

        return acc;
      },
      { FRONTLINE: 0, MIDLINE: 0, BACKLINE: 0 }
    ),
    state,
    dispatch,
  };
}