// 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.Collections.Generic;
using Mono.Cecil;
 
namespace Mono.Linker.Steps
{
    public class MarkStepContext : MarkContext
    {
 
        public List<Action<AssemblyDefinition>> MarkAssemblyActions { get; }
        public List<Action<TypeDefinition>> MarkTypeActions { get; }
        public List<Action<MethodDefinition>> MarkMethodActions { get; }
 
        public MarkStepContext()
        {
            MarkAssemblyActions = new List<Action<AssemblyDefinition>>();
            MarkTypeActions = new List<Action<TypeDefinition>>();
            MarkMethodActions = new List<Action<MethodDefinition>>();
        }
 
        public override void RegisterMarkAssemblyAction(Action<AssemblyDefinition> action)
        {
            MarkAssemblyActions.Add(action);
        }
 
        public override void RegisterMarkTypeAction(Action<TypeDefinition> action)
        {
            MarkTypeActions.Add(action);
        }
 
        public override void RegisterMarkMethodAction(Action<MethodDefinition> action)
        {
            MarkMethodActions.Add(action);
        }
    }
}