﻿// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#nullable disable

using System;
using System.Reflection;
using Microsoft.CodeAnalysis;

namespace Microsoft.VisualStudio.LanguageServices.Implementation.Diagnostics
{
    internal partial class VisualStudioDiagnosticAnalyzerProvider
    {
        private sealed class Loader : IAnalyzerAssemblyLoader
        {
            private readonly IAnalyzerAssemblyLoader _fallbackLoader;

            public Loader()
            {
                _fallbackLoader = new DefaultAnalyzerAssemblyLoader();
            }

            public void AddDependencyLocation(string fullPath)
            {
                _fallbackLoader.AddDependencyLocation(fullPath);
            }

            public Assembly LoadFromPath(string fullPath)
            {
                try
                {
                    // We want to load the analyzer assembly assets in default context.
                    // Use Assembly.Load instead of Assembly.LoadFrom to ensure that if the assembly is ngen'ed, then the native image gets loaded.
                    return Assembly.Load(AssemblyName.GetAssemblyName(fullPath));
                }
                catch (Exception)
                {
                    // Use the fallback loader if we fail to load the assembly in the default context.
                    return _fallbackLoader.LoadFromPath(fullPath);
                }
            }
        }
    }
}
