package main

import (
	"flag"
	"log"
	"net/http"
	"os"
)

var mux = http.NewServeMux()

var ThumbsDir string

func main() {
	bind := flag.String("bind", ":8080", "Bind port/socket")
	htpasswd_file := flag.String("htpasswd", "", "Admin passwords file, Apache htpasswd format")
	publishedImgDir := flag.String("publishedimgdir", "published/", "Directory where save published pictures")
	nextImgDir := flag.String("nextimgdir", "next/", "Directory where save pictures to review")
	flag.StringVar(&ThumbsDir, "thumbsdir", "./images/thumbs/", "Directory where generate thumbs")
	flag.Parse()

	htpasswd := &Htpasswd{}

	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 if *nextImgDir == "./images/next/" {
		log.Println("Disable admin interface, images will be published without moderation")
		nextImgDir = publishedImgDir
	}

	log.Println("Checking paths...")
	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) }

	if err := sanitizeStaticOptions(); err != nil {
		log.Fatal(err)
	}

	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		r.URL.Path = "/"
		http.FileServer(Assets).ServeHTTP(w, r)
	})

	pe := &PictureExplorer{
		FileBackend:     &LocalFileBackend{"./images/"},
		PublishedImgDir: *publishedImgDir,
		NextImgDir:      *nextImgDir,
	}

	mux.Handle("/api/", http.StripPrefix("/api", NewAPIHandler(pe, authFunc)))

	mux.Handle("/images/", http.StripPrefix("/images", ImagesHandler(
		"images",
		func(*http.Request) bool { return true },
		pe.ServeFile(),
		pe.GetPublishedImage,
	)))
	mux.Handle("/images/next/", http.StripPrefix("/images/next", ImagesHandler(
		"next",
		func(r *http.Request) bool { return authFunc(r) != nil },
		pe.ServeFile(),
		pe.GetNextImage,
	)))

	mux.Handle("/images/thumbs/", http.StripPrefix("/images/thumbs", ImagesHandler(
		"thumbs",
		func(*http.Request) bool { return true },
		http.FileServer(http.Dir(ThumbsDir)),
		pe.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)),
		pe.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 {
			r.URL.Path = "/admin.html"
			http.FileServer(Assets).ServeHTTP(w, r)
		}
	})

	log.Println("Ready, listening on", *bind)
	if err := http.ListenAndServe(*bind, mux); err != nil {
		log.Fatal("Unable to listen and serve: ", err)
	}
}