Initial commit

1 files changed, 46 insertions, 0 deletions

diff --git a/main.go b/main.go
new file mode 100644
index 0000000..7d172b7
--- /dev/null
+++ b/main.go
@@ -0,0 +1,46 @@
+package main
+
+import (
+	"log"
+	"net/http"
+	"time"
+)
+
+type App struct {
+	SessionStore
+}
+
+func logMux(handler http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		t := time.Now()
+		handler.ServeHTTP(w, r)
+		log.Println(r.Method, r.URL.Path, time.Since(t))
+	})
+}
+
+func (app *App) rootHandler(w http.ResponseWriter, r *http.Request) {
+	c, err := r.Cookie("id")
+	//log.Println(r.Header.Get("X-Original-URI"))
+	//log.Println(r.Host)
+	if err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+	} else {
+		if s, ok := app.Get(c.Value); ok {
+			w.Header().Set("X-Remote-User", s.UserId)
+			w.WriteHeader(http.StatusOK)
+		} else {
+			log.Println("Session does not exist:", c.Value)
+			w.WriteHeader(http.StatusForbidden)
+		}
+	}
+}
+
+func main() {
+	//log.SetFlags(log.LstdFlags)
+	store := NewPgStore()
+	app := &App{store}
+	http.HandleFunc("/", app.rootHandler)
+	if err := http.ListenAndServe(":8080", logMux(http.DefaultServeMux)); err != nil {
+		panic(err)
+	}
+}