import { file, Glob } from "bun";

import { basename, dirname, extname, resolve } from "node:path";

import { parse, type PlistValue } from "plist";
import { z } from "zod";

const plistStrictSchema = z.object({
  /** @type {string | undefined} */
  CFBundleName: z.unknown().optional(),
  CFBundleExecutable: z.string(),
  UnityBuildNumber: z.unknown().optional(),
  /** @type {string | undefined} */
  CFBundleShortVersionString: z.unknown().optional(),
  /** @type {string | undefined} */
  CFBundleGetInfoString: z.unknown().optional(),
  /** @type {string | undefined} */
  CFBundleIconFile: z.string().optional(),
});
type PlistStrict = z.infer<typeof plistStrictSchema>;

/** Zod schema for working with macOS Unity app Info.plist files. */
export const plistSchema = plistStrictSchema.passthrough();

/** A parsed macOS Unity app Info.plist. */
export type Plist = z.infer<typeof plistSchema>;

type Key = keyof PlistStrict | (string & {});

/**
 * Retrieves the paths to the `Info.plist` file for every macOS Application in
 * a given directory, recursively. Also takes into account the case where the
 * distributed Application is missing the '[name].app' wrapping directory.
 *
 * @param path The directory path to look at.
 * @returns An async generator of `string` paths.
 */
export const search = async function* (path: string) {
  const resolved = resolve(path);
  const glob = new Glob("**/Info.plist");
  for await (
    const relPath of glob.scan({
      onlyFiles: true,
      cwd: resolved,
    })
  ) {
    const plist = resolve(resolved, relPath);
    if (
      basename(dirname(plist)) === "Contents" &&
      new Set([".app", ""]).has(extname(dirname(dirname(plist))))
    ) {
      yield plist;
    }
  }
};

/**
 * Parses the plist file at `path` and returns it.
 *
 * @param path The path to an `Info.plist` file to parse.
 */
export const parsePlistFromFile = async (path: string) =>
  plistSchema.parse(parse(await file(path).text()));

/**
 * Retrieves a value from the plist file at `path` with given `key`.
 *
 * @param path The path to an `Info.plist` file to parse.
 * @param key The key to look up in the `Info.plist` file.
 *
 * @returns A string representation of the value in the plist file matching
 * `key`, or `undefined` if no matching key was found.
 */
export const getValue = (path: string, key: Key) =>
  parsePlistFromFile(path)
    .then((plist) => plist[key] as PlistValue)
    .catch(() => undefined);