Files
teleport/integrations/operator/controllers/resources/openssheiceserverv2_controller.go
T
Hugo ShakaandNoah Stride 29e1e5bc14 Refactor operator to support several types of resources + propagate labels (#38992)
* Refactor operator resource reconciler to support resource153 + add label propagation

I initially just wanted to add label propagation (which is required for
OpenSSH and OpenSSHEICE servers).

However I ran into an issue: not all Teleport resources implement the
`ResourceWithLabels` interface.  Basically we have 4 cases:
- resources implementing `ResourceWithLabels` (e.g. User, Role, ...)
- resources implementing `ResourceWithOrigin` but not
  `ResourceWithLabels` (e.g. GitHub connector, ProvisionToken, ...)
- resources that have been tweaked to implement only the operator
  `TeleportResource` interface (`LoginRule`)
- resources implementing `Resource153`

To support all those resoucres, we can either:
- edit each resource so they expose the same methods (e.g.
  ProvisionToken exposes `GetLabels` instead of `GetStaticLabels`)
- wrap each ressource in a wrapper that implements the operator
  TeleportResource interface
- add multiple reconcilers, one for each resource kind

I don't think it's sane for the operator to force changes on Teleport
resources. Especially as those are known to be non consistent. I also
don't want the operator to maintain a copy of each resource, this is not
viable on the long term.

For those reasons I chose the 3rd approach and built 3 reconcilers. Then
I deducplicated the code to end up with a generic reconciler that can be
tuned by passing an adapter. The adapter is an empty struct that has all
the methods to extract the information the operator requires from the
resource.

The resulting PR is large and heavily uses generics but the amount of
logical changes is quite low (the only diff is that we now copy labels).
I can walk you through the PR if needed.

With this PR, the operator supports properly any resource implementing
`types.Resource153`, this will be useful to add `TeleportBotV1` support.

* Update integrations/operator/controllers/reconcilers/base.go

Co-authored-by: Noah Stride <noah.stride@goteleport.com>

* use structured logging

---------

Co-authored-by: Noah Stride <noah.stride@goteleport.com>
2024-03-07 21:17:13 +00:00

88 lines
3.1 KiB
Go

/*
* Teleport
* Copyright (C) 2023 Gravitational, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package resources
import (
"context"
"github.com/gravitational/trace"
kclient "sigs.k8s.io/controller-runtime/pkg/client"
"github.com/gravitational/teleport/api/client"
"github.com/gravitational/teleport/api/defaults"
"github.com/gravitational/teleport/api/types"
resourcesv1 "github.com/gravitational/teleport/integrations/operator/apis/resources/v1"
"github.com/gravitational/teleport/integrations/operator/controllers"
"github.com/gravitational/teleport/integrations/operator/controllers/reconcilers"
)
// openSSHEICEServerClient implements TeleportResourceClient and offers CRUD
// methods needed to reconcile OpenSSH EC2 ICE servers.
type openSSHEICEServerClient struct {
teleportClient *client.Client
}
// Get gets the Teleport OpenSSHEICE server of a given name.
func (r openSSHEICEServerClient) Get(ctx context.Context, name string) (types.Server, error) {
server, err := r.teleportClient.GetNode(ctx, defaults.Namespace, name)
if err != nil {
return server, trace.Wrap(err)
}
if subKind := server.GetSubKind(); subKind != types.SubKindOpenSSHEICENode {
return nil, trace.CompareFailed(
"Wrong server subKind, was expecting %q, got %q",
types.SubKindOpenSSHEICENode,
subKind,
)
}
return server, nil
}
// Create creates a Teleport OpenSSHEICE server.
func (r openSSHEICEServerClient) Create(ctx context.Context, server types.Server) error {
_, err := r.teleportClient.UpsertNode(ctx, server)
return trace.Wrap(err)
}
// Update updates a Teleport OpenSSHEICE server.
func (r openSSHEICEServerClient) Update(ctx context.Context, server types.Server) error {
_, err := r.teleportClient.UpsertNode(ctx, server)
return trace.Wrap(err)
}
// Delete deletes a Teleport OpenSSHEICE server.
func (r openSSHEICEServerClient) Delete(ctx context.Context, name string) error {
return trace.Wrap(r.teleportClient.DeleteNode(ctx, defaults.Namespace, name))
}
// NewOpenSSHEICEServerV2Reconciler instantiates a new Kubernetes controller
// reconciling OpenSSHEICE server resources.
func NewOpenSSHEICEServerV2Reconciler(client kclient.Client, tClient *client.Client) (controllers.Reconciler, error) {
serverClient := &openSSHEICEServerClient{
teleportClient: tClient,
}
resourceReconciler, err := reconcilers.NewTeleportResourceWithLabelsReconciler[types.Server, *resourcesv1.TeleportOpenSSHEICEServerV2](
client,
serverClient,
)
return resourceReconciler, trace.Wrap(err, "building teleport resource reconciler")
}