securityos/utils/spawnSheep.ts

import { loadFiles } from "utils/functions";

type SheepOptions = {
  allowPopup: string;
  collisionsWith: string[];
  spawnContainer: HTMLElement;
};

declare global {
  interface Window {
    Sheep?: new (options: SheepOptions) => {
      Start: (animationXmlUrl: string) => void;
    };
  }
}

const PETS: Record<string, [string, number]> = {
  blue_sheep: ["/Program Files/eSheep/blue_sheep.xml", 12],
  eSheep: ["/Program Files/eSheep/eSheep.xml", 12],
  fox: ["/Program Files/eSheep/fox.xml", 4],
  green_sheep: ["/Program Files/eSheep/green_sheep.xml", 12],
  mimiko: ["/Program Files/eSheep/mimiko.xml", 4],
  neko: ["/Program Files/eSheep/neko.xml", 10],
  orange_sheep: ["/Program Files/eSheep/orange_sheep.xml", 12],
  pingus: ["/Program Files/eSheep/pingus.xml", 10],
  red_sheep: ["/Program Files/eSheep/red_sheep.xml", 12],
  yellow_sheep: ["/Program Files/eSheep/yellow_sheep.xml", 12],
};

let oneSheepLaunched = false;

const pickRandomPet = (): string => {
  const petNames = Object.keys(PETS).flatMap((pet) => {
    const [, probability] = PETS[pet];

    return Array.from({ length: probability }).fill(pet) as string[];
  });
  const randomPet = Math.floor(Math.random() * petNames.length);
  const [petPath] = PETS[petNames[randomPet]];

  return petPath;
};

const spawnSheep = (): Promise<void> =>
  loadFiles(["/Program Files/eSheep/eSheep.js"]).then(() => {
    if (window.Sheep) {
      const sheep = new window.Sheep({
        allowPopup: "no",
        collisionsWith: ["nav", "section"],
        spawnContainer: document.querySelector("main") as HTMLElement,
      });

      if (oneSheepLaunched) {
        sheep.Start(pickRandomPet());
      } else {
        oneSheepLaunched = true;
        sheep.Start("/Program Files/eSheep/eSheep.xml");
      }
    }
  });

export default spawnSheep;