-
Notifications
You must be signed in to change notification settings - Fork 26
/
factory.go
114 lines (99 loc) · 4.75 KB
/
factory.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
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package access
import (
"github.com/Azure/azure-sdk-for-go/sdk/azcore"
"github.com/Azure/azure-sdk-for-go/sdk/azcore/arm"
"github.com/Azure/azure-sdk-for-go/sdk/azidentity"
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/compute/armcompute/v5"
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/marketplaceordering/armmarketplaceordering"
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork/v4"
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/resourcegraph/armresourcegraph"
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/resources/armresources"
)
// TokenCredentialProvider is a function which gets azcore.TokenCredential using the passed ConnectConfig.
// This allows unit tests to pass their own fake provider for token credentials.
type TokenCredentialProvider func(connectConfig ConnectConfig) (azcore.TokenCredential, error)
// defaultFactory implements Factory interface.
type defaultFactory struct {
clientOptions *arm.ClientOptions
tokenCredentialProvider TokenCredentialProvider
}
// NewDefaultAccessFactory creates a new instance of Factory.
func NewDefaultAccessFactory() Factory {
return NewAccessFactoryWithOptions(nil, GetDefaultTokenCredentials)
}
// NewAccessFactoryWithOptions creates an instance of Factory which uses a fixed tokenCredential
// passed as an argument and also takes custom arm.ClientOptions.
func NewAccessFactoryWithOptions(clientOptions *arm.ClientOptions, tokenCredentialProvider TokenCredentialProvider) Factory {
return defaultFactory{
clientOptions: clientOptions,
tokenCredentialProvider: tokenCredentialProvider,
}
}
// GetDefaultTokenCredentials provides the azure token credentials using the ConnectConfig passed as an argument.
func GetDefaultTokenCredentials(connectConfig ConnectConfig) (azcore.TokenCredential, error) {
return azidentity.NewClientSecretCredential(
connectConfig.TenantID,
connectConfig.ClientID,
connectConfig.ClientSecret,
&azidentity.ClientSecretCredentialOptions{ClientOptions: connectConfig.ClientOptions},
)
}
func (f defaultFactory) GetResourceGroupsAccess(connectConfig ConnectConfig) (*armresources.ResourceGroupsClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armresources.NewResourceGroupsClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetVirtualMachinesAccess(connectConfig ConnectConfig) (*armcompute.VirtualMachinesClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armcompute.NewVirtualMachinesClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetNetworkInterfacesAccess(connectConfig ConnectConfig) (*armnetwork.InterfacesClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armnetwork.NewInterfacesClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetSubnetAccess(connectConfig ConnectConfig) (*armnetwork.SubnetsClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armnetwork.NewSubnetsClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetDisksAccess(connectConfig ConnectConfig) (*armcompute.DisksClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armcompute.NewDisksClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetResourceGraphAccess(connectConfig ConnectConfig) (*armresourcegraph.Client, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armresourcegraph.NewClient(tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetVirtualMachineImagesAccess(connectConfig ConnectConfig) (*armcompute.VirtualMachineImagesClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armcompute.NewVirtualMachineImagesClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}
func (f defaultFactory) GetMarketPlaceAgreementsAccess(connectConfig ConnectConfig) (*armmarketplaceordering.MarketplaceAgreementsClient, error) {
tokenCredential, err := f.tokenCredentialProvider(connectConfig)
if err != nil {
return nil, err
}
return armmarketplaceordering.NewMarketplaceAgreementsClient(connectConfig.SubscriptionID, tokenCredential, f.clientOptions)
}