go-sample-webpage/pkg/user/user.go

104 lines
2.2 KiB
Go
Raw Normal View History

2021-11-04 01:14:51 +00:00
package user
import (
"embed"
"fmt"
"github.com/go-chi/chi/v5"
"github.com/go-chi/jwtauth/v5"
"github.com/unrolled/render"
"net/http"
"time"
)
// content holds our static web server content.
//go:embed templates/*
var webserver embed.FS
var ren *render.Render
var tokenAuth *jwtauth.JWTAuth
2021-11-08 01:25:36 +00:00
var uc *UserClient
func Register(router chi.Router, token *jwtauth.JWTAuth, userClient UserClient) {
2021-11-04 01:14:51 +00:00
ren = render.New(render.Options{
//Layout: "layout",
FileSystem: &render.EmbedFileSystem{
FS: webserver,
},
})
router.Get("/login", loginForm)
router.Post("/login", login)
router.Get("/register", loginForm)
router.Post("/register", register)
2021-11-04 02:04:05 +00:00
router.Get("/logout", logout)
2021-11-04 01:14:51 +00:00
2021-11-08 01:25:36 +00:00
//tokenAuth = jwtauth.New("HS256", []byte("secret"), nil)
tokenAuth = token
uc = &userClient
2021-11-04 01:14:51 +00:00
router.Group(func(r chi.Router) {
r.Use(jwtauth.Verifier(tokenAuth))
r.Get("/me", func(w http.ResponseWriter, r *http.Request) {
_, claims, _ := jwtauth.FromContext(r.Context())
if val, ok := claims["username"]; ok {
w.Write([]byte(fmt.Sprintf("hi %v", val)))
} else {
w.Write([]byte("Du bist nicht eingeloggt"))
}
})
})
}
func loginForm(w http.ResponseWriter, r *http.Request) {
ren.HTML(w, http.StatusOK, "login", nil)
}
func login(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
uc := GetUserClient()
res, _ := uc.login(r.FormValue("username"), r.FormValue("password"))
if res {
_, tokenstring, err := tokenAuth.Encode(map[string]interface{}{"username": r.FormValue("username")})
if err != nil {
panic(err)
}
expiration := time.Now().Add(365 * 24 * time.Hour)
cookie := http.Cookie{Name: "jwt",Value:tokenstring,Expires:expiration}
http.SetCookie(w, &cookie)
w.Write([]byte("Login ok"))
} else {
w.Write([]byte("Login failed"))
}
}
2021-11-04 02:04:05 +00:00
func logout(w http.ResponseWriter, r *http.Request) {
cookie := http.Cookie{
Name: "jwt",
Value: "",
}
http.SetCookie(w, &cookie)
w.Write([]byte("Du wurdest ausgeloggt"))
}
2021-11-04 01:14:51 +00:00
func register(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
uc := GetUserClient()
res, _ := uc.register(r.FormValue("username"), r.FormValue("password"))
if res {
w.Write([]byte("Register ok"))
} else {
w.Write([]byte("Register failed"))
}
}