// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.AspNetCore.SignalR.Internal;
using Microsoft.AspNetCore.SignalR.Internal.Protocol;

namespace Microsoft.AspNetCore.SignalR.Common.Tests.Internal.Protocol
{
    public class CompositeTestBinder : IInvocationBinder
    {
        private readonly HubMessage[] _hubMessages;
        private int index = 0;

        public CompositeTestBinder(HubMessage[] hubMessages)
        {
            _hubMessages = hubMessages.Where(IsBindableMessage).ToArray();
        }

        public IReadOnlyList<Type> GetParameterTypes(string methodName)
        {
            index++;
            return new TestBinder(_hubMessages[index - 1]).GetParameterTypes(methodName);
        }

        public Type GetReturnType(string invocationId)
        {
            index++;
            return new TestBinder(_hubMessages[index - 1]).GetReturnType(invocationId);
        }

        private bool IsBindableMessage(HubMessage arg)
        {
            return arg is CompletionMessage ||
                arg is InvocationMessage ||
                arg is StreamItemMessage ||
                arg is StreamInvocationMessage;
        }
    }
}