From f3882881ce840533b74fa9b2acda4530a91cf525 Mon Sep 17 00:00:00 2001
From: Las Zenow <zenow@tormail.org>
Date: Thu, 13 Sep 2012 00:05:21 +0200
Subject: [PATCH] Store or remove all the books at once

---
 admin.go           | 92 ++++++++++++++++++++++++++++------------------
 templates/new.html | 14 ++++++-
 2 files changed, 70 insertions(+), 36 deletions(-)

diff --git a/admin.go b/admin.go
index fa8bf05..fc675b5 100644
--- a/admin.go
+++ b/admin.go
@@ -16,28 +16,41 @@ func deleteHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	// cutre hack: /delete/ and /delnew/ have the same lenght:
-	id := bson.ObjectIdHex(r.URL.Path[len("/delete/"):])
-	books, _, err := db.GetBooks(bson.M{"_id": id})
-	if err != nil {
-		http.NotFound(w, r)
-		return
+	var titles []string
+	var isNew bool
+	ids:= strings.Split(r.URL.Path[len("/delete/"):], "/")
+	for _, idStr := range ids {
+		if idStr == "" {
+			continue
+		}
+
+		id := bson.ObjectIdHex(idStr)
+		books, _, err := db.GetBooks(bson.M{"_id": id})
+		if err != nil {
+			sess.Notify("Book not found!", "The book with id '"+idStr+"' is not there", "error")
+			return
+		}
+		book := books[0]
+		if book.Cover != "" {
+			os.RemoveAll(book.Cover[1:])
+		}
+		if book.CoverSmall != "" {
+			os.RemoveAll(book.CoverSmall[1:])
+		}
+		os.RemoveAll(book.Path)
+		db.RemoveBook(id)
+
+		if ! book.Active {
+			isNew = true
+		}
+		titles = append(titles, book.Title)
 	}
-	book := books[0]
-	if book.Cover != "" {
-		os.RemoveAll(book.Cover[1:])
-	}
-	if book.CoverSmall != "" {
-		os.RemoveAll(book.CoverSmall[1:])
-	}
-	os.RemoveAll(book.Path)
-	db.RemoveBook(id)
-	sess.Notify("Removed book!", "The book '"+book.Title+"' it's completly removed", "success")
+	sess.Notify("Removed books!", "The books "+ strings.Join(titles, ", ") +" are completly removed", "success")
 	sess.Save(w, r)
-	if book.Active {
-		http.Redirect(w, r, "/", 307)
-	} else {
+	if isNew {
 		http.Redirect(w, r, "/new/", 307)
+	} else {
+		http.Redirect(w, r, "/", 307)
 	}
 }
 
@@ -168,23 +181,32 @@ func storeHandler(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	id := bson.ObjectIdHex(r.URL.Path[len("/store/"):])
-	books, _, err := db.GetBooks(bson.M{"_id": id})
-	if err != nil {
-		http.NotFound(w, r)
-		return
+	var titles []string
+	ids := strings.Split(r.URL.Path[len("/store/"):], "/")
+	for _, idStr := range ids {
+		if idStr == "" {
+			continue
+		}
+
+		id := bson.ObjectIdHex(idStr)
+		books, _, err := db.GetBooks(bson.M{"_id": id})
+		if err != nil {
+			sess.Notify("Book not found!", "The book with id '"+idStr+"' is not there", "error")
+			return
+		}
+		book := books[0]
+
+		title := book.Title
+		path := ValidFileName(BOOKS_PATH + title[:1], title, ".epub")
+
+		oldPath := book.Path
+		os.Mkdir(BOOKS_PATH+title[:1], os.ModePerm)
+		cmd := exec.Command("mv", oldPath, path)
+		cmd.Run()
+		db.UpdateBook(id, bson.M{"active": true, "path": path})
+		titles = append(titles, book.Title)
 	}
-	book := books[0]
-
-	title := book.Title
-	path := ValidFileName(BOOKS_PATH + title[:1], title, ".epub")
-
-	oldPath := book.Path
-	os.Mkdir(BOOKS_PATH+title[:1], os.ModePerm)
-	cmd := exec.Command("mv", oldPath, path)
-	cmd.Run()
-	db.UpdateBook(id, bson.M{"active": true, "path": path})
-	sess.Notify("Store book!", "The book '"+title+"' it's stored for public download", "success")
+	sess.Notify("Store books!", "The books '"+ strings.Join(titles, ", ") +"' are stored for public download", "success")
 	sess.Save(w, r)
 	http.Redirect(w, r, "/new/", 307)
 }
diff --git a/templates/new.html b/templates/new.html
index 301e875..e2714d8 100644
--- a/templates/new.html
+++ b/templates/new.html
@@ -1,6 +1,12 @@
 {{template "header.html" .S}}
 
-<p class="centered">Found {{.Found}} books.</p>
+{{if .Books}}
+<div class="centered btn-group">
+	<a href="/store/{{range .Books}}{{.B.Id}}/{{end}}" class="btn btn-large btn-success"><i class="icon-ok"></i> Save All</a>
+	<a href="/delete/{{range .Books}}{{.B.Id}}/{{end}}" class="btn btn-large btn-danger"><i class="icon-remove"></i> Delete All</a>
+</div>
+{{end}}
+<p class="centered">Found {{.Found}} books.<br /></p>
 
 {{range .Books}}
 	{{$titleFound  := .TitleFound}}
@@ -35,5 +41,11 @@
 	</div>
 	{{end}}
 {{end}}
+{{if .Books}}
+<div class="centered btn-group">
+	<a href="/store/{{range .Books}}{{.B.Id}}/{{end}}" class="btn btn-large btn-success"><i class="icon-ok"></i> Save All</a>
+	<a href="/delete/{{range .Books}}{{.B.Id}}/{{end}}" class="btn btn-large btn-danger"><i class="icon-remove"></i> Delete All</a>
+</div>
+{{end}}
 
 {{template "footer.html"}}