1st pass at visualize-nodes

1 files changed, 44 insertions, 0 deletions

diff --git a/cmd/btrfs-rec/inspect_visualizenodes.go b/cmd/btrfs-rec/inspect_visualizenodes.go
new file mode 100644
index 0000000..89c8ad6
--- /dev/null
+++ b/cmd/btrfs-rec/inspect_visualizenodes.go
@@ -0,0 +1,44 @@
+// Copyright (C) 2022  Luke Shumaker <lukeshu@lukeshu.com>
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+package main
+
+import (
+	"bufio"
+	"os"
+
+	"github.com/datawire/dlib/dlog"
+	"github.com/datawire/ocibuild/pkg/cliutil"
+	"github.com/spf13/cobra"
+
+	"git.lukeshu.com/btrfs-progs-ng/lib/btrfs"
+	"git.lukeshu.com/btrfs-progs-ng/lib/btrfsprogs/btrfsinspect/rebuildnodes"
+)
+
+func init() {
+	inspectors = append(inspectors, subcommand{
+		Command: cobra.Command{
+			Use:  "visualize-nodes NODESCAN.json",
+			Args: cliutil.WrapPositionalArgs(cobra.ExactArgs(1)),
+		},
+		RunE: func(fs *btrfs.FS, cmd *cobra.Command, args []string) (err error) {
+			ctx := cmd.Context()
+
+			dlog.Infof(ctx, "Reading %q...", args[0])
+			nodeScanResults, err := readScanResults(args[0])
+			if err != nil {
+				return err
+			}
+			dlog.Infof(ctx, "... done reading %q", args[0])
+
+			buffer := bufio.NewWriter(os.Stdout)
+			defer func() {
+				if _err := buffer.Flush(); err == nil && _err != nil {
+					err = _err
+				}
+			}()
+			return rebuildnodes.VisualizeNodes(ctx, buffer, fs, nodeScanResults)
+		},
+	})
+}