38 lines
619 B
Go
38 lines
619 B
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"github.com/satori/go.uuid"
|
||
|
"net/http"
|
||
|
)
|
||
|
|
||
|
func getUser(w http.ResponseWriter, req *http.Request) user {
|
||
|
// get cookie
|
||
|
c, err := req.Cookie("session")
|
||
|
if err != nil {
|
||
|
sID, _ := uuid.NewV4()
|
||
|
c = &http.Cookie{
|
||
|
Name: "session",
|
||
|
Value: sID.String(),
|
||
|
}
|
||
|
|
||
|
}
|
||
|
http.SetCookie(w, c)
|
||
|
|
||
|
// if the user exists already, get user
|
||
|
var u user
|
||
|
if un, ok := dbSessions[c.Value]; ok {
|
||
|
u = dbUsers[un]
|
||
|
}
|
||
|
return u
|
||
|
}
|
||
|
|
||
|
func alreadyLoggedIn(req *http.Request) bool {
|
||
|
c, err := req.Cookie("session")
|
||
|
if err != nil {
|
||
|
return false
|
||
|
}
|
||
|
un := dbSessions[c.Value]
|
||
|
_, ok := dbUsers[un]
|
||
|
return ok
|
||
|
}
|