From 8727285bc504f18c0b32b1ee1626b625a587622e Mon Sep 17 00:00:00 2001
From: James Mills <prologic@shortcircuit.net.au>
Date: Thu, 26 Mar 2020 14:34:19 +1000
Subject: Refactored thumbnail generation with configurable timeout and in-sycn
 with video transcoding

---
 app/app.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'app/app.go')

diff --git a/app/app.go b/app/app.go
index ac377be..5150d74 100644
--- a/app/app.go
+++ b/app/app.go
@@ -241,6 +241,8 @@ func (a *App) uploadHandler(w http.ResponseWriter, r *http.Request) {
 			a.Library.Paths[collection].Path,
 			fmt.Sprintf("%s.mp4", shortuuid.New()),
 		)
+		thumbFn1 := fmt.Sprintf("%s.jpg", strings.TrimSuffix(tf.Name(), filepath.Ext(tf.Name())))
+		thumbFn2 := fmt.Sprintf("%s.jpg", strings.TrimSuffix(vf, filepath.Ext(vf)))
 
 		// TODO: Use a proper Job Queue and make this async
 		if err := utils.RunCmd(
@@ -260,6 +262,27 @@ func (a *App) uploadHandler(w http.ResponseWriter, r *http.Request) {
 			return
 		}
 
+		if err := utils.RunCmd(
+			a.Config.Thumbnailer.Timeout,
+			"mt",
+			"-b",
+			"-s",
+			"-n", "1",
+			tf.Name(),
+		); err != nil {
+			err := fmt.Errorf("error generating thumbnail: %w", err)
+			log.Error(err)
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		if err := os.Rename(thumbFn1, thumbFn2); err != nil {
+			err := fmt.Errorf("error renaming generated thumbnail: %w", err)
+			log.Error(err)
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
 		if err := os.Rename(tf.Name(), vf); err != nil {
 			err := fmt.Errorf("error renaming transcoded video: %w", err)
 			log.Error(err)
-- 
cgit v1.2.3