// Copyright 2026 Google LLC

//

// 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

//

// https://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.

// Code generated by protoc-gen-go_gapic. DO NOT EDIT.

package kms

import (

"bytes"

"context"

"fmt"

"log/slog"

"math"

"net/http"

"net/url"

"time"

iampb "cloud.google.com/go/iam/apiv1/iampb"

kmspb "cloud.google.com/go/kms/apiv1/kmspb"

"cloud.google.com/go/longrunning"

lroauto "cloud.google.com/go/longrunning/autogen"

longrunningpb "cloud.google.com/go/longrunning/autogen/longrunningpb"

gax "github.com/googleapis/gax-go/v2"

"github.com/googleapis/gax-go/v2/callctx"

"google.golang.org/api/iterator"

"google.golang.org/api/option"

"google.golang.org/api/option/internaloption"

gtransport "google.golang.org/api/transport/grpc"

httptransport "google.golang.org/api/transport/http"

locationpb "google.golang.org/genproto/googleapis/cloud/location"

"google.golang.org/grpc"

"google.golang.org/grpc/codes"

"google.golang.org/protobuf/encoding/protojson"

"google.golang.org/protobuf/proto"

)

var newAutokeyClientHook clientHook

// AutokeyCallOptions contains the retry settings for each method of AutokeyClient.

type AutokeyCallOptions struct {

CreateKeyHandle []gax.CallOption

GetKeyHandle []gax.CallOption

ListKeyHandles []gax.CallOption

GetLocation []gax.CallOption

ListLocations []gax.CallOption

GetIamPolicy []gax.CallOption

SetIamPolicy []gax.CallOption

TestIamPermissions []gax.CallOption

GetOperation []gax.CallOption

}

func defaultAutokeyGRPCClientOptions() []option.ClientOption {

return []option.ClientOption{

internaloption.WithDefaultEndpoint("cloudkms.googleapis.com:443"),

internaloption.WithDefaultEndpointTemplate("cloudkms.UNIVERSE_DOMAIN:443"),

internaloption.WithDefaultMTLSEndpoint("cloudkms.mtls.googleapis.com:443"),

internaloption.WithDefaultUniverseDomain("googleapis.com"),

internaloption.WithDefaultAudience("https://cloudkms.googleapis.com/"),

internaloption.WithDefaultScopes(DefaultAuthScopes()...),

internaloption.EnableJwtWithScope(),

internaloption.AllowHardBoundTokens("MTLS_S2A"),

internaloption.EnableNewAuthLibrary(),

option.WithGRPCDialOption(grpc.WithDefaultCallOptions(

grpc.MaxCallRecvMsgSize(math.MaxInt32))),

}

}

func defaultAutokeyCallOptions() *AutokeyCallOptions {

return &AutokeyCallOptions{

CreateKeyHandle: []gax.CallOption{

gax.WithTimeout(60000 * time.Millisecond),

},

GetKeyHandle: []gax.CallOption{

gax.WithTimeout(60000 * time.Millisecond),

gax.WithRetry(func() gax.Retryer {

return gax.OnCodes([]codes.Code{

codes.Unavailable,

codes.DeadlineExceeded,

}, gax.Backoff{

Initial: 100 * time.Millisecond,

Max: 60000 * time.Millisecond,

Multiplier: 1.30,

})

}),

},

ListKeyHandles: []gax.CallOption{

gax.WithTimeout(60000 * time.Millisecond),

gax.WithRetry(func() gax.Retryer {

return gax.OnCodes([]codes.Code{

codes.Unavailable,

codes.DeadlineExceeded,

}, gax.Backoff{

Initial: 100 * time.Millisecond,

Max: 60000 * time.Millisecond,

Multiplier: 1.30,

})

}),

},

GetLocation: []gax.CallOption{},

ListLocations: []gax.CallOption{},

GetIamPolicy: []gax.CallOption{},

SetIamPolicy: []gax.CallOption{},

TestIamPermissions: []gax.CallOption{},

GetOperation: []gax.CallOption{},

}

}

func defaultAutokeyRESTCallOptions() *AutokeyCallOptions {

return &AutokeyCallOptions{

CreateKeyHandle: []gax.CallOption{

gax.WithTimeout(60000 * time.Millisecond),

},

GetKeyHandle: []gax.CallOption{

gax.WithTimeout(60000 * time.Millisecond),

gax.WithRetry(func() gax.Retryer {

return gax.OnHTTPCodes(gax.Backoff{

Initial: 100 * time.Millisecond,

Max: 60000 * time.Millisecond,

Multiplier: 1.30,

},

http.StatusServiceUnavailable,

http.StatusGatewayTimeout)

}),

},

ListKeyHandles: []gax.CallOption{

gax.WithTimeout(60000 * time.Millisecond),

gax.WithRetry(func() gax.Retryer {

return gax.OnHTTPCodes(gax.Backoff{

Initial: 100 * time.Millisecond,

Max: 60000 * time.Millisecond,

Multiplier: 1.30,

},

http.StatusServiceUnavailable,

http.StatusGatewayTimeout)

}),

},

GetLocation: []gax.CallOption{},

ListLocations: []gax.CallOption{},

GetIamPolicy: []gax.CallOption{},

SetIamPolicy: []gax.CallOption{},

TestIamPermissions: []gax.CallOption{},

GetOperation: []gax.CallOption{},

}

}

// internalAutokeyClient is an interface that defines the methods available from Cloud Key Management Service (KMS) API.

type internalAutokeyClient interface {

Close() error

setGoogleClientInfo(...string)

Connection() *grpc.ClientConn

CreateKeyHandle(context.Context, *kmspb.CreateKeyHandleRequest, ...gax.CallOption) (*CreateKeyHandleOperation, error)

CreateKeyHandleOperation(name string) *CreateKeyHandleOperation

GetKeyHandle(context.Context, *kmspb.GetKeyHandleRequest, ...gax.CallOption) (*kmspb.KeyHandle, error)

ListKeyHandles(context.Context, *kmspb.ListKeyHandlesRequest, ...gax.CallOption) *KeyHandleIterator

GetLocation(context.Context, *locationpb.GetLocationRequest, ...gax.CallOption) (*locationpb.Location, error)

ListLocations(context.Context, *locationpb.ListLocationsRequest, ...gax.CallOption) *LocationIterator

GetIamPolicy(context.Context, *iampb.GetIamPolicyRequest, ...gax.CallOption) (*iampb.Policy, error)

SetIamPolicy(context.Context, *iampb.SetIamPolicyRequest, ...gax.CallOption) (*iampb.Policy, error)

TestIamPermissions(context.Context, *iampb.TestIamPermissionsRequest, ...gax.CallOption) (*iampb.TestIamPermissionsResponse, error)

GetOperation(context.Context, *longrunningpb.GetOperationRequest, ...gax.CallOption) (*longrunningpb.Operation, error)

}

// AutokeyClient is a client for interacting with Cloud Key Management Service (KMS) API.

// Methods, except Close, may be called concurrently. However, fields must not be modified concurrently with method calls.

//

// Provides interfaces for using Cloud KMS

// Autokey (at https://cloud.google.com/kms/help/autokey) to provision new

// CryptoKeys, ready for Customer Managed

// Encryption Key (CMEK) use, on-demand. To support certain client tooling, this

// feature is modeled around a KeyHandle

// resource: creating a KeyHandle in a resource

// project and given location triggers Cloud KMS Autokey to provision a

// CryptoKey in the configured key project and

// the same location.

//

// Prior to use in a given resource project,

// UpdateAutokeyConfig

// should have been called on an ancestor folder, setting the key project where

// Cloud KMS Autokey should create new

// CryptoKeys. See documentation for additional

// prerequisites. To check what key project, if any, is currently configured on

// a resource project’s ancestor folder, see

// ShowEffectiveAutokeyConfig.

type AutokeyClient struct {

// The internal transport-dependent client.

internalClient internalAutokeyClient

// The call options for this service.

CallOptions *AutokeyCallOptions

// LROClient is used internally to handle long-running operations.

// It is exposed so that its CallOptions can be modified if required.

// Users should not Close this client.

LROClient *lroauto.OperationsClient

}

// Wrapper methods routed to the internal client.

// Close closes the connection to the API service. The user should invoke this when

// the client is no longer required.

func (c *AutokeyClient) Close() error {

return c.internalClient.Close()

}

// setGoogleClientInfo sets the name and version of the application in

// the `x-goog-api-client` header passed on each request. Intended for

// use by Google-written clients.

func (c *AutokeyClient) setGoogleClientInfo(keyval ...string) {

c.internalClient.setGoogleClientInfo(keyval...)

}

// Connection returns a connection to the API service.

//

// Deprecated: Connections are now pooled so this method does not always

// return the same resource.

func (c *AutokeyClient) Connection() *grpc.ClientConn {

return c.internalClient.Connection()

}

// CreateKeyHandle creates a new KeyHandle, triggering the

// provisioning of a new CryptoKey for CMEK

// use with the given resource type in the configured key project and the same

// location. GetOperation should

// be used to resolve the resulting long-running operation and get the

// resulting KeyHandle and

// CryptoKey.

func (c *AutokeyClient) CreateKeyHandle(ctx context.Context, req *kmspb.CreateKeyHandleRequest, opts ...gax.CallOption) (*CreateKeyHandleOperation, error) {

return c.internalClient.CreateKeyHandle(ctx, req, opts...)

}

// CreateKeyHandleOperation returns a new CreateKeyHandleOperation from a given name.

// The name must be that of a previously created CreateKeyHandleOperation, possibly from a different process.

func (c *AutokeyClient) CreateKeyHandleOperation(name string) *CreateKeyHandleOperation {

return c.internalClient.CreateKeyHandleOperation(name)

}

// GetKeyHandle returns the KeyHandle.

func (c *AutokeyClient) GetKeyHandle(ctx context.Context, req *kmspb.GetKeyHandleRequest, opts ...gax.CallOption) (*kmspb.KeyHandle, error) {

return c.internalClient.GetKeyHandle(ctx, req, opts...)

}

// ListKeyHandles lists KeyHandles.

func (c *AutokeyClient) ListKeyHandles(ctx context.Context, req *kmspb.ListKeyHandlesRequest, opts ...gax.CallOption) *KeyHandleIterator {

return c.internalClient.ListKeyHandles(ctx, req, opts...)

}

// GetLocation gets information about a location.

func (c *AutokeyClient) GetLocation(ctx context.Context, req *locationpb.GetLocationRequest, opts ...gax.CallOption) (*locationpb.Location, error) {

return c.internalClient.GetLocation(ctx, req, opts...)

}

// ListLocations lists information about the supported locations for this service.

// This method can be called in two ways:

//

// List all public locations: Use the path GET /v1/locations.

//

// List project-visible locations: Use the path

// GET /v1/projects/{project_id}/locations. This may include public

// locations as well as private or other locations specifically visible

// to the project.

func (c *AutokeyClient) ListLocations(ctx context.Context, req *locationpb.ListLocationsRequest, opts ...gax.CallOption) *LocationIterator {

return c.internalClient.ListLocations(ctx, req, opts...)

}

// GetIamPolicy gets the access control policy for a resource. Returns an empty policy

// if the resource exists and does not have a policy set.

func (c *AutokeyClient) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest, opts ...gax.CallOption) (*iampb.Policy, error) {

return c.internalClient.GetIamPolicy(ctx, req, opts...)

}

// SetIamPolicy sets the access control policy on the specified resource. Replaces

// any existing policy.

//

// Can return NOT_FOUND, INVALID_ARGUMENT, and PERMISSION_DENIED

// errors.

func (c *AutokeyClient) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest, opts ...gax.CallOption) (*iampb.Policy, error) {

return c.internalClient.SetIamPolicy(ctx, req, opts...)

}

// TestIamPermissions returns permissions that a caller has on the specified resource. If the

// resource does not exist, this will return an empty set of

// permissions, not a NOT_FOUND error.

//

// Note: This operation is designed to be used for building

// permission-aware UIs and command-line tools, not for authorization

// checking. This operation may “fail open” without warning.

func (c *AutokeyClient) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest, opts ...gax.CallOption) (*iampb.TestIamPermissionsResponse, error) {

return c.internalClient.TestIamPermissions(ctx, req, opts...)

}

// GetOperation is a utility method from google.longrunning.Operations.

func (c *AutokeyClient) GetOperation(ctx context.Context, req *longrunningpb.GetOperationRequest, opts ...gax.CallOption) (*longrunningpb.Operation, error) {

return c.internalClient.GetOperation(ctx, req, opts...)

}

// autokeyGRPCClient is a client for interacting with Cloud Key Management Service (KMS) API over gRPC transport.

//

// Methods, except Close, may be called concurrently. However, fields must not be modified concurrently with method calls.

type autokeyGRPCClient struct {

// Connection pool of gRPC connections to the service.

connPool gtransport.ConnPool

// Points back to the CallOptions field of the containing AutokeyClient

CallOptions **AutokeyCallOptions

// The gRPC API client.

autokeyClient kmspb.AutokeyClient

// LROClient is used internally to handle long-running operations.

// It is exposed so that its CallOptions can be modified if required.

// Users should not Close this client.

LROClient **lroauto.OperationsClient

operationsClient longrunningpb.OperationsClient

iamPolicyClient iampb.IAMPolicyClient

locationsClient locationpb.LocationsClient

// The x-goog-* metadata to be sent with each request.

xGoogHeaders []string

logger *slog.Logger

}

// NewAutokeyClient creates a new autokey client based on gRPC.

// The returned client must be Closed when it is done being used to clean up its underlying connections.

//

// Provides interfaces for using Cloud KMS

// Autokey (at https://cloud.google.com/kms/help/autokey) to provision new

// CryptoKeys, ready for Customer Managed

// Encryption Key (CMEK) use, on-demand. To support certain client tooling, this

// feature is modeled around a KeyHandle

// resource: creating a KeyHandle in a resource

// project and given location triggers Cloud KMS Autokey to provision a

// CryptoKey in the configured key project and

// the same location.

//

// Prior to use in a given resource project,

// UpdateAutokeyConfig

// should have been called on an ancestor folder, setting the key project where

// Cloud KMS Autokey should create new

// CryptoKeys. See documentation for additional

// prerequisites. To check what key project, if any, is currently configured on

// a resource project’s ancestor folder, see

// ShowEffectiveAutokeyConfig.

func NewAutokeyClient(ctx context.Context, opts ...option.ClientOption) (*AutokeyClient, error) {

clientOpts := defaultAutokeyGRPCClientOptions()

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

clientOpts = append(clientOpts, internaloption.WithTelemetryAttributes(map[string]string{

"gcp.client.service": "cloudkms",

"gcp.client.version": getVersionClient(),

"gcp.client.repo": "googleapis/google-cloud-go",

"gcp.client.artifact": "cloud.google.com/go/kms/apiv1",

"gcp.client.language": "go",

"url.domain": "cloudkms.googleapis.com",

}))

}

if newAutokeyClientHook != nil {

hookOpts, err := newAutokeyClientHook(ctx, clientHookParams{})

if err != nil {

return nil, err

}

clientOpts = append(clientOpts, hookOpts...)

}

connPool, err := gtransport.DialPool(ctx, append(clientOpts, opts...)...)

if err != nil {

return nil, err

}

client := AutokeyClient{CallOptions: defaultAutokeyCallOptions()}

c := &autokeyGRPCClient{

connPool: connPool,

autokeyClient: kmspb.NewAutokeyClient(connPool),

CallOptions: &client.CallOptions,

logger: internaloption.GetLogger(opts),

operationsClient: longrunningpb.NewOperationsClient(connPool),

iamPolicyClient: iampb.NewIAMPolicyClient(connPool),

locationsClient: locationpb.NewLocationsClient(connPool),

}

c.setGoogleClientInfo()

if gax.IsFeatureEnabled("METRICS") {

metrics := gax.NewClientMetrics(

gax.WithTelemetryLogger(c.logger),

gax.WithTelemetryAttributes(map[string]string{

gax.ClientService: "cloudkms",

gax.ClientVersion: getVersionClient(),

gax.ClientArtifact: "cloud.google.com/go/kms/apiv1",

gax.RPCSystem: "grpc",

gax.URLDomain: "cloudkms.googleapis.com",

}),

)

client.CallOptions.CreateKeyHandle = append(client.CallOptions.CreateKeyHandle, gax.WithClientMetrics(metrics))

client.CallOptions.GetKeyHandle = append(client.CallOptions.GetKeyHandle, gax.WithClientMetrics(metrics))

client.CallOptions.ListKeyHandles = append(client.CallOptions.ListKeyHandles, gax.WithClientMetrics(metrics))

client.CallOptions.GetLocation = append(client.CallOptions.GetLocation, gax.WithClientMetrics(metrics))

client.CallOptions.ListLocations = append(client.CallOptions.ListLocations, gax.WithClientMetrics(metrics))

client.CallOptions.GetIamPolicy = append(client.CallOptions.GetIamPolicy, gax.WithClientMetrics(metrics))

client.CallOptions.SetIamPolicy = append(client.CallOptions.SetIamPolicy, gax.WithClientMetrics(metrics))

client.CallOptions.TestIamPermissions = append(client.CallOptions.TestIamPermissions, gax.WithClientMetrics(metrics))

client.CallOptions.GetOperation = append(client.CallOptions.GetOperation, gax.WithClientMetrics(metrics))

}

client.internalClient = c

client.LROClient, err = lroauto.NewOperationsClient(ctx, gtransport.WithConnPool(connPool))

if err != nil {

// This error "should not happen", since we are just reusing old connection pool

// and never actually need to dial.

// If this does happen, we could leak connp. However, we cannot close conn:

// If the user invoked the constructor with option.WithGRPCConn,

// we would close a connection that's still in use.

// TODO: investigate error conditions.

return nil, err

}

c.LROClient = &client.LROClient

return &client, nil

}

// Connection returns a connection to the API service.

//

// Deprecated: Connections are now pooled so this method does not always

// return the same resource.

func (c *autokeyGRPCClient) Connection() *grpc.ClientConn {

return c.connPool.Conn()

}

// setGoogleClientInfo sets the name and version of the application in

// the `x-goog-api-client` header passed on each request. Intended for

// use by Google-written clients.

func (c *autokeyGRPCClient) setGoogleClientInfo(keyval ...string) {

kv := append([]string{"gl-go", gax.GoVersion}, keyval...)

kv = append(kv, "gapic", getVersionClient(), "gax", gax.Version, "grpc", grpc.Version, "pb", protoVersion)

c.xGoogHeaders = []string{

"x-goog-api-client", gax.XGoogHeader(kv...),

}

}

// Close closes the connection to the API service. The user should invoke this when

// the client is no longer required.

func (c *autokeyGRPCClient) Close() error {

return c.connPool.Close()

}

// Methods, except Close, may be called concurrently. However, fields must not be modified concurrently with method calls.

type autokeyRESTClient struct {

// The http endpoint to connect to.

endpoint string

// The http client.

httpClient *http.Client

// LROClient is used internally to handle long-running operations.

// It is exposed so that its CallOptions can be modified if required.

// Users should not Close this client.

LROClient **lroauto.OperationsClient

// The x-goog-* headers to be sent with each request.

xGoogHeaders []string

// Points back to the CallOptions field of the containing AutokeyClient

CallOptions **AutokeyCallOptions

logger *slog.Logger

}

// NewAutokeyRESTClient creates a new autokey rest client.

//

// Provides interfaces for using Cloud KMS

// Autokey (at https://cloud.google.com/kms/help/autokey) to provision new

// CryptoKeys, ready for Customer Managed

// Encryption Key (CMEK) use, on-demand. To support certain client tooling, this

// feature is modeled around a KeyHandle

// resource: creating a KeyHandle in a resource

// project and given location triggers Cloud KMS Autokey to provision a

// CryptoKey in the configured key project and

// the same location.

//

// Prior to use in a given resource project,

// UpdateAutokeyConfig

// should have been called on an ancestor folder, setting the key project where

// Cloud KMS Autokey should create new

// CryptoKeys. See documentation for additional

// prerequisites. To check what key project, if any, is currently configured on

// a resource project’s ancestor folder, see

// ShowEffectiveAutokeyConfig.

func NewAutokeyRESTClient(ctx context.Context, opts ...option.ClientOption) (*AutokeyClient, error) {

clientOpts := append(defaultAutokeyRESTClientOptions(), opts...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

clientOpts = append(clientOpts, internaloption.WithTelemetryAttributes(map[string]string{

"gcp.client.service": "cloudkms",

"gcp.client.version": getVersionClient(),

"gcp.client.repo": "googleapis/google-cloud-go",

"gcp.client.artifact": "cloud.google.com/go/kms/apiv1",

"gcp.client.language": "go",

"url.domain": "cloudkms.googleapis.com",

}))

}

httpClient, endpoint, err := httptransport.NewClient(ctx, clientOpts...)

if err != nil {

return nil, err

}

callOpts := defaultAutokeyRESTCallOptions()

c := &autokeyRESTClient{

endpoint: endpoint,

httpClient: httpClient,

CallOptions: &callOpts,

logger: internaloption.GetLogger(opts),

}

c.setGoogleClientInfo()

if gax.IsFeatureEnabled("METRICS") {

metrics := gax.NewClientMetrics(

gax.WithTelemetryLogger(c.logger),

gax.WithTelemetryAttributes(map[string]string{

gax.ClientService: "cloudkms",

gax.ClientVersion: getVersionClient(),

gax.ClientArtifact: "cloud.google.com/go/kms/apiv1",

gax.RPCSystem: "http",

gax.URLDomain: "cloudkms.googleapis.com",

}),

)

callOpts.CreateKeyHandle = append(callOpts.CreateKeyHandle, gax.WithClientMetrics(metrics))

callOpts.GetKeyHandle = append(callOpts.GetKeyHandle, gax.WithClientMetrics(metrics))

callOpts.ListKeyHandles = append(callOpts.ListKeyHandles, gax.WithClientMetrics(metrics))

callOpts.GetLocation = append(callOpts.GetLocation, gax.WithClientMetrics(metrics))

callOpts.ListLocations = append(callOpts.ListLocations, gax.WithClientMetrics(metrics))

callOpts.GetIamPolicy = append(callOpts.GetIamPolicy, gax.WithClientMetrics(metrics))

callOpts.SetIamPolicy = append(callOpts.SetIamPolicy, gax.WithClientMetrics(metrics))

callOpts.TestIamPermissions = append(callOpts.TestIamPermissions, gax.WithClientMetrics(metrics))

callOpts.GetOperation = append(callOpts.GetOperation, gax.WithClientMetrics(metrics))

}

lroOpts := []option.ClientOption{

option.WithHTTPClient(httpClient),

option.WithEndpoint(endpoint),

}

opClient, err := lroauto.NewOperationsRESTClient(ctx, lroOpts...)

if err != nil {

return nil, err

}

c.LROClient = &opClient

return &AutokeyClient{internalClient: c, CallOptions: callOpts}, nil

}

func defaultAutokeyRESTClientOptions() []option.ClientOption {

return []option.ClientOption{

internaloption.WithDefaultEndpoint("https://cloudkms.googleapis.com"),

internaloption.WithDefaultEndpointTemplate("https://cloudkms.UNIVERSE_DOMAIN"),

internaloption.WithDefaultMTLSEndpoint("https://cloudkms.mtls.googleapis.com"),

internaloption.WithDefaultUniverseDomain("googleapis.com"),

internaloption.WithDefaultAudience("https://cloudkms.googleapis.com/"),

internaloption.WithDefaultScopes(DefaultAuthScopes()...),

internaloption.EnableNewAuthLibrary(),

}

}

// setGoogleClientInfo sets the name and version of the application in

// the `x-goog-api-client` header passed on each request. Intended for

// use by Google-written clients.

func (c *autokeyRESTClient) setGoogleClientInfo(keyval ...string) {

kv := append([]string{"gl-go", gax.GoVersion}, keyval...)

kv = append(kv, "gapic", getVersionClient(), "gax", gax.Version, "rest", "UNKNOWN", "pb", protoVersion)

c.xGoogHeaders = []string{

"x-goog-api-client", gax.XGoogHeader(kv...),

}

}

// Close closes the connection to the API service. The user should invoke this when

// the client is no longer required.

func (c *autokeyRESTClient) Close() error {

// Replace httpClient with nil to force cleanup.

c.httpClient = nil

return nil

}

// Connection returns a connection to the API service.

//

// Deprecated: This method always returns nil.

func (c *autokeyRESTClient) Connection() *grpc.ClientConn {

return nil

}

func (c *autokeyGRPCClient) CreateKeyHandle(ctx context.Context, req *kmspb.CreateKeyHandleRequest, opts ...gax.CallOption) (*CreateKeyHandleOperation, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "parent", url.QueryEscape(req.GetParent()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//cloudkms.googleapis.com/%v", req.GetParent()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.kms.v1.Autokey/CreateKeyHandle")

}

opts = append((*c.CallOptions).CreateKeyHandle[0:len((*c.CallOptions).CreateKeyHandle):len((*c.CallOptions).CreateKeyHandle)], opts...)

var resp *longrunningpb.Operation

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.autokeyClient.CreateKeyHandle, req, settings.GRPC, c.logger, "CreateKeyHandle")

return err

}, opts...)

if err != nil {

return nil, err

}

return &CreateKeyHandleOperation{

lro: longrunning.InternalNewOperation(*c.LROClient, resp),

}, nil

}

func (c *autokeyGRPCClient) GetKeyHandle(ctx context.Context, req *kmspb.GetKeyHandleRequest, opts ...gax.CallOption) (*kmspb.KeyHandle, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "name", url.QueryEscape(req.GetName()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//cloudkms.googleapis.com/%v", req.GetName()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.kms.v1.Autokey/GetKeyHandle")

}

opts = append((*c.CallOptions).GetKeyHandle[0:len((*c.CallOptions).GetKeyHandle):len((*c.CallOptions).GetKeyHandle)], opts...)

var resp *kmspb.KeyHandle

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.autokeyClient.GetKeyHandle, req, settings.GRPC, c.logger, "GetKeyHandle")

return err

}, opts...)

if err != nil {

return nil, err

}

return resp, nil

}

func (c *autokeyGRPCClient) ListKeyHandles(ctx context.Context, req *kmspb.ListKeyHandlesRequest, opts ...gax.CallOption) *KeyHandleIterator {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "parent", url.QueryEscape(req.GetParent()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//cloudkms.googleapis.com/%v", req.GetParent()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.kms.v1.Autokey/ListKeyHandles")

}

opts = append((*c.CallOptions).ListKeyHandles[0:len((*c.CallOptions).ListKeyHandles):len((*c.CallOptions).ListKeyHandles)], opts...)

it := &KeyHandleIterator{}

req = proto.Clone(req).(*kmspb.ListKeyHandlesRequest)

it.InternalFetch = func(pageSize int, pageToken string) ([]*kmspb.KeyHandle, string, error) {

resp := &kmspb.ListKeyHandlesResponse{}

if pageToken != "" {

req.PageToken = pageToken

}

if pageSize > math.MaxInt32 {

req.PageSize = math.MaxInt32

} else if pageSize != 0 {

req.PageSize = int32(pageSize)

}

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.autokeyClient.ListKeyHandles, req, settings.GRPC, c.logger, "ListKeyHandles")

return err

}, opts...)

if err != nil {

return nil, "", err

}

it.Response = resp

return resp.GetKeyHandles(), resp.GetNextPageToken(), nil

}

fetch := func(pageSize int, pageToken string) (string, error) {

items, nextPageToken, err := it.InternalFetch(pageSize, pageToken)

if err != nil {

return "", err

}

it.items = append(it.items, items...)

return nextPageToken, nil

}

it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)

it.pageInfo.MaxSize = int(req.GetPageSize())

it.pageInfo.Token = req.GetPageToken()

return it

}

func (c *autokeyGRPCClient) GetLocation(ctx context.Context, req *locationpb.GetLocationRequest, opts ...gax.CallOption) (*locationpb.Location, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "name", url.QueryEscape(req.GetName()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.location.Locations/GetLocation")

}

opts = append((*c.CallOptions).GetLocation[0:len((*c.CallOptions).GetLocation):len((*c.CallOptions).GetLocation)], opts...)

var resp *locationpb.Location

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.locationsClient.GetLocation, req, settings.GRPC, c.logger, "GetLocation")

return err

}, opts...)

if err != nil {

return nil, err

}

return resp, nil

}

func (c *autokeyGRPCClient) ListLocations(ctx context.Context, req *locationpb.ListLocationsRequest, opts ...gax.CallOption) *LocationIterator {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "name", url.QueryEscape(req.GetName()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.location.Locations/ListLocations")

}

opts = append((*c.CallOptions).ListLocations[0:len((*c.CallOptions).ListLocations):len((*c.CallOptions).ListLocations)], opts...)

it := &LocationIterator{}

req = proto.Clone(req).(*locationpb.ListLocationsRequest)

it.InternalFetch = func(pageSize int, pageToken string) ([]*locationpb.Location, string, error) {

resp := &locationpb.ListLocationsResponse{}

if pageToken != "" {

req.PageToken = pageToken

}

if pageSize > math.MaxInt32 {

req.PageSize = math.MaxInt32

} else if pageSize != 0 {

req.PageSize = int32(pageSize)

}

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.locationsClient.ListLocations, req, settings.GRPC, c.logger, "ListLocations")

return err

}, opts...)

if err != nil {

return nil, "", err

}

it.Response = resp

return resp.GetLocations(), resp.GetNextPageToken(), nil

}

fetch := func(pageSize int, pageToken string) (string, error) {

items, nextPageToken, err := it.InternalFetch(pageSize, pageToken)

if err != nil {

return "", err

}

it.items = append(it.items, items...)

return nextPageToken, nil

}

it.pageInfo, it.nextFunc = iterator.NewPageInfo(fetch, it.bufLen, it.takeBuf)

it.pageInfo.MaxSize = int(req.GetPageSize())

it.pageInfo.Token = req.GetPageToken()

return it

}

func (c *autokeyGRPCClient) GetIamPolicy(ctx context.Context, req *iampb.GetIamPolicyRequest, opts ...gax.CallOption) (*iampb.Policy, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "resource", url.QueryEscape(req.GetResource()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//iam-meta-api.googleapis.com/%v", req.GetResource()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.iam.v1.IAMPolicy/GetIamPolicy")

}

opts = append((*c.CallOptions).GetIamPolicy[0:len((*c.CallOptions).GetIamPolicy):len((*c.CallOptions).GetIamPolicy)], opts...)

var resp *iampb.Policy

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.iamPolicyClient.GetIamPolicy, req, settings.GRPC, c.logger, "GetIamPolicy")

return err

}, opts...)

if err != nil {

return nil, err

}

return resp, nil

}

func (c *autokeyGRPCClient) SetIamPolicy(ctx context.Context, req *iampb.SetIamPolicyRequest, opts ...gax.CallOption) (*iampb.Policy, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "resource", url.QueryEscape(req.GetResource()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//iam-meta-api.googleapis.com/%v", req.GetResource()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.iam.v1.IAMPolicy/SetIamPolicy")

}

opts = append((*c.CallOptions).SetIamPolicy[0:len((*c.CallOptions).SetIamPolicy):len((*c.CallOptions).SetIamPolicy)], opts...)

var resp *iampb.Policy

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.iamPolicyClient.SetIamPolicy, req, settings.GRPC, c.logger, "SetIamPolicy")

return err

}, opts...)

if err != nil {

return nil, err

}

return resp, nil

}

func (c *autokeyGRPCClient) TestIamPermissions(ctx context.Context, req *iampb.TestIamPermissionsRequest, opts ...gax.CallOption) (*iampb.TestIamPermissionsResponse, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "resource", url.QueryEscape(req.GetResource()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//iam-meta-api.googleapis.com/%v", req.GetResource()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.iam.v1.IAMPolicy/TestIamPermissions")

}

opts = append((*c.CallOptions).TestIamPermissions[0:len((*c.CallOptions).TestIamPermissions):len((*c.CallOptions).TestIamPermissions)], opts...)

var resp *iampb.TestIamPermissionsResponse

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.iamPolicyClient.TestIamPermissions, req, settings.GRPC, c.logger, "TestIamPermissions")

return err

}, opts...)

if err != nil {

return nil, err

}

return resp, nil

}

func (c *autokeyGRPCClient) GetOperation(ctx context.Context, req *longrunningpb.GetOperationRequest, opts ...gax.CallOption) (*longrunningpb.Operation, error) {

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "name", url.QueryEscape(req.GetName()))}

hds = append(c.xGoogHeaders, hds...)

ctx = gax.InsertMetadataIntoOutgoingContext(ctx, hds...)

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.longrunning.Operations/GetOperation")

}

opts = append((*c.CallOptions).GetOperation[0:len((*c.CallOptions).GetOperation):len((*c.CallOptions).GetOperation)], opts...)

var resp *longrunningpb.Operation

err := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

var err error

resp, err = executeRPC(ctx, c.operationsClient.GetOperation, req, settings.GRPC, c.logger, "GetOperation")

return err

}, opts...)

if err != nil {

return nil, err

}

return resp, nil

}

// CreateKeyHandle creates a new KeyHandle, triggering the

// provisioning of a new CryptoKey for CMEK

// use with the given resource type in the configured key project and the same

// location. GetOperation should

// be used to resolve the resulting long-running operation and get the

// resulting KeyHandle and

// CryptoKey.

func (c *autokeyRESTClient) CreateKeyHandle(ctx context.Context, req *kmspb.CreateKeyHandleRequest, opts ...gax.CallOption) (*CreateKeyHandleOperation, error) {

m := protojson.MarshalOptions{AllowPartial: true, UseEnumNumbers: true}

body := req.GetKeyHandle()

jsonReq, err := m.Marshal(body)

if err != nil {

return nil, err

}

baseUrl, err := url.Parse(c.endpoint)

if err != nil {

return nil, err

}

baseUrl.Path += fmt.Sprintf("/v1/%v/keyHandles", req.GetParent())

params := url.Values{}

params.Add("$alt", "json;enum-encoding=int")

if req.GetKeyHandleId() != "" {

params.Add("keyHandleId", fmt.Sprintf("%v", req.GetKeyHandleId()))

}

baseUrl.RawQuery = params.Encode()

// Build HTTP headers from client and context metadata.

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "parent", url.QueryEscape(req.GetParent()))}

hds = append(c.xGoogHeaders, hds...)

hds = append(hds, "Content-Type", "application/json")

headers := gax.BuildHeaders(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//cloudkms.googleapis.com/%v", req.GetParent()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.kms.v1.Autokey/CreateKeyHandle")

ctx = callctx.WithTelemetryContext(ctx, "url_template", "/v1/{parent=projects/*/locations/*}/keyHandles")

}

unm := protojson.UnmarshalOptions{AllowPartial: true, DiscardUnknown: true}

resp := &longrunningpb.Operation{}

e := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

if settings.Path != "" {

baseUrl.Path = settings.Path

}

httpReq, err := http.NewRequest("POST", baseUrl.String(), bytes.NewReader(jsonReq))

if err != nil {

return err

}

httpReq = httpReq.WithContext(ctx)

httpReq.Header = headers

buf, err := executeHTTPRequest(ctx, c.httpClient, httpReq, c.logger, jsonReq, "CreateKeyHandle")

if err != nil {

return err

}

if err := unm.Unmarshal(buf, resp); err != nil {

return err

}

return nil

}, opts...)

if e != nil {

return nil, e

}

override := fmt.Sprintf("/v1/%s", resp.GetName())

return &CreateKeyHandleOperation{

lro: longrunning.InternalNewOperation(*c.LROClient, resp),

pollPath: override,

}, nil

}

// GetKeyHandle returns the KeyHandle.

func (c *autokeyRESTClient) GetKeyHandle(ctx context.Context, req *kmspb.GetKeyHandleRequest, opts ...gax.CallOption) (*kmspb.KeyHandle, error) {

baseUrl, err := url.Parse(c.endpoint)

if err != nil {

return nil, err

}

baseUrl.Path += fmt.Sprintf("/v1/%v", req.GetName())

params := url.Values{}

params.Add("$alt", "json;enum-encoding=int")

baseUrl.RawQuery = params.Encode()

// Build HTTP headers from client and context metadata.

hds := []string{"x-goog-request-params", fmt.Sprintf("%s=%v", "name", url.QueryEscape(req.GetName()))}

hds = append(c.xGoogHeaders, hds...)

hds = append(hds, "Content-Type", "application/json")

headers := gax.BuildHeaders(ctx, hds...)

if gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "resource_name", fmt.Sprintf("//cloudkms.googleapis.com/%v", req.GetName()))

}

if gax.IsFeatureEnabled("METRICS") || gax.IsFeatureEnabled("TRACING") || gax.IsFeatureEnabled("LOGGING") {

ctx = callctx.WithTelemetryContext(ctx, "rpc_method", "google.cloud.kms.v1.Autokey/GetKeyHandle")

ctx = callctx.WithTelemetryContext(ctx, "url_template", "/v1/{name=projects/*/locations/*/keyHandles/*}")

}

opts = append((*c.CallOptions).GetKeyHandle[0:len((*c.CallOptions).GetKeyHandle):len((*c.CallOptions).GetKeyHandle)], opts...)

unm := protojson.UnmarshalOptions{AllowPartial: true, DiscardUnknown: true}

resp := &kmspb.KeyHandle{}

e := gax.Invoke(ctx, func(ctx context.Context, settings gax.CallSettings) error {

if settings.Path != "" {

baseUrl.Path = settings.Path

}

httpReq, err := http.NewRequest("GET", baseUrl.String(), nil)

if err != nil {

return err

}

httpReq = httpReq.WithContext(ctx)

httpReq.Header = headers

buf, err := executeHTTPRequest(ctx, c.httpClient, httpReq, c.logger, nil, "GetKeyHandle")

if err != nil {

return err

}

if err := unm.Unmarshal(buf, resp); err != nil {

return err

}

return nil

}, opts...)

if e != nil {

return nil, e

}

return resp, nil

}

// ListKeyHandles lists KeyHandles.

func (c *autokeyRESTClient) ListKeyHandles(ctx context.Context, req *kmspb.ListKeyHandlesRequest, opts ...gax.CallOption) *KeyHandleIterator {

it := &KeyHandleIterator{}

req = proto.Clone(req).(*kmspb.ListKeyHandlesRequest)