PowerToys/src/modules/cmdpal/Microsoft.CmdPal.UI/Helpers/MarkdownImageProviders/ImageProvider.cs

// Copyright (c) Microsoft Corporation
// The Microsoft Corporation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using CommunityToolkit.WinUI.Controls;
using Microsoft.Extensions.Logging;
using Microsoft.UI.Xaml.Controls;

namespace Microsoft.CmdPal.UI.Helpers.MarkdownImageProviders;

public sealed partial class ImageProvider : IImageProvider
{
    private readonly CompositeImageSourceProvider _compositeProvider = new();
    private readonly ILogger _logger;

    public ImageProvider(ILogger logger)
    {
        _logger = logger;
    }

    public async Task<Image> GetImage(string url)
    {
        try
        {
            ImageSourceFactory.Initialize();

            var imageSource = await _compositeProvider.GetImageSource(url);
            return RtbInlineImageFactory.Create(imageSource.ImageSource, new RtbInlineImageFactory.InlineImageOptions
            {
                DownscaleOnly = imageSource.Hints.DownscaleOnly ?? true,
                FitColumnWidth = imageSource.Hints.FitMode == "fit",
                MaxWidthDip = imageSource.Hints.MaxPixelWidth,
                MaxHeightDip = imageSource.Hints.MaxPixelHeight,
                WidthDip = imageSource.Hints.DesiredPixelWidth,
                HeightDip = imageSource.Hints.DesiredPixelHeight,
            });
        }
        catch (Exception ex)
        {
            Log_FailedToProvideImageFromUrl(url, ex);
            return null!;
        }
    }

    public bool ShouldUseThisProvider(string url)
    {
        return _compositeProvider.ShouldUseThisProvider(url);
    }

    [LoggerMessage(level: LogLevel.Error, message: "Failed to provide an image from URI '{url}'")]
    partial void Log_FailedToProvideImageFromUrl(string url, Exception ex);
}