Add MediaInfo to movies and generate action

ffmpeg/ffmpeg.go

@@ -0,0 +1,50 @@
+package ffmpeg
+
+import (
+	"fmt"
+	"os/exec"
+	"strings"
+)
+
+const (
+	inputFileIdentifier = "INPUT_VIDEO_FILE"
+)
+
+var (
+	Debug = true
+)
+
+func MediaInfo(path string) (string, error) {
+	out, err := execCommand(
+		fmt.Sprintf("-i %s", inputFileIdentifier),
+		path)
+	if err.Error() == "exit status 1" && strings.HasSuffix(out, "At least one output file must be specified") {
+		err = nil
+	}
+	return out, err
+}
+
+func execCommand(cmdStr, path string) (string, error) {
+	cmdSplit := strings.Split(cmdStr, " ")
+	for i, s := range cmdSplit {
+		if s == inputFileIdentifier {
+			cmdSplit[i] = path
+		}
+	}
+
+	cmd := exec.Command("ffmpeg", cmdSplit...)
+
+	if Debug {
+		fmt.Println("Executing:", cmd)
+	}
+	out, err := cmd.CombinedOutput()
+	if err != nil {
+		fmt.Printf("Error: %#v\n", err.Error())
+		return strings.TrimSuffix(string(out), "\n"), err
+	}
+	if Debug {
+		fmt.Printf("Output: %s\n", string(out))
+	}
+
+	return strings.TrimSuffix(string(out), "\n"), err
+}

movies.go

@@ -2,8 +2,11 @@ package movies
 
 import (
 	"encoding/json"
+	"fmt"
 	"strconv"
 	"strings"
+
+	"gogs.gildas.ch/gildas/movies/ffmpeg"
 )
 
 type Movie struct {
@@ -18,6 +21,7 @@ type Movie struct {
 
 	Files     []string
 	MovieFile string
+	MediaInfo string
 	Subtitles string
 
 	OMDB OMDBMovie
@@ -127,3 +131,17 @@ func (m *Movie) FillFromOMDB() error {
 
 	return nil
 }
+
+func (m *Movie) GenerateMediaInfo() error {
+	if m.MovieFile == "" {
+		return fmt.Errorf("no movie file set")
+	}
+
+	mediaInfo, err := ffmpeg.MediaInfo(m.MovieFile)
+	if err != nil {
+		return fmt.Errorf("error generating media info from %q: %w", m.MovieFile, err)
+	}
+
+	m.MediaInfo = mediaInfo
+	return nil
+}

pages/pages.go

@@ -83,12 +83,14 @@ func Home(c *movies.Collection) http.HandlerFunc {
 			}
 		}
 
-		t.Execute(w, map[string]interface{}{
+		if err := t.Execute(w, map[string]interface{}{
 			"Collection": c,
 			"IMDBID":     r.URL.Query().Get("imdb_id"),
 			"OMDBString": omdbString,
 			"Errors":     errs,
-		})
+		}); err != nil {
+			fmt.Println(err)
+		}
 	})
 }
 
@@ -135,6 +137,14 @@ func Movie(c *movies.Collection) http.HandlerFunc {
 				fmt.Println("Update with", m)
 				c.Update(m)
 			}
+
+			if r.FormValue("generate_mediainfo") != "" {
+				if err := m.GenerateMediaInfo(); err != nil {
+					errs = append(errs, err)
+				} else {
+					c.Update(m)
+				}
+			}
 		}
 
 		fileQuery := r.URL.Query().Get("file_query")
@@ -148,14 +158,16 @@ func Movie(c *movies.Collection) http.HandlerFunc {
 			errs = append(errs, err)
 		}
 
-		t.Execute(w, map[string]interface{}{
+		if err := t.Execute(w, map[string]interface{}{
 			"Movie":       m,
 			"MovieJSON":   string(b),
 			"Files":       strings.Join(m.Files, "\n"),
 			"FileQuery":   fileQuery,
 			"FileResults": fileResults,
 			"Errors":      errs,
-		})
+		}); err != nil {
+			fmt.Println(err)
+		}
 	})
 }
 
@@ -206,7 +218,7 @@ func List(c *movies.Collection) http.HandlerFunc {
 			}
 		}
 
-		t.Execute(w, map[string]interface{}{
+		if err := t.Execute(w, map[string]interface{}{
 			"List": l,
 			"Description": template.HTML(
 				blackfriday.Run(
@@ -215,7 +227,9 @@ func List(c *movies.Collection) http.HandlerFunc {
 			"Movies":   ms,
 			"ListJSON": string(b),
 			"Errors":   errs,
-		})
+		}); err != nil {
+			fmt.Println(err)
+		}
 	})
 }
 
@@ -244,10 +258,12 @@ func Files(c *movies.Collection) http.HandlerFunc {
 			}
 		}
 
-		t.Execute(w, map[string]interface{}{
+		if err := t.Execute(w, map[string]interface{}{
 			"Files":  c.AllFiles(),
 			"Errors": errs,
-		})
+		}); err != nil {
+			fmt.Println(err)
+		}
 	})
 }
 

templates/movie.html

@@ -18,12 +18,20 @@
             <div>{{ .Movie.Director }}</div>
         </div>
 
+        <p>Movie file: <a href="/files/{{ .Movie.MovieFile }}">{{ .Movie.MovieFile }}</a></p>
+
+        <pre>{{ .Movie.MediaInfo }}</pre>
+
         <ul class="file-list">
             {{ range $f := .Movie.Files }}
             <li><a href="/files/{{ $f }}">{{ $f }}</a></li>
             {{ end }}
         </ul>
 
+        <div>
+            <form method="post"><input type="submit" name="generate_mediainfo" value="Generate MediaInfo" /></form>
+        </div>
+
         <div>
             <form method="post">
                 <p>Update file list: <input type="submit" /><br />