mirror of
https://github.com/davidallendj/opaal.git
synced 2025-12-20 03:27:02 -07:00
Major refactoring and code restructure
This commit is contained in:
parent
72adbe1f0d
commit
6d63211d35
10 changed files with 454 additions and 859 deletions
227
internal/oauth/client.go
Normal file
227
internal/oauth/client.go
Normal file
|
|
@ -0,0 +1,227 @@
|
|||
package oauth
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/cookiejar"
|
||||
"net/url"
|
||||
"slices"
|
||||
"strings"
|
||||
|
||||
"github.com/davidallendj/go-utils/httpx"
|
||||
"github.com/davidallendj/go-utils/util"
|
||||
"golang.org/x/net/publicsuffix"
|
||||
)
|
||||
|
||||
type Client struct {
|
||||
http.Client
|
||||
Id string `db:"id" yaml:"id"`
|
||||
Secret string `db:"secret" yaml:"secret"`
|
||||
Name string `db:"name" yaml:"name"`
|
||||
Description string `db:"description" yaml:"description"`
|
||||
Issuer string `db:"issuer" yaml:"issuer"`
|
||||
RegistrationAccessToken string `db:"registration_access_token" yaml:"registration-access-token"`
|
||||
RedirectUris []string `db:"redirect_uris" yaml:"redirect-uris"`
|
||||
Scope []string `db:"scope" yaml:"scope"`
|
||||
Audience []string `db:"audience" yaml:"audience"`
|
||||
FlowId string
|
||||
CsrfToken string
|
||||
}
|
||||
|
||||
func NewClient() *Client {
|
||||
return &Client{}
|
||||
}
|
||||
|
||||
func (client *Client) ClearCookies() {
|
||||
jar, _ := cookiejar.New(&cookiejar.Options{PublicSuffixList: publicsuffix.List})
|
||||
client.Jar = jar
|
||||
}
|
||||
|
||||
func (client *Client) IsOAuthClientRegistered(clientUrl string) (bool, error) {
|
||||
_, _, err := httpx.MakeHttpRequest(clientUrl, http.MethodGet, nil, nil)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("failed to make request: %v", err)
|
||||
}
|
||||
// TODO: need to check contents of actual response
|
||||
return true, nil
|
||||
}
|
||||
|
||||
func (client *Client) GetOAuthClient(clientUrl string) error {
|
||||
_, b, err := httpx.MakeHttpRequest(clientUrl, http.MethodGet, nil, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to make request: %v", err)
|
||||
}
|
||||
|
||||
fmt.Printf("GetOAuthClient: %v\n", string(b))
|
||||
|
||||
var data []map[string]any
|
||||
err = json.Unmarshal(b, &data)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to unmarshal JSON: %v", err)
|
||||
}
|
||||
|
||||
index := slices.IndexFunc(data, func(c map[string]any) bool {
|
||||
if c["client_id"] == nil {
|
||||
return false
|
||||
}
|
||||
return c["client_id"].(string) == client.Id
|
||||
})
|
||||
if index < 0 {
|
||||
return fmt.Errorf("client not found")
|
||||
}
|
||||
|
||||
// cast the redirect_uris from []any to []string and extract registration token
|
||||
foundClient := data[index]
|
||||
for _, uri := range foundClient["redirect_uris"].([]any) {
|
||||
client.RedirectUris = append(client.RedirectUris, uri.(string))
|
||||
}
|
||||
if foundClient["registration-access-token"] != nil {
|
||||
client.RegistrationAccessToken = foundClient["registration-access-token"].(string)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (client *Client) CreateOAuthClient(registerUrl string) ([]byte, error) {
|
||||
// hydra endpoint: POST /clients
|
||||
audience := util.QuoteArrayStrings(client.Audience)
|
||||
body := httpx.Body(fmt.Sprintf(`{
|
||||
"client_id": "%s",
|
||||
"client_name": "%s",
|
||||
"client_secret": "%s",
|
||||
"token_endpoint_auth_method": "client_secret_post",
|
||||
"scope": "%s",
|
||||
"grant_types": ["urn:ietf:params:oauth:grant-type:jwt-bearer"],
|
||||
"response_types": ["token"],
|
||||
"redirect_uris": ["http://127.0.0.1:3333/callback"],
|
||||
"state": 12345678910,
|
||||
"audience": [%s]
|
||||
}`, client.Id, client.Id, client.Secret, strings.Join(client.Scope, " "), strings.Join(audience, ","),
|
||||
))
|
||||
headers := httpx.Headers{
|
||||
"Content-Type": "application/json",
|
||||
}
|
||||
|
||||
_, b, err := httpx.MakeHttpRequest(registerUrl, http.MethodPost, []byte(body), headers)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to make request: %v", err)
|
||||
}
|
||||
var rjson map[string]any
|
||||
err = json.Unmarshal(b, &rjson)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to unmarshal response body: %v", err)
|
||||
}
|
||||
|
||||
// check for error first
|
||||
errJson := rjson["error"]
|
||||
if errJson == nil {
|
||||
// set the client ID and secret of registered client
|
||||
client.Id = rjson["client_id"].(string)
|
||||
client.Secret = rjson["client_secret"].(string)
|
||||
client.RegistrationAccessToken = rjson["registration_access_token"].(string)
|
||||
} else {
|
||||
return b, nil
|
||||
}
|
||||
|
||||
return b, err
|
||||
}
|
||||
|
||||
func (client *Client) RegisterOAuthClient(registerUrl string) ([]byte, error) {
|
||||
// hydra endpoint: POST /oauth2/register
|
||||
if registerUrl == "" {
|
||||
return nil, fmt.Errorf("no URL provided")
|
||||
}
|
||||
audience := util.QuoteArrayStrings(client.Audience)
|
||||
body := httpx.Body(fmt.Sprintf(`{
|
||||
"client_name": "opaal",
|
||||
"token_endpoint_auth_method": "client_secret_post",
|
||||
"scope": "%s",
|
||||
"grant_types": ["urn:ietf:params:oauth:grant-type:jwt-bearer"],
|
||||
"response_types": ["token"],
|
||||
"redirect_uris": ["http://127.0.0.1:3333/callback"],
|
||||
"state": 12345678910,
|
||||
"audience": [%s]
|
||||
}`, strings.Join(client.Scope, " "), strings.Join(audience, ","),
|
||||
))
|
||||
headers := httpx.Headers{
|
||||
"Content-Type": "application/json",
|
||||
}
|
||||
_, b, err := httpx.MakeHttpRequest(registerUrl, http.MethodPost, body, headers)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to make request: %v", err)
|
||||
}
|
||||
var rjson map[string]any
|
||||
err = json.Unmarshal(b, &rjson)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to unmarshal response body: %v", err)
|
||||
}
|
||||
|
||||
// check for error first
|
||||
errJson := rjson["error"]
|
||||
if errJson == nil {
|
||||
// set the client ID and secret of registered client
|
||||
client.Id = rjson["client_id"].(string)
|
||||
client.Secret = rjson["client_secret"].(string)
|
||||
client.RegistrationAccessToken = rjson["registration_access_token"].(string)
|
||||
} else {
|
||||
return b, nil
|
||||
}
|
||||
return b, err
|
||||
}
|
||||
|
||||
func (client *Client) AuthorizeOAuthClient(authorizeUrl string) ([]byte, error) {
|
||||
// set the authorization header
|
||||
body := []byte("grant_type=" + url.QueryEscape("urn:ietf:params:oauth:grant-type:jwt-bearer") +
|
||||
"&scope=" + strings.Join(client.Scope, "+") +
|
||||
"&client_id=" + client.Id +
|
||||
"&client_secret=" + client.Secret +
|
||||
"&redirect_uri=" + url.QueryEscape("http://127.0.0.1:3333/callback") + // FIXME: needs to not be hardcorded
|
||||
"&response_type=token" +
|
||||
"&state=12345678910",
|
||||
)
|
||||
headers := httpx.Headers{
|
||||
"Authorization": "Bearer " + client.RegistrationAccessToken,
|
||||
"Content-Type": "application/x-www-form-urlencoded",
|
||||
}
|
||||
_, b, err := httpx.MakeHttpRequest(authorizeUrl, http.MethodPost, body, headers)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to make HTTP request: %v", err)
|
||||
}
|
||||
|
||||
return b, nil
|
||||
}
|
||||
|
||||
func (client *Client) PerformTokenGrant(clientUrl string, encodedJwt string) ([]byte, error) {
|
||||
// hydra endpoint: /oauth/token
|
||||
body := "grant_type=" + url.QueryEscape("urn:ietf:params:oauth:grant-type:jwt-bearer") +
|
||||
"&client_id=" + client.Id +
|
||||
"&client_secret=" + client.Secret +
|
||||
"&redirect_uri=" + url.QueryEscape("http://127.0.0.1:3333/callback")
|
||||
// add optional params if valid
|
||||
if encodedJwt != "" {
|
||||
body += "&assertion=" + encodedJwt
|
||||
}
|
||||
if client.Scope != nil || len(client.Scope) > 0 {
|
||||
body += "&scope=" + strings.Join(client.Scope, "+")
|
||||
}
|
||||
headers := httpx.Headers{
|
||||
"Content-Type": "application/x-www-form-urlencoded",
|
||||
"Authorization": "Bearer " + client.RegistrationAccessToken,
|
||||
}
|
||||
|
||||
_, b, err := httpx.MakeHttpRequest(clientUrl, http.MethodPost, []byte(body), headers)
|
||||
|
||||
// set flow ID back to empty string to indicate a completed flow
|
||||
client.FlowId = ""
|
||||
|
||||
return b, err
|
||||
}
|
||||
|
||||
func (client *Client) DeleteOAuthClient(clientUrl string) error {
|
||||
_, _, err := httpx.MakeHttpRequest(clientUrl+"/"+client.Id, http.MethodDelete, nil, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to make request: %v", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue