From 5be9c3f8f94906cc33578ee22262e8f3d43ab537 Mon Sep 17 00:00:00 2001
From: DaoYuan <txp1035@gmail.com>
Date: Tue, 7 Feb 2023 14:50:53 +0800
Subject: [PATCH] feat: customInstall

---
 packages/utils/src/loader/fs.ts          |  7 +++--
 packages/utils/src/loader/install-pkg.ts | 39 ++++++++++++++++--------
 packages/utils/src/loader/node-loader.ts |  5 +--
 packages/utils/src/loader/types.ts       |  1 +
 4 files changed, 32 insertions(+), 20 deletions(-)

diff --git a/packages/utils/src/loader/fs.ts b/packages/utils/src/loader/fs.ts
index bd906e1..7a2a338 100644
--- a/packages/utils/src/loader/fs.ts
+++ b/packages/utils/src/loader/fs.ts
@@ -1,4 +1,5 @@
 import { promises as fs, Stats } from 'fs';
+import type { IconifyLoaderOptions } from './types';
 import { isPackageExists, resolveModule } from 'local-pkg';
 import type { IconifyJSON } from '@iconify/types';
 import { tryInstallPkg } from './install-pkg';
@@ -8,7 +9,7 @@ const isLegacyExists = isPackageExists('@iconify/json');
 
 export async function loadCollectionFromFS(
 	name: string,
-	autoInstall = false
+	options?: IconifyLoaderOptions
 ): Promise<IconifyJSON | undefined> {
 	if (!(await _collections[name])) {
 		_collections[name] = task();
@@ -21,8 +22,8 @@ export async function loadCollectionFromFS(
 			jsonPath = resolveModule(`@iconify/json/json/${name}.json`);
 		}
 
-		if (!jsonPath && !isLegacyExists && autoInstall) {
-			await tryInstallPkg(`@iconify-json/${name}`);
+		if (!jsonPath && !isLegacyExists && options?.autoInstall) {
+			await tryInstallPkg(`@iconify-json/${name}`, options);
 			jsonPath = resolveModule(`@iconify-json/${name}/icons.json`);
 		}
 
diff --git a/packages/utils/src/loader/install-pkg.ts b/packages/utils/src/loader/install-pkg.ts
index fa248a8..1dfe547 100644
--- a/packages/utils/src/loader/install-pkg.ts
+++ b/packages/utils/src/loader/install-pkg.ts
@@ -2,11 +2,15 @@ import { installPackage } from '@antfu/install-pkg';
 import { sleep } from '@antfu/utils';
 import { cyan } from 'kolorist';
 import { warnOnce } from './warn';
+import type { IconifyLoaderOptions } from './types';
 
 let pending: Promise<void> | undefined;
 const tasks: Record<string, Promise<void> | undefined> = {};
 
-export async function tryInstallPkg(name: string): Promise<void | undefined> {
+export async function tryInstallPkg(
+	name: string,
+	options?: IconifyLoaderOptions
+): Promise<void | undefined> {
 	if (pending) {
 		await pending;
 	}
@@ -14,19 +18,28 @@ export async function tryInstallPkg(name: string): Promise<void | undefined> {
 	if (!tasks[name]) {
 		// eslint-disable-next-line no-console
 		console.log(cyan(`Installing ${name}...`));
-		tasks[name] = pending = installPackage(name, {
-			dev: true,
-			preferOffline: true,
-		})
-			.then(() => sleep(300))
-			// eslint-disable-next-line
-			.catch((e: any) => {
-				warnOnce(`Failed to install ${name}`);
-				console.error(e);
+		if (options?.customInstall) {
+			tasks[name] = pending = options
+				.customInstall(name)
+				.then(() => sleep(300))
+				.finally(() => {
+					pending = undefined;
+				});
+		} else {
+			tasks[name] = pending = installPackage(name, {
+				dev: true,
+				preferOffline: true,
 			})
-			.finally(() => {
-				pending = undefined;
-			});
+				.then(() => sleep(300))
+				// eslint-disable-next-line
+				.catch((e: any) => {
+					warnOnce(`Failed to install ${name}`);
+					console.error(e);
+				})
+				.finally(() => {
+					pending = undefined;
+				});
+		}
 	}
 
 	return tasks[name];
diff --git a/packages/utils/src/loader/node-loader.ts b/packages/utils/src/loader/node-loader.ts
index 5ee221a..3809e27 100644
--- a/packages/utils/src/loader/node-loader.ts
+++ b/packages/utils/src/loader/node-loader.ts
@@ -14,10 +14,7 @@ export const loadNodeIcon: UniversalIconLoader = async (
 		return result;
 	}
 
-	const iconSet = await loadCollectionFromFS(
-		collection,
-		options?.autoInstall
-	);
+	const iconSet = await loadCollectionFromFS(collection, options);
 	if (iconSet) {
 		// possible icon names
 		const ids = [
diff --git a/packages/utils/src/loader/types.ts b/packages/utils/src/loader/types.ts
index c694300..1757116 100644
--- a/packages/utils/src/loader/types.ts
+++ b/packages/utils/src/loader/types.ts
@@ -156,6 +156,7 @@ export type IconifyLoaderOptions = {
 	 */
 	autoInstall?: boolean;
 
+	customInstall?: Function;
 	/**
 	 * The additional icon properties applied to the svg.
 	 *