using DNS.Client;
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using System;
using System.Linq;
using System.Net;
using System.Threading;
using System.Threading.Tasks;

namespace FastGithub.ReverseProxy
{
    /// <summary>
    /// 域名解析器
    /// </summary> 
    sealed class DomainResolver
    {
        private DnsClient dnsClient;
        private readonly IMemoryCache memoryCache;
        private readonly IOptionsMonitor<FastGithubOptions> options;
        private readonly ILogger<DomainResolver> logger;
        private readonly TimeSpan cacheTimeSpan = TimeSpan.FromSeconds(10d);

        /// <summary>
        /// 域名解析器
        /// </summary>
        /// <param name="memoryCache"></param>
        /// <param name="fastGithubConfig"></param>
        public DomainResolver(
            IMemoryCache memoryCache,
            IOptionsMonitor<FastGithubOptions> options,
            ILogger<DomainResolver> logger)
        {
            this.memoryCache = memoryCache;
            this.options = options;
            this.logger = logger;

            this.dnsClient = new DnsClient(options.CurrentValue.FastDns.ToIPEndPoint());
            options.OnChange(opt => this.dnsClient = new DnsClient(opt.FastDns.ToIPEndPoint()));
        }

        /// <summary>
        /// 解析指定的域名
        /// </summary>
        /// <param name="domain"></param>
        /// <returns></returns>
        /// <exception cref="FastGithubException"></exception>
        public async Task<IPAddress> ResolveAsync(string domain, CancellationToken cancellationToken)
        {
            // 缓存以避免做不必要的并发查询
            var key = $"domain:{domain}";
            var address = await this.memoryCache.GetOrCreateAsync(key, e =>
            {
                e.SetAbsoluteExpiration(this.cacheTimeSpan);
                return this.LookupAsync(domain, cancellationToken);
            });
            return address;
        }

        /// <summary>
        /// 查找ip
        /// </summary>
        /// <param name="domain"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        /// <exception cref="FastGithubException">
        private async Task<IPAddress> LookupAsync(string domain, CancellationToken cancellationToken)
        {
            try
            {
                var addresses = await dnsClient.Lookup(domain, DNS.Protocol.RecordType.A, cancellationToken);
                var address = addresses?.FirstOrDefault();
                if (address == null)
                {
                    throw new Exception($"解析不到{domain}的ip");
                }

                // 受干扰的dns，常常返回127.0.0.1来阻断请求
                // 如果解析到的ip为本机ip，会产生反向代理请求死循环
                if (address.Equals(IPAddress.Loopback))
                {
                    throw new Exception($"dns被污染，解析{domain}为{address}");
                }

                this.logger.LogInformation($"[{domain}->{address}]");
                return address;
            }
            catch (Exception ex)
            {
                var dns = this.options.CurrentValue.PureDns;
                throw new FastGithubException($"dns({dns})服务器异常：{ex.Message}", ex);
            }
        }
    }
}