safedep/vet
1
0
Fork 0
mirror of https://github.com/safedep/vet.git synced 2025-12-10 13:43:01 -06:00
Code Issues Packages Projects Releases 68 Wiki Activity

fix: Tenant domain handling in cloud commands

Browse Source
This commit is contained in:
abhisek 2024-10-11 10:07:01 +05:30
parent ad6340e60d
commit 69e32d99cc
No known key found for this signature in database
GPG Key ID: CB92A4990C02A88F
4 changed files with 33 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmd/cloud/login.go
View File

@ -12,10 +12,6 @@ import (
"github.com/spf13/cobra"
)
var (
loginTenantDomain string
)
func newCloudLoginCommand() *cobra.Command {
cmd := &cobra.Command{
Use: "login",
@ -30,9 +26,6 @@ func newCloudLoginCommand() *cobra.Command {
},
}
cmd.Flags().StringVar(&loginTenantDomain, "tenant", "",
"Domain of the tenant to login to for existing users")
return cmd
}
@ -59,5 +52,6 @@ func executeDeviceAuthFlow() error {
return fmt.Errorf("failed to authenticate: %w", err)
}
return auth.PersistCloudTokens(token.Token, token.RefreshToken, loginTenantDomain)
return auth.PersistCloudTokens(token.Token,
token.RefreshToken, tenantDomain)
}

16
cmd/cloud/main.go
View File

@ -1,6 +1,11 @@
package cloud
import "github.com/spf13/cobra"
import (
"github.com/safedep/vet/internal/auth"
"github.com/spf13/cobra"
)
var tenantDomain string
func NewCloudCommand() *cobra.Command {
cmd := &cobra.Command{
@ -11,6 +16,9 @@ func NewCloudCommand() *cobra.Command {
},
}
cmd.PersistentFlags().StringVar(&tenantDomain, "tenant", "",
"Tenant domain to use for the command")
cmd.AddCommand(newCloudLoginCommand())
cmd.AddCommand(newRegisterCommand())
cmd.AddCommand(newQueryCommand())
@ -18,5 +26,11 @@ func NewCloudCommand() *cobra.Command {
cmd.AddCommand(newWhoamiCommand())
cmd.AddCommand(newKeyCommand())
cmd.PersistentPreRun = func(cmd *cobra.Command, args []string) {
if tenantDomain != "" {
auth.SetRuntimeCloudTenant(tenantDomain)
}
}
return cmd
}

10
internal/auth/auth.go
View File

@ -15,6 +15,7 @@ const (
apiKeyEnvKey = "VET_INSIGHTS_API_KEY"
apiKeyAlternateEnvKey = "VET_API_KEY"
communityModeEnvKey = "VET_COMMUNITY_MODE"
controlTowerTenantEnvKey = "VET_CONTROL_TOWER_TENANT_ID"
defaultApiUrl = "https://api.safedep.io/insights/v1"
defaultCommunityApiUrl = "https://api.safedep.io/insights-community/v1"
@ -168,6 +169,11 @@ func ControlTowerUrl() string {
}
func TenantDomain() string {
tenantFromEnv := os.Getenv(controlTowerTenantEnvKey)
if tenantFromEnv != "" {
return tenantFromEnv
}
if globalConfig != nil {
return globalConfig.TenantDomain
}
@ -226,6 +232,10 @@ func SetRuntimeCommunityMode() {
os.Setenv(communityModeEnvKey, "true")
}
func SetRuntimeCloudTenant(domain string) {
os.Setenv(controlTowerTenantEnvKey, domain)
}
func loadConfiguration() error {
path, err := os.UserHomeDir()
if err != nil {

11
internal/auth/grpc.go
View File

@ -35,17 +35,10 @@ func cloudClientConnection(name, loc, tok string) (*grpc.ClientConn, error) {
logger.Debugf("ControlTower host: %s, port: %s", host, port)
vetTenantId := TenantDomain()
tenantIdOverride := os.Getenv("VET_CONTROL_TOWER_TENANT_ID")
if tenantIdOverride != "" {
vetTenantId = tenantIdOverride
}
headers := http.Header{}
headers.Set("x-tenant-id", vetTenantId)
headers.Set("x-tenant-id", TenantDomain())
vetTenantMockUser := os.Getenv("VET_CONTROL_TOWER_MOCK_USER")
vetTenantMockUser := os.Getenv(controlTowerTenantEnvKey)
if vetTenantMockUser != "" {
headers.Set("x-mock-user", vetTenantMockUser)
}