112 lines
3.5 KiB
Go
112 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
var PublishedImgDir string
|
|
var NextImgDir string
|
|
var ThumbsDir string
|
|
|
|
func main() {
|
|
bind := flag.String("bind", ":8080", "Bind port/socket")
|
|
htpasswd_file := flag.String("htpasswd", "", "Admin passwords file, Apache htpasswd format")
|
|
flag.StringVar(&PublishedImgDir, "publishedimgdir", "./images/published/", "Directory where save published pictures")
|
|
flag.StringVar(&NextImgDir, "nextimgdir", "./images/next/", "Directory where save pictures to review")
|
|
flag.StringVar(&ThumbsDir, "thumbsdir", "./images/thumbs/", "Directory where generate thumbs")
|
|
flag.Parse()
|
|
|
|
htpasswd := &Htpasswd{
|
|
entries: map[string]string{"admin": "2fClb0C8dIphk"},
|
|
}
|
|
|
|
if htpasswd_file != nil && *htpasswd_file != "" {
|
|
log.Println("Reading htpasswd file...")
|
|
var err error
|
|
if htpasswd, err = NewHtpasswd(*htpasswd_file); htpasswd == nil {
|
|
log.Fatal("Unable to parse htpasswd:", err)
|
|
}
|
|
} else {
|
|
log.Println("Using default credentials for administrative part: admin:admin")
|
|
}
|
|
|
|
log.Println("Checking paths...")
|
|
if _, err := os.Stat(PublishedImgDir); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(PublishedImgDir, 0755); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
if _, err := os.Stat(NextImgDir); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(NextImgDir, 0755); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
if _, err := os.Stat(ThumbsDir); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(ThumbsDir, 0755); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
log.Println("Registering handlers...")
|
|
authFunc := func (r *http.Request) (*User){ return Authenticate(*htpasswd, r) }
|
|
|
|
staticDir, _ := filepath.Abs("static")
|
|
if _, err := os.Stat(staticDir); os.IsNotExist(err) {
|
|
staticDir, _ = filepath.Abs(filepath.Join(filepath.Dir(os.Args[0]), "static"))
|
|
if _, err := os.Stat(staticDir); os.IsNotExist(err) {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { http.ServeFile(w, r, filepath.Join(staticDir, "index.html")) })
|
|
mux.Handle("/css/", http.FileServer(http.Dir(staticDir)))
|
|
mux.Handle("/js/", http.FileServer(http.Dir(staticDir)))
|
|
|
|
mux.Handle("/api/", http.StripPrefix("/api", ApiHandler(authFunc)))
|
|
|
|
mux.Handle("/images/", http.StripPrefix("/images", ImagesHandler(
|
|
"images",
|
|
func (*http.Request) (bool){ return true; },
|
|
http.FileServer(http.Dir(PublishedImgDir)),
|
|
GetPublishedImage,
|
|
)))
|
|
mux.Handle("/images/next/", http.StripPrefix("/images/next", ImagesHandler(
|
|
"next",
|
|
func (r *http.Request) (bool){ return authFunc(r) != nil; },
|
|
http.FileServer(http.Dir(NextImgDir)),
|
|
GetNextImage,
|
|
)))
|
|
|
|
mux.Handle("/images/thumbs/", http.StripPrefix("/images/thumbs", ImagesHandler(
|
|
"thumbs",
|
|
func (*http.Request) (bool){ return true; },
|
|
http.FileServer(http.Dir(ThumbsDir)),
|
|
GetPublishedImage,
|
|
)))
|
|
mux.Handle("/images/next/thumbs/", http.StripPrefix("/images/next/thumbs", ImagesHandler(
|
|
"nexthumbs",
|
|
func (r *http.Request) (bool){ return authFunc(r) != nil; },
|
|
http.FileServer(http.Dir(ThumbsDir)),
|
|
GetNextImage,
|
|
)))
|
|
|
|
mux.HandleFunc("/admin/", func(w http.ResponseWriter, r *http.Request) {
|
|
if authFunc(r) == nil {
|
|
w.Header().Set("WWW-Authenticate", "Basic realm=\"YouP0m\"")
|
|
http.Error(w, "You are not allowed to perform this request.", http.StatusUnauthorized)
|
|
} else {
|
|
http.ServeFile(w, r, filepath.Join(staticDir, "admin.html"))
|
|
}
|
|
})
|
|
|
|
log.Println("Ready, listening on", *bind)
|
|
if err := http.ListenAndServe(*bind, mux); err != nil {
|
|
log.Fatal("Unable to listen and serve: ", err)
|
|
}
|
|
}
|