common functions for grpc

pkg/analyser/client.go

@@ -2,24 +2,13 @@ package analyser
 
 import (
 	"fmt"
-	"net/http"
-	"os"
 
 	"buf.build/gen/go/safedep/api/grpc/go/safedep/services/malysis/v1/malysisv1grpc"
-	drygrpc "github.com/safedep/dry/adapters/grpc"
-	"google.golang.org/grpc"
+	"github.com/safedep/pmg/pkg/common"
 )
 
 func GetMalwareAnalysisClient() (malysisv1grpc.MalwareAnalysisServiceClient, error) {
-	tok := os.Getenv("SAFEDEP_API_KEY")
-	tenantId := os.Getenv("SAFEDEP_TENANT_ID")
-	if tok == "" || tenantId == "" {
-		return nil, fmt.Errorf("SAFEDEP_API_KEY and SAFEDEP_TENANT_ID must be set")
-	}
-	headers := http.Header{}
-	headers.Set("x-tenant-id", tenantId)
-	cc, err := drygrpc.GrpcClient("pmg-pkg-scan", "api.safedep.io", "443",
-		tok, headers, []grpc.DialOption{})
+	cc, err := common.NewCloudClientConnection()
 	if err != nil {
 		return nil, fmt.Errorf("failed to create gRPC client: %v", err)
 	}