kro cli generate diagram (#589)

* kro-cli generate diagram command

Author: DhairyaMajmudar

cmd/kro/commands/generate.go

@@ -1,10 +1,26 @@
+// Copyright 2025 The Kube Resource Orchestrator Authors
+//
+// 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.
+
 package commands
 
 import (
 	"encoding/json"
 	"fmt"
 	"os"
 
+	"github.com/go-echarts/go-echarts/v2/charts"
+	"github.com/go-echarts/go-echarts/v2/opts"
 	"github.com/kro-run/kro/api/v1alpha1"
 	kroclient "github.com/kro-run/kro/pkg/client"
 	"github.com/kro-run/kro/pkg/graph"
@@ -64,7 +80,7 @@ var generateCRDCmd = &cobra.Command{
 
 var generateInstanceCmd = &cobra.Command{
 	Use:   "instance",
-	Short: "Generate a ResourceGraphDefinition (RGD) instance",
+	Short: "Generate an instance of the ResourceGraphDefinition",
 	Long: "Generate a ResourceGraphDefinition (RGD) instance from a " +
 		"ResourceGraphDefinition file. This command reads the " +
 		"ResourceGraphDefinition and outputs the corresponding RGD instance",
@@ -84,7 +100,36 @@ var generateInstanceCmd = &cobra.Command{
 		}
 
 		if err = generateInstance(&rgd); err != nil {
-			return fmt.Errorf("failed to generate RGD instance: %w", err)
+			return fmt.Errorf("failed to generate instance: %w", err)
+		}
+
+		return nil
+	},
+}
+
+var generateDiagramCmd = &cobra.Command{
+	Use:   "diagram",
+	Short: "Generate a diagram from a ResourceGraphDefinition",
+	Long: "Generate a diagram from a ResourceGraphDefinition file. This command reads the " +
+		"ResourceGraphDefinition and outputs the corresponding diagram " +
+		"in the specified format.",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		if config.resourceGraphDefinitionFile == "" {
+			return fmt.Errorf("ResourceGraphDefinition file is required")
+		}
+
+		data, err := os.ReadFile(config.resourceGraphDefinitionFile)
+		if err != nil {
+			return fmt.Errorf("failed to read ResourceGraphDefinition file: %w", err)
+		}
+
+		var rgd v1alpha1.ResourceGraphDefinition
+		if err = yaml.Unmarshal(data, &rgd); err != nil {
+			return fmt.Errorf("failed to unmarshal ResourceGraphDefinition: %w", err)
+		}
+
+		if err = generateDiagram(&rgd); err != nil {
+			return fmt.Errorf("failed to generate diagram: %w", err)
 		}
 
 		return nil
@@ -110,6 +155,93 @@ func generateCRD(rgd *v1alpha1.ResourceGraphDefinition) error {
 	return nil
 }
 
+func generateDiagram(rgd *v1alpha1.ResourceGraphDefinition) error {
+	rgdGraph, err := createGraphBuilder(rgd)
+	if err != nil {
+		return fmt.Errorf("failed to setup rgd graph: %w", err)
+	}
+
+	graph := charts.NewGraph()
+
+	// Graph layout
+	graph.SetGlobalOptions(
+		charts.WithInitializationOpts(opts.Initialization{
+			Width:  "70%",
+			Height: "90vh",
+		}),
+		charts.WithTitleOpts(opts.Title{
+			Title:    fmt.Sprintf("Resource Graph: %s", rgd.Name),
+			Subtitle: fmt.Sprintf("Topological Order: %v", rgdGraph.TopologicalOrder),
+			TitleStyle: &opts.TextStyle{
+				FontSize:   18,
+				FontWeight: "bold",
+			},
+			SubtitleStyle: &opts.TextStyle{
+				FontSize: 14,
+			},
+		}),
+		charts.WithLegendOpts(opts.Legend{
+			Show: opts.Bool(false),
+		}),
+	)
+
+	resourceOrder := make(map[string]int)
+	for i, resourceName := range rgdGraph.TopologicalOrder {
+		resourceOrder[resourceName] = i + 1
+	}
+
+	// Graph nodes
+	nodes := make([]opts.GraphNode, 0, len(rgdGraph.Resources))
+	for resourceName := range rgdGraph.Resources {
+		order := resourceOrder[resourceName]
+
+		node := opts.GraphNode{
+			Name:       fmt.Sprintf("%s\n(%d)", resourceName, order),
+			SymbolSize: 40.0,
+			ItemStyle: &opts.ItemStyle{
+				Color:       "#269103",
+				BorderColor: "#333",
+				BorderWidth: 1,
+			},
+		}
+		nodes = append(nodes, node)
+	}
+
+	// Graph links
+	links := make([]opts.GraphLink, 0)
+	for resourceName, resource := range rgdGraph.Resources {
+		targetOrder := resourceOrder[resourceName]
+
+		for _, dependency := range resource.GetDependencies() {
+			sourceOrder := resourceOrder[dependency]
+
+			link := opts.GraphLink{
+				Source: fmt.Sprintf("%s\n(%d)", dependency, sourceOrder),
+				Target: fmt.Sprintf("%s\n(%d)", resourceName, targetOrder),
+				LineStyle: &opts.LineStyle{
+					Color: "#000",
+				},
+			}
+			links = append(links, link)
+		}
+	}
+
+	graph.AddSeries("resource", nodes, links).SetSeriesOptions(
+		charts.WithGraphChartOpts(
+			opts.GraphChart{
+				Force: &opts.GraphForce{
+					Repulsion: 5000,
+				},
+				Roam:           opts.Bool(true),
+				EdgeSymbol:     "arrow",
+				EdgeSymbolSize: []int{0, 7},
+			},
+		),
+	)
+
+	return graph.Render(os.Stdout)
+}
+
 func generateInstance(rgd *v1alpha1.ResourceGraphDefinition) error {
 	rgdGraph, err := createGraphBuilder(rgd)
 	if err != nil {
@@ -180,6 +312,7 @@ func marshalObject(obj interface{}, outputFormat string) ([]byte, error) {
 
 func AddGenerateCommands(rootCmd *cobra.Command) {
 	generateCmd.AddCommand(generateCRDCmd)
+	generateCmd.AddCommand(generateDiagramCmd)
 	generateCmd.AddCommand(generateInstanceCmd)
 	rootCmd.AddCommand(generateCmd)
 }

go.mod

@@ -3,6 +3,7 @@ module github.com/kro-run/kro
 go 1.24.0
 
 require (
+	github.com/go-echarts/go-echarts/v2 v2.6.1
 	github.com/go-logr/logr v1.4.2
 	github.com/gobuffalo/flect v1.0.2
 	github.com/google/cel-go v0.24.1

go.sum

@@ -85,6 +85,8 @@ github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nos
 github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=
 github.com/fxamacker/cbor/v2 v2.7.0 h1:iM5WgngdRBanHcxugY4JySA0nk1wZorNOpTgCMedv5E=
 github.com/fxamacker/cbor/v2 v2.7.0/go.mod h1:pxXPTn3joSm21Gbwsv0w9OSA2y1HFR9qXEeXQVeNoDQ=
+github.com/go-echarts/go-echarts/v2 v2.6.1 h1:UjyovbU7sbALakMYaoFsSKimT1Sm3kHCJcJSu6U5JoU=
+github.com/go-echarts/go-echarts/v2 v2.6.1/go.mod h1:56YlvzhW/a+du15f3S2qUGNDfKnFOeJSThBIrVFHDtI=
 github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=
 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20191125211704-12ad95a8df72/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=
 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=