fix: send correct MIME for rss file

closes #144
hexojs · Nov 10, 2020 · dd10f2a · dd10f2a
1 parent 571d235
commit dd10f2a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/lib/middlewares/route.js b/lib/middlewares/route.js
@@ -6,7 +6,7 @@ const mime = require('mime');
 module.exports = function(app) {
   const { config, route } = this;
   const { args = {} } = this.env;
-  const { root } = config;
+  const { root, feed } = config;
 
   if (args.s || args.static) return;
 
@@ -63,7 +63,17 @@ module.exports = function(app) {
       return;
     }
 
-    res.setHeader('Content-Type', extname ? mime.getType(extname) : 'application/octet-stream');
+    if (feed && feed.path === url) {
+      if (feed.type === 'atom') {
+        res.setHeader('Content-Type', 'application/atom+xml');
+      } else if (feed.type === 'rss') {
+        res.setHeader('Content-Type', 'application/rss+xml');
+      } else {
+        res.setHeader('Content-Type', extname ? mime.getType(extname) : 'application/octet-stream');
+      }
+    } else {
+      res.setHeader('Content-Type', extname ? mime.getType(extname) : 'application/octet-stream');
+    }
 
     if (method === 'GET') {
       data.pipe(res).on('error', next);