//----------------------------------------------------------------------------
// Copyright (c) Microsoft Corporation.  All rights reserved.
//----------------------------------------------------------------------------

#pragma warning disable 1634 // Stops compiler from warning about unknown warnings (for Presharp)
namespace System.ServiceModel.Channels
{
    using System;
    using System.IO;
    using System.Runtime;
    using System.ServiceModel;
    using System.Xml;
    using DiagnosticUtility = System.ServiceModel.DiagnosticUtility;

    class HttpStreamMessage : Message
    {
        internal const string StreamElementName = "Binary";
        BodyWriter bodyWriter;
        MessageHeaders headers;
        MessageProperties properties;

        public HttpStreamMessage(BodyWriter writer)
        {
            if (writer == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("writer");
            }
            this.bodyWriter = writer;
            this.headers = new MessageHeaders(MessageVersion.None, 1);
            this.properties = new MessageProperties();
        }

        public HttpStreamMessage(MessageHeaders headers, MessageProperties properties, BodyWriter bodyWriter)
        {
            if (bodyWriter == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperArgumentNull("bodyWriter");
            }
            this.headers = new MessageHeaders(headers);
            this.properties = new MessageProperties(properties);
            this.bodyWriter = bodyWriter;
        }

        public override MessageHeaders Headers
        {
            get
            {
                if (IsDisposed)
                {
#pragma warning suppress 56503
                    throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(CreateDisposedException());
                }
                return headers;
            }
        }

        public override bool IsEmpty
        {
            get
            {
                return false;
            }
        }

        public override bool IsFault
        {
            get { return false; }
        }

        public override MessageProperties Properties
        {
            get
            {
                if (IsDisposed)
                {
#pragma warning suppress 56503
                    throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(CreateDisposedException());
                }
                return properties;
            }
        }

        public override MessageVersion Version
        {
            get
            {
                if (IsDisposed)
                {
#pragma warning suppress 56503
                    throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(CreateDisposedException());
                }
                return MessageVersion.None;
            }
        }

        protected override void OnBodyToString(XmlDictionaryWriter writer)
        {
            if (this.bodyWriter.IsBuffered)
            {
                bodyWriter.WriteBodyContents(writer);
            }
            else
            {
                writer.WriteString(SR2.GetString(SR2.MessageBodyIsStream));
            }
        }

        protected override void OnClose()
        {
            Exception ex = null;
            try
            {
                base.OnClose();
            }
            catch (Exception e)
            {
                if (Fx.IsFatal(e))
                {
                    throw;
                }
                ex = e;
            }

            try
            {
                if (properties != null)
                {
                    properties.Dispose();
                }
            }
            catch (Exception e)
            {
                if (Fx.IsFatal(e))
                {
                    throw;
                }
                if (ex == null)
                {
                    ex = e;
                }
            }

            if (ex != null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(ex);
            }

            this.bodyWriter = null;
        }

        protected override MessageBuffer OnCreateBufferedCopy(int maxBufferSize)
        {
            BodyWriter bufferedBodyWriter;
            if (this.bodyWriter.IsBuffered)
            {
                bufferedBodyWriter = this.bodyWriter;
            }
            else
            {
                bufferedBodyWriter = this.bodyWriter.CreateBufferedCopy(maxBufferSize);
            }
            return new HttpStreamMessageBuffer(this.Headers, new MessageProperties(this.Properties), bufferedBodyWriter);
        }

        protected override void OnWriteBodyContents(XmlDictionaryWriter writer)
        {
            this.bodyWriter.WriteBodyContents(writer);
        }

        Exception CreateDisposedException()
        {
            return new ObjectDisposedException("", SR2.GetString(SR2.MessageClosed));
        }

        class HttpStreamMessageBuffer : MessageBuffer
        {
            BodyWriter bodyWriter;
            bool closed;
            MessageHeaders headers;
            MessageProperties properties;
            object thisLock = new object();

            public HttpStreamMessageBuffer(MessageHeaders headers,
                MessageProperties properties, BodyWriter bodyWriter)
                : base()
            {
                this.bodyWriter = bodyWriter;
                this.headers = headers;
                this.properties = properties;
            }

            public override int BufferSize
            {
                get { return 0; }
            }

            object ThisLock
            {
                get { return thisLock; }
            }

            public override void Close()
            {
                lock (ThisLock)
                {
                    if (!closed)
                    {
                        closed = true;
                        bodyWriter = null;
                        headers = null;
                        properties = null;
                    }
                }
            }

            public override Message CreateMessage()
            {
                lock (ThisLock)
                {
                    if (closed)
                    {
                        throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(CreateDisposedException());
                    }
                    return new HttpStreamMessage(this.headers, this.properties, this.bodyWriter);
                }
            }

            Exception CreateDisposedException()
            {
                return new ObjectDisposedException("", SR2.GetString(SR2.MessageBufferIsClosed));
            }
        }
    }
}