840fd3ed47
The collection new is removed, now the books have a field "active" set to true
if the book was aproved.
For update the database:
db.books.update({}, {$set: {active: true}}, true, true)
141 lines
3.1 KiB
Go
141 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"labix.org/v2/mgo/bson"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
type aboutData struct {
|
|
S Status
|
|
}
|
|
|
|
func aboutHandler(w http.ResponseWriter, r *http.Request) {
|
|
var data aboutData
|
|
data.S = GetStatus(w, r)
|
|
data.S.About = true
|
|
loadTemplate(w, "about", data)
|
|
}
|
|
|
|
func logoutHandler(w http.ResponseWriter, r *http.Request) {
|
|
sess := GetSession(r)
|
|
sess.LogOut()
|
|
sess.Notify("Log out!", "Bye bye "+sess.User, "success")
|
|
sess.Save(w, r)
|
|
http.Redirect(w, r, "/", 307)
|
|
}
|
|
|
|
func loginHandler(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "POST" {
|
|
user := r.FormValue("user")
|
|
pass := r.FormValue("pass")
|
|
sess := GetSession(r)
|
|
if db.UserValid(user, pass) {
|
|
sess.LogIn(user)
|
|
sess.Notify("Successful login!", "Welcome "+user, "success")
|
|
} else {
|
|
sess.Notify("Invalid login!", "user or password invalid", "error")
|
|
}
|
|
sess.Save(w, r)
|
|
}
|
|
http.Redirect(w, r, r.Referer(), 307)
|
|
}
|
|
|
|
type bookData struct {
|
|
S Status
|
|
Book Book
|
|
}
|
|
|
|
func bookHandler(w http.ResponseWriter, r *http.Request) {
|
|
var data bookData
|
|
data.S = GetStatus(w, r)
|
|
id := bson.ObjectIdHex(r.URL.Path[len("/book/"):])
|
|
books, _, err := db.GetBooks(bson.M{"_id": id})
|
|
if err != nil || len(books) == 0 {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
data.Book = books[0]
|
|
loadTemplate(w, "book", data)
|
|
}
|
|
|
|
func fileHandler(path string) {
|
|
h := http.FileServer(http.Dir(path[1:]))
|
|
http.Handle(path, http.StripPrefix(path, h))
|
|
}
|
|
|
|
type indexData struct {
|
|
S Status
|
|
Books []Book
|
|
Count int
|
|
Tags []string
|
|
}
|
|
|
|
func indexHandler(w http.ResponseWriter, r *http.Request) {
|
|
var data indexData
|
|
|
|
/* get the tags */
|
|
tags, err := db.GetTags()
|
|
if err == nil {
|
|
length := len(tags)
|
|
if length > TAGS_DISPLAY {
|
|
length = TAGS_DISPLAY
|
|
}
|
|
data.Tags = make([]string, length)
|
|
for i, tag := range tags {
|
|
if i == TAGS_DISPLAY {
|
|
break /* display only 50 */
|
|
}
|
|
if tag.Subject != "" {
|
|
data.Tags[i] = tag.Subject
|
|
}
|
|
}
|
|
}
|
|
|
|
data.S = GetStatus(w, r)
|
|
data.S.Home = true
|
|
data.Books, data.Count, _ = db.GetBooks(bson.M{"active": true}, 6)
|
|
loadTemplate(w, "index", data)
|
|
}
|
|
|
|
func main() {
|
|
db = initDB()
|
|
defer db.Close()
|
|
|
|
/* create the needed folders */
|
|
var err error
|
|
_, err = os.Stat(BOOKS_PATH)
|
|
if err != nil {
|
|
os.Mkdir(BOOKS_PATH, os.ModePerm)
|
|
}
|
|
_, err = os.Stat(COVER_PATH)
|
|
if err != nil {
|
|
os.Mkdir(COVER_PATH, os.ModePerm)
|
|
}
|
|
_, err = os.Stat(NEW_PATH)
|
|
if err != nil {
|
|
os.Mkdir(NEW_PATH, os.ModePerm)
|
|
}
|
|
|
|
/* set up web handlers */
|
|
http.HandleFunc("/book/", bookHandler)
|
|
http.HandleFunc("/search/", searchHandler)
|
|
http.HandleFunc("/upload/", uploadHandler)
|
|
http.HandleFunc("/login/", loginHandler)
|
|
http.HandleFunc("/logout/", logoutHandler)
|
|
http.HandleFunc("/new/", newHandler)
|
|
http.HandleFunc("/store/", storeHandler)
|
|
http.HandleFunc("/read/", readHandler)
|
|
http.HandleFunc("/content/", contentHandler)
|
|
http.HandleFunc("/edit/", editHandler)
|
|
http.HandleFunc("/save/", saveHandler)
|
|
http.HandleFunc("/delete/", deleteHandler)
|
|
http.HandleFunc("/about/", aboutHandler)
|
|
fileHandler("/img/")
|
|
fileHandler("/cover/")
|
|
fileHandler("/books/")
|
|
fileHandler("/css/")
|
|
fileHandler("/js/")
|
|
http.HandleFunc("/", indexHandler)
|
|
panic(http.ListenAndServe(":"+PORT, nil))
|
|
}
|