cli/object: Share file open code in get and range commands

Signed-off-by: Leonard Lyubich <[email protected]>

Author: cthulhu-rider

cmd/neofs-cli/modules/object/get.go

@@ -57,9 +57,9 @@ func getObject(cmd *cobra.Command, _ []string) error {
 	if filename == "" {
 		out = os.Stdout
 	} else {
-		f, err := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0o644)
+		f, err := openFileForPayload(filename)
 		if err != nil {
-			return fmt.Errorf("can't open file '%s': %w", filename, err)
+			return err
 		}
 
 		defer func() { _ = f.Close() }()

cmd/neofs-cli/modules/object/range.go

@@ -69,9 +69,9 @@ func getObjectRange(cmd *cobra.Command, _ []string) error {
 	if filename == "" {
 		out = os.Stdout
 	} else {
-		f, err := os.OpenFile(filename, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0o644)
+		f, err := openFileForPayload(filename)
 		if err != nil {
-			return fmt.Errorf("can't open file '%s': %w", filename, err)
+			return err
 		}
 
 		defer f.Close()

cmd/neofs-cli/modules/object/util.go

@@ -5,6 +5,7 @@ import (
 	"crypto/ecdsa"
 	"errors"
 	"fmt"
+	"io"
 	"os"
 	"strings"
 
@@ -361,3 +362,11 @@ func finalizeSession(cmd *cobra.Command, dst SessionPrm, tok *session.Object, ke
 func initFlagSession(cmd *cobra.Command, verb string) {
 	commonflags.InitSession(cmd, "object "+verb)
 }
+
+func openFileForPayload(name string) (io.WriteCloser, error) {
+	f, err := os.OpenFile(name, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0o644)
+	if err != nil {
+		return nil, fmt.Errorf("can't open file '%s': %w", name, err)
+	}
+	return f, nil
+}