// 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 ILLink.Shared;
using Mono.Cecil;
 
namespace Mono.Linker.Steps
{
	public static class MarkExportedTypesTarget
	{
		public static void ProcessAssembly (AssemblyDefinition assembly, LinkContext context)
		{
			if (!assembly.MainModule.HasExportedTypes)
				return;
 
			foreach (var type in assembly.MainModule.ExportedTypes)
				InitializeExportedType (type, context, assembly);
		}
 
		static void InitializeExportedType (ExportedType exportedType, LinkContext context, AssemblyDefinition assembly)
		{
			if (!context.Annotations.IsMarked (exportedType))
				return;
 
			if (!context.Annotations.TryGetPreservedMembers (exportedType, out TypePreserveMembers members))
				return;
 
			TypeDefinition? type = context.TryResolve (exportedType);
			if (type == null) {
				if (!context.IgnoreUnresolved)
					context.LogError (null, DiagnosticId.ExportedTypeCannotBeResolved, exportedType.Name);
 
				return;
			}
 
			context.Annotations.Mark (type, new DependencyInfo (DependencyKind.ExportedType, exportedType), new MessageOrigin (assembly));
			context.Annotations.SetMembersPreserve (type, members);
		}
	}
}