36 lines
1 KiB
Go
36 lines
1 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/zitadel/oidc/v4/pkg/client/rp"
|
|
httphelper "github.com/zitadel/oidc/v4/pkg/http"
|
|
"github.com/zitadel/oidc/v4/pkg/oidc"
|
|
)
|
|
|
|
const (
|
|
loginPath = "/login"
|
|
)
|
|
|
|
func CodeFlow[C oidc.IDClaims](ctx context.Context, relyingParty rp.RelyingParty, callbackPath, port string, stateProvider func() string) *oidc.Tokens[C] {
|
|
codeflowCtx, codeflowCancel := context.WithCancel(ctx)
|
|
defer codeflowCancel()
|
|
|
|
tokenChan := make(chan *oidc.Tokens[C], 1)
|
|
|
|
callback := func(w http.ResponseWriter, r *http.Request, tokens *oidc.Tokens[C], state string, rp rp.RelyingParty) {
|
|
tokenChan <- tokens
|
|
msg := "<p><strong>Success!</strong></p>"
|
|
msg = msg + "<p>You are authenticated and can now return to the CLI.</p>"
|
|
w.Write([]byte(msg))
|
|
}
|
|
http.Handle(loginPath, rp.AuthURLHandler(stateProvider, relyingParty))
|
|
http.Handle(callbackPath, rp.CodeExchangeHandler(callback, relyingParty))
|
|
|
|
httphelper.StartServer(codeflowCtx, ":"+port)
|
|
|
|
OpenBrowser("http://localhost:" + port + loginPath)
|
|
|
|
return <-tokenChan
|
|
}
|