mirror of
https://github.com/argoproj/argo-cd
synced 2026-04-22 01:17:16 +00:00
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package generator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
"github.com/argoproj/argo-cd/v3/hack/gen-resources/util"
|
|
"github.com/argoproj/argo-cd/v3/pkg/apis/application/v1alpha1"
|
|
appclientset "github.com/argoproj/argo-cd/v3/pkg/client/clientset/versioned"
|
|
)
|
|
|
|
type ProjectGenerator struct {
|
|
clientSet *appclientset.Clientset
|
|
}
|
|
|
|
func NewProjectGenerator(clientSet *appclientset.Clientset) Generator {
|
|
return &ProjectGenerator{clientSet}
|
|
}
|
|
|
|
func (pg *ProjectGenerator) Generate(opts *util.GenerateOpts) error {
|
|
projects := pg.clientSet.ArgoprojV1alpha1().AppProjects(opts.Namespace)
|
|
for i := 0; i < opts.ProjectOpts.Samples; i++ {
|
|
log.Printf("Generate project #%v", i)
|
|
_, err := projects.Create(context.TODO(), &v1alpha1.AppProject{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
GenerateName: "project-",
|
|
Namespace: opts.Namespace,
|
|
Labels: labels,
|
|
},
|
|
Spec: v1alpha1.AppProjectSpec{
|
|
Description: "generated-project",
|
|
},
|
|
}, metav1.CreateOptions{})
|
|
if err != nil {
|
|
log.Printf("Project #%v failed to generate", i)
|
|
return fmt.Errorf("error in generated-project: %w", err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (pg *ProjectGenerator) Clean(opts *util.GenerateOpts) error {
|
|
log.Print("Clean projects")
|
|
projects := pg.clientSet.ArgoprojV1alpha1().AppProjects(opts.Namespace)
|
|
return projects.DeleteCollection(context.TODO(), metav1.DeleteOptions{}, metav1.ListOptions{
|
|
LabelSelector: "app.kubernetes.io/generated-by=argocd-generator",
|
|
})
|
|
}
|