Files
teleport/integrations/operator/controllers/resources/provision_token_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

73 lines
2.7 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/types"
resourcesv2 "github.com/gravitational/teleport/integrations/operator/apis/resources/v2"
"github.com/gravitational/teleport/integrations/operator/controllers"
"github.com/gravitational/teleport/integrations/operator/controllers/reconcilers"
)
// provisionTokenClient implements TeleportResourceClient and offers CRUD methods needed to reconcile provision tokens
type provisionTokenClient struct {
teleportClient *client.Client
}
// Get gets the Teleport provision token of a given name
func (r provisionTokenClient) Get(ctx context.Context, name string) (types.ProvisionToken, error) {
token, err := r.teleportClient.GetToken(ctx, name)
return token, trace.Wrap(err)
}
// Create creates a Teleport provision token
func (r provisionTokenClient) Create(ctx context.Context, token types.ProvisionToken) error {
return trace.Wrap(r.teleportClient.UpsertToken(ctx, token))
}
// Update updates a Teleport provision token
func (r provisionTokenClient) Update(ctx context.Context, token types.ProvisionToken) error {
return trace.Wrap(r.teleportClient.UpsertToken(ctx, token))
}
// Delete deletes a Teleport provision token
func (r provisionTokenClient) Delete(ctx context.Context, name string) error {
return trace.Wrap(r.teleportClient.DeleteToken(ctx, name))
}
// NewProvisionTokenReconciler instantiates a new Kubernetes controller reconciling provision token resources
func NewProvisionTokenReconciler(client kclient.Client, tClient *client.Client) (controllers.Reconciler, error) {
tokenClient := &provisionTokenClient{
teleportClient: tClient,
}
resourceReconciler, err := reconcilers.NewTeleportResourceWithoutLabelsReconciler[types.ProvisionToken, *resourcesv2.TeleportProvisionToken](
client,
tokenClient,
)
return resourceReconciler, trace.Wrap(err, "building teleport resource reconciler")
}