using System;
using Microsoft.AspNetCore.Http;
using NzbDrone.Common.EnvironmentInfo;
using NzbDrone.Common.Extensions;

namespace Prowlarr.Http.Middleware
{
    public interface ICacheableSpecification
    {
        bool IsCacheable(HttpContext context);
    }

    public class CacheableSpecification : ICacheableSpecification
    {
        public bool IsCacheable(HttpContext context)
        {
            if (!RuntimeInfo.IsProduction)
            {
                return false;
            }

            if (context.Request.Query.ContainsKey("h"))
            {
                return true;
            }

            if (context.Request.Path.StartsWithSegments("/api", StringComparison.CurrentCultureIgnoreCase))
            {
                if (context.Request.Path.ToString().ContainsIgnoreCase("/MediaCover"))
                {
                    return true;
                }

                return false;
            }

            if (context.Request.Path.StartsWithSegments("/signalr", StringComparison.CurrentCultureIgnoreCase))
            {
                return false;
            }

            if (context.Request.Path.Equals("/index.js"))
            {
                return false;
            }

            if (context.Request.Path.Equals("/initialize.js"))
            {
                return false;
            }

            if (context.Request.Path.StartsWithSegments("/feed", StringComparison.CurrentCultureIgnoreCase))
            {
                return false;
            }

            if (context.Request.Path.StartsWithSegments("/log", StringComparison.CurrentCultureIgnoreCase) &&
                context.Request.Path.ToString().EndsWith(".txt", StringComparison.CurrentCultureIgnoreCase))
            {
                return false;
            }

            if (context.Response != null)
            {
                if (context.Response.ContentType?.Contains("text/html") ?? false || context.Response.StatusCode >= 400)
                {
                    return false;
                }
            }

            return true;
        }
    }
}