iconify/packages/utils/src/loader/utils.ts

import type { Awaitable } from '@antfu/utils';
import type { IconCustomizer } from './types';

export const isNode = typeof process < 'u' && typeof process.stdout < 'u'

export async function mergeIconProps(
	svg: string,
	collection: string,
	icon: string,
	additionalProps: Record<string, string | undefined>,
	scale: number,
	propsProvider?: () => Awaitable<Record<string, string>>,
	iconCustomizer?: IconCustomizer,
): Promise<string> {
	const props: Record<string, string> = (await propsProvider?.()) ?? {
		height: `${scale ?? 1}em`,
		width: `${scale ?? 1}em`,
	};
	await iconCustomizer?.(collection, icon, props);
	Object.keys(additionalProps).forEach((p) => {
		const v = additionalProps[p];
		if (v !== undefined && v !== null) props[p] = v;
	});
	// add svg xmlns if missing
	if (!svg.includes(' xmlns=') && !additionalProps['xmlns']) {
		additionalProps['xmlns'] = 'http://www.w3.org/2000/svg';
	}
	// add xmlns:xlink if xlink present and the xmlns missing
	if (!svg.includes('xmlns:xlink') && svg.includes('xlink:') && !additionalProps['xmlns:xlink']) {
		additionalProps['xmlns:xlink'] = 'http://www.w3.org/1999/xlink';
	}
	const replacement = svg.startsWith('<svg ') ? '<svg ' : '<svg';
	return svg.replace(
		replacement,
		`${replacement}${Object.keys(props).map((p) => `${p}="${props[p]}"`).join(' ')}`
	);
}