// 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 Mono.Cecil;
 
namespace Mono.Linker
{
	public class MarkingHelpers
	{
		protected readonly LinkContext _context;
 
		public MarkingHelpers (LinkContext context)
		{
			_context = context;
		}
 
		public void MarkMatchingExportedType (TypeDefinition typeToMatch, AssemblyDefinition assembly, in DependencyInfo reason, in MessageOrigin origin)
		{
			if (typeToMatch == null || assembly == null)
				return;
 
			if (assembly.MainModule.GetMatchingExportedType (typeToMatch, _context, out var exportedType))
				MarkExportedType (exportedType, assembly.MainModule, reason, origin);
		}
 
		public void MarkExportedType (ExportedType exportedType, ModuleDefinition module, in DependencyInfo reason, in MessageOrigin origin)
		{
			if (!_context.Annotations.MarkProcessed (exportedType, reason))
				return;
 
			_context.Annotations.Mark (module, reason, origin);
		}
 
		public void MarkForwardedScope (TypeReference typeReference, in MessageOrigin origin)
		{
			if (typeReference == null)
				return;
 
			if (typeReference.Scope is AssemblyNameReference) {
				var assembly = _context.Resolve (typeReference.Scope);
				if (assembly != null &&
					_context.TryResolve (typeReference) is TypeDefinition typeDefinition &&
					assembly.MainModule.GetMatchingExportedType (typeDefinition, _context, out var exportedType))
					MarkExportedType (exportedType, assembly.MainModule, new DependencyInfo (DependencyKind.ExportedType, typeReference), origin);
			}
		}
	}
}