This repository has been archived on 2025-03-01. You can view files and clone it, but cannot push or open issues or pull requests.
trantor/trantor.go

157 lines
3.9 KiB
Go
Raw Normal View History

2012-07-30 23:23:38 +02:00
package main
import (
"labix.org/v2/mgo/bson"
2012-10-29 21:53:04 +01:00
"log"
2012-07-30 23:23:38 +02:00
"net/http"
"os"
2012-07-30 23:23:38 +02:00
)
2012-08-18 02:06:43 +02:00
type aboutData struct {
S Status
}
2012-08-15 13:17:27 +02:00
func aboutHandler(w http.ResponseWriter, r *http.Request) {
2012-08-18 02:06:43 +02:00
var data aboutData
2012-08-19 02:29:34 +02:00
data.S = GetStatus(w, r)
2012-08-21 10:57:54 +02:00
data.S.About = true
2012-08-18 02:06:43 +02:00
loadTemplate(w, "about", data)
}
func logoutHandler(w http.ResponseWriter, r *http.Request) {
2012-08-19 02:29:34 +02:00
sess := GetSession(r)
sess.LogOut()
sess.Notify("Log out!", "Bye bye "+sess.User, "success")
sess.Save(w, r)
2012-10-29 21:53:04 +01:00
log.Println("User", sess.User, "log out")
http.Redirect(w, r, "/", http.StatusFound)
2012-08-18 02:06:43 +02:00
}
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) {
log.Println("User", user, "log in")
sess.LogIn(user)
sess.Notify("Successful login!", "Welcome "+user, "success")
} else {
log.Println("User", user, "bad user or password")
sess.Notify("Invalid login!", "user or password invalid", "error")
2012-08-18 02:06:43 +02:00
}
sess.Save(w, r)
2012-08-18 02:06:43 +02:00
}
http.Redirect(w, r, r.Referer(), http.StatusFound)
2012-08-18 02:06:43 +02:00
}
type bookData struct {
2012-08-19 02:29:34 +02:00
S Status
Book Book
2012-08-15 13:17:27 +02:00
}
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
2012-08-15 13:58:16 +02:00
}
db.IncVisit(id)
data.Book = books[0]
loadTemplate(w, "book", data)
2012-07-30 23:23:38 +02:00
}
func downloadHandler(w http.ResponseWriter, r *http.Request) {
file := BOOKS_PATH + r.URL.Path[len("/books/"):]
db.IncDownload(file)
http.ServeFile(w, r, file)
}
2012-08-15 14:26:10 +02:00
type indexData struct {
2012-10-26 13:44:08 +02:00
S Status
Books []Book
VisitedBooks []Book
DownloadedBooks []Book
2012-10-26 13:44:08 +02:00
Count int
Tags []string
2012-08-20 17:19:27 +02:00
}
func indexHandler(w http.ResponseWriter, r *http.Request) {
var data indexData
2012-08-20 17:19:27 +02:00
/* 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 */
2012-08-20 17:28:53 +02:00
}
if tag.Subject != "" {
data.Tags[i] = tag.Subject
2012-08-20 17:19:27 +02:00
}
}
2012-08-15 14:26:10 +02:00
}
data.S = GetStatus(w, r)
data.S.Home = true
data.Books, data.Count, _ = db.GetBooks(bson.M{"active": true}, 6)
data.VisitedBooks, _ = db.GetVisitedBooks(6)
data.DownloadedBooks, _ = db.GetDownloadedBooks(6)
loadTemplate(w, "index", data)
2012-08-15 14:26:10 +02:00
}
2012-07-30 23:23:38 +02:00
func main() {
db = initDB()
defer db.Close()
2012-07-30 23:23:38 +02:00
/* 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)
2012-08-18 02:06:43 +02:00
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)
2012-08-15 13:17:27 +02:00
http.HandleFunc("/about/", aboutHandler)
http.HandleFunc("/books/", downloadHandler)
http.HandleFunc("/settings/", settingsHandler)
2012-10-28 19:21:42 +01:00
h := http.FileServer(http.Dir(IMG_PATH))
http.Handle("/img/", http.StripPrefix("/img/", h))
h = http.FileServer(http.Dir(COVER_PATH))
http.Handle("/cover/", http.StripPrefix("/cover/", h))
h = http.FileServer(http.Dir(CSS_PATH))
http.Handle("/css/", http.StripPrefix("/css/", h))
h = http.FileServer(http.Dir(JS_PATH))
http.Handle("/js/", http.StripPrefix("/js/", h))
http.HandleFunc("/", indexHandler)
2012-08-31 01:00:08 +02:00
panic(http.ListenAndServe(":"+PORT, nil))
2012-07-30 23:23:38 +02:00
}