// 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.Collections.Generic;
using ILLink.Shared.DataFlow;
using ILLink.Shared.TrimAnalysis;
using Mono.Linker.Steps;
 
namespace Mono.Linker.Dataflow
{
	public readonly struct TrimAnalysisPatternStore
	{
		readonly Dictionary<(MessageOrigin, int?), TrimAnalysisAssignmentPattern> AssignmentPatterns;
		readonly Dictionary<MessageOrigin, TrimAnalysisMethodCallPattern> MethodCallPatterns;
		readonly ValueSetLattice<SingleValue> Lattice;
		readonly LinkContext _context;
 
		public TrimAnalysisPatternStore (ValueSetLattice<SingleValue> lattice, LinkContext context)
		{
			AssignmentPatterns = new Dictionary<(MessageOrigin, int?), TrimAnalysisAssignmentPattern> ();
			MethodCallPatterns = new Dictionary<MessageOrigin, TrimAnalysisMethodCallPattern> ();
			Lattice = lattice;
			_context = context;
		}
 
		public void Add (TrimAnalysisAssignmentPattern pattern)
		{
			var key = (pattern.Origin, pattern.ParameterIndex);
			if (!AssignmentPatterns.TryGetValue (key, out var existingPattern)) {
				AssignmentPatterns.Add (key, pattern);
				return;
			}
 
			AssignmentPatterns[key] = pattern.Merge (Lattice, existingPattern);
		}
 
		public void Add (TrimAnalysisMethodCallPattern pattern)
		{
			if (!MethodCallPatterns.TryGetValue (pattern.Origin, out var existingPattern)) {
				MethodCallPatterns.Add (pattern.Origin, pattern);
				return;
			}
 
			MethodCallPatterns[pattern.Origin] = pattern.Merge (Lattice, existingPattern);
		}
 
		public void MarkAndProduceDiagnostics (ReflectionMarker reflectionMarker, MarkStep markStep)
		{
			foreach (var pattern in AssignmentPatterns.Values)
				pattern.MarkAndProduceDiagnostics (reflectionMarker, _context);
 
			foreach (var pattern in MethodCallPatterns.Values)
				pattern.MarkAndProduceDiagnostics (reflectionMarker, markStep, _context);
		}
	}
}