mirror of
https://github.com/safedep/vet.git
synced 2025-12-10 00:22:08 -06:00
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
|
|
"github.com/safedep/vet/pkg/common/logger"
|
|
"google.golang.org/grpc"
|
|
|
|
drygrpc "github.com/safedep/dry/adapters/grpc"
|
|
)
|
|
|
|
// Create a gRPC client connection for the control plane
|
|
// based on available configuration
|
|
func ControlPlaneClientConnection(name string) (*grpc.ClientConn, error) {
|
|
parsedUrl, err := url.Parse(DefaultControlTowerUrl())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
host, port := parsedUrl.Hostname(), parsedUrl.Port()
|
|
if port == "" {
|
|
port = "443"
|
|
}
|
|
|
|
logger.Debugf("ControlTower host: %s, port: %s", host, port)
|
|
|
|
// For local development, we use the mock user.
|
|
vetTenantId := os.Getenv("VET_CONTROL_TOWER_TENANT_ID")
|
|
vetTenantMockUser := os.Getenv("VET_CONTROL_TOWER_MOCK_USER")
|
|
|
|
headers := http.Header{}
|
|
headers.Set("x-tenant-id", vetTenantId)
|
|
headers.Set("x-mock-user", vetTenantMockUser)
|
|
|
|
client, err := drygrpc.GrpcClient(name, host, port,
|
|
ApiKey(), headers, []grpc.DialOption{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create gRPC client: %w", err)
|
|
}
|
|
|
|
return client, nil
|
|
}
|