import type { PlatformIconIOS as ResolvedPlatformIconIOS } from '../../../../../fabric/gamma/stack/StackHeaderItemIOSNativeComponent';
import type { PlatformIconIOS } from '../../../../../types';
import { Image } from 'react-native';

export function resolveIconAssetSources(
  icon: PlatformIconIOS | undefined,
): ResolvedPlatformIconIOS | undefined {
  if (icon == null) {
    return undefined;
  }
  if (icon.type === 'imageSource') {
    const resolvedImageSource = Image.resolveAssetSource(icon.imageSource);

    if (!resolvedImageSource) {
      return undefined;
    }

    return {
      type: 'imageSource',
      imageSource: resolvedImageSource,
    };
  }
  if (icon.type === 'templateSource') {
    const resolvedTemplateSource = Image.resolveAssetSource(
      icon.templateSource,
    );

    if (!resolvedTemplateSource) {
      return undefined;
    }

    return {
      type: 'templateSource',
      templateSource: resolvedTemplateSource,
    };
  }
  return icon;
}