forked from TrueCloudLab/frostfs-node
1c30414a6c
Core changes: * avoid package-colliding variable naming * avoid using pointers to IDs where unnecessary * avoid using `idSDK` import alias pattern * use `EncodeToString` for protocol string calculation and `String` for printing Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package common
|
|
|
|
import (
|
|
"encoding/json"
|
|
"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
|
|
}
|
|
|
|
data, err := os.ReadFile(path)
|
|
ExitOnErr(cmd, "can't read bearer token file: %w", err)
|
|
|
|
var tok bearer.Token
|
|
if err := tok.UnmarshalJSON(data); err != nil {
|
|
err = tok.Unmarshal(data)
|
|
ExitOnErr(cmd, "can't decode bearer token: %w", err)
|
|
|
|
PrintVerbose("Using binary encoded bearer token")
|
|
} else {
|
|
PrintVerbose("Using JSON encoded bearer token")
|
|
}
|
|
|
|
return &tok
|
|
}
|
|
|
|
// ReadSessionToken calls ReadSessionTokenErr and exists on error.
|
|
func ReadSessionToken(cmd *cobra.Command, dst json.Unmarshaler, fPath string) {
|
|
ExitOnErr(cmd, "", ReadSessionTokenErr(dst, fPath))
|
|
}
|
|
|
|
// ReadSessionTokenErr reads session token as JSON file with session token
|
|
// from path provided in a specified flag.
|
|
func ReadSessionTokenErr(dst json.Unmarshaler, fPath string) error {
|
|
// try to read session token from file
|
|
data, err := os.ReadFile(fPath)
|
|
if err != nil {
|
|
return fmt.Errorf("could not open file with session token <%s>: %w", fPath, err)
|
|
}
|
|
|
|
err = dst.UnmarshalJSON(data)
|
|
if err != nil {
|
|
return fmt.Errorf("could not unmarshal session token from file: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|