From 28e9d6101e70ac1256f698dd21bb41ec9c42eb20 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marco=20Vidal=20Garc=C3=ADa?= <marco.vidal@cern.ch>
Date: Fri, 14 Feb 2020 17:14:17 +0100
Subject: [PATCH] rest: add preview flag to download file

---
 docs/openapi.json                              |  7 +++++++
 .../rest/workflows_workspace.py                | 18 +++++++++++++++++-
 2 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/docs/openapi.json b/docs/openapi.json
index d0dd9587..8d6ac795 100644
--- a/docs/openapi.json
+++ b/docs/openapi.json
@@ -1218,6 +1218,13 @@
             "name": "file_name",
             "required": true,
             "type": "string"
+          },
+          {
+            "description": "Optional flag to return a previewable response of the file (corresponding mime-type).",
+            "in": "query",
+            "name": "preview",
+            "required": false,
+            "type": "boolean"
           }
         ],
         "produces": [
diff --git a/reana_workflow_controller/rest/workflows_workspace.py b/reana_workflow_controller/rest/workflows_workspace.py
index d8af10e8..fe519c93 100644
--- a/reana_workflow_controller/rest/workflows_workspace.py
+++ b/reana_workflow_controller/rest/workflows_workspace.py
@@ -8,6 +8,8 @@
 
 """REANA Workflow Controller workspaces REST API."""
 
+import json
+import mimetypes
 import os
 
 from flask import Blueprint, current_app, jsonify, request, send_from_directory
@@ -181,6 +183,13 @@ def download_file(workflow_id_or_name, file_name):  # noqa
           description: Required. Name (or path) of the file to be downloaded.
           required: true
           type: string
+        - name: preview
+          in: query
+          description: >-
+            Optional flag to return a previewable response of the file
+            (corresponding mime-type).
+          required: false
+          type: boolean
       responses:
         200:
           description: >-
@@ -220,9 +229,16 @@ def download_file(workflow_id_or_name, file_name):  # noqa
         absolute_workflow_workspace_path = os.path.join(
             current_app.config['SHARED_VOLUME_PATH'],
             workflow.workspace_path)
+
+        preview = json.loads(request.args.get('preview', 'false').lower())
+        response_mime_type = 'multipart/form-data'
+        file_mime_type = mimetypes.guess_type(file_name)[0]
+        # Only display image files as preview
+        if preview and file_mime_type.startswith('image'):
+            response_mime_type = file_mime_type
         return send_from_directory(absolute_workflow_workspace_path,
                                    file_name,
-                                   mimetype='multipart/form-data',
+                                   mimetype=response_mime_type,
                                    as_attachment=True), 200
 
     except ValueError: