towk/opaal
1
0
Fork 0
mirror of https://github.com/davidallendj/opaal.git synced 2025-12-20 03:27:02 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Removed unused code, wrapped verbose prints, and added error page template

Browse source
This commit is contained in:
David Allen 2024-03-13 00:40:02 -06:00
parent 35cf37836d
commit 56ea3290a6
No known key found for this signature in database
GPG key ID: 1D2A29322FBB6FCB
1 changed files with 23 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

39
internal/server/server.go
View file

@ -7,7 +7,6 @@ import (
"encoding/json"
"fmt"
"net/http"
"os"
"github.com/davidallendj/go-utils/httpx"
"github.com/go-chi/chi/v5"
@ -71,7 +70,9 @@ func (s *Server) Login(buttons string, provider *oidc.IdentityProvider, client *
// get the code from the OIDC provider
if r != nil {
code = r.URL.Query().Get("code")
if params.Verbose {
fmt.Printf("Authorization code: %v\n", code)
}
// use code from response and exchange for bearer token (with ID token)
bearerToken, err := client.FetchTokenFromAuthenticationServer(
@ -120,7 +121,9 @@ func (s *Server) Login(buttons string, provider *oidc.IdentityProvider, client *
http.Redirect(w, r, "/success", http.StatusSeeOther)
})
r.HandleFunc("/success", func(w http.ResponseWriter, r *http.Request) {
if params.Verbose {
fmt.Printf("Serving success page.\n")
}
template, err := gonja.FromFile("pages/success.html")
if err != nil {
panic(err)
@ -135,7 +138,9 @@ func (s *Server) Login(buttons string, provider *oidc.IdentityProvider, client *
}
// try and send access code to target if set
if target != "" {
fmt.Printf("Send access token to target: %s\n", target)
if params.Verbose {
fmt.Printf("Sending access token to target: %s\n", target)
}
_, _, err := httpx.MakeHttpRequest(target, http.MethodPost, nil, httpx.Headers{"access_token": accessToken})
if err != nil {
fmt.Printf("failed to make request: %v", err)
@ -143,30 +148,22 @@ func (s *Server) Login(buttons string, provider *oidc.IdentityProvider, client *
}
})
r.HandleFunc("/error", func(w http.ResponseWriter, r *http.Request) {
if params.Verbose {
fmt.Printf("Serving error page.")
errorPage, err := os.ReadFile("pages/error.html")
}
template, err := gonja.FromFile("pages/success.html")
if err != nil {
fmt.Printf("failed to load error page: %v\n", err)
panic(err)
}
w.Write(errorPage)
data := exec.NewContext(map[string]interface{}{
"index": fmt.Sprintf("<a href=\"%s\">try logging in again?</a>", s.Addr),
})
s.Handler = r
return s.ListenAndServe()
}
func (s *Server) Serve(data chan []byte) error {
output, ok := <-data
if !ok {
return fmt.Errorf("failed to receive data")
if err = template.Execute(w, data); err != nil { // Prints: Hello Bob!
panic(err)
}
fmt.Printf("Received data: %v\n", string(output))
// http.HandleFunc("/redirect", func(w http.ResponseWriter, r *http.Request) {
// })
r := chi.NewRouter()
})
r.Handle("/static/*", http.StripPrefix("/static/", http.FileServer(http.Dir("./pages/static"))))
s.Handler = r
return s.ListenAndServe()
}