Add mime type to StorageItem

sitkoru · Dec 17, 2020 · 50ae697 · 50ae697
1 parent 36d04cc
commit 50ae697
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/src/Sitko.Core.Storage/Sitko.Core.Storage.csproj b/src/Sitko.Core.Storage/Sitko.Core.Storage.csproj
@@ -25,6 +25,7 @@
   </ItemGroup>
   <ItemGroup>
     <PackageReference Include="Microsoft.SourceLink.GitHub" Version="1.0.0" PrivateAssets="All" />
+    <PackageReference Include="MimeMapping" Version="1.0.1.30" />
   </ItemGroup>
 
   <ItemGroup Condition=" '$(TargetFramework)' == 'netcoreapp3.1' ">

diff --git a/src/Sitko.Core.Storage/Storage.cs b/src/Sitko.Core.Storage/Storage.cs
@@ -87,15 +87,16 @@ private StorageItem CreateStorageItem(string destinationPath,
             long fileSize,
             StorageItemMetadata metadata)
         {
+            var fileName = metadata.FileName ?? Path.GetFileName(destinationPath);
             var storageItem = new StorageItem
             {
                 Path = PreparePath(Path.GetDirectoryName(destinationPath))!,
-                FileName =
-                    metadata.FileName ?? Path.GetFileName(destinationPath),
+                FileName = fileName,
                 LastModified = date,
                 FileSize = fileSize,
                 FilePath = destinationPath,
-                MetadataJson = metadata.Data
+                MetadataJson = metadata.Data,
+                MimeType = MimeMapping.MimeUtility.GetMimeMapping(fileName)
             };
             return storageItem;
         }

diff --git a/src/Sitko.Core.Storage/StorageItem.cs b/src/Sitko.Core.Storage/StorageItem.cs
@@ -14,6 +14,11 @@ public sealed record StorageItem
         /// Size of uploaded file
         /// </summary>
         public long FileSize { get; set; }
+
+        /// <summary>
+        /// MimeType of uploaded file
+        /// </summary>
+        public string MimeType { get; set; } = string.Empty;
 
         /// <summary>
         /// Full path to uploaded file in storage

diff --git a/tests/Sitko.Core.Storage.Tests/BasicTests.cs b/tests/Sitko.Core.Storage.Tests/BasicTests.cs
@@ -34,6 +34,7 @@ public async Task UploadFile()
             Assert.NotNull(uploaded);
             Assert.NotEqual(0, uploaded.FileSize);
             Assert.Equal(fileName, uploaded.FileName);
+            Assert.Equal("text/plain", uploaded.MimeType);
         }
 
         [Fact]