feat: diagram just test target

adds output name option

adds just targets for different vlab diagrams
adds just target for lab-ci env

Signed-off-by: Pau Capdevila <pau@githedgehog.com>

Author: pau-hedgehog

cmd/hhfab/main.go

@@ -605,6 +605,11 @@ func Run(ctx context.Context) error {
 							func(item diagram.StyleType, _ int) string { return string(item) }), ", "),
 						Value: string(diagram.StyleTypeDefault),
 					},
+					&cli.StringFlag{
+						Name:    "output",
+						Aliases: []string{"o"},
+						Usage:   "output file path for the generated diagram (default: result/diagram.{format})",
+					},
 					&cli.BoolFlag{
 						Name:  "live",
 						Usage: "load resources from actually running API instead of the config file (fab.yaml and include/*)",
@@ -614,7 +619,7 @@ func Run(ctx context.Context) error {
 				Action: func(c *cli.Context) error {
 					format := diagram.Format(strings.ToLower(c.String("format")))
 					styleType := diagram.StyleType(c.String("style"))
-					if err := hhfab.Diagram(ctx, workDir, cacheDir, c.Bool("live"), format, styleType); err != nil {
+					if err := hhfab.Diagram(ctx, workDir, cacheDir, c.Bool("live"), format, styleType, c.String("output")); err != nil {
 						return fmt.Errorf("failed to generate %s diagram: %w", format, err)
 					}
 

justfile

@@ -165,3 +165,122 @@ zot := localbin / "zot" + "-" + zot_os + "-" + zot_arch + "-" + zot_version
 _localreg: _zot
   ./hack/localreg.sh
   {{zot}} serve .zot/config.json 2>&1 | tee .zot/log
+
+#
+# Generate diagrams for multiple environments in different formats and styles
+#
+test-diagram:
+  @mkdir -p test-diagram
+  @echo "==============================================="
+  @echo "Diagram generation test - various topologies, formats, and styles"
+  @echo "==============================================="
+  
+  # Check if a VLAB is actually running
+  @echo "=== Checking for running VLAB ==="
+  @VLAB_PIDS=$(pgrep -f "[h]hfab vlab up" 2>/dev/null || echo ""); \
+  if [ -n "$VLAB_PIDS" ] && [ -f "vlab/kubeconfig" ]; then \
+    echo "=== Detected running VLAB, generating live diagrams ==="; \
+    bin/hhfab diagram --format drawio --style default --live --output test-diagram/live-drawio-default.drawio || echo "Failed to generate live DrawIO diagram"; \
+    bin/hhfab diagram --format drawio --style cisco --live --output test-diagram/live-drawio-cisco.drawio || echo "Failed to generate live DrawIO cisco diagram"; \
+    bin/hhfab diagram --format drawio --style hedgehog --live --output test-diagram/live-drawio-hedgehog.drawio || echo "Failed to generate live DrawIO hedgehog diagram"; \
+    bin/hhfab diagram --format dot --live --output test-diagram/live-dot.dot || echo "Failed to generate live DOT diagram"; \
+    if command -v dot >/dev/null 2>&1 && [ -f "test-diagram/live-dot.dot" ]; then \
+      dot -Tpng test-diagram/live-dot.dot -o test-diagram/live-dot.png; \
+    fi; \
+    bin/hhfab diagram --format mermaid --live --output test-diagram/live-mermaid.mermaid || echo "Failed to generate live Mermaid diagram"; \
+    if [ -f "test-diagram/live-mermaid.mermaid" ]; then \
+      echo '# Live Network Diagram' > test-diagram/live-mermaid.md; \
+      echo '```mermaid' >> test-diagram/live-mermaid.md; \
+      cat test-diagram/live-mermaid.mermaid >> test-diagram/live-mermaid.md; \
+      echo '```' >> test-diagram/live-mermaid.md; \
+    fi; \
+  else \
+    echo "No running VLAB detected, skipping live diagrams"; \
+  fi
+
+  @echo -n "This will generate diagrams from multiple environments. Continue? [y/N] " && read ans && [ "$ans" = "y" -o "$ans" = "Y" ]
+  
+  @echo "=== Generating diagrams for default VLAB topology ==="
+  bin/hhfab init -f --dev --gw
+  bin/hhfab vlab gen
+  
+  # Generate all formats and styles for default topology
+  bin/hhfab diagram --format drawio --style default --output test-diagram/default-drawio-default.drawio
+  bin/hhfab diagram --format drawio --style cisco --output test-diagram/default-drawio-cisco.drawio
+  bin/hhfab diagram --format drawio --style hedgehog --output test-diagram/default-drawio-hedgehog.drawio
+  bin/hhfab diagram --format dot --output test-diagram/default-dot.dot
+  bin/hhfab diagram --format mermaid --output test-diagram/default-mermaid.mermaid
+  
+  @echo "=== Generating diagrams for variant 3-spine topology ==="
+  bin/hhfab vlab gen --spines-count 3 --mclag-leafs-count 2 --orphan-leafs-count 1 --eslag-leaf-groups 2
+  
+  # Generate all formats and styles for 3-spine topology
+  bin/hhfab diagram --format drawio --style default --output test-diagram/3spine-drawio-default.drawio
+  bin/hhfab diagram --format drawio --style cisco --output test-diagram/3spine-drawio-cisco.drawio
+  bin/hhfab diagram --format drawio --style hedgehog --output test-diagram/3spine-drawio-hedgehog.drawio
+  bin/hhfab diagram --format dot --output test-diagram/3spine-dot.dot
+  bin/hhfab diagram --format mermaid --output test-diagram/3spine-mermaid.mermaid
+  
+  # Convert DOT files to PNG if GraphViz is installed
+  @echo "=== Converting DOT files to PNG if GraphViz is installed ==="
+  @if command -v dot >/dev/null 2>&1; then \
+    for DOT_FILE in test-diagram/*-dot.dot; do \
+      PNG_FILE="${DOT_FILE%.dot}.png"; \
+      echo "Converting $DOT_FILE to $PNG_FILE"; \
+      dot -Tpng "$DOT_FILE" -o "$PNG_FILE"; \
+    done; \
+  else \
+    echo "GraphViz dot not installed, skipping PNG conversion"; \
+  fi
+  
+  # Create markdown files with embedded mermaid diagrams
+  @echo "=== Creating Markdown files with embedded Mermaid diagrams ==="
+  @for MERMAID_FILE in test-diagram/*-mermaid.mermaid; do \
+    MD_FILE="${MERMAID_FILE%.mermaid}.md"; \
+    BASE_NAME=$(basename "$MERMAID_FILE" -mermaid.mermaid); \
+    echo "Creating $MD_FILE"; \
+    echo "# $BASE_NAME Network Diagram" > "$MD_FILE"; \
+    echo '```mermaid' >> "$MD_FILE"; \
+    cat "$MERMAID_FILE" >> "$MD_FILE"; \
+    echo '```' >> "$MD_FILE"; \
+  done
+  
+  @echo ""
+  @echo "All diagrams generated in test-diagram/ directory"
+  @ls -la test-diagram/
+  @echo ""
+  @echo "Summary of generated files:"
+  @echo "- Default VLAB topology: default-*"
+  @echo "- 3-spine VLAB topology: 3spine-*"
+  @echo "- Live diagrams (if VLAB running): live-*"
+  @echo ""
+  @echo "For each topology, these formats are available:"
+  @echo "- DrawIO: *-drawio-{default,cisco,hedgehog}.drawio"
+  @echo "- DOT: *-dot.dot (and PNG if GraphViz was installed)"
+  @echo "- Mermaid: *-mermaid.mermaid (and embedded in markdown *.md)"
+
+# Test lab-ci diagram
+test-diagram-labci:
+  @mkdir -p test-diagram
+  @echo "==============================================="
+  @echo "Diagram generation test - lab-ci"
+  @echo "==============================================="
+
+  # Check if the lab-ci repo is available
+  @if [ ! -d "lab-ci" ]; then \
+    echo "lab-ci repository not found. Please clone it:"; \
+    echo "git clone https://github.com/githedgehog/lab-ci.git"; \
+    exit 1; \
+  fi
+
+  @echo "=== Generating diagrams for lab-ci environment 1 ==="
+  bin/hhfab init -f -c lab-ci/envs/env-ci-1.l/hhfab.yaml -w lab-ci/envs/env-ci-1.l/wiring-0-base.yaml -w lab-ci/envs/env-ci-1.l/wiring-1-spine-leaf.yaml
+  bin/hhfab diagram --format drawio --style default --output test-diagram/labci1-drawio-default.drawio
+  bin/hhfab diagram --format drawio --style cisco --output test-diagram/labci1-drawio-cisco.drawio
+  bin/hhfab diagram --format drawio --style hedgehog --output test-diagram/labci1-drawio-hedgehog.drawio
+  bin/hhfab diagram --format dot --output test-diagram/labci1-dot.dot
+  bin/hhfab diagram --format mermaid --output test-diagram/labci1-mermaid.mermaid
+
+  @echo ""
+  @echo "All diagrams generated in test-diagram/ directory"
+  @ls -la test-diagram/

pkg/hhfab/cmddiagram.go

@@ -16,7 +16,7 @@ import (
 	kclient "sigs.k8s.io/controller-runtime/pkg/client"
 )
 
-func Diagram(ctx context.Context, workDir, cacheDir string, live bool, format diagram.Format, style diagram.StyleType) error {
+func Diagram(ctx context.Context, workDir, cacheDir string, live bool, format diagram.Format, style diagram.StyleType, outputPath string) error {
 	resultDir := filepath.Join(workDir, ResultDir)
 	if err := os.MkdirAll(resultDir, 0o755); err != nil {
 		return fmt.Errorf("creating result directory: %w", err)
@@ -44,7 +44,7 @@ func Diagram(ctx context.Context, workDir, cacheDir string, live bool, format di
 		client = kube
 	}
 
-	if err := diagram.Generate(ctx, resultDir, client, format, style); err != nil {
+	if err := diagram.Generate(ctx, resultDir, client, format, style, outputPath); err != nil {
 		return fmt.Errorf("generating diagram: %w", err)
 	}
 

pkg/hhfab/diagram/diagram.go

@@ -27,7 +27,7 @@ var Formats = []Format{
 	FormatMermaid,
 }
 
-func Generate(ctx context.Context, resultDir string, client kclient.Reader, format Format, style StyleType) error {
+func Generate(ctx context.Context, resultDir string, client kclient.Reader, format Format, style StyleType, outputPath string) error {
 	if !slices.Contains(Formats, format) {
 		return fmt.Errorf("unsupported diagram format: %s", format) //nolint:goerr113
 	}
@@ -40,39 +40,84 @@ func Generate(ctx context.Context, resultDir string, client kclient.Reader, form
 		return fmt.Errorf("getting topology: %w", err)
 	}
 
+	var filePath string
+	var displayPath string
+
 	switch format {
 	case FormatDrawio:
 		slog.Debug("Generating draw.io diagram", "style", style)
-		if err := GenerateDrawio(resultDir, topo, style); err != nil {
+		if err := GenerateDrawio(resultDir, topo, style, outputPath); err != nil {
 			return fmt.Errorf("generating draw.io diagram: %w", err)
 		}
-		filePath := filepath.Join("result", DrawioFilename)
-		slog.Info("Generated draw.io diagram", "file", filePath, "style", style)
+		if outputPath != "" {
+			filePath = outputPath
+		} else {
+			filePath = filepath.Join(resultDir, DrawioFilename)
+		}
+
+		displayPath = filePath
+		workDir, err := filepath.Abs(".")
+		if err == nil {
+			rel, err := filepath.Rel(workDir, filePath)
+			if err == nil {
+				displayPath = rel
+			}
+		}
+
+		slog.Info("Generated draw.io diagram", "file", displayPath, "style", style)
 		fmt.Printf("To use this diagram:\n")
 		fmt.Printf("1. Open with https://app.diagrams.net/ or the desktop Draw.io application\n")
 		fmt.Printf("2. You can edit the diagram and export to PNG, SVG, PDF or other formats\n")
 	case FormatDot:
 		slog.Debug("Generating DOT diagram")
-		if err := GenerateDOT(resultDir, topo); err != nil {
+		if err := GenerateDOT(resultDir, topo, outputPath); err != nil {
 			return fmt.Errorf("generating DOT diagram: %w", err)
 		}
-		filePath := filepath.Join("result", DotFilename)
-		slog.Info("Generated graphviz diagram", "file", filePath)
+		if outputPath != "" {
+			filePath = outputPath
+		} else {
+			filePath = filepath.Join(resultDir, DotFilename)
+		}
+
+		displayPath = filePath
+		workDir, err := filepath.Abs(".")
+		if err == nil {
+			rel, err := filepath.Rel(workDir, filePath)
+			if err == nil {
+				displayPath = rel
+			}
+		}
+
+		slog.Info("Generated graphviz diagram", "file", displayPath)
 		fmt.Printf("To render this diagram with Graphviz:\n")
 		fmt.Printf("1. Install Graphviz: https://graphviz.org/download/\n")
-		fmt.Printf("2. Convert to PNG: dot -Tpng %s -o diagram.png\n", filePath)
-		fmt.Printf("3. Convert to SVG: dot -Tsvg %s -o diagram.svg\n", filePath)
-		fmt.Printf("4. Convert to PDF: dot -Tpdf %s -o diagram.pdf\n", filePath)
+		fmt.Printf("2. Convert to PNG: dot -Tpng %s -o diagram.png\n", displayPath)
+		fmt.Printf("3. Convert to SVG: dot -Tsvg %s -o diagram.svg\n", displayPath)
+		fmt.Printf("4. Convert to PDF: dot -Tpdf %s -o diagram.pdf\n", displayPath)
 	case FormatMermaid:
 		slog.Debug("Generating Mermaid diagram")
-		if err := GenerateMermaid(resultDir, topo); err != nil {
+		if err := GenerateMermaid(resultDir, topo, outputPath); err != nil {
 			return fmt.Errorf("generating Mermaid diagram: %w", err)
 		}
-		filePath := filepath.Join("result", MermaidFilename)
-		slog.Info("Generated Mermaid diagram", "file", filePath)
+		if outputPath != "" {
+			filePath = outputPath
+		} else {
+			filePath = filepath.Join(resultDir, MermaidFilename)
+		}
+
+		displayPath = filePath
+		workDir, err := filepath.Abs(".")
+		if err == nil {
+			rel, err := filepath.Rel(workDir, filePath)
+			if err == nil {
+				displayPath = rel
+			}
+		}
+
+		slog.Info("Generated Mermaid diagram", "file", displayPath)
 		fmt.Printf("To render this diagram with Mermaid:\n")
 		fmt.Printf("1. Visit https://mermaid.live/ or use a Markdown editor with Mermaid support\n")
-		fmt.Printf("2. Copy the contents of %s into the editor\n", filePath)
+		fmt.Printf("2. Copy the contents of %s into the editor\n", displayPath)
 	default:
 		return fmt.Errorf("unsupported diagram format: %s", format) //nolint:goerr113
 	}

pkg/hhfab/diagram/drawio.go

@@ -72,9 +72,15 @@ var nodeConnectionsMap map[string][]float64
 
 var nodes []Node
 
-func GenerateDrawio(workDir string, topo Topology, styleType StyleType) error {
+func GenerateDrawio(workDir string, topo Topology, styleType StyleType, outputPath string) error {
 	slog.Debug("Initializing new diagram generation")
-	outputFile := filepath.Join(workDir, DrawioFilename)
+
+	var finalOutputPath string
+	if outputPath != "" {
+		finalOutputPath = outputPath
+	} else {
+		finalOutputPath = filepath.Join(workDir, DrawioFilename)
+	}
 
 	style := GetStyle(styleType)
 
@@ -86,7 +92,12 @@ func GenerateDrawio(workDir string, topo Topology, styleType StyleType) error {
 		return fmt.Errorf("marshaling XML: %w", err)
 	}
 	xmlContent := []byte(xml.Header + string(outputXML))
-	if err := os.WriteFile(outputFile, xmlContent, 0o600); err != nil {
+
+	if err := os.MkdirAll(filepath.Dir(finalOutputPath), 0o755); err != nil {
+		return fmt.Errorf("creating output directory: %w", err)
+	}
+
+	if err := os.WriteFile(finalOutputPath, xmlContent, 0o600); err != nil {
 		return fmt.Errorf("writing draw.io file: %w", err)
 	}
 

pkg/hhfab/diagram/graphviz.go

@@ -22,11 +22,21 @@ const (
 	ColorDefault   = "black"
 )
 
-func GenerateDOT(workDir string, topo Topology) error {
-	outputFile := filepath.Join(workDir, DotFilename)
+func GenerateDOT(workDir string, topo Topology, outputPath string) error {
+	var finalOutputPath string
+	if outputPath != "" {
+		finalOutputPath = outputPath
+	} else {
+		finalOutputPath = filepath.Join(workDir, DotFilename)
+	}
 
 	dot := generateDOT(topo)
-	if err := os.WriteFile(outputFile, []byte(dot), 0o600); err != nil {
+
+	if err := os.MkdirAll(filepath.Dir(finalOutputPath), 0o755); err != nil {
+		return fmt.Errorf("creating output directory: %w", err)
+	}
+
+	if err := os.WriteFile(finalOutputPath, []byte(dot), 0o600); err != nil {
 		return fmt.Errorf("writing DOT file: %w", err)
 	}
 

pkg/hhfab/diagram/mermaid.go

@@ -12,10 +12,21 @@ import (
 	"strings"
 )
 
-func GenerateMermaid(workDir string, topo Topology) error {
-	outputFile := filepath.Join(workDir, MermaidFilename)
+func GenerateMermaid(workDir string, topo Topology, outputPath string) error {
+	var finalOutputPath string
+	if outputPath != "" {
+		finalOutputPath = outputPath
+	} else {
+		finalOutputPath = filepath.Join(workDir, MermaidFilename)
+	}
+
 	mermaid := generateMermaid(topo)
-	if err := os.WriteFile(outputFile, []byte(mermaid), 0o600); err != nil {
+
+	if err := os.MkdirAll(filepath.Dir(finalOutputPath), 0o755); err != nil {
+		return fmt.Errorf("creating output directory: %w", err)
+	}
+
+	if err := os.WriteFile(finalOutputPath, []byte(mermaid), 0o600); err != nil {
 		return fmt.Errorf("writing Mermaid file: %w", err)
 	}