openpitrix crd

Signed-off-by: LiHui <andrewli@yunify.com> delete helm repo, release and app Signed-off-by: LiHui <andrewli@yunify.com> Fix Dockerfile Signed-off-by: LiHui <andrewli@yunify.com> add unit test for category controller Signed-off-by: LiHui <andrewli@yunify.com> resource api Signed-off-by: LiHui <andrewli@yunify.com> miscellaneous Signed-off-by: LiHui <andrewli@yunify.com> resource api Signed-off-by: LiHui <andrewli@yunify.com> add s3 repo indx Signed-off-by: LiHui <andrewli@yunify.com> attachment api Signed-off-by: LiHui <andrewli@yunify.com> repo controller test Signed-off-by: LiHui <andrewli@yunify.com> application controller test Signed-off-by: LiHui <andrewli@yunify.com> release metric Signed-off-by: LiHui <andrewli@yunify.com> helm release controller test Signed-off-by: LiHui <andrewli@yunify.com> move constants to /pkg/apis/application Signed-off-by: LiHui <andrewli@yunify.com> remove unused code Signed-off-by: LiHui <andrewli@yunify.com> add license header Signed-off-by: LiHui <andrewli@yunify.com> Fix bugs Signed-off-by: LiHui <andrewli@yunify.com> cluster cluent Signed-off-by: LiHui <andrewli@yunify.com> format code Signed-off-by: LiHui <andrewli@yunify.com> move workspace,cluster from spec to labels Signed-off-by: LiHui <andrewli@yunify.com> add license header Signed-off-by: LiHui <andrewli@yunify.com> openpitrix test Signed-off-by: LiHui <andrewli@yunify.com> add worksapce labels for app in appstore Signed-off-by: LiHui <andrewli@yunify.com>
2020-12-23 15:24:30 +08:00
parent 737639020b
commit 83e6221f3a
193 changed files with 19634 additions and 6039 deletions
--- a/pkg/controller/workspacetemplate/workspacetemplate_controller.go
+++ b/pkg/controller/workspacetemplate/workspacetemplate_controller.go
@@ -25,16 +25,19 @@ import (
 	rbacv1 "k8s.io/api/rbac/v1"
 	"k8s.io/apimachinery/pkg/api/errors"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/labels"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/apimachinery/pkg/util/yaml"
 	"k8s.io/client-go/kubernetes/scheme"
 	"k8s.io/client-go/tools/record"
+	"kubesphere.io/kubesphere/pkg/apis/application/v1alpha1"
 	iamv1alpha2 "kubesphere.io/kubesphere/pkg/apis/iam/v1alpha2"
 	tenantv1alpha1 "kubesphere.io/kubesphere/pkg/apis/tenant/v1alpha1"
 	tenantv1alpha2 "kubesphere.io/kubesphere/pkg/apis/tenant/v1alpha2"
 	typesv1beta1 "kubesphere.io/kubesphere/pkg/apis/types/v1beta1"
 	"kubesphere.io/kubesphere/pkg/constants"
 	controllerutils "kubesphere.io/kubesphere/pkg/controller/utils/controller"
+	"kubesphere.io/kubesphere/pkg/utils/sliceutil"
 	"reflect"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
@@ -43,7 +46,8 @@ import (
 )

 const (
-	controllerName = "workspacetemplate-controller"
+	controllerName             = "workspacetemplate-controller"
+	workspaceTemplateFinalizer = "finalizers.workspacetemplate.kubesphere.io"
 )

 // Reconciler reconciles a WorkspaceRoleBinding object
@@ -88,6 +92,37 @@ func (r *Reconciler) Reconcile(req ctrl.Request) (ctrl.Result, error) {
 		return ctrl.Result{}, client.IgnoreNotFound(err)
 	}

+	if workspaceTemplate.ObjectMeta.DeletionTimestamp.IsZero() {
+		// The object is not being deleted, so if it does not have our finalizer,
+		// then lets add the finalizer and update the object.
+		if !sliceutil.HasString(workspaceTemplate.ObjectMeta.Finalizers, workspaceTemplateFinalizer) {
+			workspaceTemplate.ObjectMeta.Finalizers = append(workspaceTemplate.ObjectMeta.Finalizers, workspaceTemplateFinalizer)
+			if err := r.Update(rootCtx, workspaceTemplate); err != nil {
+				return ctrl.Result{}, err
+			}
+		}
+	} else {
+		// The object is being deleted
+		if sliceutil.HasString(workspaceTemplate.ObjectMeta.Finalizers, workspaceTemplateFinalizer) {
+			if err := r.deleteOpenPitrixResourcesInWorkspace(rootCtx, workspaceTemplate.Name); err != nil {
+				logger.Error(err, "delete resource in workspace template failed")
+				return ctrl.Result{}, err
+			}
+
+			// remove our finalizer from the list and update it.
+			workspaceTemplate.ObjectMeta.Finalizers = sliceutil.RemoveString(workspaceTemplate.ObjectMeta.Finalizers, func(item string) bool {
+				return item == workspaceTemplateFinalizer
+			})
+			logger.V(4).Info("update workspace template")
+			if err := r.Update(rootCtx, workspaceTemplate); err != nil {
+				logger.Error(err, "update workspace template failed")
+				return ctrl.Result{}, err
+			}
+		}
+		// Our finalizer has finished, so the reconciler can do nothing.
+		return ctrl.Result{}, nil
+	}
+
 	if r.MultiClusterEnabled {
 		if err := r.multiClusterSync(rootCtx, logger, workspaceTemplate); err != nil {
 			return ctrl.Result{}, err
@@ -301,6 +336,81 @@ func (r *Reconciler) initManagerRoleBinding(ctx context.Context, logger logr.Log

 	return nil
 }
+func (r *Reconciler) deleteOpenPitrixResourcesInWorkspace(ctx context.Context, ws string) error {
+	if len(ws) == 0 {
+		return nil
+	}
+
+	var err error
+	// helm release, apps and appVersion only exist in host cluster. Delete these resource in workspace template controller
+	if err = r.deleteHelmReleases(ctx, ws); err != nil {
+		return err
+	}
+
+	if err = r.deleteHelmApps(ctx, ws); err != nil {
+		return err
+	}
+
+	if err = r.deleteHelmRepos(ctx, ws); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (r *Reconciler) deleteHelmApps(ctx context.Context, ws string) error {
+	if len(ws) == 0 {
+		return nil
+	}
+
+	apps := v1alpha1.HelmApplicationList{}
+	err := r.List(ctx, &apps, &client.ListOptions{LabelSelector: labels.SelectorFromSet(map[string]string{
+		constants.WorkspaceLabelKey: ws}),
+	})
+	if err != nil {
+		return err
+	}
+	for i := range apps.Items {
+		state := apps.Items[i].Status.State
+		// active and suspended applications belong to app store, they should not be removed here.
+		if !(state == v1alpha1.StateActive || state == v1alpha1.StateSuspended) {
+			err = r.Delete(ctx, &apps.Items[i])
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+// Delete all helm releases in the workspace ws
+func (r *Reconciler) deleteHelmReleases(ctx context.Context, ws string) error {
+	if len(ws) == 0 {
+		return nil
+	}
+	rls := &v1alpha1.HelmRelease{}
+	err := r.DeleteAllOf(ctx, rls, &client.DeleteAllOfOptions{
+		ListOptions: client.ListOptions{LabelSelector: labels.SelectorFromSet(map[string]string{
+			constants.WorkspaceLabelKey: ws,
+		}),
+		}})
+	return err
+}
+
+func (r *Reconciler) deleteHelmRepos(ctx context.Context, ws string) error {
+	if len(ws) == 0 {
+		return nil
+	}
+	rls := &v1alpha1.HelmRepo{}
+	err := r.DeleteAllOf(ctx, rls, &client.DeleteAllOfOptions{
+		ListOptions: client.ListOptions{LabelSelector: labels.SelectorFromSet(map[string]string{
+			constants.WorkspaceLabelKey: ws,
+		}),
+		}})
+
+	return err
+}

 func workspaceRoleBindingChanger(workspaceRoleBinding *iamv1alpha2.WorkspaceRoleBinding, workspace, username, workspaceRoleName string) controllerutil.MutateFn {
 	return func() error {