mirror of
https://github.com/arangodb/kube-arangodb.git
synced 2024-12-15 17:51:03 +00:00
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
//
|
|
// DISCLAIMER
|
|
//
|
|
// Copyright 2023 ArangoDB GmbH, Cologne, Germany
|
|
//
|
|
// 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.
|
|
//
|
|
// Copyright holder is ArangoDB GmbH, Cologne, Germany
|
|
//
|
|
|
|
package license
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
|
|
"k8s.io/apimachinery/pkg/api/errors"
|
|
meta "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
|
|
api "github.com/arangodb/kube-arangodb/pkg/apis/deployment/v1"
|
|
"github.com/arangodb/kube-arangodb/pkg/util/constants"
|
|
"github.com/arangodb/kube-arangodb/pkg/util/k8sutil"
|
|
)
|
|
|
|
func NewArangoDeploymentLicenseLoader(client kubernetes.Interface, deployment *api.ArangoDeployment) Loader {
|
|
return arangoDeploymentLicenseLoader{
|
|
client: client,
|
|
deployment: deployment,
|
|
}
|
|
}
|
|
|
|
type arangoDeploymentLicenseLoader struct {
|
|
client kubernetes.Interface
|
|
|
|
deployment *api.ArangoDeployment
|
|
}
|
|
|
|
func (a arangoDeploymentLicenseLoader) Refresh(ctx context.Context) (string, bool, error) {
|
|
spec := a.deployment.GetAcceptedSpec()
|
|
|
|
if !spec.License.HasSecretName() {
|
|
return "", false, nil
|
|
}
|
|
|
|
secret, err := a.client.CoreV1().Secrets(a.deployment.GetNamespace()).Get(ctx, spec.License.GetSecretName(), meta.GetOptions{})
|
|
if err != nil {
|
|
if errors.IsNotFound(err) {
|
|
return "", false, nil
|
|
}
|
|
|
|
return "", false, err
|
|
}
|
|
|
|
var licenseData []byte
|
|
|
|
if lic, ok := secret.Data[constants.SecretKeyV2License]; ok {
|
|
licenseData = lic
|
|
} else if lic2, ok := secret.Data[constants.SecretKeyV2Token]; ok {
|
|
licenseData = lic2
|
|
}
|
|
|
|
if len(licenseData) == 0 {
|
|
return "", false, nil
|
|
}
|
|
|
|
if !k8sutil.IsJSON(licenseData) {
|
|
d, err := base64.StdEncoding.DecodeString(string(licenseData))
|
|
if err != nil {
|
|
return "", false, err
|
|
}
|
|
|
|
licenseData = d
|
|
}
|
|
|
|
return string(licenseData), true, nil
|
|
}
|