collection.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"os"
 	"sync"
 )
 
@@ -14,6 +15,8 @@ type Collection struct {
 	Lists   []*List
 	listMap map[string]*List
 
+	Files *FileSource
+
 	mutex      sync.RWMutex
 	hasChanged bool
 }
@@ -163,3 +166,30 @@ func (c *Collection) UpdateList(l *List) {
 
 	c.hasChanged = true
 }
+
+func (c *Collection) AllFiles() []string {
+	if c.Files == nil {
+		return nil
+	}
+	return c.Files.Files
+}
+
+func (c *Collection) SearchFiles(query string) []string {
+	return c.Files.Search(query)
+}
+
+func (c *Collection) ScanFiles(root string) error {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+
+	if c.Files == nil {
+		c.Files = &FileSource{}
+	}
+
+	if err := c.Files.Scan(os.Getenv("FILE_ROOT")); err != nil {
+		return err
+	}
+
+	c.hasChanged = true
+	return nil
+}

files.go

@@ -35,6 +35,10 @@ func (fs *FileSource) Scan(root string) error {
 }
 
 func (fs *FileSource) Search(query string) []string {
+	if fs == nil {
+		return nil
+	}
+
 	split := strings.Split(query, " ")
 
 	var res []string

movies.go

@@ -16,6 +16,8 @@ type Movie struct {
 	Year     string
 	Runtime  string
 
+	Files []string
+
 	OMDB OMDBMovie
 }
 

pages/pages.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"html/template"
 	"net/http"
+	"os"
 	"strings"
 
 	"gogs.gildas.ch/gildas/movies"
@@ -16,6 +17,7 @@ func Router(c *movies.Collection) http.HandlerFunc {
 	homeHandler := Home(c)
 	movieHandler := Movie(c)
 	listHandler := List(c)
+	filesHandler := Files(c)
 
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		path := r.URL.Path
@@ -23,6 +25,9 @@ func Router(c *movies.Collection) http.HandlerFunc {
 		case path == "/":
 			homeHandler(w, r)
 			return
+		case path == "/files":
+			filesHandler(w, r)
+			return
 		case path == "/style.css":
 			http.ServeFile(w, r, "templates/style.css")
 			return
@@ -108,15 +113,28 @@ func Movie(c *movies.Collection) http.HandlerFunc {
 		}
 
 		if r.Method == "POST" {
-			updated, err := movies.Unmarshal([]byte(r.FormValue("movie_json")))
-			if err != nil {
-				errs = append(errs, err)
-			} else if imdbID != updated.IMDBID {
-				errs = append(errs, fmt.Errorf("you cannot change the imdb id."))
-			} else {
-				m = updated
-				c.Update(updated)
+			if movieJSON := r.FormValue("movie_json"); movieJSON != "" {
+				updated, err := movies.Unmarshal([]byte(movieJSON))
+				if err != nil {
+					errs = append(errs, err)
+				} else if imdbID != updated.IMDBID {
+					errs = append(errs, fmt.Errorf("you cannot change the imdb id."))
+				} else {
+					m = updated
+					c.Update(updated)
+				}
 			}
+
+			if files := r.FormValue("files"); files != "" {
+				m.Files = strings.Split(files, "\n")
+				c.Update(m)
+			}
+		}
+
+		fileQuery := r.URL.Query().Get("file_query")
+		var fileResults string
+		if fileQuery != "" {
+			fileResults = strings.Join(c.SearchFiles(fileQuery), "\n")
 		}
 
 		b, err := json.MarshalIndent(m, "", "  ")
@@ -125,9 +143,12 @@ func Movie(c *movies.Collection) http.HandlerFunc {
 		}
 
 		t.Execute(w, map[string]interface{}{
-			"Movie":     m,
-			"MovieJSON": string(b),
-			"Errors":    errs,
+			"Movie":       m,
+			"MovieJSON":   string(b),
+			"Files":       strings.Join(m.Files, "\n"),
+			"FileQuery":   fileQuery,
+			"FileResults": fileResults,
+			"Errors":      errs,
 		})
 	})
 }
@@ -191,3 +212,35 @@ func List(c *movies.Collection) http.HandlerFunc {
 		})
 	})
 }
+
+func Files(c *movies.Collection) http.HandlerFunc {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if query := r.URL.Query().Get("query"); query != "" {
+			matches := c.SearchFiles(query)
+
+			for _, match := range matches {
+				fmt.Fprintf(w, "%s\n", match)
+			}
+			return
+		}
+
+		t, err := template.ParseFiles("templates/files.html")
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+
+		var errs []error
+
+		if r.Method == "POST" && r.FormValue("scan") == "1" {
+			if err := c.ScanFiles(os.Getenv("FILE_ROOT")); err != nil {
+				errs = append(errs, err)
+			}
+		}
+
+		t.Execute(w, map[string]interface{}{
+			"Files":  c.AllFiles(),
+			"Errors": errs,
+		})
+	})
+}

templates/files.html

@@ -0,0 +1,25 @@
+<html>
+    <head>
+        <title>gildas.ch</title>
+
+        <link href="https://fonts.googleapis.com/css2?family=Roboto&display=swap" rel="stylesheet">
+        <link rel="stylesheet" href="/style.css">
+    </head>
+    <body>
+        {{ range $e := .Errors }}
+        <div class="error">{{ $e }}</div>
+        {{ end }}
+
+        <form method="post">
+            <p>Scan files: <input type="submit" />
+                <input type="hidden" name="scan" value="1" />
+            </p>
+        </form>
+
+        <ul class="file-list">
+            {{ range $f := .Files }}
+            <li>{{ $f }}</li>
+            {{ end }}
+        </ul>
+    </body>
+</html>

templates/movie.html

@@ -18,6 +18,28 @@
             <div>{{ .Movie.Director }}</div>
         </div>
 
+        <ul class="file-list">
+            {{ range $f := .Movie.Files }}
+            <li>{{ $f }}</li>
+            {{ end }}
+        </ul>
+
+        <div>
+            <form method="post">
+                <p>Update file list: <input type="submit" /><br />
+                    <textarea name="files" style="width:500px;height:140px;">{{ .Files }}</textarea>
+                </p>
+            </form>
+
+            <form method="get">
+                <p>
+                    Search by filename: <input type="text" name="file_query" value="{{ .FileQuery }}" />
+                    <input type="submit" />
+                </p>
+            </form>
+            <textarea style="width:500px;height:140px;">{{ .FileResults }}</textarea>
+        </div>
+
         <form method="post">
             <p>Update movie: <input type="submit" /><br />
                 <textarea name="movie_json" style="width:90%;height:500px;">{{ .MovieJSON }}</textarea>