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)) }