// Copyright (c) .NET Foundation and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

using System;
using System.Linq;
using Mono.Cecil;

namespace Mono.Linker.Steps
{
    public class RemoveResourcesStep : BaseStep
    {
        protected override void ProcessAssembly(AssemblyDefinition assembly)
        {
            if (!ShouldProcess(assembly)) return;

            RemoveFSharpCompilationResources(assembly);
        }

        private bool ShouldProcess(AssemblyDefinition assembly)
        {
            if (!assembly.MainModule.HasResources)
                return false;

            var action = Annotations.GetAction(assembly);
            return action == AssemblyAction.Link || action == AssemblyAction.Save;
        }

        private void RemoveFSharpCompilationResources(AssemblyDefinition assembly)
        {
            var resourcesInAssembly = assembly.MainModule.Resources.OfType<EmbeddedResource>();
            foreach (var resource in resourcesInAssembly.Where(IsFSharpCompilationResource))
            {
                Annotations.AddResourceToRemove(assembly, resource);
            }

            static bool IsFSharpCompilationResource(Resource resource)
                => resource.Name.StartsWith("FSharpSignatureData", StringComparison.Ordinal)
                || resource.Name.StartsWith("FSharpOptimizationData", StringComparison.Ordinal);
        }
    }
}