Enable videos for uploading additional files

Author: zubeydecivelek

cds/modules/deposit/api.py

@@ -860,15 +860,7 @@ def _rename_subtitles(self):
                 subtitle_obj_key = "{}_{}.vtt".format(
                     self["report_number"][0], match.group("iso_lang")
                 )
-                obj = ObjectVersion.create(
-                    bucket=subtitle_obj.bucket,
-                    key=subtitle_obj_key,
-                    _file_id=subtitle_obj.file_id,
-                )
-                # copy tags to the newly created object version
-                for tag in subtitle_obj.tags:
-                    tag.object_version = obj
-                subtitle_obj.remove()
+                subtitle_obj.key = subtitle_obj_key
 
     def _rename_master_file(self, master_file):
         """Rename master file."""

cds/modules/deposit/ext.py

@@ -25,6 +25,7 @@
 """CDSDeposit app for Webhook receivers."""
 
 import re
+import mimetypes
 
 from invenio_base.signals import app_loaded
 from invenio_db import db
@@ -45,38 +46,33 @@
 
 def _create_tags(obj):
     """Create additional tags for file."""
-    # Subtitle file
-    pattern = re.compile(".*_([a-zA-Z]{2})\.vtt$")
+    pattern_subtitle = re.compile(r".*_([a-zA-Z]{2})\.vtt$")
+    pattern_poster = re.compile(r"^poster\.(jpg|png)$")
+    
+    # Get the media_type and content_type(file ext)
+    file_name = obj.key
+    mimetypes.add_type("subtitle/vtt", ".vtt")
+    media_type = mimetypes.guess_type(file_name)[0].split("/")[0]
+    file_ext = file_name.rsplit(".", 1)[-1].lower()
+
     with db.session.begin_nested():
-        # language tag
-        found = pattern.findall(obj.key)
-        if len(found) == 1:
-            lang = found[0]
-            ObjectVersionTag.create_or_update(obj, "language", lang)
-        else:
-            # clean to be sure there is no some previous value
-            ObjectVersionTag.delete(obj, "language")
-        # other tags
-        ObjectVersionTag.create_or_update(obj, "content_type", "vtt")
-        ObjectVersionTag.create_or_update(obj, "context_type", "subtitle")
-        ObjectVersionTag.create_or_update(obj, "media_type", "subtitle")
-        # refresh object
-        db.session.add(obj)
+        ObjectVersionTag.create_or_update(obj, "content_type", file_ext)
+        ObjectVersionTag.create_or_update(obj, "media_type", media_type)
+        if file_ext == "vtt":
+            # language tag
+            match = pattern_subtitle.search(file_name)
+            if match:
+                ObjectVersionTag.create_or_update(obj, "language", match.group(1))
+            else:
+                ObjectVersionTag.delete(obj, "language")
+            # other tags
+            ObjectVersionTag.create_or_update(obj, "content_type", "vtt")
+            ObjectVersionTag.create_or_update(obj, "context_type", "subtitle")
+        # poster tag 
+        elif pattern_poster.match(file_name):
+            ObjectVersionTag.create_or_update(obj, "context_type", "poster")
 
-        # Poster frame
-        pattern = re.compile("^poster\.(jpg|png)$")
-        try:
-            poster = pattern.findall(obj.key)
-            if poster:
-                ext = pattern.findall(poster.key)[0]
-                # frame tags
-                ObjectVersionTag.create_or_update(poster, "content_type", ext)
-                ObjectVersionTag.create_or_update(poster, "context_type", "poster")
-                ObjectVersionTag.create_or_update(poster, "media_type", "image")
-                # refresh object
-                db.session.add(poster)
-        except IndexError:
-            return
+        db.session.add(obj)
 
 
 def create_tags_on_file_upload(sender, obj):

cds/modules/deposit/static/templates/cds_deposit/deposits.html

@@ -39,6 +39,7 @@ <h5 class="text-muted"><strong>Tips</strong></h5>
                     <h3>Click here to select videos to upload</h3>
                   </div>
                   <p class="cds-deposit-box-upload-description">You can also <strong>Drag & Drop</strong> video files here</p>
+                  <p class="text-muted mt-20">supported files <mark>{{ $ctrl.videoExtensions }}</mark></p>
                 </div>
               </div>
             </div>

cds/modules/deposit/static/templates/cds_deposit/types/video/uploader.html

@@ -283,6 +283,7 @@ <h5 class="text-muted"><strong>Tips and suggestions</strong></h5>
   </ul>
 </div>
 <!-- Error alert -->
+
 <div class="cds-deposit-box" ng-if="!$ctrl.cdsDepositCtrl.isPublished()">
   <div
        ngf-drag-over-class="'cds-deposit-dragover'"
@@ -309,13 +310,36 @@ <h4>Upload complimentary files for this video</h4>
   <div class="text-muted">
     <h5 class="text-muted"><strong>Tips and suggestions</strong></h5>
     <ul>
-      <li>To replace the video file, just upload another video.</li>
+      <li>To replace the video file, just upload another video to here.</li>
     </ul>
+    <div
+      ngf-drag-over-class="{accept: 'cds-deposit-dragover',  delay:100}"
+      ngf-drop=""
+      ngf-change="$ctrl.replaceMasterFile($newFiles, $invalidFiles)"
+      ngf-select=""
+      ngf-model-options="{allowInvalid: false}"
+      ngf-max-size="500GB"
+      ngf-multiple="true"
+      ngf-accept="'{{$ctrl.cdsDepositsCtrl.videoExtensions}}'"
+      ngf-pattern="'{{$ctrl.cdsDepositsCtrl.videoExtensions}}'"
+  >
+  <div class="pa-10 cds-deposit-box-upload-wrapper text-center">
+    <p class="cds-deposit-box-upload-icon mb-20">
+      <i class="fa fa-2x fa-video-camera" aria-hidden="true"></i>
+    </p>
+    <div class="cds-deposit-box-upload-content">
+      <div class="cds-deposit-box-upload-title">
+        <h4>To replace the video file, just upload a video to here.</h4>
+      </div>
+      <p class="cds-deposit-box-upload-description"> Or Drag & Drop files</p>
+    </div>
+  </div>
+  </div>
   </div>
 </div>
 <div ng-if="$ctrl.cdsDepositCtrl.isPublished()" class="cds-deposit-box text-muted">
   <h5 class="text-muted"><strong>Tips and suggestions</strong></h5>
   <ul>
     <li>Click the <strong>Edit</strong> button on the top right corner to add more files.</li>
   </ul>
-</div>
+</div>

cds/modules/theme/assets/bootstrap3/js/cds_deposit/avc/components/cdsUploader.js

@@ -115,7 +115,10 @@ function cdsUploaderCtrl(
         if (!upload.key) {
           upload.key = upload.name;
         }
-        if (that.cdsDepositsCtrl.isVideoFile(upload.key)) {
+        if (
+          !upload.isAdditional &&
+          that.cdsDepositsCtrl.isVideoFile(upload.key)
+        ) {
           _subpromise = Upload.http(_startWorkflow(upload, response));
         } else {
           var d = $q.defer();
@@ -282,21 +285,16 @@ function cdsUploaderCtrl(
       angular.forEach(_files, function (file) {
         file.key = file.name;
         file.local = !file.receiver;
+        file.isAdditional = true;
         // Add any extra paramemters to the files
         if (extraHeaders) {
           file.headers = extraHeaders;
         }
+        file.headers = {
+          "X-Invenio-File-Tags": "context_type=additional_file",
+        };
       });
 
-      // Add the files to the list
-      var masterFile = that.cdsDepositCtrl.findMasterFile() || {};
-      var videoFiles = _.values(
-        that.cdsDepositsCtrl.filterOutFiles(_files).videos
-      );
-
-      // Exclude video files
-      _files = _.difference(_files, videoFiles);
-
       // Find if any of the existing files has been replaced
       // (file with same filename), and if yes remove it from the existing
       // file list (aka from the interface).
@@ -323,6 +321,44 @@ function cdsUploaderCtrl(
       Array.prototype.push.apply(that.files, _files);
       // Add the files to the queue
       Array.prototype.push.apply(that.queue, _files);
+
+      // Start upload automatically if the option is selected
+      if (that.autoStartUpload) {
+        that.upload();
+      }
+    };
+
+    this.replaceMasterFile = function (_files, invalidFiles) {
+      // Do nothing if files array is empty
+      if (!_files) {
+        return;
+      }
+      // Remove any invalid files
+      _files = _.difference(_files, invalidFiles || []);
+      // Make sure they have proper metadata
+      angular.forEach(_files, function (file) {
+        file.key = file.name;
+        file.local = !file.receiver;
+      });
+
+      // Add the files to the list
+      var masterFile = that.cdsDepositCtrl.findMasterFile() || {};
+      var videoFiles = _.values(
+        that.cdsDepositsCtrl.filterOutFiles(_files).videos
+      );
+
+      if ((invalidFiles || []).length > 0) {
+        // Push a notification
+        toaster.pop({
+          type: "error",
+          title:
+            "Invalid file(s) for " +
+            (that.cdsDepositCtrl.record.title.title || "video."),
+          body: _.map(invalidFiles, "name").join(", "),
+          bodyOutputType: "trustedHtml",
+        });
+      }
+
       if (!that.cdsDepositCtrl.master) {
         // Check for new master file
         var newMasterFile = videoFiles[0];
@@ -358,11 +394,6 @@ function cdsUploaderCtrl(
           });
         }
       }
-
-      // Start upload automatically if the option is selected
-      if (that.autoStartUpload) {
-        that.upload();
-      }
     };
 
     // Prepare file request