using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Emby.Server.Implementations.IO;
using MediaBrowser.Common.Configuration;
using MediaBrowser.Common.Extensions;
using MediaBrowser.Common.Net;
using MediaBrowser.Model.IO;
using Microsoft.Extensions.Logging;
using MediaBrowser.Model.Net;
using MediaBrowser.Controller.IO;
namespace Emby.Server.Implementations.HttpClientManager
{
    /// 
    /// Class HttpClientManager
    /// 
    public class HttpClientManager : IHttpClient
    {
        /// 
        /// When one request to a host times out, we'll ban all other requests for this period of time, to prevent scans from stalling
        /// 
        private const int TimeoutSeconds = 30;
        /// 
        /// The _logger
        /// 
        private readonly ILogger _logger;
        /// 
        /// The _app paths
        /// 
        private readonly IApplicationPaths _appPaths;
        private readonly IFileSystem _fileSystem;
        private readonly Func _defaultUserAgentFn;
        /// 
        /// Initializes a new instance of the  class.
        /// 
        public HttpClientManager(IApplicationPaths appPaths, ILogger logger, IFileSystem fileSystem, Func defaultUserAgentFn)
        {
            if (appPaths == null)
            {
                throw new ArgumentNullException("appPaths");
            }
            if (logger == null)
            {
                throw new ArgumentNullException("logger");
            }
            _logger = logger;
            _fileSystem = fileSystem;
            _appPaths = appPaths;
            _defaultUserAgentFn = defaultUserAgentFn;
            // http://stackoverflow.com/questions/566437/http-post-returns-the-error-417-expectation-failed-c
            ServicePointManager.Expect100Continue = false;
#if NET46
// Trakt requests sometimes fail without this
            ServicePointManager.SecurityProtocol = SecurityProtocolType.Ssl3 | SecurityProtocolType.Tls;
#endif
        }
        /// 
        /// Holds a dictionary of http clients by host.  Use GetHttpClient(host) to retrieve or create a client for web requests.
        /// DON'T dispose it after use.
        /// 
        /// The HTTP clients.
        private readonly ConcurrentDictionary _httpClients = new ConcurrentDictionary();
        /// 
        /// Gets
        /// 
        /// The host.
        /// if set to true [enable HTTP compression].
        /// HttpClient.
        /// host
        private HttpClientInfo GetHttpClient(string host, bool enableHttpCompression)
        {
            if (string.IsNullOrEmpty(host))
            {
                throw new ArgumentNullException("host");
            }
            HttpClientInfo client;
            var key = host + enableHttpCompression;
            if (!_httpClients.TryGetValue(key, out client))
            {
                client = new HttpClientInfo();
                _httpClients.TryAdd(key, client);
            }
            return client;
        }
        private WebRequest CreateWebRequest(string url)
        {
            try
            {
                return WebRequest.Create(url);
            }
            catch (NotSupportedException)
            {
                //Webrequest creation does fail on MONO randomly when using WebRequest.Create
                //the issue occurs in the GetCreator method here: http://www.oschina.net/code/explore/mono-2.8.1/mcs/class/System/System.Net/WebRequest.cs
                var type = Type.GetType("System.Net.HttpRequestCreator, System, Version=4.0.0.0,Culture=neutral, PublicKeyToken=b77a5c561934e089");
                var creator = Activator.CreateInstance(type, nonPublic: true) as IWebRequestCreate;
                return creator.Create(new Uri(url)) as HttpWebRequest;
            }
        }
        private WebRequest GetRequest(HttpRequestOptions options, string method)
        {
            var url = options.Url;
            var uriAddress = new Uri(url);
            var userInfo = uriAddress.UserInfo;
            if (!string.IsNullOrWhiteSpace(userInfo))
            {
                _logger.LogInformation("Found userInfo in url: {0} ... url: {1}", userInfo, url);
                url = url.Replace(userInfo + "@", string.Empty);
            }
            var request = CreateWebRequest(url);
            var httpWebRequest = request as HttpWebRequest;
            if (httpWebRequest != null)
            {
                AddRequestHeaders(httpWebRequest, options);
                if (options.EnableHttpCompression)
                {
                    if (options.DecompressionMethod.HasValue)
                    {
                        httpWebRequest.AutomaticDecompression = options.DecompressionMethod.Value == CompressionMethod.Gzip
                            ? DecompressionMethods.GZip
                            : DecompressionMethods.Deflate;
                    }
                    else
                    {
                        httpWebRequest.AutomaticDecompression = DecompressionMethods.Deflate;
                    }
                }
                else
                {
                    httpWebRequest.AutomaticDecompression = DecompressionMethods.None;
                }
            }
            request.CachePolicy = new System.Net.Cache.RequestCachePolicy(System.Net.Cache.RequestCacheLevel.BypassCache);
            if (httpWebRequest != null)
            {
                if (options.EnableKeepAlive)
                {
                    httpWebRequest.KeepAlive = true;
                }
            }
            request.Method = method;
            request.Timeout = options.TimeoutMs;
            if (httpWebRequest != null)
            {
                if (!string.IsNullOrEmpty(options.Host))
                {
                    httpWebRequest.Host = options.Host;
                }
                if (!string.IsNullOrEmpty(options.Referer))
                {
                    httpWebRequest.Referer = options.Referer;
                }
            }
            if (!string.IsNullOrWhiteSpace(userInfo))
            {
                var parts = userInfo.Split(':');
                if (parts.Length == 2)
                {
                    request.Credentials = GetCredential(url, parts[0], parts[1]);
                    // TODO: .net core ??
                    request.PreAuthenticate = true;
                }
            }
            return request;
        }
        private CredentialCache GetCredential(string url, string username, string password)
        {
            //ServicePointManager.SecurityProtocol = SecurityProtocolType.Ssl3;
            CredentialCache credentialCache = new CredentialCache();
            credentialCache.Add(new Uri(url), "Basic", new NetworkCredential(username, password));
            return credentialCache;
        }
        private void AddRequestHeaders(HttpWebRequest request, HttpRequestOptions options)
        {
            var hasUserAgent = false;
            foreach (var header in options.RequestHeaders.ToList())
            {
                if (string.Equals(header.Key, "Accept", StringComparison.OrdinalIgnoreCase))
                {
                    request.Accept = header.Value;
                }
                else if (string.Equals(header.Key, "User-Agent", StringComparison.OrdinalIgnoreCase))
                {
                    SetUserAgent(request, header.Value);
                    hasUserAgent = true;
                }
                else
                {
                    request.Headers.Set(header.Key, header.Value);
                }
            }
            if (!hasUserAgent && options.EnableDefaultUserAgent)
            {
                SetUserAgent(request, _defaultUserAgentFn());
            }
        }
        private void SetUserAgent(HttpWebRequest request, string userAgent)
        {
            request.UserAgent = userAgent;
        }
        /// 
        /// Gets the response internal.
        /// 
        /// The options.
        /// Task{HttpResponseInfo}.
        public Task GetResponse(HttpRequestOptions options)
        {
            return SendAsync(options, "GET");
        }
        /// 
        /// Performs a GET request and returns the resulting stream
        /// 
        /// The options.
        /// Task{Stream}.
        public async Task Get(HttpRequestOptions options)
        {
            var response = await GetResponse(options).ConfigureAwait(false);
            return response.Content;
        }
        /// 
        /// send as an asynchronous operation.
        /// 
        /// The options.
        /// The HTTP method.
        /// Task{HttpResponseInfo}.
        /// 
        /// 
        public async Task SendAsync(HttpRequestOptions options, string httpMethod)
        {
            if (options.CacheMode == CacheMode.None)
            {
                return await SendAsyncInternal(options, httpMethod).ConfigureAwait(false);
            }
            var url = options.Url;
            var urlHash = url.ToLower().GetMD5().ToString("N");
            var responseCachePath = Path.Combine(_appPaths.CachePath, "httpclient", urlHash);
            var response = await GetCachedResponse(responseCachePath, options.CacheLength, url).ConfigureAwait(false);
            if (response != null)
            {
                return response;
            }
            response = await SendAsyncInternal(options, httpMethod).ConfigureAwait(false);
            if (response.StatusCode == HttpStatusCode.OK)
            {
                await CacheResponse(response, responseCachePath).ConfigureAwait(false);
            }
            return response;
        }
        private async Task GetCachedResponse(string responseCachePath, TimeSpan cacheLength, string url)
        {
            try
            {
                if (_fileSystem.GetLastWriteTimeUtc(responseCachePath).Add(cacheLength) > DateTime.UtcNow)
                {
                    using (var stream = _fileSystem.GetFileStream(responseCachePath, FileOpenMode.Open, FileAccessMode.Read, FileShareMode.Read, true))
                    {
                        var memoryStream = new MemoryStream();
                        await stream.CopyToAsync(memoryStream).ConfigureAwait(false);
                        memoryStream.Position = 0;
                        return new HttpResponseInfo
                        {
                            ResponseUrl = url,
                            Content = memoryStream,
                            StatusCode = HttpStatusCode.OK,
                            ContentLength = memoryStream.Length
                        };
                    }
                }
            }
            catch (FileNotFoundException)
            {
            }
            catch (DirectoryNotFoundException)
            {
            }
            return null;
        }
        private async Task CacheResponse(HttpResponseInfo response, string responseCachePath)
        {
            _fileSystem.CreateDirectory(_fileSystem.GetDirectoryName(responseCachePath));
            using (var responseStream = response.Content)
            {
                var memoryStream = new MemoryStream();
                await responseStream.CopyToAsync(memoryStream).ConfigureAwait(false);
                memoryStream.Position = 0;
                using (var fileStream = _fileSystem.GetFileStream(responseCachePath, FileOpenMode.Create, FileAccessMode.Write, FileShareMode.None, true))
                {
                    await memoryStream.CopyToAsync(fileStream).ConfigureAwait(false);
                    memoryStream.Position = 0;
                    response.Content = memoryStream;
                }
            }
        }
        private async Task SendAsyncInternal(HttpRequestOptions options, string httpMethod)
        {
            ValidateParams(options);
            options.CancellationToken.ThrowIfCancellationRequested();
            var client = GetHttpClient(GetHostFromUrl(options.Url), options.EnableHttpCompression);
            if ((DateTime.UtcNow - client.LastTimeout).TotalSeconds < TimeoutSeconds)
            {
                throw new HttpException(string.Format("Cancelling connection to {0} due to a previous timeout.", options.Url))
                {
                    IsTimedOut = true
                };
            }
            var httpWebRequest = GetRequest(options, httpMethod);
            if (options.RequestContentBytes != null ||
                !string.IsNullOrEmpty(options.RequestContent) ||
                string.Equals(httpMethod, "post", StringComparison.OrdinalIgnoreCase))
            {
                try
                {
                    // TODO: We can always put this in the options object if needed
                    var requestEncoding = Encoding.UTF8;
                    var bytes = options.RequestContentBytes ?? requestEncoding.GetBytes(options.RequestContent ?? string.Empty);
                    var contentType = options.RequestContentType ?? "application/x-www-form-urlencoded";
                    if (options.AppendCharsetToMimeType)
                    {
                        contentType = contentType.TrimEnd(';') + "; charset=\"utf-8\"";
                    }
                    httpWebRequest.ContentType = contentType;
                    httpWebRequest.ContentLength = bytes.Length;
                    (await httpWebRequest.GetRequestStreamAsync().ConfigureAwait(false)).Write(bytes, 0, bytes.Length);
                }
                catch (Exception ex)
                {
                    throw new HttpException(ex.Message) { IsTimedOut = true };
                }
            }
            if (options.ResourcePool != null)
            {
                await options.ResourcePool.WaitAsync(options.CancellationToken).ConfigureAwait(false);
            }
            if ((DateTime.UtcNow - client.LastTimeout).TotalSeconds < TimeoutSeconds)
            {
                if (options.ResourcePool != null)
                {
                    options.ResourcePool.Release();
                }
                throw new HttpException(string.Format("Connection to {0} timed out", options.Url)) { IsTimedOut = true };
            }
            if (options.LogRequest)
            {
                if (options.LogRequestAsDebug)
                {
                    _logger.LogDebug("HttpClientManager {0}: {1}", httpMethod.ToUpper(), options.Url);
                }
                else
                {
                    _logger.LogInformation("HttpClientManager {0}: {1}", httpMethod.ToUpper(), options.Url);
                }
            }
            try
            {
                options.CancellationToken.ThrowIfCancellationRequested();
                if (!options.BufferContent)
                {
                    var response = await GetResponseAsync(httpWebRequest, TimeSpan.FromMilliseconds(options.TimeoutMs)).ConfigureAwait(false);
                    var httpResponse = (HttpWebResponse)response;
                    EnsureSuccessStatusCode(client, httpResponse, options);
                    options.CancellationToken.ThrowIfCancellationRequested();
                    return GetResponseInfo(httpResponse, httpResponse.GetResponseStream(), GetContentLength(httpResponse), httpResponse);
                }
                using (var response = await GetResponseAsync(httpWebRequest, TimeSpan.FromMilliseconds(options.TimeoutMs)).ConfigureAwait(false))
                {
                    var httpResponse = (HttpWebResponse)response;
                    EnsureSuccessStatusCode(client, httpResponse, options);
                    options.CancellationToken.ThrowIfCancellationRequested();
                    using (var stream = httpResponse.GetResponseStream())
                    {
                        var memoryStream = new MemoryStream();
                        await stream.CopyToAsync(memoryStream).ConfigureAwait(false);
                        memoryStream.Position = 0;
                        return GetResponseInfo(httpResponse, memoryStream, memoryStream.Length, null);
                    }
                }
            }
            catch (OperationCanceledException ex)
            {
                throw GetCancellationException(options, client, options.CancellationToken, ex);
            }
            catch (Exception ex)
            {
                throw GetException(ex, options, client);
            }
            finally
            {
                if (options.ResourcePool != null)
                {
                    options.ResourcePool.Release();
                }
            }
        }
        private HttpResponseInfo GetResponseInfo(HttpWebResponse httpResponse, Stream content, long? contentLength, IDisposable disposable)
        {
            var responseInfo = new HttpResponseInfo(disposable)
            {
                Content = content,
                StatusCode = httpResponse.StatusCode,
                ContentType = httpResponse.ContentType,
                ContentLength = contentLength,
                ResponseUrl = httpResponse.ResponseUri.ToString()
            };
            if (httpResponse.Headers != null)
            {
                SetHeaders(httpResponse.Headers, responseInfo);
            }
            return responseInfo;
        }
        private HttpResponseInfo GetResponseInfo(HttpWebResponse httpResponse, string tempFile, long? contentLength)
        {
            var responseInfo = new HttpResponseInfo
            {
                TempFilePath = tempFile,
                StatusCode = httpResponse.StatusCode,
                ContentType = httpResponse.ContentType,
                ContentLength = contentLength
            };
            if (httpResponse.Headers != null)
            {
                SetHeaders(httpResponse.Headers, responseInfo);
            }
            return responseInfo;
        }
        private void SetHeaders(WebHeaderCollection headers, HttpResponseInfo responseInfo)
        {
            foreach (var key in headers.AllKeys)
            {
                responseInfo.Headers[key] = headers[key];
            }
        }
        public Task Post(HttpRequestOptions options)
        {
            return SendAsync(options, "POST");
        }
        /// 
        /// Performs a POST request
        /// 
        /// The options.
        /// Params to add to the POST data.
        /// stream on success, null on failure
        public async Task Post(HttpRequestOptions options, Dictionary postData)
        {
            options.SetPostData(postData);
            var response = await Post(options).ConfigureAwait(false);
            return response.Content;
        }
        /// 
        /// Downloads the contents of a given url into a temporary location
        /// 
        /// The options.
        /// Task{System.String}.
        public async Task GetTempFile(HttpRequestOptions options)
        {
            var response = await GetTempFileResponse(options).ConfigureAwait(false);
            return response.TempFilePath;
        }
        public async Task GetTempFileResponse(HttpRequestOptions options)
        {
            ValidateParams(options);
            _fileSystem.CreateDirectory(_appPaths.TempDirectory);
            var tempFile = Path.Combine(_appPaths.TempDirectory, Guid.NewGuid() + ".tmp");
            if (options.Progress == null)
            {
                throw new ArgumentNullException("progress");
            }
            options.CancellationToken.ThrowIfCancellationRequested();
            var httpWebRequest = GetRequest(options, "GET");
            if (options.ResourcePool != null)
            {
                await options.ResourcePool.WaitAsync(options.CancellationToken).ConfigureAwait(false);
            }
            options.Progress.Report(0);
            if (options.LogRequest)
            {
                if (options.LogRequestAsDebug)
                {
                    _logger.LogDebug("HttpClientManager.GetTempFileResponse url: {0}", options.Url);
                }
                else
                {
                    _logger.LogInformation("HttpClientManager.GetTempFileResponse url: {0}", options.Url);
                }
            }
            var client = GetHttpClient(GetHostFromUrl(options.Url), options.EnableHttpCompression);
            try
            {
                options.CancellationToken.ThrowIfCancellationRequested();
                using (var response = await httpWebRequest.GetResponseAsync().ConfigureAwait(false))
                {
                    var httpResponse = (HttpWebResponse)response;
                    EnsureSuccessStatusCode(client, httpResponse, options);
                    options.CancellationToken.ThrowIfCancellationRequested();
                    var contentLength = GetContentLength(httpResponse);
                    if (!contentLength.HasValue)
                    {
                        // We're not able to track progress
                        using (var stream = httpResponse.GetResponseStream())
                        {
                            using (var fs = _fileSystem.GetFileStream(tempFile, FileOpenMode.Create, FileAccessMode.Write, FileShareMode.Read, true))
                            {
                                await stream.CopyToAsync(fs, StreamDefaults.DefaultCopyToBufferSize, options.CancellationToken).ConfigureAwait(false);
                            }
                        }
                    }
                    else
                    {
                        using (var fs = _fileSystem.GetFileStream(tempFile, FileOpenMode.Create, FileAccessMode.Write, FileShareMode.Read, true))
                        {
                            await httpResponse.GetResponseStream().CopyToAsync(fs, StreamDefaults.DefaultCopyToBufferSize, options.CancellationToken).ConfigureAwait(false);
                        }
                    }
                    options.Progress.Report(100);
                    return GetResponseInfo(httpResponse, tempFile, contentLength);
                }
            }
            catch (Exception ex)
            {
                DeleteTempFile(tempFile);
                throw GetException(ex, options, client);
            }
            finally
            {
                if (options.ResourcePool != null)
                {
                    options.ResourcePool.Release();
                }
            }
        }
        private long? GetContentLength(HttpWebResponse response)
        {
            var length = response.ContentLength;
            if (length == 0)
            {
                return null;
            }
            return length;
        }
        protected static readonly CultureInfo UsCulture = new CultureInfo("en-US");
        private Exception GetException(Exception ex, HttpRequestOptions options, HttpClientInfo client)
        {
            if (ex is HttpException)
            {
                return ex;
            }
            var webException = ex as WebException
                               ?? ex.InnerException as WebException;
            if (webException != null)
            {
                if (options.LogErrors)
                {
                    _logger.LogError(webException, "Error {status} getting response from {url}", webException.Status, options.Url);
                }
                var exception = new HttpException(webException.Message, webException);
                var response = webException.Response as HttpWebResponse;
                if (response != null)
                {
                    exception.StatusCode = response.StatusCode;
                    if ((int)response.StatusCode == 429)
                    {
                        client.LastTimeout = DateTime.UtcNow;
                    }
                }
                if (!exception.StatusCode.HasValue)
                {
                    if (webException.Status == WebExceptionStatus.NameResolutionFailure ||
                        webException.Status == WebExceptionStatus.ConnectFailure)
                    {
                        exception.IsTimedOut = true;
                    }
                }
                return exception;
            }
            var operationCanceledException = ex as OperationCanceledException
                                             ?? ex.InnerException as OperationCanceledException;
            if (operationCanceledException != null)
            {
                return GetCancellationException(options, client, options.CancellationToken, operationCanceledException);
            }
            if (options.LogErrors)
            {
                _logger.LogError(ex, "Error getting response from {url}", options.Url);
            }
            return ex;
        }
        private void DeleteTempFile(string file)
        {
            try
            {
                _fileSystem.DeleteFile(file);
            }
            catch (IOException)
            {
                // Might not have been created at all. No need to worry.
            }
        }
        private void ValidateParams(HttpRequestOptions options)
        {
            if (string.IsNullOrEmpty(options.Url))
            {
                throw new ArgumentNullException("options");
            }
        }
        /// 
        /// Gets the host from URL.
        /// 
        /// The URL.
        /// System.String.
        private string GetHostFromUrl(string url)
        {
            var index = url.IndexOf("://", StringComparison.OrdinalIgnoreCase);
            if (index != -1)
            {
                url = url.Substring(index + 3);
                var host = url.Split(new[] { '/' }, StringSplitOptions.RemoveEmptyEntries).FirstOrDefault();
                if (!string.IsNullOrWhiteSpace(host))
                {
                    return host;
                }
            }
            return url;
        }
        /// 
        /// Throws the cancellation exception.
        /// 
        /// The options.
        /// The client.
        /// The cancellation token.
        /// The exception.
        /// Exception.
        private Exception GetCancellationException(HttpRequestOptions options, HttpClientInfo client, CancellationToken cancellationToken, OperationCanceledException exception)
        {
            // If the HttpClient's timeout is reached, it will cancel the Task internally
            if (!cancellationToken.IsCancellationRequested)
            {
                var msg = string.Format("Connection to {0} timed out", options.Url);
                if (options.LogErrors)
                {
                    _logger.LogError(msg);
                }
                client.LastTimeout = DateTime.UtcNow;
                // Throw an HttpException so that the caller doesn't think it was cancelled by user code
                return new HttpException(msg, exception)
                {
                    IsTimedOut = true
                };
            }
            return exception;
        }
        private void EnsureSuccessStatusCode(HttpClientInfo client, HttpWebResponse response, HttpRequestOptions options)
        {
            var statusCode = response.StatusCode;
            var isSuccessful = statusCode >= HttpStatusCode.OK && statusCode <= (HttpStatusCode)299;
            if (!isSuccessful)
            {
                if (options.LogErrorResponseBody)
                {
                    try
                    {
                        using (var stream = response.GetResponseStream())
                        {
                            if (stream != null)
                            {
                                using (var reader = new StreamReader(stream))
                                {
                                    var msg = reader.ReadToEnd();
                                    _logger.LogError(msg);
                                }
                            }
                        }
                    }
                    catch
                    {
                    }
                }
                throw new HttpException(response.StatusDescription)
                {
                    StatusCode = response.StatusCode
                };
            }
        }
        private Task GetResponseAsync(WebRequest request, TimeSpan timeout)
        {
            var taskCompletion = new TaskCompletionSource();
            Task asyncTask = Task.Factory.FromAsync(request.BeginGetResponse, request.EndGetResponse, null);
            ThreadPool.RegisterWaitForSingleObject((asyncTask as IAsyncResult).AsyncWaitHandle, TimeoutCallback, request, timeout, true);
            var callback = new TaskCallback { taskCompletion = taskCompletion };
            asyncTask.ContinueWith(callback.OnSuccess, TaskContinuationOptions.NotOnFaulted);
            // Handle errors
            asyncTask.ContinueWith(callback.OnError, TaskContinuationOptions.OnlyOnFaulted);
            return taskCompletion.Task;
        }
        private static void TimeoutCallback(object state, bool timedOut)
        {
            if (timedOut)
            {
                WebRequest request = (WebRequest)state;
                if (state != null)
                {
                    request.Abort();
                }
            }
        }
        private class TaskCallback
        {
            public TaskCompletionSource taskCompletion;
            public void OnSuccess(Task task)
            {
                taskCompletion.TrySetResult(task.Result);
            }
            public void OnError(Task task)
            {
                if (task.Exception != null)
                {
                    taskCompletion.TrySetException(task.Exception);
                }
                else
                {
                    taskCompletion.TrySetException(new List());
                }
            }
        }
    }
}