astro-iconify/lib/createIconPack.ts

Fork of astro-icon. Lets you easily use the up to date iconify service as a straight forward astro icon component.

import { statSync, promises as fs } from "node:fs";
import { fileURLToPath, pathToFileURL } from "node:url";
import resolvePackage from "resolve-pkg";

export interface CreateIconPackOptions {
  package?: string;
  dir?: string;
  url?: string;
}

export function createIconPack({
  package: pkg,
  dir,
  url,
}: CreateIconPackOptions) {
  if (pkg) {
    return async (name: string) => {
      const baseUrl = new URL(pathToFileURL(resolvePackage(pkg)) + "/");
      const path = fileURLToPath(
        new URL(dir ? `${dir}/${name}.svg` : `${name}.svg`, baseUrl)
      );
      if (!exists(path)) {
        throw new Error(
          `[astro-icon] Unable to load "${path}"! Does the file exist?"`
        );
      }
      const svg = await fs.readFile(path).then((res) => res.toString());
      return svg;
    };
  }

  if (url) {
    const baseUrl = new URL(url);
    const fetchCache = new Map();
    return async (name: string) => {
      const url = new URL(name + ".svg", baseUrl).toString();
      if (fetchCache.has(url)) {
        return fetchCache.get(url);
      }
      const res = await fetch(url);
      if (!res.ok) {
        const body = await res.text();
        throw new Error(`Failed to fetch "${url}"!\n${body}`);
      }
      const svg = await res.text();
      fetchCache.set(url, svg);
      return svg;
    };
  }
}

const exists = (path: string): boolean => {
  try {
    return statSync(path).isFile();
  } catch (e) {}
  return false;
};