Explicitly register supported mime types (#82)

getmango · Jul 1, 2020 · f5cdf8b · f5cdf8b
1 parent fe082e7
commit f5cdf8b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/library.cr b/src/library.cr
@@ -536,6 +536,8 @@ class Library
 
   def initialize
     @storage = Storage.default
+    register_mime_types
+
     @dir = Config.current.library_path
     @scan_interval = Config.current.scan_interval
     # explicitly initialize @titles to bypass the compiler check. it will

diff --git a/src/util.cr b/src/util.cr
@@ -41,8 +41,6 @@ def send_json(env, json)
 end
 
 def send_attachment(env, path)
-  MIME.register ".cbz", "application/vnd.comicbook+zip"
-  MIME.register ".cbr", "application/vnd.comicbook-rar"
   send_file env, path, filename: File.basename(path), disposition: "attachment"
 end
 
@@ -164,3 +162,14 @@ def escape_xml(str)
     '&'  => "&amp;",
   })
 end
+
+def register_mime_types
+  {
+    ".zip" => "application/zip",
+    ".rar" => "application/x-rar-compressed",
+    ".cbz" => "application/vnd.comicbook+zip",
+    ".cbr" => "application/vnd.comicbook-rar",
+  }.each do |k, v|
+    MIME.register k, v
+  end
+end