From 7ed94fa7e829a7360a025ca0ca8d20d081859fdc Mon Sep 17 00:00:00 2001 From: Luke Shumaker Date: Mon, 5 Sep 2022 10:39:48 -0600 Subject: Rework visualize-nodes --- cmd/btrfs-rec/inspect_visualizenodes.go | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) (limited to 'cmd/btrfs-rec/inspect_visualizenodes.go') diff --git a/cmd/btrfs-rec/inspect_visualizenodes.go b/cmd/btrfs-rec/inspect_visualizenodes.go index 89c8ad6..4c31100 100644 --- a/cmd/btrfs-rec/inspect_visualizenodes.go +++ b/cmd/btrfs-rec/inspect_visualizenodes.go @@ -5,9 +5,6 @@ package main import ( - "bufio" - "os" - "github.com/datawire/dlib/dlog" "github.com/datawire/ocibuild/pkg/cliutil" "github.com/spf13/cobra" @@ -19,8 +16,8 @@ import ( func init() { inspectors = append(inspectors, subcommand{ Command: cobra.Command{ - Use: "visualize-nodes NODESCAN.json", - Args: cliutil.WrapPositionalArgs(cobra.ExactArgs(1)), + Use: "visualize-nodes NODESCAN.json OUTPUT_DIR", + Args: cliutil.WrapPositionalArgs(cobra.ExactArgs(2)), }, RunE: func(fs *btrfs.FS, cmd *cobra.Command, args []string) (err error) { ctx := cmd.Context() @@ -32,13 +29,7 @@ func init() { } 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) + return rebuildnodes.VisualizeNodes(ctx, args[1], fs, nodeScanResults) }, }) } -- cgit v1.2.3-2-g168b