2017-08-05 20:30:20 +02:00
|
|
|
package utils
|
|
|
|
|
2017-10-22 10:07:36 +02:00
|
|
|
// Copyright 2017 Microsoft Corporation
|
|
|
|
//
|
|
|
|
// 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.
|
|
|
|
|
2017-08-05 20:30:20 +02:00
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
|
|
|
|
"github.com/Azure/go-autorest/autorest"
|
|
|
|
"github.com/Azure/go-autorest/autorest/adal"
|
|
|
|
"github.com/Azure/go-autorest/autorest/azure"
|
|
|
|
)
|
|
|
|
|
|
|
|
// GetAuthorizer gets an Azure Service Principal authorizer.
|
|
|
|
// This func assumes "AZURE_TENANT_ID", "AZURE_CLIENT_ID",
|
|
|
|
// "AZURE_CLIENT_SECRET" are set as environment variables.
|
|
|
|
func GetAuthorizer(env azure.Environment) (*autorest.BearerAuthorizer, error) {
|
|
|
|
tenantID := GetEnvVarOrExit("AZURE_TENANT_ID")
|
|
|
|
|
|
|
|
oauthConfig, err := adal.NewOAuthConfig(env.ActiveDirectoryEndpoint, tenantID)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
clientID := GetEnvVarOrExit("AZURE_CLIENT_ID")
|
|
|
|
clientSecret := GetEnvVarOrExit("AZURE_CLIENT_SECRET")
|
|
|
|
|
|
|
|
spToken, err := adal.NewServicePrincipalToken(*oauthConfig, clientID, clientSecret, env.ResourceManagerEndpoint)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return autorest.NewBearerAuthorizer(spToken), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetEnvVarOrExit returns the value of specified environment variable or terminates if it's not defined.
|
|
|
|
func GetEnvVarOrExit(varName string) string {
|
|
|
|
value := os.Getenv(varName)
|
|
|
|
if value == "" {
|
|
|
|
fmt.Printf("Missing environment variable %s\n", varName)
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
return value
|
|
|
|
}
|