167 lines
3.9 KiB
Go
167 lines
3.9 KiB
Go
package main
|
|
|
|
import (
|
|
"labix.org/v2/mgo"
|
|
"labix.org/v2/mgo/bson"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
)
|
|
|
|
func deleteHandler(coll *mgo.Collection, url string) func(http.ResponseWriter, *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
sess := GetSession(r)
|
|
if sess.User == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
// cutre hack: /delete/ and /delnew/ have the same lenght:
|
|
id := bson.ObjectIdHex(r.URL.Path[len("/delete/"):])
|
|
books, _, err := GetBook(coll, bson.M{"_id": id})
|
|
if err != nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
book := books[0]
|
|
if book.Cover != "" {
|
|
os.RemoveAll(book.Cover[1:])
|
|
}
|
|
if book.CoverSmall != "" {
|
|
os.RemoveAll(book.CoverSmall[1:])
|
|
}
|
|
os.RemoveAll(book.Path)
|
|
coll.Remove(bson.M{"_id": id})
|
|
sess.Notify("Removed book!", "The book '"+book.Title+"' it's completly removed", "success")
|
|
sess.Save(w, r)
|
|
http.Redirect(w, r, url, 307)
|
|
}
|
|
}
|
|
|
|
func editHandler(coll *mgo.Collection) func(http.ResponseWriter, *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
sess := GetSession(r)
|
|
if sess.User == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
id := bson.ObjectIdHex(r.URL.Path[len("/edit/"):])
|
|
books, _, err := GetBook(coll, bson.M{"_id": id})
|
|
if err != nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
var data bookData
|
|
data.Book = books[0]
|
|
data.S = GetStatus(w, r)
|
|
loadTemplate(w, "edit", data)
|
|
}
|
|
}
|
|
|
|
func cleanEmptyStr(s []string) []string {
|
|
var res []string
|
|
for _, v := range s {
|
|
if v != "" {
|
|
res = append(res, v)
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
func saveHandler(coll *mgo.Collection) func(http.ResponseWriter, *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != "POST" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
sess := GetSession(r)
|
|
if sess.User == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
id := bson.ObjectIdHex(r.URL.Path[len("/save/"):])
|
|
title := r.FormValue("title")
|
|
publisher := r.FormValue("publisher")
|
|
date := r.FormValue("date")
|
|
description := r.FormValue("description")
|
|
author := cleanEmptyStr(r.Form["author"])
|
|
subject := cleanEmptyStr(r.Form["subject"])
|
|
lang := cleanEmptyStr(r.Form["lang"])
|
|
err := coll.Update(bson.M{"_id": id}, bson.M{"$set": bson.M{"title": title,
|
|
"publisher": publisher,
|
|
"date": date,
|
|
"description": description,
|
|
"author": author,
|
|
"subject": subject,
|
|
"lang": lang}})
|
|
if err != nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
sess.Notify("Book Modified!", "", "success")
|
|
sess.Save(w, r)
|
|
http.Redirect(w, r, "/book/"+title, 307)
|
|
}
|
|
}
|
|
|
|
type newData struct {
|
|
S Status
|
|
Found int
|
|
Books []Book
|
|
}
|
|
|
|
func newHandler(coll *mgo.Collection) func(http.ResponseWriter, *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
sess := GetSession(r)
|
|
if sess.User == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
res, num, _ := GetBook(coll, bson.M{})
|
|
var data newData
|
|
data.S = GetStatus(w, r)
|
|
data.Found = num
|
|
data.Books = res
|
|
loadTemplate(w, "new", data)
|
|
}
|
|
}
|
|
|
|
func storeHandler(newColl, coll *mgo.Collection) func(http.ResponseWriter, *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
sess := GetSession(r)
|
|
if sess.User == "" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
|
|
id := bson.ObjectIdHex(r.URL.Path[len("/store/"):])
|
|
books, _, err := GetBook(newColl, bson.M{"_id": id})
|
|
if err != nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
book := books[0]
|
|
|
|
path := BOOKS_PATH + book.Title[:1] + "/" + book.Title + ".epub"
|
|
_, err = os.Stat(path)
|
|
for i := 0; err == nil; i++ {
|
|
path := BOOKS_PATH + book.Title[:1] + "/" + book.Title + "_" + strconv.Itoa(i) + ".epub"
|
|
_, err = os.Stat(path)
|
|
}
|
|
|
|
os.Mkdir(BOOKS_PATH+book.Title[:1], os.ModePerm)
|
|
cmd := exec.Command("mv", book.Path, path)
|
|
cmd.Run()
|
|
book.Path = path
|
|
coll.Insert(book)
|
|
newColl.Remove(bson.M{"_id": id})
|
|
sess.Notify("Store book!", "The book '"+book.Title+"' it's stored for public download", "success")
|
|
sess.Save(w, r)
|
|
http.Redirect(w, r, "/new/", 307)
|
|
}
|
|
}
|