using System;

using System.Collections;

using System.Collections.Generic;

using System.IO;

using System.Text;



namespace BestHTTP

{

    using BestHTTP.Authentication;

    using BestHTTP.Extensions;

    using BestHTTP.Forms;



#if !BESTHTTP_DISABLE_COOKIES

    using BestHTTP.Cookies;

#endif



    using BestHTTP.Core;

    using BestHTTP.PlatformSupport.Memory;

    using BestHTTP.Connections;

    using BestHTTP.Logger;

    using BestHTTP.Timings;



    /// <summary>

    /// Possible logical states of a HTTTPRequest object.

    /// </summary>

    public enum HTTPRequestStates

    {

        /// <summary>

        /// Initial status of a request. No callback will be called with this status.

        /// </summary>

        Initial,



        /// <summary>

        /// The request queued for processing.

        /// </summary>

        Queued,



        /// <summary>

        /// Processing of the request started. In this state the client will send the request, and parse the response. No callback will be called with this status.

        /// </summary>

        Processing,



        /// <summary>

        /// The request finished without problem. Parsing the response done, the result can be used. The user defined callback will be called with a valid response object. The request¡¯s Exception property will be null.

        /// </summary>

        Finished,



        /// <summary>

        /// The request finished with an unexpected error. The user defined callback will be called with a null response object. The request's Exception property may contain more info about the error, but it can be null.

        /// </summary>

        Error,



        /// <summary>

        /// The request aborted by the client(HTTPRequest¡¯s Abort() function). The user defined callback will be called with a null response. The request¡¯s Exception property will be null.

        /// </summary>

        Aborted,



        /// <summary>

        /// Connecting to the server timed out. The user defined callback will be called with a null response. The request¡¯s Exception property will be null.

        /// </summary>

        ConnectionTimedOut,



        /// <summary>

        /// The request didn't finished in the given time. The user defined callback will be called with a null response. The request¡¯s Exception property will be null.

        /// </summary>

        TimedOut

    }



    public delegate void OnRequestFinishedDelegate(HTTPRequest originalRequest, HTTPResponse response);

    public delegate void OnProgressDelegate(HTTPRequest originalRequest, long progress, long length);

    public delegate bool OnBeforeRedirectionDelegate(HTTPRequest originalRequest, HTTPResponse response, Uri redirectUri);

    public delegate void OnHeaderEnumerationDelegate(string header, List<string> values);

    public delegate void OnBeforeHeaderSendDelegate(HTTPRequest req);

    public delegate void OnHeadersReceivedDelegate(HTTPRequest originalRequest, HTTPResponse response, Dictionary<string, List<string>> headers);



    /// <summary>

    /// Called for every fragment of data downloaded from the server. Its return value indicates whether the plugin free to reuse the dataFragment array.

    /// </summary>

    /// <param name="request">The parent HTTPRequest object</param>

    /// <param name="response">The HTTPResponse object.</param>

    /// <param name="dataFragment">The downloaded data. The byte[] can be larger than the actual payload! Its valid length that can be used is in the dataFragmentLength param.</param>

    /// <param name="dataFragmentLength">Length of the downloaded data.</param>

    public delegate bool OnStreamingDataDelegate(HTTPRequest request, HTTPResponse response, byte[] dataFragment, int dataFragmentLength);



    public sealed class HTTPRequest : IEnumerator, IEnumerator<HTTPRequest>

    {

        #region Statics



        public static readonly byte[] EOL = { HTTPResponse.CR, HTTPResponse.LF };



        /// <summary>

        /// Cached uppercase values to save some cpu cycles and GC alloc per request.

        /// </summary>

        public static readonly string[] MethodNames = {

                                                          HTTPMethods.Get.ToString().ToUpper(),

                                                          HTTPMethods.Head.ToString().ToUpper(),

                                                          HTTPMethods.Post.ToString().ToUpper(),

                                                          HTTPMethods.Put.ToString().ToUpper(),

                                                          HTTPMethods.Delete.ToString().ToUpper(),

                                                          HTTPMethods.Patch.ToString().ToUpper(),

                                                          HTTPMethods.Merge.ToString().ToUpper(),

                                                          HTTPMethods.Options.ToString().ToUpper(),

                                                          HTTPMethods.Connect.ToString().ToUpper(),

                                                          HTTPMethods.Query.ToString().ToUpper()

                                                      };



        /// <summary>

        /// Size of the internal buffer, and upload progress will be fired when this size of data sent to the wire. Its default value is 4 KiB.

        /// </summary>

        public static int UploadChunkSize = 4 * 1024;



        #endregion



        #region Properties



        /// <summary>

        /// The original request's Uri.

        /// </summary>

        public Uri Uri { get; set; }



        /// <summary>

        /// The method that how we want to process our request the server.

        /// </summary>

        public HTTPMethods MethodType { get; set; }



        /// <summary>

        /// The raw data to send in a POST request. If it set all other fields that added to this request will be ignored.

        /// </summary>

        public byte[] RawData { get; set; }



        /// <summary>

        /// The stream that the plugin will use to get the data to send out the server. When this property is set, no forms or the RawData property will be used

        /// </summary>

        public Stream UploadStream { get; set; }



        /// <summary>

        /// When set to true(its default value) the plugin will call the UploadStream's Dispose() function when finished uploading the data from it. Default value is true.

        /// </summary>

        public bool DisposeUploadStream { get; set; }



        /// <summary>

        /// If it's true, the plugin will use the Stream's Length property. Otherwise the plugin will send the data chunked. Default value is true.

        /// </summary>

        public bool UseUploadStreamLength { get; set; }



        /// <summary>

        /// Called after data sent out to the wire.

        /// </summary>

        public OnProgressDelegate OnUploadProgress;



        /// <summary>

        /// Indicates that the connection should be open after the response received. If its true, then the internal TCP connections will be reused if it's possible. Default value is true.

        /// The default value can be changed in the HTTPManager class. If you make rare request to the server it's should be changed to false.

        /// </summary>

        public bool IsKeepAlive

        {

            get { return isKeepAlive; }

            set

            {

                if (State == HTTPRequestStates.Processing)

                    throw new NotSupportedException("Changing the IsKeepAlive property while processing the request is not supported.");

                isKeepAlive = value;

            }

        }



#if !BESTHTTP_DISABLE_CACHING

        /// <summary>

        /// With this property caching can be enabled/disabled on a per-request basis.

        /// </summary>

        public bool DisableCache

        {

            get { return disableCache; }

            set

            {

                if (State == HTTPRequestStates.Processing)

                    throw new NotSupportedException("Changing the DisableCache property while processing the request is not supported.");

                disableCache = value;

            }

        }



        /// <summary>

        /// It can be used with streaming. When set to true, no OnStreamingData event is called, the streamed content will be saved straight to the cache if all requirements are met(caching is enabled and there's a caching headers).

        /// </summary>

        public bool CacheOnly

        {

            get { return cacheOnly; }

            set

            {

                if (State == HTTPRequestStates.Processing)

                    throw new NotSupportedException("Changing the CacheOnly property while processing the request is not supported.");

                cacheOnly = value;

            }

        }

#endif



        /// <summary>

        /// Maximum size of a data chunk that we want to receive when streaming is set. Its default value is 1 MB.

        /// </summary>

        public int StreamFragmentSize

        {

            get{ return streamFragmentSize; }

            set

            {

                if (State == HTTPRequestStates.Processing)

                    throw new NotSupportedException("Changing the StreamFragmentSize property while processing the request is not supported.");



                if (value < 1)

                    throw new System.ArgumentException("StreamFragmentSize must be at least 1.");



                streamFragmentSize = value;

            }

        }



        /// <summary>

        /// When set to true, StreamFragmentSize will be ignored and downloaded chunks will be sent immediately.

        /// </summary>

        public bool StreamChunksImmediately { get; set; }



        /// <summary>

        /// This property can be used to force the HTTPRequest to use an exact sized read buffer.

        /// </summary>

        public int ReadBufferSizeOverride { get; set; }



        /// <summary>

        /// Maximum unprocessed fragments allowed to queue up. 

        /// </summary>

        public int MaxFragmentQueueLength { get; set; }



        /// <summary>

        /// The callback function that will be called when a request is fully processed or when any downloaded fragment is available if UseStreaming is true. Can be null for fire-and-forget requests.

        /// </summary>

        public OnRequestFinishedDelegate Callback { get; set; }



        /// <summary>

        /// When the request is queued for processing.

        /// </summary>

        public DateTime QueuedAt { get; internal set; }



        public bool IsConnectTimedOut { get { return this.QueuedAt != DateTime.MinValue && DateTime.UtcNow - this.QueuedAt > this.ConnectTimeout; } }



        /// <summary>

        /// When the processing of the request started

        /// </summary>

        public DateTime ProcessingStarted { get; internal set; }



        /// <summary>

        /// Returns true if the time passed the Timeout setting since processing started.

        /// </summary>

        public bool IsTimedOut

        {

            get

            {

                DateTime now = DateTime.UtcNow;



                return (!this.UseStreaming || (this.UseStreaming && this.EnableTimoutForStreaming)) &&

                    ((this.ProcessingStarted != DateTime.MinValue && now - this.ProcessingStarted > this.Timeout) ||

                     this.IsConnectTimedOut);

            }

        }



        /// <summary>

        /// Called for every fragment of data downloaded from the server. Return true if dataFrament is processed and the plugin can recycle the byte[].

        /// </summary>

        public OnStreamingDataDelegate OnStreamingData;



        /// <summary>

        /// This event is called when the plugin received and parsed all headers.

        /// </summary>

        public OnHeadersReceivedDelegate OnHeadersReceived;



        /// <summary>

        /// Number of times that the plugin retried the request.

        /// </summary>

        public int Retries { get; internal set; }



        /// <summary>

        /// Maximum number of tries allowed. To disable it set to 0. Its default value is 1 for GET requests, otherwise 0.

        /// </summary>

        public int MaxRetries { get; set; }



        /// <summary>

        /// True if Abort() is called on this request.

        /// </summary>

        public bool IsCancellationRequested { get; internal set; }



        /// <summary>

        /// Called when new data downloaded from the server.

        /// The first parameter is the original HTTTPRequest object itself, the second parameter is the downloaded bytes while the third parameter is the content length.

        /// <remarks>There are download modes where we can't figure out the exact length of the final content. In these cases we just guarantee that the third parameter will be at least the size of the second one.</remarks>

        /// </summary>

        public OnProgressDelegate OnDownloadProgress;



        /// <summary>

        /// Indicates that the request is redirected. If a request is redirected, the connection that served it will be closed regardless of the value of IsKeepAlive.

        /// </summary>

        public bool IsRedirected { get; internal set; }



        /// <summary>

        /// The Uri that the request redirected to.

        /// </summary>

        public Uri RedirectUri { get; internal set; }



        /// <summary>

        /// If redirected it contains the RedirectUri.

        /// </summary>

        public Uri CurrentUri { get { return IsRedirected ? RedirectUri : Uri; } }



        /// <summary>

        /// The response to the query.

        /// <remarks>If an exception occurred during reading of the response stream or can't connect to the server, this will be null!</remarks>

        /// </summary>

        public HTTPResponse Response { get; internal set; }



#if !BESTHTTP_DISABLE_PROXY && (!UNITY_WEBGL || UNITY_EDITOR)

        /// <summary>

        /// Response from the Proxy server. It's null with transparent proxies.

        /// </summary>

        public HTTPResponse ProxyResponse { get; internal set; }

#endif



        /// <summary>

        /// It there is an exception while processing the request or response the Response property will be null, and the Exception will be stored in this property.

        /// </summary>

        public Exception Exception { get; internal set; }



        /// <summary>

        /// Any object can be passed with the request with this property. (eq. it can be identified, etc.)

        /// </summary>

        public object Tag { get; set; }



        /// <summary>

        /// The UserName, Password pair that the plugin will use to authenticate to the remote server.

        /// </summary>

        public Credentials Credentials { get; set; }



#if !BESTHTTP_DISABLE_PROXY && (!UNITY_WEBGL || UNITY_EDITOR)

        /// <summary>

        /// True, if there is a Proxy object.

        /// </summary>

        public bool HasProxy { get { return Proxy != null && Proxy.UseProxyForAddress(this.CurrentUri); } }



        /// <summary>

        /// A web proxy's properties where the request must pass through.

        /// </summary>

        public Proxy Proxy { get; set; }

#endif



        /// <summary>

        /// How many redirection supported for this request. The default is 10. 0 or a negative value means no redirection supported.

        /// </summary>

        public int MaxRedirects { get; set; }



#if !BESTHTTP_DISABLE_COOKIES



        /// <summary>

        /// If true cookies will be added to the headers (if any), and parsed from the response. If false, all cookie operations will be ignored. It's default value is HTTPManager's IsCookiesEnabled.

        /// </summary>

        public bool IsCookiesEnabled { get; set; }



        /// <summary>

        /// Cookies that are added to this list will be sent to the server alongside withe the server sent ones. If cookies are disabled only these cookies will be sent.

        /// </summary>

        public List<Cookie> Cookies

        {

            get

            {

                if (customCookies == null)

                    customCookies = new List<Cookie>();

                return customCookies;

            }

            set { customCookies = value; }

        }



        private List<Cookie> customCookies;

#endif



        /// <summary>

        /// What form should used. Its default value is Automatic.

        /// </summary>

        public HTTPFormUsage FormUsage { get; set; }



        /// <summary>

        /// Current state of this request.

        /// </summary>

        public HTTPRequestStates State {

            get { return this._state; }

            internal set {

                lock (this)

                {

                    if (this._state != value)

                    {

                        //if (this._state >= HTTPRequestStates.Finished && value >= HTTPRequestStates.Finished)

                        //    return;



                        this._state = value;



                        RequestEventHelper.EnqueueRequestEvent(new RequestEventInfo(this, this._state));

                    }

                }

            }

        }

        private volatile HTTPRequestStates _state;



        /// <summary>

        /// How many times redirected.

        /// </summary>

        public int RedirectCount { get; internal set; }



        /// <summary>

        /// Maximum time we wait to establish the connection to the target server. If set to TimeSpan.Zero or lower, no connect timeout logic is executed. Default value is 20 seconds.

        /// </summary>

        public TimeSpan ConnectTimeout { get; set; }



        /// <summary>

        /// Maximum time we want to wait to the request to finish after the connection is established. Default value is 60 seconds.

        /// <remarks>It's disabled for streaming requests! See <see cref="EnableTimoutForStreaming"/>.</remarks>

        /// </summary>

        public TimeSpan Timeout { get; set; }



        /// <summary>

        /// Set to true to enable Timeouts on streaming request. Default value is false.

        /// </summary>

        public bool EnableTimoutForStreaming { get; set; }



        /// <summary>

        /// Enables safe read method when the response's length of the content is unknown. Its default value is enabled (true).

        /// </summary>

        public bool EnableSafeReadOnUnknownContentLength { get; set; }



        /// <summary>

        /// It's called before the plugin will do a new request to the new uri. The return value of this function will control the redirection: if it's false the redirection is aborted.

        /// This function is called on a thread other than the main Unity thread!

        /// </summary>

        public event OnBeforeRedirectionDelegate OnBeforeRedirection

        {

            add { onBeforeRedirection += value; }

            remove { onBeforeRedirection -= value; }

        }

        private OnBeforeRedirectionDelegate onBeforeRedirection;



        /// <summary>

        /// This event will be fired before the plugin will write headers to the wire. New headers can be added in this callback. This event is called on a non-Unity thread!

        /// </summary>

        public event OnBeforeHeaderSendDelegate OnBeforeHeaderSend

        {

            add { _onBeforeHeaderSend += value; }

            remove { _onBeforeHeaderSend -= value; }

        }

        private OnBeforeHeaderSendDelegate _onBeforeHeaderSend;



        /// <summary>

        /// Logging context of the request.

        /// </summary>

        public LoggingContext Context { get; private set; }



        /// <summary>

        /// Timing information.

        /// </summary>

        public TimingCollector Timing { get; private set; }



#if UNITY_WEBGL

        /// <summary>

        /// Its value will be set to the XmlHTTPRequest's withCredentials field. Its default value is HTTPManager.IsCookiesEnabled's value.

        /// </summary>

        public bool WithCredentials { get; set; }

#endif



#if !UNITY_WEBGL || UNITY_EDITOR

        /// <summary>

        /// Called when the current protocol is upgraded to an other. (HTTP => WebSocket for example)

        /// </summary>

        internal OnRequestFinishedDelegate OnUpgraded;

#endif



        #region Internal Properties For Progress Report Support



        /// <summary>

        /// If it's true, the Callback will be called every time if we can send out at least one fragment.

        /// </summary>

        internal bool UseStreaming { get { return this.OnStreamingData != null; } }



        /// <summary>

        /// Will return the length of the UploadStream, or -1 if it's not supported.

        /// </summary>

        internal long UploadStreamLength

        {

            get

            {

                if (UploadStream == null || !UseUploadStreamLength)

                    return -1;



                try

                {

                    // This may will throw a NotSupportedException

                    return UploadStream.Length;

                }

                catch

                {

                    // We will fall back to chunked

                    return -1;

                }

            }

        }



#if !UNITY_WEBGL || UNITY_EDITOR

        /// <summary>

        /// This action is called when a user calls the Abort function. Do not use it outside of the plugin!

        /// </summary>

        internal Action<HTTPRequest> OnCancellationRequested;

#endif



        #endregion



        #endregion



        #region Privates



        private bool isKeepAlive;

#if !BESTHTTP_DISABLE_CACHING

        private bool disableCache;

        private bool cacheOnly;

#endif

        private int streamFragmentSize;



        private Dictionary<string, List<string>> Headers { get; set; }



        /// <summary>

        /// We will collect the fields and values to the FieldCollector through the AddField and AddBinaryData functions.

        /// </summary>

        private HTTPFormBase FieldCollector;



        /// <summary>

        /// When the request about to send the request we will create a specialized form implementation(url-encoded, multipart, or the legacy WWWForm based).

        /// And we will use this instance to create the data that we will send to the server.

        /// </summary>

        private HTTPFormBase FormImpl;



        #endregion



        #region Constructors



        #region Default Get Constructors



        public HTTPRequest(Uri uri)

            : this(uri, HTTPMethods.Get, HTTPManager.KeepAliveDefaultValue,

#if !BESTHTTP_DISABLE_CACHING

            HTTPManager.IsCachingDisabled

#else

            true

#endif

            , null)

        {

        }



        public HTTPRequest(Uri uri, OnRequestFinishedDelegate callback)

            : this(uri, HTTPMethods.Get, HTTPManager.KeepAliveDefaultValue,

#if !BESTHTTP_DISABLE_CACHING

            HTTPManager.IsCachingDisabled

#else

            true

#endif

            , callback)

        {

        }



        public HTTPRequest(Uri uri, bool isKeepAlive, OnRequestFinishedDelegate callback)

            : this(uri, HTTPMethods.Get, isKeepAlive,

#if !BESTHTTP_DISABLE_CACHING

            HTTPManager.IsCachingDisabled

#else

            true

#endif



            , callback)

        {

        }

        public HTTPRequest(Uri uri, bool isKeepAlive, bool disableCache, OnRequestFinishedDelegate callback)

            : this(uri, HTTPMethods.Get, isKeepAlive, disableCache, callback)

        {

        }



        #endregion



        public HTTPRequest(Uri uri, HTTPMethods methodType)

            : this(uri, methodType, HTTPManager.KeepAliveDefaultValue,

#if !BESTHTTP_DISABLE_CACHING

            HTTPManager.IsCachingDisabled || methodType != HTTPMethods.Get

#else

            true

#endif

            , null)

        {

        }



        public HTTPRequest(Uri uri, HTTPMethods methodType, OnRequestFinishedDelegate callback)

            : this(uri, methodType, HTTPManager.KeepAliveDefaultValue,

#if !BESTHTTP_DISABLE_CACHING

            HTTPManager.IsCachingDisabled || methodType != HTTPMethods.Get

#else

            true

#endif

            , callback)

        {

        }



        public HTTPRequest(Uri uri, HTTPMethods methodType, bool isKeepAlive, OnRequestFinishedDelegate callback)

            : this(uri, methodType, isKeepAlive,

#if !BESTHTTP_DISABLE_CACHING

            HTTPManager.IsCachingDisabled || methodType != HTTPMethods.Get

#else

            true

#endif

            , callback)

        {

        }



        public HTTPRequest(Uri uri, HTTPMethods methodType, bool isKeepAlive, bool disableCache, OnRequestFinishedDelegate callback)

        {

            this.Uri = uri;

            this.MethodType = methodType;

            this.IsKeepAlive = isKeepAlive;

#if !BESTHTTP_DISABLE_CACHING

            this.DisableCache = disableCache;

#endif

            this.Callback = callback;

            this.StreamFragmentSize = 1024 * 1024;

            this.MaxFragmentQueueLength = 10;



            this.MaxRetries = methodType == HTTPMethods.Get ? 1 : 0;

            this.MaxRedirects = 10;

            this.RedirectCount = 0;

#if !BESTHTTP_DISABLE_COOKIES

            this.IsCookiesEnabled = HTTPManager.IsCookiesEnabled;

#endif

            

            this.State = HTTPRequestStates.Initial;



            this.ConnectTimeout = HTTPManager.ConnectTimeout;

            this.Timeout = HTTPManager.RequestTimeout;

            this.EnableTimoutForStreaming = false;



            this.EnableSafeReadOnUnknownContentLength = true;



#if !BESTHTTP_DISABLE_PROXY && (!UNITY_WEBGL || UNITY_EDITOR)

            this.Proxy = HTTPManager.Proxy;

#endif



            this.UseUploadStreamLength = true;

            this.DisposeUploadStream = true;



#if UNITY_WEBGL && !BESTHTTP_DISABLE_COOKIES

            this.WithCredentials = this.IsCookiesEnabled;

#endif



            this.Context = new LoggingContext(this);

            this.Timing = new TimingCollector(this);

        }



        #endregion



        #region Public Field Functions



        /// <summary>

        /// Add a field with a given string value.

        /// </summary>

        public void AddField(string fieldName, string value)

        {

            AddField(fieldName, value, System.Text.Encoding.UTF8);

        }



        /// <summary>

        /// Add a field with a given string value.

        /// </summary>

        public void AddField(string fieldName, string value, System.Text.Encoding e)

        {

            if (FieldCollector == null)

                FieldCollector = new HTTPFormBase();



            FieldCollector.AddField(fieldName, value, e);

        }



        /// <summary>

        /// Add a field with binary content to the form.

        /// </summary>

        public void AddBinaryData(string fieldName, byte[] content)

        {

            AddBinaryData(fieldName, content, null, null);

        }



        /// <summary>

        /// Add a field with binary content to the form.

        /// </summary>

        public void AddBinaryData(string fieldName, byte[] content, string fileName)

        {

            AddBinaryData(fieldName, content, fileName, null);

        }



        /// <summary>

        /// Add a field with binary content to the form.

        /// </summary>

        public void AddBinaryData(string fieldName, byte[] content, string fileName, string mimeType)

        {

            if (FieldCollector == null)

                FieldCollector = new HTTPFormBase();



            FieldCollector.AddBinaryData(fieldName, content, fileName, mimeType);

        }



        /// <summary>

        /// Manually set a HTTP Form.

        /// </summary>

        public void SetForm(HTTPFormBase form)

        {

            FormImpl = form;

        }



        /// <summary>

        /// Returns with the added form-fields or null if no one added.

        /// </summary>

        public List<HTTPFieldData> GetFormFields()

        {

            if (this.FieldCollector == null || this.FieldCollector.IsEmpty)

                return null;



            return new List<HTTPFieldData>(this.FieldCollector.Fields);

        }



        /// <summary>

        /// Clears all data from the form.

        /// </summary>

        public void ClearForm()

        {

            FormImpl = null;

            FieldCollector = null;

        }



        /// <summary>

        /// Will create the form implementation based on the value of the FormUsage property.

        /// </summary>

        private HTTPFormBase SelectFormImplementation()

        {

            // Our form already created with a previous

            if (FormImpl != null)

                return FormImpl;



            // No field added to this request yet

            if (FieldCollector == null)

                return null;



            switch (FormUsage)

            {

                case HTTPFormUsage.Automatic:

                    // A really simple decision making: if there are at least one field with binary data, or a 'long' string value then we will choose a Multipart form.

                    //  Otherwise Url Encoded form will be used.

                    if (FieldCollector.HasBinary || FieldCollector.HasLongValue)

                        goto case HTTPFormUsage.Multipart;

                    else

                        goto case HTTPFormUsage.UrlEncoded;



                case HTTPFormUsage.UrlEncoded:  FormImpl = new HTTPUrlEncodedForm(); break;

                case HTTPFormUsage.Multipart:   FormImpl = new HTTPMultiPartForm(); break;

            }



            // Copy the fields, and other properties to the new implementation

            FormImpl.CopyFrom(FieldCollector);



            return FormImpl;

        }



        #endregion



        #region Header Management



        #region General Management



        /// <summary>

        /// Adds a header and value pair to the Headers. Use it to add custom headers to the request.

        /// </summary>

        /// <example>AddHeader("User-Agent', "FooBar 1.0")</example>

        public void AddHeader(string name, string value)

        {

            if (Headers == null)

                Headers = new Dictionary<string, List<string>>();



            List<string> values;

            if (!Headers.TryGetValue(name, out values))

                Headers.Add(name, values = new List<string>(1));



            values.Add(value);

        }



        /// <summary>

        /// Removes any previously added values, and sets the given one.

        /// </summary>

        public void SetHeader(string name, string value)

        {

            if (Headers == null)

                Headers = new Dictionary<string, List<string>>();



            List<string> values;

            if (!Headers.TryGetValue(name, out values))

                Headers.Add(name, values = new List<string>(1));



            values.Clear();

            values.Add(value);

        }



        /// <summary>

        /// Removes the specified header. Returns true, if the header found and succesfully removed.

        /// </summary>

        /// <param name="name"></param>

        /// <returns></returns>

        public bool RemoveHeader(string name)

        {

            if (Headers == null)

                return false;



            return Headers.Remove(name);

        }



        /// <summary>

        /// Returns true if the given head name is already in the Headers.

        /// </summary>

        public bool HasHeader(string name)

        {

            return Headers != null && Headers.ContainsKey(name);

        }



        /// <summary>

        /// Returns the first header or null for the given header name.

        /// </summary>

        public string GetFirstHeaderValue(string name)

        {

            if (Headers == null)

                return null;



            List<string> headers = null;

            if (Headers.TryGetValue(name, out headers) && headers.Count > 0)

                return headers[0];



            return null;

        }



        /// <summary>

        /// Returns all header values for the given header or null.

        /// </summary>

        public List<string> GetHeaderValues(string name)

        {

            if (Headers == null)

                return null;



            List<string> headers = null;

            if (Headers.TryGetValue(name, out headers) && headers.Count > 0)

                return headers;



            return null;

        }



        /// <summary>

        /// Removes all headers.

        /// </summary>

        public void RemoveHeaders()

        {

            if (Headers == null)

                return;



            Headers.Clear();

        }



        #endregion



        #region Range Headers



        /// <summary>

        /// Sets the Range header to download the content from the given byte position. See http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.35

        /// </summary>

        /// <param name="firstBytePos">Start position of the download.</param>

        public void SetRangeHeader(long firstBytePos)

        {

            SetHeader("Range", string.Format("bytes={0}-", firstBytePos));

        }



        /// <summary>

        /// Sets the Range header to download the content from the given byte position to the given last position. See http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.35

        /// </summary>

        /// <param name="firstBytePos">Start position of the download.</param>

        /// <param name="lastBytePos">The end position of the download.</param>

        public void SetRangeHeader(long firstBytePos, long lastBytePos)

        {

            SetHeader("Range", string.Format("bytes={0}-{1}", firstBytePos, lastBytePos));

        }



        #endregion



        public void EnumerateHeaders(OnHeaderEnumerationDelegate callback)

        {

            EnumerateHeaders(callback, false);

        }



        public void EnumerateHeaders(OnHeaderEnumerationDelegate callback, bool callBeforeSendCallback)

        {

#if !UNITY_WEBGL || UNITY_EDITOR

            if (!HasHeader("Host"))

            {

                if (CurrentUri.Port == 80 || CurrentUri.Port == 443)

                    SetHeader("Host", CurrentUri.Host);

                else

                    SetHeader("Host", CurrentUri.Authority);

            }



            if (IsRedirected && !HasHeader("Referer"))

                AddHeader("Referer", Uri.ToString());



            Decompression.DecompressorFactory.SetupHeaders(this);



#if !BESTHTTP_DISABLE_PROXY && (!UNITY_WEBGL || UNITY_EDITOR)

            if (!HTTPProtocolFactory.IsSecureProtocol(this.CurrentUri) && HasProxy && !HasHeader("Proxy-Connection"))

                AddHeader("Proxy-Connection", IsKeepAlive ? "Keep-Alive" : "Close");

            #endif



            if (!HasHeader("Connection"))

                AddHeader("Connection", IsKeepAlive ? "Keep-Alive, TE" : "Close, TE");



            if (IsKeepAlive && !HasHeader("Keep-Alive"))

            {

                // Send the server a slightly larger value to make sure it's not going to close sooner than the client

                int seconds = (int)Math.Ceiling(HTTPManager.MaxConnectionIdleTime.TotalSeconds + 1);



                AddHeader("Keep-Alive", "timeout=" + seconds);

            }



            if (!HasHeader("TE"))

                AddHeader("TE", "identity");



            if (!string.IsNullOrEmpty(HTTPManager.UserAgent) && !HasHeader("User-Agent"))

                AddHeader("User-Agent", HTTPManager.UserAgent);

#endif

            long contentLength = -1;



            if (UploadStream == null)

            {

                byte[] entityBody = GetEntityBody();

                contentLength = entityBody != null ? entityBody.Length : 0;



                if (RawData == null && (FormImpl != null || (FieldCollector != null && !FieldCollector.IsEmpty)))

                {

                    SelectFormImplementation();

                    if (FormImpl != null)

                        FormImpl.PrepareRequest(this);

                }

            }

            else

            {

                contentLength = UploadStreamLength;



                if (contentLength == -1)

                    SetHeader("Transfer-Encoding", "Chunked");



                if (!HasHeader("Content-Type"))

                    SetHeader("Content-Type", "application/octet-stream");

            }



            // Always set the Content-Length header if possible

            // http://tools.ietf.org/html/rfc2616#section-4.4 : For compatibility with HTTP/1.0 applications, HTTP/1.1 requests containing a message-body MUST include a valid Content-Length header field unless the server is known to be HTTP/1.1 compliant.

            // 2018.06.03: Changed the condition so that content-length header will be included for zero length too.

            // 2022.05.25: Don't send a Content-Length (: 0) header if there's an Upgrade header. Upgrade is set for websocket, and it might be not true that the client doesn't send any bytes.

            if (

#if !UNITY_WEBGL || UNITY_EDITOR

                contentLength >= 0

#else

                contentLength != -1

#endif

                && !HasHeader("Content-Length")

                && !HasHeader("Upgrade"))

                SetHeader("Content-Length", contentLength.ToString());



#if !UNITY_WEBGL || UNITY_EDITOR

            #if !BESTHTTP_DISABLE_PROXY && (!UNITY_WEBGL || UNITY_EDITOR)

            // Proxy Authentication

            if (!HTTPProtocolFactory.IsSecureProtocol(this.CurrentUri) && HasProxy && Proxy.Credentials != null)

            {

                switch (Proxy.Credentials.Type)

                {

                    case AuthenticationTypes.Basic:

                        // With Basic authentication we don't want to wait for a challenge, we will send the hash with the first request

                        SetHeader("Proxy-Authorization", string.Concat("Basic ", Convert.ToBase64String(Encoding.UTF8.GetBytes(Proxy.Credentials.UserName + ":" + Proxy.Credentials.Password))));

                        break;



                    case AuthenticationTypes.Unknown:

                    case AuthenticationTypes.Digest:

                        var digest = DigestStore.Get(Proxy.Address);

                        if (digest != null)

                        {

                            string authentication = digest.GenerateResponseHeader(this, Proxy.Credentials);

                            if (!string.IsNullOrEmpty(authentication))

                                SetHeader("Proxy-Authorization", authentication);

                        }



                        break;

                }

            }

            #endif



#endif



            // Server authentication

            if (Credentials != null)

            {

                switch (Credentials.Type)

                {

                    case AuthenticationTypes.Basic:

                        // With Basic authentication we don't want to wait for a challenge, we will send the hash with the first request

                        SetHeader("Authorization", string.Concat("Basic ", Convert.ToBase64String(Encoding.UTF8.GetBytes(Credentials.UserName + ":" + Credentials.Password))));

                        break;



                    case AuthenticationTypes.Unknown:

                    case AuthenticationTypes.Digest:

                        var digest = DigestStore.Get(this.CurrentUri);

                        if (digest != null)

                        {

                            string authentication = digest.GenerateResponseHeader(this, Credentials);

                            if (!string.IsNullOrEmpty(authentication))

                                SetHeader("Authorization", authentication);

                        }



                        break;

                }

            }



            // Cookies.

#if !BESTHTTP_DISABLE_COOKIES

            // User added cookies are sent even when IsCookiesEnabled is set to false

            List<Cookie> cookies = IsCookiesEnabled ? CookieJar.Get(CurrentUri) : null;



            // Merge server sent cookies with user-set cookies

            if (cookies == null || cookies.Count == 0)

                cookies = this.customCookies;

            else if (this.customCookies != null)

            {

                // Merge

                int idx = 0;

                while (idx < this.customCookies.Count)

                {

                    Cookie customCookie = customCookies[idx];



                    int foundIdx = cookies.FindIndex(c => c.Name.Equals(customCookie.Name));

                    if (foundIdx >= 0)

                        cookies[foundIdx] = customCookie;

                    else

                        cookies.Add(customCookie);



                    idx++;

                }

            }



            // http://tools.ietf.org/html/rfc6265#section-5.4

            //  -When the user agent generates an HTTP request, the user agent MUST NOT attach more than one Cookie header field.

            if (cookies != null && cookies.Count > 0)

            {

                // Room for improvement:

                //   2. The user agent SHOULD sort the cookie-list in the following order:

                //      *  Cookies with longer paths are listed before cookies with shorter paths.

                //      *  Among cookies that have equal-length path fields, cookies with earlier creation-times are listed before cookies with later creation-times.



                bool first = true;

                string cookieStr = string.Empty;



                bool isSecureProtocolInUse = HTTPProtocolFactory.IsSecureProtocol(CurrentUri);



                foreach (var cookie in cookies)

                    if (!cookie.IsSecure || (cookie.IsSecure && isSecureProtocolInUse))

                    {

                        if (!first)

                            cookieStr += "; ";

                        else

                            first = false;



                        cookieStr += cookie.ToString();



                        // 3. Update the last-access-time of each cookie in the cookie-list to the current date and time.

                        cookie.LastAccess = DateTime.UtcNow;

                    }



                if (!string.IsNullOrEmpty(cookieStr))

                    SetHeader("Cookie", cookieStr);

            }

#endif



            if (callBeforeSendCallback && _onBeforeHeaderSend != null)

            {

                try

                {

                    _onBeforeHeaderSend(this);

                }

                catch(Exception ex)

                {

                    HTTPManager.Logger.Exception("HTTPRequest", "OnBeforeHeaderSend", ex, this.Context);

                }

            }



            // Write out the headers to the stream

            if (callback != null && Headers != null)

                foreach (var kvp in Headers)

                    callback(kvp.Key, kvp.Value);

        }



        /// <summary>

        /// Writes out the Headers to the stream.

        /// </summary>

        private void SendHeaders(Stream stream)

        {

            EnumerateHeaders((header, values) =>

                {

                    if (string.IsNullOrEmpty(header) || values == null)

                        return;



                    var headerName = string.Concat(header, ": ").GetASCIIBytes();



                    for (int i = 0; i < values.Count; ++i)

                    {

                        if (string.IsNullOrEmpty(values[i]))

                        {

                            HTTPManager.Logger.Warning("HTTPRequest", string.Format("Null/empty value for header: {0}", header), this.Context);

                            continue;

                        }



                        if (HTTPManager.Logger.Level <= Logger.Loglevels.Information)

                            VerboseLogging("Header - '" + header + "': '" + values[i] + "'");



                        var valueBytes = values[i].GetASCIIBytes();



                        stream.WriteBufferSegment(headerName);

                        stream.WriteBufferSegment(valueBytes);

                        stream.WriteArray(EOL);



                        BufferPool.Release(valueBytes);

                    }



                    BufferPool.Release(headerName);

                }, /*callBeforeSendCallback:*/ true);

        }



        /// <summary>

        /// Returns a string representation of the headers.

        /// </summary>

        public string DumpHeaders()

        {

            using (var ms = new BufferPoolMemoryStream(5 * 1024))

            {

                SendHeaders(ms);

                return ms.ToArray().AsciiToString();

            }

        }



        /// <summary>

        /// Returns with the bytes that will be sent to the server as the request's payload.

        /// </summary>

        /// <remarks>Call this only after all form-fields are added!</remarks>

        public byte[] GetEntityBody()

        {

            if (RawData != null)

                return RawData;



            if (FormImpl != null || (FieldCollector != null && !FieldCollector.IsEmpty))

            {

                SelectFormImplementation();

                if (FormImpl != null)

                    return FormImpl.GetData();

            }



            return null;

        }



        public struct UploadStreamInfo

        {

            public readonly Stream Stream;

            public readonly long Length;



            public UploadStreamInfo(Stream stream, long length)

            {

                this.Stream = stream;

                this.Length = length;

            }

        }



        internal UploadStreamInfo GetUpStream()

        {

            byte[] data = RawData;



            // We are sending forms? Then convert the form to a byte array

            if (data == null && FormImpl != null)

                data = FormImpl.GetData();



            if (data != null || UploadStream != null)

            {

                // Make a new reference, as we will check the UploadStream property in the HTTPManager

                Stream uploadStream = UploadStream;



                long UploadLength = 0;



                if (uploadStream == null)

                {

                    // Make stream from the data. A BufferPoolMemoryStream could be used here,

                    // but because data comes from outside, we don't have control on its lifetime

                    // and might be gets reused without our knowledge.

                    uploadStream = new MemoryStream(data, 0, data.Length);



                    // Initialize progress report variable

                    UploadLength = data.Length;

                }

                else

                    UploadLength = UseUploadStreamLength ? UploadStreamLength : -1;



                return new UploadStreamInfo(uploadStream, UploadLength);

            }



            return new UploadStreamInfo(null, 0);

        }



        #endregion



        #region Internal Helper Functions



        internal void SendOutTo(Stream stream)

        {

            // Under WEBGL EnumerateHeaders and GetEntityBody are used instead of this function.

#if !UNITY_WEBGL || UNITY_EDITOR

            string requestPathAndQuery =

#if !BESTHTTP_DISABLE_PROXY && (!UNITY_WEBGL || UNITY_EDITOR)

                    HasProxy ? this.Proxy.GetRequestPath(CurrentUri) :

#endif

                    CurrentUri.GetRequestPathAndQueryURL();



            string requestLine = string.Format("{0} {1} HTTP/1.1", MethodNames[(byte)MethodType], requestPathAndQuery);



            if (HTTPManager.Logger.Level <= Logger.Loglevels.Information)

                HTTPManager.Logger.Information("HTTPRequest", string.Format("Sending request: '{0}'", requestLine), this.Context);



            // Create a buffer stream that will not close 'stream' when disposed or closed.

            // buffersize should be larger than UploadChunkSize as it might be used for uploading user data and

            //  it should have enough room for UploadChunkSize data and additional chunk information.

            using (WriteOnlyBufferedStream bufferStream = new WriteOnlyBufferedStream(stream, (int)(UploadChunkSize * 1.5f)))

            {

                var requestLineBytes = requestLine.GetASCIIBytes();

                bufferStream.WriteBufferSegment(requestLineBytes);

                bufferStream.WriteArray(EOL);



                BufferPool.Release(requestLineBytes);



                // Write headers to the buffer

                SendHeaders(bufferStream);

                bufferStream.WriteArray(EOL);



                // Send remaining data to the wire

                bufferStream.Flush();



                byte[] data = RawData;



                // We are sending forms? Then convert the form to a byte array

                if (data == null && FormImpl != null)

                    data = FormImpl.GetData();



                if (data != null || UploadStream != null)

                {

                    // Make a new reference, as we will check the UploadStream property in the HTTPManager

                    Stream uploadStream = UploadStream;



                    long UploadLength = 0;



                    if (uploadStream == null)

                    {

                        // Make stream from the data. A BufferPoolMemoryStream could be used here,

                        // but because data comes from outside, we don't have control on it's lifetime

                        // and might be gets reused without our knowledge.

                        uploadStream = new MemoryStream(data, 0, data.Length);



                        // Initialize progress report variable

                        UploadLength = data.Length;

                    }

                    else

                        UploadLength = UseUploadStreamLength ? UploadStreamLength : -1;



                    // Initialize the progress report variables

                    long Uploaded = 0;



                    // Upload buffer. First we will read the data into this buffer from the UploadStream, then write this buffer to our outStream

                    byte[] buffer = BufferPool.Get(UploadChunkSize, true);



                    // How many bytes was read from the UploadStream

                    int count = 0;

                    while ((count = uploadStream.Read(buffer, 0, buffer.Length)) > 0)

                    {

                        // If we don't know the size, send as chunked

                        if (!UseUploadStreamLength)

                        {

                            var countBytes = count.ToString("X").GetASCIIBytes();

                            bufferStream.WriteBufferSegment(countBytes);

                            bufferStream.WriteArray(EOL);



                            BufferPool.Release(countBytes);

                        }



                        // write out the buffer to the wire

                        bufferStream.Write(buffer, 0, count);



                        // chunk trailing EOL

                        if (!UseUploadStreamLength)

                            bufferStream.WriteArray(EOL);



                        // update how many bytes are uploaded

                        Uploaded += count;



                        // Write to the wire

                        bufferStream.Flush();



                        if (this.OnUploadProgress != null)

                            RequestEventHelper.EnqueueRequestEvent(new RequestEventInfo(this, RequestEvents.UploadProgress, Uploaded, UploadLength));



                        if (this.IsCancellationRequested)

                            return;

                    }



                    BufferPool.Release(buffer);



                    // All data from the stream are sent, write the 'end' chunk if necessary

                    if (!UseUploadStreamLength)

                    {

                        byte[] noMoreChunkBytes = BufferPool.Get(1, true);

                        noMoreChunkBytes[0] = (byte)'0';

                        bufferStream.Write(noMoreChunkBytes, 0, 1);

                        bufferStream.WriteArray(EOL);

                        bufferStream.WriteArray(EOL);



                        BufferPool.Release(noMoreChunkBytes);

                    }



                    // Make sure all remaining data will be on the wire

                    bufferStream.Flush();



                    // Dispose the MemoryStream

                    if (UploadStream == null && uploadStream != null)

                        uploadStream.Dispose();

                }

                else

                    bufferStream.Flush();

            } // bufferStream.Dispose



            HTTPManager.Logger.Information("HTTPRequest", "'" + requestLine + "' sent out", this.Context);

#endif

        }



#if !UNITY_WEBGL || UNITY_EDITOR

        internal void UpgradeCallback()

        {

            if (Response == null || !Response.IsUpgraded)

                return;



            try

            {

                if (OnUpgraded != null)

                    OnUpgraded(this, Response);

            }

            catch (Exception ex)

            {

                HTTPManager.Logger.Exception("HTTPRequest", "UpgradeCallback", ex, this.Context);

            }

        }

#endif



        internal bool CallOnBeforeRedirection(Uri redirectUri)

        {

            if (onBeforeRedirection != null)

                return onBeforeRedirection(this, this.Response, redirectUri);



            return true;

        }



        /// <summary>

        /// Called on Unity's main thread just before processing it.

        /// </summary>

        internal void Prepare()

        {



        }



#endregion



        /// <summary>

        /// Starts processing the request.

        /// </summary>

        public HTTPRequest Send()

        {

            this.IsCancellationRequested = false;

            this.Exception = null;



            return HTTPManager.SendRequest(this);

        }



        /// <summary>

        /// Aborts an already established connection, so no further download or upload are done.

        /// </summary>

        public void Abort()

        {

            VerboseLogging("Abort request!");



            lock (this)

            {

                if (this.State >= HTTPRequestStates.Finished)

                    return;



                this.IsCancellationRequested = true;



                // If the response is an IProtocol implementation, call the protocol's cancellation.

                IProtocol protocol = this.Response as IProtocol;

                if (protocol != null)

                    protocol.CancellationRequested();



                // There's a race-condition here, another thread might set it too.

                this.Response = null;



                // There's a race-condition here too, another thread might set it too.

                //  In this case, both state going to be queued up that we have to handle in RequestEvents.cs.

                if (this.IsTimedOut)

                {

                    this.State = this.IsConnectTimedOut ? HTTPRequestStates.ConnectionTimedOut : HTTPRequestStates.TimedOut;

                }

                else

                    this.State = HTTPRequestStates.Aborted;



#if !UNITY_WEBGL || UNITY_EDITOR

                if (this.OnCancellationRequested != null)

                {

                    try

                    {

                        this.OnCancellationRequested(this);

                    }

                    catch { }

                }

#endif

            }

        }



        /// <summary>

        /// Resets the request for a state where switching MethodType is possible.

        /// </summary>

        public void Clear()

        {

            ClearForm();

            RemoveHeaders();



            this.IsRedirected = false;

            this.RedirectCount = 0;

            this.Exception = null;

        }



        private void VerboseLogging(string str)

        {

            HTTPManager.Logger.Verbose("HTTPRequest", str, this.Context);

        }



        #region System.Collections.IEnumerator implementation



        public object Current { get { return null; } }



        public bool MoveNext()

        {

            return this.State < HTTPRequestStates.Finished;

        }



        public void Reset()

        {

            throw new NotImplementedException();

        }



#endregion



        HTTPRequest IEnumerator<HTTPRequest>.Current

        {

            get { return this; }

        }



        public void Dispose()

        {

            if (UploadStream != null && DisposeUploadStream)

            {

                UploadStream.Dispose();

                UploadStream = null;

            }



            if (Response != null)

                Response.Dispose();

        }

    }

}