forked from TrueCloudLab/frostfs-node
[#1461] cli: Reorganize common functions
Make `ParseEpoch` and `GetCurrentEpoch` funcs public and move the first one to the `common` package. Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
fed9e6679d
commit
595fc60d72
3 changed files with 66 additions and 61 deletions
|
@ -1,9 +1,13 @@
|
||||||
package internal
|
package internal
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"crypto/ecdsa"
|
"crypto/ecdsa"
|
||||||
|
"crypto/elliptic"
|
||||||
|
"crypto/rand"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common"
|
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common"
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/network"
|
"github.com/nspcc-dev/neofs-node/pkg/network"
|
||||||
|
@ -54,3 +58,32 @@ func GetSDKClient(key *ecdsa.PrivateKey, addr network.Address) (*client.Client,
|
||||||
|
|
||||||
return &c, nil
|
return &c, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetCurrentEpoch returns current epoch.
|
||||||
|
func GetCurrentEpoch(endpoint string) (uint64, error) {
|
||||||
|
var addr network.Address
|
||||||
|
|
||||||
|
if err := addr.FromString(endpoint); err != nil {
|
||||||
|
return 0, fmt.Errorf("can't parse RPC endpoint: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
key, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("can't generate key to sign query: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
c, err := GetSDKClient(key, addr)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
ni, err := c.NetworkInfo(ctx, client.PrmNetworkInfo{})
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return ni.Info().CurrentEpoch(), nil
|
||||||
|
}
|
||||||
|
|
28
cmd/neofs-cli/internal/common/epoch.go
Normal file
28
cmd/neofs-cli/internal/common/epoch.go
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ParseEpoch parses epoch argument. Second return value is true if
|
||||||
|
// the specified epoch is relative, and false otherwise.
|
||||||
|
func ParseEpoch(cmd *cobra.Command, flag string) (uint64, bool, error) {
|
||||||
|
s, _ := cmd.Flags().GetString(flag)
|
||||||
|
if len(s) == 0 {
|
||||||
|
return 0, false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
relative := s[0] == '+'
|
||||||
|
if relative {
|
||||||
|
s = s[1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
epoch, err := strconv.ParseUint(s, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return 0, relative, fmt.Errorf("can't parse epoch for %s argument: %w", flag, err)
|
||||||
|
}
|
||||||
|
return epoch, relative, nil
|
||||||
|
}
|
|
@ -1,21 +1,14 @@
|
||||||
package bearer
|
package bearer
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"crypto/ecdsa"
|
|
||||||
"crypto/elliptic"
|
|
||||||
"crypto/rand"
|
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
internalclient "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/client"
|
internalclient "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/client"
|
||||||
|
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/common"
|
||||||
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/commonflags"
|
"github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/commonflags"
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/network"
|
|
||||||
"github.com/nspcc-dev/neofs-sdk-go/bearer"
|
"github.com/nspcc-dev/neofs-sdk-go/bearer"
|
||||||
"github.com/nspcc-dev/neofs-sdk-go/client"
|
|
||||||
eaclSDK "github.com/nspcc-dev/neofs-sdk-go/eacl"
|
eaclSDK "github.com/nspcc-dev/neofs-sdk-go/eacl"
|
||||||
"github.com/nspcc-dev/neofs-sdk-go/user"
|
"github.com/nspcc-dev/neofs-sdk-go/user"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
|
@ -63,21 +56,21 @@ func init() {
|
||||||
}
|
}
|
||||||
|
|
||||||
func createToken(cmd *cobra.Command, _ []string) error {
|
func createToken(cmd *cobra.Command, _ []string) error {
|
||||||
iat, iatRelative, err := parseEpoch(cmd, issuedAtFlag)
|
iat, iatRelative, err := common.ParseEpoch(cmd, issuedAtFlag)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
exp, expRelative, err := parseEpoch(cmd, expireAtFlag)
|
exp, expRelative, err := common.ParseEpoch(cmd, expireAtFlag)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
nvb, nvbRelative, err := parseEpoch(cmd, notValidBeforeFlag)
|
nvb, nvbRelative, err := common.ParseEpoch(cmd, notValidBeforeFlag)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if iatRelative || expRelative || nvbRelative {
|
if iatRelative || expRelative || nvbRelative {
|
||||||
endpoint, _ := cmd.Flags().GetString(commonflags.RPC)
|
endpoint, _ := cmd.Flags().GetString(commonflags.RPC)
|
||||||
currEpoch, err := getCurrentEpoch(endpoint)
|
currEpoch, err := internalclient.GetCurrentEpoch(endpoint)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -140,52 +133,3 @@ func createToken(cmd *cobra.Command, _ []string) error {
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// parseEpoch parses epoch argument. Second return value is true if
|
|
||||||
// the specified epoch is relative, and false otherwise.
|
|
||||||
func parseEpoch(cmd *cobra.Command, flag string) (uint64, bool, error) {
|
|
||||||
s, _ := cmd.Flags().GetString(flag)
|
|
||||||
if len(s) == 0 {
|
|
||||||
return 0, false, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
relative := s[0] == '+'
|
|
||||||
if relative {
|
|
||||||
s = s[1:]
|
|
||||||
}
|
|
||||||
|
|
||||||
epoch, err := strconv.ParseUint(s, 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
return 0, relative, fmt.Errorf("can't parse epoch for %s argument: %w", flag, err)
|
|
||||||
}
|
|
||||||
return epoch, relative, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// getCurrentEpoch returns current epoch.
|
|
||||||
func getCurrentEpoch(endpoint string) (uint64, error) {
|
|
||||||
var addr network.Address
|
|
||||||
|
|
||||||
if err := addr.FromString(endpoint); err != nil {
|
|
||||||
return 0, fmt.Errorf("can't parse RPC endpoint: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
key, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
|
|
||||||
if err != nil {
|
|
||||||
return 0, fmt.Errorf("can't generate key to sign query: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
c, err := internalclient.GetSDKClient(key, addr)
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
ni, err := c.NetworkInfo(ctx, client.PrmNetworkInfo{})
|
|
||||||
if err != nil {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return ni.Info().CurrentEpoch(), nil
|
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in a new issue