ps2.live-old/app/utils/saerro.ts

export const saerroFetch = async <T>(query: string): Promise<T> => {
  const response = await fetch(
    `https://saerro.ps2.live/graphql?query=${query}`,
    {
      cf: {
        cacheTtl: 60,
      },
    }
  );
  const json: { data: T } = await response.json();
  return json.data;
};

export type Population = {
  total: number;
  nc: number;
  tr: number;
  vs: number;
};

export type Zone = {
  id: string;
  name: string;
  population: Population;
  vehicles?: Record<(typeof allVehicles)[number], Population> & {
    total: number;
  };
  classes?: Record<(typeof allClasses)[number], Population>;
};

export type World = {
  id: number;
  name: string;
  population: Population;
  zones: {
    all: Zone[];
  };
};

export type Health = {
  ingestReachable: string;
  ingest: string;
  database: string;
  worlds: {
    name: string;
    status: string;
    lastEvent: string;
  }[];
};

export type IndexResponse = {
  health: Health;
  allWorlds: World[];
};

export const indexQuery = async (): Promise<IndexResponse> => {
  const query = `{
    health {
      worlds {
        name
        status
        lastEvent
      }
    }
    allWorlds {
      id
      name
      population {
        nc
        tr
        vs
      }
      zones {
        all {
          id
          name
          population {
            nc
            tr
            vs
          }
        }
      }
    }
  }`;

  const indexData: IndexResponse = await saerroFetch(query);

  indexData.allWorlds.sort((a, b) => a.id - b.id);

  return indexData;
};

export type WorldResponse = {
  world: World;
};

export const allVehicles = [
  "flash",
  "sunderer",
  "lightning",
  "scythe",
  "vanguard",
  "prowler",
  "reaver",
  "mosquito",
  "galaxy",
  "valkyrie",
  "liberator",
  "ant",
  "harasser",
  "dervish",
  "chimera",
  "javelin",
  "corsair",
  "magrider",
];

export const allClasses = [
  "infiltrator",
  "lightAssault",
  "combatMedic",
  "engineer",
  "heavyAssault",
  "max",
];

export const worldQuery = async (worldID: string): Promise<WorldResponse> => {
  const query = `{
    world(by: {id: ${Number(worldID)}}) {
      id
      name
      population {
        total
        nc
        tr
        vs
      }
      zones {
        all {
          id
          name
          classes {
            ${allClasses.map((cls) => `${cls} { total nc tr vs }`).join(" ")}
          }
          vehicles {
            total
            ${allVehicles
              .map((vehicle) => `${vehicle} { total nc tr vs }`)
              .join(" ")}
          }
          population {
            total
            nc
            tr
            vs
          }
        }
      }
    }
  }`;

  const worldData: WorldResponse = await saerroFetch(query);

  return worldData;
};

export const totalPopulation = ({ nc, vs, tr }: Population): number =>
  nc + vs + tr;