facilmap/leaflet/rollup-icons.ts

83 wiersze
3.2 KiB
TypeScript

import type { Plugin } from "rollup";
// eslint-disable-next-line import/no-named-as-default
import glob from "fast-glob";
import { readFile } from "fs/promises";
import { createRequire } from "module";
import { dirname } from "path";
import { fileURLToPath } from "url";
// https://stackoverflow.com/a/62499498/242365
const require = createRequire(import.meta.url);
const coreIcons = [
"arrow-left", "arrow-right", "car", "check", "circle-info", "cog", "copy", "info-sign",
"menu-hamburger", "minus", "new-window", "person-biking", "person-walking", "plus",
"question-sign", "qrcode", "remove", "resize-horizontal", "resize-vertical", "screenshot",
"search", "slash", "triangle-bottom", "triangle-top", "unchecked", "zoom-in"
];
async function getIconFilenames(): Promise<Record<string, Record<string, string>>> {
const icons: Record<string, Record<string, string>> = {};
for (const path of await glob(`${dirname(fileURLToPath(import.meta.url))}/assets/icons/*/*.svg`)) {
const [set, fname] = path.split("/").slice(-2);
if (!icons[set])
icons[set] = {};
icons[set][fname.replace(/\.svg$/, "")] = path;
}
icons["fontawesome"] = {};
for (const name of ["arrow-left", "arrow-right", "person-biking", "car", "chart-line", "copy", "circle-info", "slash", "person-walking"]) {
icons["fontawesome"][name] = require.resolve(`@fortawesome/fontawesome-free/svgs/solid/${name}.svg`);
}
return icons;
}
function pickIcons(icons: Record<string, Record<string, string>>, pick: (set: string, key: string) => boolean): Record<string, Record<string, string>> {
return Object.fromEntries(Object.entries(icons).map(([k1, v2]) => (
[k1, Object.fromEntries(Object.entries(v2).filter(([k2]) => pick(k1, k2)))]
)));
}
async function getIcons(iconFilenames: Record<string, Record<string, string>>): Promise<Record<string, Record<string, string>>> {
const icons: Record<string, Record<string, string>> = {};
for (const [set, fnames] of Object.entries(iconFilenames)) {
icons[set] = {};
for (const [key, fname] of Object.entries(fnames)) {
icons[set][key] = (await readFile(fname)).toString();
}
}
return icons;
}
export default function iconPlugin(): Plugin {
return {
name: 'virtual:icons',
resolveId: (id) => {
if (['virtual:icons:core', 'virtual:icons:extra', 'virtual:icons:keys'].includes(id)) {
return id;
}
},
load: async (id) => {
if (id === 'virtual:icons:keys') {
const iconFilenames = await getIconFilenames();
const icons = Object.fromEntries(Object.entries(iconFilenames).map(([set, fnames]) => [set, Object.keys(fnames)]));
return (
`export const coreIconKeys = ${JSON.stringify(coreIcons)};\n\n` +
`export default ${JSON.stringify(icons, undefined, '\t')}`
);
} else if (id === 'virtual:icons:core') {
const iconFilenames = pickIcons(await getIconFilenames(), (set, key) => coreIcons.includes(key));
const icons = await getIcons(iconFilenames);
return `export default ${JSON.stringify(icons, undefined, '\t')}`;
} else if (id === 'virtual:icons:extra') {
const iconFilenames = pickIcons(await getIconFilenames(), (set, key) => !coreIcons.includes(key));
const icons = await getIcons(iconFilenames);
return `export default ${JSON.stringify(icons, undefined, '\t')}`;
}
}
}
}