Add KubeProvision service implementation.

api/client/client.go

@@ -52,6 +52,7 @@ import (
 	crownjewelapi "github.com/gravitational/teleport/api/client/crownjewel"
 	"github.com/gravitational/teleport/api/client/discoveryconfig"
 	"github.com/gravitational/teleport/api/client/externalauditstorage"
+	"github.com/gravitational/teleport/api/client/kubeprovision"
 	kubewaitingcontainerclient "github.com/gravitational/teleport/api/client/kubewaitingcontainer"
 	"github.com/gravitational/teleport/api/client/okta"
 	"github.com/gravitational/teleport/api/client/proto"
@@ -72,6 +73,7 @@ import (
 	externalauditstoragev1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/externalauditstorage/v1"
 	integrationpb "github.com/gravitational/teleport/api/gen/proto/go/teleport/integration/v1"
 	kubeproto "github.com/gravitational/teleport/api/gen/proto/go/teleport/kube/v1"
+	kubeprovisionv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/kubeprovision/v1"
 	kubewaitingcontainerpb "github.com/gravitational/teleport/api/gen/proto/go/teleport/kubewaitingcontainer/v1"
 	loginrulepb "github.com/gravitational/teleport/api/gen/proto/go/teleport/loginrule/v1"
 	machineidv1pb "github.com/gravitational/teleport/api/gen/proto/go/teleport/machineid/v1"
@@ -3131,6 +3133,16 @@ func (c *Client) DeleteAllKubernetesClusters(ctx context.Context) error {
 	return trace.Wrap(err)
 }
 
+// GetKubeProvision returns the requested KubeProvision resource.
+func (c *Client) GetKubeProvision(ctx context.Context, name string) (*kubeprovisionv1.KubeProvision, error) {
+	return c.KubeProvisionServiceClient().GetKubeProvision(ctx, name)
+}
+
+// ListKubeProvisions lists KubeProvisions resources
+func (c *Client) ListKubeProvisions(ctx context.Context, pageSize int, pageToken string) ([]*kubeprovisionv1.KubeProvision, string, error) {
+	return c.KubeProvisionServiceClient().ListKubeProvisions(ctx, pageSize, pageToken)
+}
+
 // GetKubernetesWaitingContainerClient an unadorned KubeWaitingContainers
 // client, using the underlying Auth gRPC connection.
 func (c *Client) GetKubernetesWaitingContainerClient() *kubewaitingcontainerclient.Client {
@@ -4604,6 +4616,11 @@ func (c *Client) DatabaseObjectClient() dbobjectv1.DatabaseObjectServiceClient {
 	return dbobjectv1.NewDatabaseObjectServiceClient(c.conn)
 }
 
+// KubeProvisionServiceClient returns a client for managing kube provisions.
+func (c *Client) KubeProvisionServiceClient() *kubeprovision.Client {
+	return kubeprovision.NewClient(kubeprovisionv1.NewKubeProvisionServiceClient(c.conn))
+}
+
 // DiscoveryConfigClient returns a DiscoveryConfig client.
 // Clients connecting to older Teleport versions, still get an DiscoveryConfig client
 // when calling this method, but all RPCs will return "not implemented" errors

api/client/events.go

@@ -21,6 +21,7 @@ import (
 	accessmonitoringrulesv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/accessmonitoringrules/v1"
 	crownjewelv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/crownjewel/v1"
 	dbobjectv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/dbobject/v1"
+	kubeprovisionv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/kubeprovision/v1"
 	kubewaitingcontainerpb "github.com/gravitational/teleport/api/gen/proto/go/teleport/kubewaitingcontainer/v1"
 	machineidv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/machineid/v1"
 	notificationsv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/notifications/v1"
@@ -77,6 +78,10 @@ func EventToGRPC(in types.Event) (*proto.Event, error) {
 			out.Resource = &proto.Event_CrownJewel{
 				CrownJewel: r,
 			}
+		case *kubeprovisionv1.KubeProvision:
+			out.Resource = &proto.Event_KubeProvision{
+				KubeProvision: r,
+			}
 		case *dbobjectv1.DatabaseObject:
 			out.Resource = &proto.Event_DatabaseObject{
 				DatabaseObject: r,
@@ -519,6 +524,9 @@ func EventFromGRPC(in *proto.Event) (*types.Event, error) {
 	} else if r := in.GetBotInstance(); r != nil {
 		out.Resource = types.Resource153ToLegacy(r)
 		return &out, nil
+	} else if r := in.GetKubeProvision(); r != nil {
+		out.Resource = types.Resource153ToLegacy(r)
+		return &out, nil
 	} else {
 		return nil, trace.BadParameter("received unsupported resource %T", in.Resource)
 	}

api/client/kubeprovision/kubeprovision.go

@@ -0,0 +1,84 @@
+// Copyright 2024 Gravitational, Inc.
+//
+// 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
+//
+//     http://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.
+
+package kubeprovision
+
+import (
+	"context"
+
+	"github.com/gravitational/trace"
+
+	kubeprovisionv1 "github.com/gravitational/teleport/api/gen/proto/go/teleport/kubeprovision/v1"
+)
+
+// Client is a client for the Kube Provision API.
+type Client struct {
+	grpcClient kubeprovisionv1.KubeProvisionServiceClient
+}
+
+// NewClient creates a new Kube Provision service client.
+func NewClient(grpcClient kubeprovisionv1.KubeProvisionServiceClient) *Client {
+	return &Client{
+		grpcClient: grpcClient,
+	}
+}
+
+// ListKubeProvisions returns a list of Kube Provisions.
+func (c *Client) ListKubeProvisions(ctx context.Context, pageSize int, nextToken string) ([]*kubeprovisionv1.KubeProvision, string, error) {
+	resp, err := c.grpcClient.ListKubeProvisions(ctx, &kubeprovisionv1.ListKubeProvisionsRequest{
+		PageSize:  int32(pageSize),
+		PageToken: nextToken,
+	})
+	if err != nil {
+		return nil, "", trace.Wrap(err)
+	}
+
+	return resp.KubeProvisions, resp.NextPageToken, nil
+}
+
+// CreateKubeProvision creates a new Kube Provision.
+func (c *Client) CreateKubeProvision(ctx context.Context, req *kubeprovisionv1.KubeProvision) (*kubeprovisionv1.KubeProvision, error) {
+	rsp, err := c.grpcClient.CreateKubeProvision(ctx, &kubeprovisionv1.CreateKubeProvisionRequest{KubeProvision: req})
+	return rsp, trace.Wrap(err)
+}
+
+// GetKubeProvision returns a Kube Provision by name.
+func (c *Client) GetKubeProvision(ctx context.Context, name string) (*kubeprovisionv1.KubeProvision, error) {
+	rsp, err := c.grpcClient.GetKubeProvision(ctx, &kubeprovisionv1.GetKubeProvisionRequest{Name: name})
+	return rsp, trace.Wrap(err)
+}
+
+// UpdateKubeProvision updates an existing Kube Provision.
+func (c *Client) UpdateKubeProvision(ctx context.Context, req *kubeprovisionv1.KubeProvision) (*kubeprovisionv1.KubeProvision, error) {
+	rsp, err := c.grpcClient.UpdateKubeProvision(ctx, &kubeprovisionv1.UpdateKubeProvisionRequest{KubeProvision: req})
+	return rsp, trace.Wrap(err)
+}
+
+// UpsertKubeProvision upserts a Kube Provision.
+func (c *Client) UpsertKubeProvision(ctx context.Context, req *kubeprovisionv1.KubeProvision) (*kubeprovisionv1.KubeProvision, error) {
+	rsp, err := c.grpcClient.UpsertKubeProvision(ctx, &kubeprovisionv1.UpsertKubeProvisionRequest{KubeProvision: req})
+	return rsp, trace.Wrap(err)
+}
+
+// DeleteKubeProvision deletes a Kube Provision.
+func (c *Client) DeleteKubeProvision(ctx context.Context, name string) error {
+	_, err := c.grpcClient.DeleteKubeProvision(ctx, &kubeprovisionv1.DeleteKubeProvisionRequest{Name: name})
+	return trace.Wrap(err)
+}
+
+// DeleteAllKubeProvisions is not implemented for the client.
+// Deprecated: Can't delete all KubeProvisions over gRPC.
+func (c *Client) DeleteAllKubeProvisions(_ context.Context) error {
+	return trace.NotImplemented("DeleteAllKubeProvisions is not implemented for gRPC.")
+}