using System;
using System.Diagnostics;
using System.Text;
using System.Net.Sockets;
using System.Reflection;
using System.Xml;
using System.Collections;

namespace Trace
{
	/// <summary>
	/// TraceListener is a specialized TraceListener that sends the output
	/// of <c>Debug.Write</c> or <c>Debug.WriteLine</c> to a server using
	/// UDP/IP. This class is intended to be used with the TraceViewer VS.Net Add-In.
	/// </summary>
    public class TraceListener : System.Diagnostics.TraceListener, IDisposable
    {
        private TcpClient client = null;
        
        [Conditional("DEBUG")]
            /// <summary>
            /// Call this method to hook to the Debug infrastructure a 
            /// TraceListener that sends its output to the specified server
            /// and port
            /// </summary>
            /// <param name="server">The IP address or name of the server</param>
            /// <param name="port">The port number the server is listening to</param>
            /// <exception cref="System.ArgumentException"></exception>
            /// <exception cref="System.ArgumentNullException"></exception>
            /// <exception cref="System.ArgumentOutOfRangeException"></exception>
            /// <exception cref="System.Net.Sockets.SocketException"></exception>
            /// <exception cref="System.ObjectDisposedException"></exception>
            /// <exception cref="System.InvalidOperationException"></exception>
        public static void InstallTcpTraceListener(string server, int port)
        {
            Debug.Listeners.Add(new TraceListener(server, port));
        }

        /// <summary>
        /// Constructor that takes a server IP address and port
        /// </summary>
        /// <param name="server">The server address or name</param>
        /// <param name="port">The server port number</param>
        public TraceListener(string server, int port)
        {
            client = new TcpClient(server, port);
        }

        public override void Write(string message)
        {
            SendString(message);
        }
		
        public override void WriteLine(string message)
        {
            SendString(message);
        }

        public override void Close()
        {
            client.Close();
            base.Close ();
        }

        public void Dispose()
        {
            this.Close();
        }

        private void SendString(string message)
        {
            NetworkStream stream = client.GetStream();
            Byte[] data = System.Text.Encoding.Unicode.GetBytes(message);
            stream.Write(data, 0, data.Length);
        }
    }
}