From 36cf628a63f1b11ce908ebd7e7f59b4c237f38d1 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Tue, 7 Aug 2018 12:49:00 -0400
Subject: [PATCH] add static files caching

---
 util/Server/Startup.cs | 40 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/util/Server/Startup.cs b/util/Server/Startup.cs
index 0edef59122..546f5b2ece 100644
--- a/util/Server/Startup.cs
+++ b/util/Server/Startup.cs
@@ -1,4 +1,7 @@
-using Microsoft.AspNetCore.Builder;
+using System.Collections.Generic;
+using System.Linq;
+using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
@@ -7,6 +10,15 @@ namespace Bit.Server
 {
     public class Startup
     {
+        private readonly List<string> _longCachedPaths = new List<string>
+        {
+            "/app/", "/locales/", "/fonts/", "/connectors/", "/scripts/"
+        };
+        private readonly List<string> _mediumCachedPaths = new List<string>
+        {
+            "/images/"
+        };
+
         public void ConfigureServices(IServiceCollection services)
         { }
 
@@ -30,7 +42,31 @@ namespace Bit.Server
             }
             else
             {
-                app.UseFileServer();
+                var options = new DefaultFilesOptions();
+                options.DefaultFileNames.Clear();
+                options.DefaultFileNames.Add("index.html");
+                app.UseDefaultFiles(options);
+                app.UseStaticFiles(new StaticFileOptions
+                {
+                    OnPrepareResponse = ctx =>
+                    {
+                        if(!ctx.Context.Request.Path.HasValue)
+                        {
+                            return;
+                        }
+                        var path = ctx.Context.Request.Path.Value;
+                        if(_longCachedPaths.Any(ext => path.StartsWith(ext)))
+                        {
+                            // 14 days
+                            ctx.Context.Response.Headers.Append("Cache-Control", "max-age=1209600");
+                        }
+                        if(_mediumCachedPaths.Any(ext => path.StartsWith(ext)))
+                        {
+                            // 7 days
+                            ctx.Context.Response.Headers.Append("Cache-Control", "max-age=604800");
+                        }
+                    }
+                });
             }
         }
     }