drop/server/internal/metadata/pcgamingwiki.ts

import type { Developer, Publisher } from "~/prisma/client";
import { MetadataSource } from "~/prisma/client";
import type { MetadataProvider } from ".";
import type {
  GameMetadataSearchResult,
  _FetchGameMetadataParams,
  GameMetadata,
  _FetchPublisherMetadataParams,
  PublisherMetadata,
  _FetchDeveloperMetadataParams,
  DeveloperMetadata,
} from "./types";
import type { AxiosRequestConfig } from "axios";
import axios from "axios";
import * as jdenticon from "jdenticon";
import { DateTime } from "luxon";

interface PCGamingWikiPage {
  PageID: string;
  PageName: string;
}

interface PCGamingWikiSearchStub extends PCGamingWikiPage {
  "Cover URL": string | null;
  Released: string | null;
  Released__precision: string | null;
}

interface PCGamingWikiGame extends PCGamingWikiSearchStub {
  Developers: string | null;
  Genres: string | null;
  Publishers: string | null;
  Themes: string | null;
  Series: string | null;
  Modes: string | null;
}

interface PCGamingWikiCompany extends PCGamingWikiPage {
  Parent: string | null;
  Founded: string | null;
  Website: string | null;
  Founded__precision: string | null;
  Defunct__precision: string | null;
}

interface PCGamingWikiCargoResult<T> {
  cargoquery: [
    {
      title: T;
    },
  ];
  error?: {
    code?: string;
    info?: string;
    errorclass?: string;
    "*"?: string;
  };
}

// Api Docs: https://www.pcgamingwiki.com/wiki/PCGamingWiki:API
// Good tool for helping build cargo queries: https://www.pcgamingwiki.com/wiki/Special:CargoQuery
export class PCGamingWikiProvider implements MetadataProvider {
  id() {
    return "pcgamingwiki";
  }
  name() {
    return "PCGamingWiki";
  }
  source() {
    return MetadataSource.PCGamingWiki;
  }

  private async request<T>(
    query: URLSearchParams,
    options?: AxiosRequestConfig,
  ) {
    const finalURL = `https://www.pcgamingwiki.com/w/api.php?${query.toString()}`;

    const overlay: AxiosRequestConfig = {
      url: finalURL,
      baseURL: "",
    };
    const response = await axios.request<PCGamingWikiCargoResult<T>>(
      Object.assign({}, options, overlay),
    );

    if (response.status !== 200)
      throw new Error(
        `Error in pcgamingwiki \nStatus Code: ${response.status}`,
      );
    else if (response.data.error !== undefined)
      throw new Error(`Error in pcgamingwiki, malformed query`);

    return response;
  }

  async search(query: string) {
    const searchParams = new URLSearchParams({
      action: "cargoquery",
      tables: "Infobox_game",
      fields:
        "Infobox_game._pageID=PageID,Infobox_game._pageName=PageName,Infobox_game.Cover_URL,Infobox_game.Released",
      where: `Infobox_game._pageName="${query}"`,
      format: "json",
    });

    const res = await this.request<PCGamingWikiSearchStub>(searchParams);

    const mapped = res.data.cargoquery.map((result) => {
      const game = result.title;

      const metadata: GameMetadataSearchResult = {
        id: game.PageID,
        name: game.PageName,
        icon: game["Cover URL"] ?? "",
        description: "", // TODO: need to render the `Introduction` template somehow (or we could just hardcode it)
        year:
          game.Released !== null && game.Released.length > 0
            ? // sometimes will provide multiple dates
              DateTime.fromISO(game.Released.split(";")[0]).year
            : 0,
      };
      return metadata;
    });

    return mapped;
  }

  /**
   * Parses the specific format that the wiki returns when specifying a company
   * @param companyStr
   * @returns
   */
  private parseCompanyStr(companyStr: string): string[] {
    const results: string[] = [];
    // provides the string as a list of companies
    // ie: "Company:Digerati Distribution,Company:Greylock Studio"
    const items = companyStr.split(",");

    items.forEach((item) => {
      // remove the `Company:` and trim and whitespace
      results.push(item.replace("Company:", "").trim());
    });

    return results;
  }

  async fetchGame({
    id,
    name,
    publisher,
    developer,
    createObject,
  }: _FetchGameMetadataParams): Promise<GameMetadata> {
    const searchParams = new URLSearchParams({
      action: "cargoquery",
      tables: "Infobox_game",
      fields:
        "Infobox_game._pageID=PageID,Infobox_game._pageName=PageName,Infobox_game.Cover_URL,Infobox_game.Developers,Infobox_game.Released,Infobox_game.Genres,Infobox_game.Publishers,Infobox_game.Themes,Infobox_game.Series,Infobox_game.Modes",
      where: `Infobox_game._pageID="${id}"`,
      format: "json",
    });

    const res = await this.request<PCGamingWikiGame>(searchParams);
    if (res.data.cargoquery.length < 1)
      throw new Error("Error in pcgamingwiki, no game");

    const game = res.data.cargoquery[0].title;

    const publishers: Publisher[] = [];
    if (game.Publishers !== null) {
      const pubListClean = this.parseCompanyStr(game.Publishers);
      for (const pub of pubListClean) {
        publishers.push(await publisher(pub));
      }
    }

    const developers: Developer[] = [];
    if (game.Developers !== null) {
      const devListClean = this.parseCompanyStr(game.Developers);
      for (const dev of devListClean) {
        developers.push(await developer(dev));
      }
    }

    const icon =
      game["Cover URL"] !== null
        ? createObject(game["Cover URL"])
        : createObject(jdenticon.toPng(name, 512));

    const metadata: GameMetadata = {
      id: game.PageID,
      name: game.PageName,
      shortDescription: "", // TODO: (again) need to render the `Introduction` template somehow (or we could just hardcode it)
      description: "",
      released: game.Released
        ? DateTime.fromISO(game.Released.split(";")[0]).toJSDate()
        : new Date(),

      reviewCount: 0,
      reviewRating: 0,

      publishers,
      developers,

      icon: icon,
      bannerId: icon,
      coverId: icon,
      images: [icon],
    };

    return metadata;
  }

  async fetchPublisher({
    query,
    createObject,
  }: _FetchPublisherMetadataParams): Promise<PublisherMetadata> {
    const searchParams = new URLSearchParams({
      action: "cargoquery",
      tables: "Company",
      fields:
        "Company.Parent,Company.Founded,Company.Defunct,Company.Website,Company._pageName=PageName,Company._pageID=pageID",
      where: `Company._pageName="Company:${query}"`,
      format: "json",
    });

    const res = await this.request<PCGamingWikiCompany>(searchParams);

    // TODO: replace
    const icon = createObject(jdenticon.toPng(query, 512));

    for (let i = 0; i < res.data.cargoquery.length; i++) {
      const company = res.data.cargoquery[i].title;

      const fixedCompanyName =
        company.PageName.split("Company:").at(1) ?? company.PageName;

      const metadata: PublisherMetadata = {
        id: company.PageID,
        name: fixedCompanyName,
        shortDescription: "",
        description: "",
        website: company?.Website ?? "",

        logo: icon,
        banner: icon,
      };
      return metadata;
    }

    throw new Error("Error in pcgamingwiki, no publisher");
  }

  async fetchDeveloper(
    params: _FetchDeveloperMetadataParams,
  ): Promise<DeveloperMetadata> {
    return await this.fetchPublisher(params);
  }
}