Add mimeTypeFromDestination to server config (#22)

lehigh-university-libraries · May 19, 2024 · 53cac41 · 53cac41
1 parent 3d9cf40
commit 53cac41
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/examples/ffmpeg/scyllaridae.yml b/examples/ffmpeg/scyllaridae.yml
@@ -1,3 +1,4 @@
+mimeTypeFromDestination: true
 allowedMimeTypes:
   - "audio/*"
   - "video/*"

diff --git a/internal/config/server.go b/internal/config/server.go
@@ -54,6 +54,11 @@ type ServerConfig struct {
 	//
 	// required: false
 	CmdByMimeType map[string]Command `yaml:"cmdByMimeType"`
+
+	// Commands and arguments ran by MIME type based on the destination file format
+	//
+	// required: false
+	MimeTypeFromDestination bool `yaml:"mimeTypeFromDestination,omitempty"`
 }
 
 // Command describes the command and arguments to execute for a specific MIME type.
@@ -116,11 +121,16 @@ func ReadConfig(yp string) (*ServerConfig, error) {
 }
 
 func BuildExecCommand(message api.Payload, c *ServerConfig) (*exec.Cmd, error) {
-	if message.Attachment.Content.SourceMimeType != "" && !IsAllowedMimeType(message.Attachment.Content.SourceMimeType, c.AllowedMimeTypes) {
-		return nil, fmt.Errorf("undefined sourceMimeType: %s", message.Attachment.Content.SourceMimeType)
+	mimeType := message.Attachment.Content.SourceMimeType
+	if c.MimeTypeFromDestination {
+		mimeType = message.Attachment.Content.DestinationMimeType
+	}
+
+	if mimeType != "" && !IsAllowedMimeType(mimeType, c.AllowedMimeTypes) {
+		return nil, fmt.Errorf("undefined mimeType to build command: %s", mimeType)
 	}
 
-	cmdConfig, exists := c.CmdByMimeType[message.Attachment.Content.SourceMimeType]
+	cmdConfig, exists := c.CmdByMimeType[mimeType]
 	if !exists {
 		cmdConfig = c.CmdByMimeType["default"]
 	}