diff options
| -rw-r--r-- | main.go | 36 |
1 files changed, 27 insertions, 9 deletions
@@ -4,10 +4,13 @@ import ( "crypto/md5" "crypto/subtle" "encoding/hex" + "flag" + "fmt" "html/template" "log" "net/http" "os" + "path/filepath" "strconv" "time" ) @@ -15,6 +18,7 @@ import ( type App struct { Store Template *template.Template + Domain string } func logMux(handler http.Handler) http.Handler { @@ -53,7 +57,7 @@ func (app *App) loginHandler(w http.ResponseWriter, r *http.Request) { u, ok := app.GetUser(username) if ok && subtle.ConstantTimeCompare(u.Password, dst) == 1 { s := app.NewSession(u.Id) - c := http.Cookie{Name: "id", Value: s.Id, Domain: ".horel.test"} + c := http.Cookie{Name: "id", Value: s.Id, Domain: "." + app.Domain} http.SetCookie(w, &c) http.Redirect(w, r, next, http.StatusSeeOther) } else { @@ -70,23 +74,37 @@ func (app *App) loginHandler(w http.ResponseWriter, r *http.Request) { } func (app *App) logoutHandler(w http.ResponseWriter, r *http.Request) { - c := http.Cookie{Name: "id", Value: "", Domain: ".horel.test", MaxAge: 0} + c := http.Cookie{Name: "id", Value: "", Domain: "." + app.Domain, MaxAge: 0} http.SetCookie(w, &c) http.Redirect(w, r, "/login", http.StatusSeeOther) } func main() { - //log.SetFlags(log.LstdFlags) - if len(os.Args) == 1 { - log.Fatalf("Usage: %s <connect-string>\n", os.Args[0]) + + flag.Usage = func() { + fmt.Fprintf( + flag.CommandLine.Output(), + "Usage: %s [OPTIONS] [DATABASE] [DOMAIN]\n", + os.Args[0], + ) + flag.PrintDefaults() + } + address := flag.String("a", ":8080", "bind the server to `address`") + templateDir := flag.String("t", "templates", "template `directory`") + flag.Parse() + if flag.NArg() < 2 { + fmt.Print("Argument missing. ") + flag.Usage() + os.Exit(1) } - store := NewPgStore(os.Args[1]) - template := template.Must(template.New("").ParseGlob("templates/*.tmpl")) - app := &App{store, template} + + store := NewPgStore(flag.Args()[0]) + template := template.Must(template.New("").ParseGlob(filepath.Join(*templateDir, "*.tmpl"))) + app := &App{store, template, flag.Args()[1]} http.HandleFunc("/validate", app.validateHandler) http.HandleFunc("/login", app.loginHandler) http.HandleFunc("/logout", app.logoutHandler) - if err := http.ListenAndServe(":8080", logMux(http.DefaultServeMux)); err != nil { + if err := http.ListenAndServe(*address, logMux(http.DefaultServeMux)); err != nil { panic(err) } } |