From da77ea463093157cd918e896230628344df9bad9 Mon Sep 17 00:00:00 2001
From: Tobias Klika <cee@live.at>
Date: Tue, 22 Sep 2020 19:28:54 +0200
Subject: [PATCH] Update MediaController.cs

---
 zero.Web/Controllers/MediaController.cs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/zero.Web/Controllers/MediaController.cs b/zero.Web/Controllers/MediaController.cs
index e252a560..8c197b60 100644
--- a/zero.Web/Controllers/MediaController.cs
+++ b/zero.Web/Controllers/MediaController.cs
@@ -1,10 +1,12 @@
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.AspNetCore.StaticFiles;
+using Microsoft.Net.Http.Headers;
 using System;
 using System.Collections.Generic;
 using System.IO;
 using System.Threading.Tasks;
+using zero.Core;
 using zero.Core.Api;
 using zero.Core.Entities;
 using zero.Core.Identity;
@@ -17,12 +19,14 @@ namespace zero.Web.Controllers
   {
     IMediaApi Api;
     IMediaFolderApi MediaFolderApi;
+    IPaths Paths;
 
 
-    public MediaController(IMediaApi api, IMediaFolderApi mediaFolderApi)
+    public MediaController(IMediaApi api, IMediaFolderApi mediaFolderApi, IPaths paths)
     {
       Api = api;
       MediaFolderApi = mediaFolderApi;
+      Paths = paths;
     }
 
   
@@ -77,8 +81,9 @@ namespace zero.Web.Controllers
     public async Task<IActionResult> StreamThumbnail(string id, [FromQuery] bool thumb = true)
     {
       string path = await Api.GetSourceById(id, thumb);
+      string fullPath = Path.Combine(Paths.Root, path?.Trim('/') ?? String.Empty);
 
-      if (path == null)
+      if (path == null || !System.IO.File.Exists(fullPath))
       {
         return NotFound();
       }
@@ -90,7 +95,7 @@ namespace zero.Web.Controllers
         contentType = "application/octet-stream";
       }
 
-      return File(path, contentType);
+      return File(path, contentType, DateTimeOffset.Now, EntityTagHeaderValue.Any);
     }
   }
 }