forked from TrueCloudLab/frostfs-node
8bba490c30
There is a need to support NeoFS-binary sessions along with JSON ones in NeoFS CLI. Provide generic `common.ReadBinaryOrJSON` functions which tries to decode NeoFS-binary structure and falls back to JSON format. Use this function in all places with token reading. Signed-off-by: Leonard Lyubich <ctulhurider@gmail.com>
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package common
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/nspcc-dev/neofs-sdk-go/bearer"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// ReadBearerToken reads bearer token from the path provided in a specified flag.
|
|
func ReadBearerToken(cmd *cobra.Command, flagname string) *bearer.Token {
|
|
path, err := cmd.Flags().GetString(flagname)
|
|
ExitOnErr(cmd, "", err)
|
|
|
|
if len(path) == 0 {
|
|
return nil
|
|
}
|
|
|
|
PrintVerbose("Reading bearer token from file [%s]...", path)
|
|
|
|
var tok bearer.Token
|
|
|
|
err = ReadBinaryOrJSON(&tok, path)
|
|
ExitOnErr(cmd, "invalid bearer token: %v", err)
|
|
|
|
return &tok
|
|
}
|
|
|
|
// BinaryOrJSON is an interface of entities which provide json.Unmarshaler
|
|
// and NeoFS binary decoder.
|
|
type BinaryOrJSON interface {
|
|
Unmarshal([]byte) error
|
|
json.Unmarshaler
|
|
}
|
|
|
|
// ReadBinaryOrJSON reads file data using provided path and decodes
|
|
// BinaryOrJSON from the data.
|
|
func ReadBinaryOrJSON(dst BinaryOrJSON, fPath string) error {
|
|
PrintVerbose("Reading file [%s]...", fPath)
|
|
|
|
// try to read session token from file
|
|
data, err := os.ReadFile(fPath)
|
|
if err != nil {
|
|
return fmt.Errorf("read file <%s>: %w", fPath, err)
|
|
}
|
|
|
|
PrintVerbose("Trying to decode binary...")
|
|
|
|
err = dst.Unmarshal(data)
|
|
if err != nil {
|
|
PrintVerbose("Failed to decode binary: %v", err)
|
|
|
|
PrintVerbose("Trying to decode JSON...")
|
|
|
|
err = dst.UnmarshalJSON(data)
|
|
if err != nil {
|
|
PrintVerbose("Failed to decode JSON: %v", err)
|
|
return errors.New("invalid format")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|