// <copyright>
// Copyright (c) Microsoft Corporation.  All rights reserved.
// </copyright> 

namespace System.ServiceModel.Channels
{
    using System;
    using System.Collections.Generic;
    using System.Diagnostics.CodeAnalysis;
    using System.Globalization;
    using System.Net;
    using System.Net.Sockets;
    using System.Runtime;
    using System.Runtime.Diagnostics;
    using System.ServiceModel.Diagnostics;
    using System.Threading;
    using System.Xml;

    internal class ClientUdpOutputChannel : UdpOutputChannel
    {
        private EndpointAddress to;

        public ClientUdpOutputChannel(ChannelManagerBase factory, IPEndPoint remoteEndPoint, MessageEncoder encoder, BufferManager bufferManager, UdpSocket[] sendSockets, UdpRetransmissionSettings retransmissionSettings, EndpointAddress to, Uri via, bool isMulticast)
            : base(factory, encoder, bufferManager, sendSockets, retransmissionSettings, via, isMulticast)
        {
            Fx.Assert(to != null, "to address can't be null for this constructor...");
            Fx.Assert(remoteEndPoint != null, "remoteEndPoint can't be null");
            
            this.RemoteEndPoint = remoteEndPoint;
            this.to = to;
        }

        public IPEndPoint RemoteEndPoint
        {
            get;
            private set;
        }

        protected override UdpSocket[] GetSendSockets(Message message, out IPEndPoint remoteEndPoint, out Exception exceptionToBeThrown)
        {
            UdpSocket[] socketList = null;
            remoteEndPoint = this.RemoteEndPoint;
            exceptionToBeThrown = null;

            if (this.IsMulticast)
            {
                // always send on all sockets...
                socketList = this.SendSockets;
            }
            else
            {
                Fx.Assert(this.SendSockets.Length == 1, "Unicast Send socket list on client should always be 1 item long");
                socketList = this.SendSockets;
            }

            return socketList;
        }

        protected override void AddHeadersTo(Message message)
        {
            Fx.Assert(message != null, "Message can't be null");

            if (message.Version.Addressing != AddressingVersion.None)
            {
                this.to.ApplyTo(message);
            }

            message.Properties.Via = this.Via;

            base.AddHeadersTo(message);
        }
    }
}