-
Notifications
You must be signed in to change notification settings - Fork 5.4k
/
project_generator.go
50 lines (44 loc) · 1.45 KB
/
project_generator.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package generator
import (
"context"
"fmt"
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"log"
"github.com/argoproj/argo-cd/v2/hack/gen-resources/util"
"github.com/argoproj/argo-cd/v2/pkg/apis/application/v1alpha1"
appclientset "github.com/argoproj/argo-cd/v2/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: v1.ObjectMeta{
GenerateName: "project-",
Namespace: opts.Namespace,
Labels: labels,
},
Spec: v1alpha1.AppProjectSpec{
Description: "generated-project",
},
}, v1.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.Printf("Clean projects")
projects := pg.clientSet.ArgoprojV1alpha1().AppProjects(opts.Namespace)
return projects.DeleteCollection(context.TODO(), v1.DeleteOptions{}, v1.ListOptions{
LabelSelector: "app.kubernetes.io/generated-by=argocd-generator",
})
}