import sharp from "sharp";
import { AssetUploader } from "../core/github/assetUploader";
import { vfs } from "./constants";
import { ConsumablesCommon, ProvisionsCommon } from "./definitions";

interface Mappings {
  Header: { version: number };
  StyleSheets: {
    selectors: string[];
    properties: { "bg-sprite": string };
  }[];
}

const listItemsPattern = /<items path="(.+)\.xml"\/>/g;

export async function consumableProvisionIcons() {
  console.log("Building consumable and provision icons...");

  using uploader = new AssetUploader("consumable and provision icons");
  const styles = [
    "UI/Styles/Lobby/Inventory/Event/InventoryBigStyles.yaml",
    "UI/Styles/Lobby/Inventory/Event/InventoryNormalStyles.yaml",
    "UI/Styles/Lobby/Inventory/InventoryNormalStyles.yaml",
    "UI/Screens3/Lobby/Inventory/Equipment/EquipmentItemImage.style.yaml",
    "UI/Screens/Battle/Styles/BattleEquipmentStyles.yaml",
  ];
  const styleSheets = await Promise.all(
    styles.map(async (path) =>
      Object.values(
        (
          await vfs.yaml<Mappings>(`Data/${path}`)
        ).StyleSheets
      )
    )
  ).then((array) => array.flat());

  for (const match of (
    await vfs.text(
      `Data/XML/item_defs/vehicles/common/consumables/list.xml`
    )
  ).matchAll(listItemsPattern)) {
    const consumablesCommon = await vfs.xml<{
      root: ConsumablesCommon;
    }>(`Data/XML/item_defs/vehicles/common/consumables/${match[1]}.xml`);

    for (const consumable of Object.values(consumablesCommon.root)) {
      const styleSheet = styleSheets.find((styleSheet) =>
        styleSheet.selectors.some((selector) =>
          selector.includes(`${consumable.icon} `)
        )
      );

      if (!styleSheet) {
        console.warn(
          `No style sheet found for consumable ${consumable.icon}; skipping...`
        );
        continue;
      }

      const configPath = styleSheet.properties["bg-sprite"]
        .replace("~res:/", "")
        .replace(".psd", "")
        .replace(".txt", "");

      if (vfs.resolve(`Data/${configPath}.packed.webp`)) {
        const image = sharp(
          await vfs.file(`Data/${configPath}.packed.webp`)
        );
        const content = await image.trim({ threshold: 100 }).toBuffer();

        await uploader.add({
          path: `icons/consumables/${consumable.id}.webp`,
          content,
        });
      } else {
        const consumablesTexture = sharp(
          await vfs.file(
            `${`Data/${configPath}`
              .split("/")
              .slice(0, -1)
              .join("/")}/texture0.packed.webp`
          )
        );
        const sizes = (await vfs.text(`Data/${configPath}.txt`))
          .split("\n")[4]
          .split(" ")
          .map(Number);
        const content = await consumablesTexture
          .clone()
          .extract({
            left: sizes[0],
            top: sizes[1],
            width: sizes[2],
            height: sizes[3],
          })
          .toBuffer();

        await uploader.add({
          path: `icons/consumables/${consumable.id}.webp`,
          content,
        });
      }
    }
  }

  for (const match of (
    await vfs.text(
      `Data/XML/item_defs/vehicles/common/provisions/list.xml`
    )
  ).matchAll(listItemsPattern)) {
    const provisionsCommon = await vfs.xml<{
      root: ProvisionsCommon;
    }>(`Data/XML/item_defs/vehicles/common/provisions/${match[1]}.xml`);

    for (const provision of Object.values(provisionsCommon.root)) {
      const styleSheet = styleSheets.find((styleSheet) =>
        styleSheet.selectors.some((selector) =>
          selector.includes(`${provision.icon} `)
        )
      );

      if (!styleSheet) {
        console.warn(
          `No style sheet found for provision ${provision.icon}; skipping...`
        );
        continue;
      }

      const configPath = styleSheet.properties["bg-sprite"]
        .replace("~res:/", "")
        .replace(".psd", "")
        .replace(".txt", "");

      if (vfs.resolve(`Data/${configPath}.packed.webp`)) {
        const image = sharp(
          await vfs.file(`Data/${configPath}.packed.webp`)
        );
        const content = await image.trim({ threshold: 100 }).toBuffer();

        await uploader.add({
          path: `icons/provisions/${provision.id}.webp`,
          content,
        });
      } else {
        const provisionsTexture = sharp(
          await vfs.file(
            `${`Data/${configPath}`
              .split("/")
              .slice(0, -1)
              .join("/")}/texture0.packed.webp`
          )
        );
        const sizes = (await vfs.text(`Data/${configPath}.txt`))
          .split("\n")[4]
          .split(" ")
          .map(Number);
        const content = await provisionsTexture
          .clone()
          .extract({
            left: sizes[0],
            top: sizes[1],
            width: sizes[2],
            height: sizes[3],
          })
          .toBuffer();

        await uploader.add({
          path: `icons/provisions/${provision.id}.webp`,
          content,
        });
      }
    }
  }

  await uploader.flush();
}