diff options
| author | Luke Shumaker <lukeshu@lukeshu.com> | 2022-09-05 17:01:24 -0600 |
|---|---|---|
| committer | Luke Shumaker <lukeshu@lukeshu.com> | 2022-09-05 17:01:24 -0600 |
| commit | 205b055a2fc7aa2742bff497f85bbde5880e0584 (patch) | |
| tree | 9b16e8845eb872f854a52feb0adee8246d1f5fa1 /cmd/btrfs-rec | |
| parent | 7ed94fa7e829a7360a025ca0ca8d20d081859fdc (diff) | |
visualize-nodes: Have the output be a zip file
Diffstat (limited to 'cmd/btrfs-rec')
| -rw-r--r-- | cmd/btrfs-rec/inspect_visualizenodes.go | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/cmd/btrfs-rec/inspect_visualizenodes.go b/cmd/btrfs-rec/inspect_visualizenodes.go index 4c31100..89c8ad6 100644 --- a/cmd/btrfs-rec/inspect_visualizenodes.go +++ b/cmd/btrfs-rec/inspect_visualizenodes.go @@ -5,6 +5,9 @@ package main import ( + "bufio" + "os" + "github.com/datawire/dlib/dlog" "github.com/datawire/ocibuild/pkg/cliutil" "github.com/spf13/cobra" @@ -16,8 +19,8 @@ import ( func init() { inspectors = append(inspectors, subcommand{ Command: cobra.Command{ - Use: "visualize-nodes NODESCAN.json OUTPUT_DIR", - Args: cliutil.WrapPositionalArgs(cobra.ExactArgs(2)), + 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() @@ -29,7 +32,13 @@ func init() { } dlog.Infof(ctx, "... done reading %q", args[0]) - return rebuildnodes.VisualizeNodes(ctx, args[1], fs, nodeScanResults) + buffer := bufio.NewWriter(os.Stdout) + defer func() { + if _err := buffer.Flush(); err == nil && _err != nil { + err = _err + } + }() + return rebuildnodes.VisualizeNodes(ctx, buffer, fs, nodeScanResults) }, }) } |