using DotNetty.Transport.Channels;
using JSMachine.WMS.Infrastructure.Helper;
using JSMachine.WMS.Netty.Common.Util;
using JSMachine.WMS.Netty.NettyUdp.Command;
using JSMachine.WMS.Netty.NettyUdp.CommandHandler;
using System;
using System.Collections.Generic;
using System.ComponentModel.Composition;
using System.ComponentModel.Composition.Hosting;
using System.Reflection;
using System.Threading.Tasks;

namespace JSMachine.WMS.Netty.NettyUdp.Transport
{
    public class NettyUdpTransport
    {
        public IChannelHandlerContext ChannelHandlerContext;
        public IPInfo RemoteIpInfo;

        [ImportMany("NettyUdp")]
        private List<BaseCmdhandler> AllCmds { get; set; }

        private Lazy<CompositionContainer> _Container = new Lazy<CompositionContainer>(() =>
        {
            AggregateCatalog catalog = new();
            catalog.Catalogs.Add(new AssemblyCatalog(Assembly.GetExecutingAssembly()));

            return new CompositionContainer(catalog);
        });

        public NettyUdpTransport(IChannelHandlerContext channel)
        {
            _Container.Value.ComposeParts(this);

            ChannelHandlerContext = channel;

            foreach (BaseCmdhandler cmd in AllCmds)
            {
                cmd.Channel = channel;
            }

            RemoteIpInfo = channel.ToIPInfo(AddressOnwer.Remote);
        }
        
        public Task ChannelReadHandle(string msg)
        {
            return Task.Run(async () =>
            {
                foreach (BaseCmdhandler cmd in AllCmds)
                {
                    bool isMsgHasBeenHandled = await cmd.Handle(msg);
                    if (isMsgHasBeenHandled)
                        break;
                }
            });
        }

        public async Task<bool> WriteMsg(BaseCmd msg)
        {
            try
            {
                if (ChannelHandlerContext.Channel.Active)
                {
                    await ChannelHandlerContext.Channel.WriteAndFlushAsync(msg);
                    return true;
                }
            }
            catch (Exception ex)
            {
                LogHelper.Error($"向TCP通道写入数据失败，错误信息 {ex.Message}");

            }

            return false;
        }
    }
}