110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package server
|
|
|
|
import (
|
|
"gothtest/internal/auth"
|
|
"encoding/json"
|
|
"log"
|
|
"fmt"
|
|
"context"
|
|
"net/http"
|
|
"html/template"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
|
|
"github.com/markbates/goth/gothic"
|
|
)
|
|
|
|
func (s *Server) RegisterRoutes() http.Handler {
|
|
r := chi.NewRouter()
|
|
r.Use(middleware.Logger)
|
|
|
|
r.Get("/hello", s.HelloWorldHandler)
|
|
|
|
r.Get("/auth/{provider}/callback", s.getAuthCallbackFunc)
|
|
r.Get("/logout/{provider}", s.getAuthLogout)
|
|
r.Get("/auth/{provider}", s.getAuthLogin)
|
|
r.Get("/", func (w http.ResponseWriter, r *http.Request) {
|
|
t, _ := template.New("foo").Parse(indexTemplate)
|
|
t.Execute(w, nil)
|
|
})
|
|
|
|
r.Get("/health", s.healthHandler)
|
|
|
|
return r
|
|
}
|
|
|
|
func (s *Server) HelloWorldHandler(w http.ResponseWriter, r *http.Request) {
|
|
resp := make(map[string]string)
|
|
resp["message"] = "Hello World"
|
|
|
|
jsonResp, err := json.Marshal(resp)
|
|
if err != nil {
|
|
log.Fatalf("error handling JSON marshal. Err: %v", err)
|
|
}
|
|
|
|
_, _ = w.Write(jsonResp)
|
|
}
|
|
|
|
func (s *Server) healthHandler(w http.ResponseWriter, r *http.Request) {
|
|
jsonResp, _ := json.Marshal(s.db.Health())
|
|
_, _ = w.Write(jsonResp)
|
|
}
|
|
|
|
func (s *Server) getAuthCallbackFunc(w http.ResponseWriter, r *http.Request) {
|
|
provider := chi.URLParam(r, "provider")
|
|
r = r.WithContext(context.WithValue(context.Background(), "provider", provider))
|
|
|
|
user, err := auth.CompleteAuthFlow(w, r)
|
|
if err != nil {
|
|
fmt.Fprintln(w, err)
|
|
return
|
|
}
|
|
|
|
fmt.Println(user)
|
|
|
|
t, _ := template.New("foo").Parse(userTemplate)
|
|
t.Execute(w, user)
|
|
//http.Redirect(w, r, "http://localhost:3000", http.StatusFound)
|
|
}
|
|
|
|
//p.Get("/logout/{provider}",
|
|
func (s *Server) getAuthLogout(res http.ResponseWriter, req *http.Request) {
|
|
provider := chi.URLParam(req, "provider")
|
|
req = req.WithContext(context.WithValue(context.Background(), "provider", provider))
|
|
gothic.Logout(res, req)
|
|
res.Header().Set("Location", "/")
|
|
res.WriteHeader(http.StatusTemporaryRedirect)
|
|
}
|
|
|
|
//p.Get("/auth/{provider}",
|
|
func (s *Server) getAuthLogin (res http.ResponseWriter, req *http.Request) {
|
|
provider := chi.URLParam(req, "provider")
|
|
req = req.WithContext(context.WithValue(context.Background(), "provider", provider))
|
|
// try to get the user without re-authenticating
|
|
if gothUser, err := gothic.CompleteUserAuth(res, req); err == nil {
|
|
t, _ := template.New("foo").Parse(userTemplate)
|
|
t.Execute(res, gothUser)
|
|
} else {
|
|
gothic.BeginAuthHandler(res, req)
|
|
}
|
|
}
|
|
//)
|
|
var indexTemplate = `
|
|
<p><a href="/auth/openid-connect">Log in with openid connect</a></p>
|
|
`
|
|
|
|
var userTemplate = `
|
|
<p><a href="/logout/{{.Provider}}">logout</a></p>
|
|
<p>Name: {{.Name}} [{{.LastName}}, {{.FirstName}}]</p>
|
|
<p>Email: {{.Email}}</p>
|
|
<p>NickName: {{.NickName}}</p>
|
|
<p>Location: {{.Location}}</p>
|
|
<p>AvatarURL: {{.AvatarURL}} <img src="{{.AvatarURL}}"></p>
|
|
<p>Description: {{.Description}}</p>
|
|
<p>UserID: {{.UserID}}</p>
|
|
<p>AccessToken: {{.AccessToken}}</p>
|
|
<p>ExpiresAt: {{.ExpiresAt}}</p>
|
|
<p>RefreshToken: {{.RefreshToken}}</p>
|
|
`
|