Introduce List page with add, view, update

collection.go

@@ -2,16 +2,20 @@ package movies
 
 import (
 	"encoding/json"
+	"fmt"
 	"io/ioutil"
 	"sync"
 )
 
 type Collection struct {
-	Movies []*Movie
+	Movies   []*Movie
+	movieMap map[string]*Movie
+
+	Lists   []*List
+	listMap map[string]*List
 
 	mutex      sync.RWMutex
 	hasChanged bool
-	movieMap   map[string]*Movie
 }
 
 func Import(filename string) (*Collection, error) {
@@ -31,6 +35,11 @@ func Import(filename string) (*Collection, error) {
 		c.movieMap[m.IMDBID] = m
 	}
 
+	c.listMap = make(map[string]*List)
+	for _, l := range c.Lists {
+		c.listMap[l.ID] = l
+	}
+
 	return &c, nil
 }
 
@@ -102,3 +111,55 @@ func (c *Collection) Update(m *Movie) {
 
 	c.hasChanged = true
 }
+
+func (c *Collection) GetList(listID string) (*List, bool) {
+	c.mutex.RLock()
+	defer c.mutex.RUnlock()
+
+	l, ok := c.listMap[listID]
+	return l, ok
+}
+
+func (c *Collection) AddList(id, title string) error {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+
+	c.hasChanged = true
+
+	for _, present := range c.Lists {
+		if id == present.ID {
+			return fmt.Errorf("there is already a list with id %q", id)
+		}
+	}
+
+	l := &List{
+		ID:    id,
+		Title: title,
+	}
+
+	c.Lists = append(c.Lists, l)
+	c.listMap[l.ID] = l
+
+	return nil
+}
+
+func (c *Collection) UpdateList(l *List) {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+
+	c.listMap[l.ID] = l
+
+	found := false
+	for i := range c.Lists {
+		if c.Lists[i].ID == l.ID {
+			c.Lists[i] = l
+			found = true
+			break
+		}
+	}
+	if !found {
+		c.Lists = append(c.Lists, l)
+	}
+
+	c.hasChanged = true
+}

list.go

@@ -0,0 +1,19 @@
+package movies
+
+import "encoding/json"
+
+type List struct {
+	ID          string
+	Title       string
+	Description string
+	Movies      []string // IMDBIDs
+}
+
+func UnmarshalList(b []byte) (*List, error) {
+	var l List
+	if err := json.Unmarshal(b, &l); err != nil {
+		return nil, err
+	}
+
+	return &l, nil
+}

pages/pages.go

@@ -14,6 +14,7 @@ import (
 func Router(c *movies.Collection) http.HandlerFunc {
 	homeHandler := Home(c)
 	movieHandler := Movie(c)
+	listHandler := List(c)
 
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		path := r.URL.Path
@@ -24,6 +25,9 @@ func Router(c *movies.Collection) http.HandlerFunc {
 		case path == "/style.css":
 			http.ServeFile(w, r, "templates/style.css")
 			return
+		case strings.HasPrefix(path, "/l/"):
+			listHandler(w, r)
+			return
 		case strings.HasPrefix(path, "/tt"):
 			movieHandler(w, r)
 			return
@@ -42,12 +46,20 @@ func Home(c *movies.Collection) http.HandlerFunc {
 		var errs []error
 
 		if r.Method == "POST" {
-			m, err := movies.Unmarshal([]byte(r.FormValue("omdb_json")))
-			if err != nil {
-				w.WriteHeader(http.StatusBadRequest)
-				errs = append(errs, err)
-			} else {
-				c.Add(m)
+			if r.FormValue("omdb_json") != "" {
+				m, err := movies.Unmarshal([]byte(r.FormValue("omdb_json")))
+				if err != nil {
+					w.WriteHeader(http.StatusBadRequest)
+					errs = append(errs, err)
+				} else {
+					c.Add(m)
+				}
+			}
+
+			if r.FormValue("list-id") != "" {
+				if err := c.AddList(r.FormValue("list-id"), r.FormValue("list-title")); err != nil {
+					errs = append(errs, err)
+				}
 			}
 		}
 
@@ -118,3 +130,52 @@ func Movie(c *movies.Collection) http.HandlerFunc {
 		})
 	})
 }
+
+func List(c *movies.Collection) http.HandlerFunc {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		t, err := template.ParseFiles("templates/list.html")
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+
+		var errs []error
+
+		splitted := strings.Split(r.URL.Path, "/")
+		if len(splitted) < 3 {
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+
+		listID := splitted[2]
+
+		l, ok := c.GetList(listID)
+		if !ok {
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+
+		if r.Method == "POST" {
+			updated, err := movies.UnmarshalList([]byte(r.FormValue("list_json")))
+			if err != nil {
+				errs = append(errs, err)
+			} else if listID != updated.ID {
+				errs = append(errs, fmt.Errorf("you cannot change the imdb id."))
+			} else {
+				l = updated
+				c.UpdateList(updated)
+			}
+		}
+
+		b, err := json.MarshalIndent(l, "", "  ")
+		if err != nil {
+			errs = append(errs, err)
+		}
+
+		t.Execute(w, map[string]interface{}{
+			"List":     l,
+			"ListJSON": string(b),
+			"Errors":   errs,
+		})
+	})
+}

templates/home.html

@@ -33,6 +33,19 @@
             {{ end }}
         </ul>
 
+        <h2>Lists</h2>
+
+        <ul class="list-list">
+            {{ range $l := .Collection.Lists }}
+            <a href="/l/{{ $l.ID }}">
+                <li class="list">
+                    <h3>{{ $l.Title }} ({{ $l.ID }})</h3>
+                    <p>{{ $l.Description }}</p>
+                </li>
+            </a>
+            {{ end }}
+        </ul>
+
         <h2>Add a movie</h2>
         <div>
             <form method="get">
@@ -48,5 +61,16 @@
                 </p>
             </form>
         </div>
+
+        <h2>Add a list</h2>
+        <div>
+            <form method="post" action="/">
+                <div>
+                    ID: <input type="text" name="list-id" />
+                    Title: <input type="text" name="list-title" />
+                    <input type="submit" />
+                </div>
+            </form>
+        </div>
     </body>
 </html>

templates/list.html

@@ -0,0 +1,23 @@
+<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 }}
+
+        <h1>{{ .List.Title }}</h1>
+
+        <p>{{ .List.Description }}</p>
+
+        <form method="post">
+            <p>Update list: <input type="submit" /><br />
+                <textarea name="list_json" style="width:90%;height:500px;">{{ .ListJSON }}</textarea>
+            </p>
+        </form>
+    </body>
+</html>