-
Notifications
You must be signed in to change notification settings - Fork 0
/
sdk.go
133 lines (113 loc) · 3.86 KB
/
sdk.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
/*
* Copyright © 2015-2018 Aeneas Rekkas <aeneas+oss@aeneas.io>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* @author Aeneas Rekkas <aeneas+oss@aeneas.io>
* @copyright 2015-2018 Aeneas Rekkas <aeneas+oss@aeneas.io>
* @license Apache-2.0
*/
package hydra
import (
"context"
"github.com/ory/hydra/sdk/go/hydra/swagger"
"github.com/pkg/errors"
"golang.org/x/oauth2"
"golang.org/x/oauth2/clientcredentials"
)
// CodeGenSDK contains all relevant API clients for interacting with ORY Hydra.
type CodeGenSDK struct {
*swagger.OAuth2Api
*swagger.JsonWebKeyApi
*swagger.WardenApi
*swagger.PolicyApi
Configuration *Configuration
oAuth2ClientConfig *clientcredentials.Config
oAuth2Config *oauth2.Config
}
// Configuration configures the CodeGenSDK.
type Configuration struct {
// EndpointURL should point to the url of ORY Hydra, for example: http://localhost:4444
EndpointURL string
// ClientID is the id of the management client. The management client should have appropriate access rights
// and the ability to request the client_credentials grant.
ClientID string
// ClientSecret is the secret of the management client.
ClientSecret string
// Scopes is a list of scopes the CodeGenSDK should request. If no scopes are given, this defaults to `hydra.*`
Scopes []string
}
func removeTrailingSlash(path string) string {
for len(path) > 0 && path[len(path)-1] == '/' {
path = path[0 : len(path)-1]
}
return path
}
func (s *CodeGenSDK) GetOAuth2ClientConfig() *clientcredentials.Config {
return s.oAuth2ClientConfig
}
func (s *CodeGenSDK) GetOAuth2Config() *oauth2.Config {
return s.oAuth2Config
}
// NewSDK instantiates a new CodeGenSDK instance or returns an error.
func NewSDK(c *Configuration) (*CodeGenSDK, error) {
if c.EndpointURL == "" {
return nil, errors.New("Please specify the ORY Hydra Endpoint URL")
}
if c.ClientSecret == "" {
return nil, errors.New("Please specify an OAuth 2.0 Client Secret")
}
if c.ClientID == "" {
return nil, errors.New("Please specify an OAuth 2.0 Client ID")
}
if len(c.Scopes) == 0 {
c.Scopes = []string{"hydra.*"}
}
c.EndpointURL = removeTrailingSlash(c.EndpointURL)
oAuth2Config := &oauth2.Config{
ClientSecret: c.ClientSecret,
ClientID: c.ClientID,
Scopes: c.Scopes,
Endpoint: oauth2.Endpoint{
TokenURL: c.EndpointURL + "/oauth2/token",
AuthURL: c.EndpointURL + "/oauth2/auth",
},
}
oAuth2ClientConfig := &clientcredentials.Config{
ClientSecret: c.ClientSecret,
ClientID: c.ClientID,
Scopes: c.Scopes,
TokenURL: c.EndpointURL + "/oauth2/token",
}
oAuth2Client := oAuth2ClientConfig.Client(context.Background())
o := swagger.NewOAuth2ApiWithBasePath(c.EndpointURL)
o.Configuration.Transport = oAuth2Client.Transport
o.Configuration.Username = c.ClientID
o.Configuration.Password = c.ClientSecret
j := swagger.NewJsonWebKeyApiWithBasePath(c.EndpointURL)
j.Configuration.Transport = oAuth2Client.Transport
w := swagger.NewWardenApiWithBasePath(c.EndpointURL)
w.Configuration.Transport = oAuth2Client.Transport
p := swagger.NewPolicyApiWithBasePath(c.EndpointURL)
p.Configuration.Transport = oAuth2Client.Transport
sdk := &CodeGenSDK{
OAuth2Api: o,
JsonWebKeyApi: j,
WardenApi: w,
PolicyApi: p,
Configuration: c,
oAuth2ClientConfig: oAuth2ClientConfig,
oAuth2Config: oAuth2Config,
}
return sdk, nil
}