import { existsSync } from 'fs';
import sharp from 'sharp';
import { readDVPLFile } from '../../src/core/blitz/readDVPLFile';
import { readStringDVPL } from '../../src/core/blitz/readStringDVPL';
import { readXMLDVPL } from '../../src/core/blitz/readXMLDVPL';
import { readYAMLDVPL } from '../../src/core/blitz/readYAMLDVPL';
import { commitAssets } from '../../src/core/blitzkit/commitAssets';
import { FileChange } from '../../src/core/blitzkit/commitMultipleFiles';
import { DATA } 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(production: boolean) {
  console.log('Building consumable and provision icons...');

  const changes: FileChange[] = [];

  const styles = [
    'UI/Styles/Lobby/Inventory/InventoryNormalStyles.yaml',
    'UI/Screens3/Lobby/Inventory/Equipment/EquipmentItemImage.style.yaml',
    'UI/Styles/Lobby/Inventory/Event/InventoryNormalStyles.yaml',
    'UI/Screens/Battle/Styles/BattleEquipmentStyles.yaml',
    'UI/Styles/Lobby/Inventory/Event/InventoryBigStyles.yaml',
  ];
  const styleSheets = await Promise.all(
    styles.map(async (path) =>
      Object.values(
        (await readYAMLDVPL<Mappings>(`${DATA}/${path}.dvpl`)).StyleSheets,
      ),
    ),
  ).then((array) => array.flat());

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

    await Promise.all(
      Object.values(consumablesCommon.root).map(async (consumable) => {
        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...`,
          );
          return;
        }

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

        if (existsSync(`${DATA}/${configPath}.packed.webp.dvpl`)) {
          const image = sharp(
            await readDVPLFile(`${DATA}/${configPath}.packed.webp.dvpl`),
          );
          const content = (
            await image.trim({ threshold: 100 }).toBuffer()
          ).toString('base64');

          changes.push({
            path: `icons/consumables/${consumable.id}.webp`,
            encoding: 'base64',
            content,
          });
        } else {
          const consumablesTexture = sharp(
            await readDVPLFile(
              `${`${DATA}/${configPath}`.split('/').slice(0, -1).join('/')}/texture0.packed.webp.dvpl`,
            ),
          );
          const sizes = (await readStringDVPL(`${DATA}/${configPath}.txt.dvpl`))
            .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()
          ).toString('base64');

          changes.push({
            path: `icons/consumables/${consumable.id}.webp`,
            encoding: 'base64',
            content,
          });
        }
      }),
    );
  }

  for (const match of (
    await readStringDVPL(
      `${DATA}/XML/item_defs/vehicles/common/provisions/list.xml.dvpl`,
    )
  ).matchAll(listItemsPattern)) {
    const provisionsCommon = await readXMLDVPL<{
      root: ProvisionsCommon;
    }>(`${DATA}/XML/item_defs/vehicles/common/provisions/${match[1]}.xml.dvpl`);
    await Promise.all(
      Object.values(provisionsCommon.root).map(async (provision) => {
        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...`,
          );
          return;
        }

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

        if (existsSync(`${DATA}/${configPath}.packed.webp.dvpl`)) {
          const image = sharp(
            await readDVPLFile(`${DATA}/${configPath}.packed.webp.dvpl`),
          );
          const content = (
            await image.trim({ threshold: 100 }).toBuffer()
          ).toString('base64');

          changes.push({
            path: `icons/provisions/${provision.id}.webp`,
            encoding: 'base64',
            content,
          });
        } else {
          const provisionsTexture = sharp(
            await readDVPLFile(
              `${`${DATA}/${configPath}`.split('/').slice(0, -1).join('/')}/texture0.packed.webp.dvpl`,
            ),
          );
          const sizes = (await readStringDVPL(`${DATA}/${configPath}.txt.dvpl`))
            .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()
          ).toString('base64');

          changes.push({
            path: `icons/provisions/${provision.id}.webp`,
            encoding: 'base64',
            content,
          });
        }
      }),
    );
  }

  await commitAssets('consumable and provision icons', changes, production);
}