From 8820f083df3a350a79f605eb26fb2705107d65be Mon Sep 17 00:00:00 2001 From: Henry Avetisyan Date: Tue, 7 Feb 2017 14:27:07 -0800 Subject: [PATCH] Make header configurable --- utils/zts-roletoken/zts-roletoken.go | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/utils/zts-roletoken/zts-roletoken.go b/utils/zts-roletoken/zts-roletoken.go index 8eb3da01506..217dfe73336 100644 --- a/utils/zts-roletoken/zts-roletoken.go +++ b/utils/zts-roletoken/zts-roletoken.go @@ -11,15 +11,14 @@ import ( "github.com/yahoo/athenz/clients/go/zts" ) -const authHeader = "Athenz-Principal-Auth" - func main() { - var domain, role, ntoken, ztsUrl string + var domain, role, ntoken, ztsUrl, hdr string flag.StringVar(&domain, "domain", "", "name of provider domain") flag.StringVar(&role, "role", "", "name of provider role") flag.StringVar(&ntoken, "ntoken", "", "service identity token") flag.StringVar(&ztsUrl, "zts", "", "url of the ZTS Service") + flag.StringVar(&hdr, "hdr", "Athenz-Principal-Auth", "Header name") flag.Parse() if domain == "" || ntoken == "" || ztsUrl == "" { @@ -28,7 +27,7 @@ func main() { // use the ntoken to talk to Athenz client := zts.NewClient(ztsUrl, nil) - client.AddCredentials(authHeader, ntoken) + client.AddCredentials(hdr, ntoken) // request a roletoken roleToken, err := client.GetRoleToken(zts.DomainName(domain), zts.EntityName(role), nil, nil, "")