Add WASM-ready flamegraph JSON export

2 files changed, 12 insertions, 0 deletions

diff --git a/internal/flags/flags.go b/internal/flags/flags.go
index a732c3a..bc1654b 100644
--- a/internal/flags/flags.go
+++ b/internal/flags/flags.go
@@ -67,6 +67,7 @@ type Flags struct {
 	LiveInterval     time.Duration
 	OpenCommand      string
 	FlamegraphName   string
+	FlamegraphJSON   bool
 	TUIExportEnable  bool
 
 	// To convert ior data into native SVG format
@@ -125,6 +126,7 @@ func parse() error {
 	flag.DurationVar(&singleton.LiveInterval, "live-interval", 200*time.Millisecond, "Live flamegraph refresh interval")
 	flag.StringVar(&singleton.OpenCommand, "open", "", "Command to open live flamegraph URL (used with -live); use {url} placeholder or URL is appended")
 	flag.StringVar(&singleton.FlamegraphName, "name", "default", "Name of the flamegraph, used to generate the SVG file")
+	flag.BoolVar(&singleton.FlamegraphJSON, "flamegraphJson", false, "Also export flamegraph tree as JSON in -ior mode (experimental WASM-ready output)")
 	flag.BoolVar(&singleton.TUIExportEnable, "tuiExport", true, "Enable writing TUI snapshot export files")
 
 	flag.StringVar(&singleton.IorDataFile, "ior", "", "IOR data file to convert into native SVG flamegraph")
diff --git a/internal/flags/flags_test.go b/internal/flags/flags_test.go
index 32af4bd..11414d8 100644
--- a/internal/flags/flags_test.go
+++ b/internal/flags/flags_test.go
@@ -104,6 +104,16 @@ func TestParseOpenFlags(t *testing.T) {
 	}
 }
 
+func TestParseFlamegraphJSONFlag(t *testing.T) {
+	cfg, err := parseForTest(t, "-flamegraphJson")
+	if err != nil {
+		t.Fatalf("parse returned error: %v", err)
+	}
+	if !cfg.FlamegraphJSON {
+		t.Fatalf("expected -flamegraphJson to enable JSON export")
+	}
+}
+
 func TestParseDefaultCollapsedFieldsOrder(t *testing.T) {
 	cfg, err := parseForTest(t)
 	if err != nil {