archives/trantor
Archived
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add a 404 page

Browse source
This commit is contained in:
Las Zenow 2013-05-09 09:42:03 +02:00
parent 137357cd12
commit d0f6b83423
6 changed files with 45 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

16
reader.go
View file

@ -131,14 +131,14 @@ func readStartHandler(w http.ResponseWriter, r *http.Request, sess *Session) {
id := mux.Vars(r)["id"]
e, _ := openReadEpub(w, r, sess)
if e == nil {
http.NotFound(w, r)
notFound(w)
return
}
defer e.Close()
it, err := e.Spine()
if err != nil {
http.NotFound(w, r)
notFound(w)
return
}
http.Redirect(w, r, "/read/"+id+"/"+it.Url(), http.StatusTemporaryRedirect)
@ -149,7 +149,7 @@ func readHandler(w http.ResponseWriter, r *http.Request, sess *Session) {
file := mux.Vars(r)["file"]
e, book := openReadEpub(w, r, sess)
if e == nil {
http.NotFound(w, r)
notFound(w)
return
}
defer e.Close()
@ -195,32 +195,32 @@ func contentHandler(w http.ResponseWriter, r *http.Request, sess *Session) {
id := vars["id"]
file := vars["file"]
if file == "" {
http.NotFound(w, r)
notFound(w)
return
}
books, _, err := db.GetBooks(bson.M{"_id": bson.ObjectIdHex(id)})
if err != nil || len(books) == 0 {
http.NotFound(w, r)
notFound(w)
return
}
book := books[0]
if !book.Active {
if sess.User == "" {
http.NotFound(w, r)
notFound(w)
return
}
}
e, err := OpenBook(book.File)
if err != nil {
http.NotFound(w, r)
notFound(w)
return
}
defer e.Close()
html, err := e.OpenFile(file)
if err != nil {
http.NotFound(w, r)
notFound(w)
return
}
defer html.Close()