Add missing http headers

mieweb · Sep 23, 2023 · 5978f41 · 5978f41
1 parent 599ef14
commit 5978f41
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/src/containers/server/routes/FolderController.ts b/src/containers/server/routes/FolderController.ts
@@ -28,7 +28,7 @@ export const extToMime = {
   'mjs': 'application/javascript',
   'css': 'text/css',
   'txt': 'text/plain',
-  'md': 'text/plain',
+  'md': 'text/x-markdown',
   'htm': 'text/html',
   'html': 'text/html',
   'svg': 'image/svg+xml'
@@ -330,13 +330,25 @@ export default class FolderController extends Controller {
       return;
     } else {
       const ext = await transformedFileSystem.guessExtension(filePath);
-      const mimeType = extToMime[ext];
+
+      const mimeType = extToMime[ext] || (isTextFileName(filePath) ? 'text/plain' : undefined);
       if (mimeType) {
         this.res.setHeader('Content-type', mimeType);
-      } else {
-        if (isTextFileName(filePath)) {
-          this.res.setHeader('Content-type', 'text/plain');
-        }
+      }
+
+      if ('md' === ext) {
+        const previewMdUrl = filePath
+          ? '/' + driveId + (userConfigService.config.hugo_theme?.id ? `/${userConfigService.config.hugo_theme?.id}` : '/_manual') + filePath
+          : '';
+
+        const previewUrl = '/preview' +
+          previewMdUrl
+            .replace(/.md$/, '')
+            .replace(/_index$/, '');
+
+        this.res.setHeader('wgd-path', filePath || '');
+        this.res.setHeader('wgd-mime-type', mimeType);
+        this.res.setHeader('wgd-preview-url', previewUrl);
       }
 
       const buffer = await transformedFileSystem.readBuffer(filePath);

diff --git a/src/containers/transform/DirectoryScanner.ts b/src/containers/transform/DirectoryScanner.ts
@@ -24,6 +24,9 @@ export function isTextFileName(fileName) {
   if (fileName.endsWith('.css')) {
     return true;
   }
+  if (fileName.endsWith('.md')) {
+    return true;
+  }
   return false;
 }