|
| 1 | +package main |
| 2 | + |
| 3 | +import ( |
| 4 | + "encoding/json" |
| 5 | + "flag" |
| 6 | + "fmt" |
| 7 | + "log" |
| 8 | + "os" |
| 9 | + "path/filepath" |
| 10 | + |
| 11 | + "github.com/perses/perses/go-sdk/dashboard" |
| 12 | + "github.com/stolostron/multicluster-observability-addon/pkg/perses/dashboards/virtualization" |
| 13 | + |
| 14 | + "github.com/kubevirt/monitoring/pkg/dashboards/transform" |
| 15 | + |
| 16 | + k8syaml "sigs.k8s.io/yaml" |
| 17 | +) |
| 18 | + |
| 19 | +type persesResource struct { |
| 20 | + APIVersion string `json:"apiVersion"` |
| 21 | + Kind string `json:"kind"` |
| 22 | + Metadata map[string]string `json:"metadata"` |
| 23 | + Spec json.RawMessage `json:"spec"` |
| 24 | +} |
| 25 | + |
| 26 | +type builderEntry struct { |
| 27 | + name string |
| 28 | + fn func(string, string) (dashboard.Builder, error) |
| 29 | + multicluster bool // true = source is multicluster; apply ToSingleCluster transform |
| 30 | +} |
| 31 | + |
| 32 | +func includedDashboardNames(builders []builderEntry, prefix string) []string { |
| 33 | + names := make([]string, len(builders)) |
| 34 | + for i, b := range builders { |
| 35 | + names[i] = prefix + "-" + b.name |
| 36 | + } |
| 37 | + return names |
| 38 | +} |
| 39 | + |
| 40 | +func buildDashboardSpec(b builderEntry, project string) ([]byte, error) { |
| 41 | + db, err := b.fn(project, "rbac-query-proxy-datasource") |
| 42 | + if err != nil { |
| 43 | + return nil, fmt.Errorf("building dashboard: %w", err) |
| 44 | + } |
| 45 | + return json.Marshal(db.Dashboard.Spec) |
| 46 | +} |
| 47 | + |
| 48 | +func wrapAsResource(name, namespace string, spec json.RawMessage) ([]byte, error) { |
| 49 | + resource := persesResource{ |
| 50 | + APIVersion: "perses.dev/v1alpha1", |
| 51 | + Kind: "PersesDashboard", |
| 52 | + Metadata: map[string]string{ |
| 53 | + "name": name, |
| 54 | + "namespace": namespace, |
| 55 | + }, |
| 56 | + Spec: spec, |
| 57 | + } |
| 58 | + |
| 59 | + resourceJSON, err := json.Marshal(resource) |
| 60 | + if err != nil { |
| 61 | + return nil, fmt.Errorf("marshalling resource: %w", err) |
| 62 | + } |
| 63 | + return k8syaml.JSONToYAML(resourceJSON) |
| 64 | +} |
| 65 | + |
| 66 | +func main() { |
| 67 | + prefix := flag.String("prefix", "cnv", "dashboard ID prefix") |
| 68 | + displayPrefix := flag.String("display-prefix", "OpenShift Virtualization", "display name prefix (replaces 'Virtualization')") |
| 69 | + outDir := flag.String("out", "dashboards/perses", "output directory for generated YAML") |
| 70 | + namespace := flag.String("namespace", "openshift-cnv", "namespace for PersesDashboard CRs") |
| 71 | + flag.Parse() |
| 72 | + |
| 73 | + builders := []builderEntry{ |
| 74 | + {"node-memory-overview", virtualization.BuildNodeMemoryOverview, true}, |
| 75 | + {"virtual-machines-inventory", virtualization.BuildVMInventory, true}, |
| 76 | + {"virtual-machines-utilization", virtualization.BuildVMUtilization, true}, |
| 77 | + {"virtual-machines-service-level", virtualization.BuildVMServiceLevel, true}, |
| 78 | + {"virtual-machines-by-time-in-status", virtualization.BuildVMByTimeInStatus, true}, |
| 79 | + } |
| 80 | + |
| 81 | + cfg := transform.Config{ |
| 82 | + DashboardPrefix: *prefix, |
| 83 | + DisplayNamePrefix: *displayPrefix, |
| 84 | + IncludedDashboards: includedDashboardNames(builders, *prefix), |
| 85 | + } |
| 86 | + |
| 87 | + if err := os.MkdirAll(*outDir, 0o755); err != nil { |
| 88 | + log.Fatalf("Failed to create output directory: %v", err) |
| 89 | + } |
| 90 | + |
| 91 | + for _, b := range builders { |
| 92 | + specJSON, err := buildDashboardSpec(b, *namespace) |
| 93 | + if err != nil { |
| 94 | + log.Fatalf("Failed to build %s: %v", b.name, err) |
| 95 | + } |
| 96 | + |
| 97 | + var outputSpec json.RawMessage |
| 98 | + if b.multicluster { |
| 99 | + outputSpec, err = transform.ToSingleCluster(specJSON, cfg) |
| 100 | + if err != nil { |
| 101 | + log.Fatalf("Failed to transform %s: %v", b.name, err) |
| 102 | + } |
| 103 | + } else { |
| 104 | + outputSpec = specJSON |
| 105 | + } |
| 106 | + |
| 107 | + dashboardName := *prefix + "-" + b.name |
| 108 | + yamlBytes, err := wrapAsResource(dashboardName, *namespace, outputSpec) |
| 109 | + if err != nil { |
| 110 | + log.Fatalf("Failed to generate YAML for %s: %v", b.name, err) |
| 111 | + } |
| 112 | + |
| 113 | + outFile := filepath.Join(*outDir, dashboardName+".yaml") |
| 114 | + if err := os.WriteFile(outFile, yamlBytes, 0o644); err != nil { |
| 115 | + log.Fatalf("Failed to write %s: %v", outFile, err) |
| 116 | + } |
| 117 | + fmt.Printf("Generated: %s\n", outFile) |
| 118 | + } |
| 119 | +} |
0 commit comments