diff --git a/trantor.go b/trantor.go index 3a2da73..8bb05cf 100644 --- a/trantor.go +++ b/trantor.go @@ -5,6 +5,7 @@ import ( "labix.org/v2/mgo" "labix.org/v2/mgo/bson" "net/http" + "os" ) type aboutData struct { @@ -118,6 +119,19 @@ func main() { userColl := session.DB(DB_NAME).C(USERS_COLL) newColl := session.DB(DB_NAME).C(NEW_BOOKS_COLL) + _, 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) + } + http.HandleFunc("/book/", bookHandler(coll)) http.HandleFunc("/search/", searchHandler(coll)) http.HandleFunc("/upload/", uploadHandler(newColl))