3cf89633e9
The cmd/storagenode/main.go is a big mess right now with so many unneeded config structures initialized and shared by several subcommands. There are many instances where the config structure of one subcommand is mistakenly used for another subcommand. This changes is an attempt to clean up the main.go by moving the subcommands to a separate `cmd_*.go` files with separate config structures for each subcommand. Resolves https://github.com/storj/storj/issues/5756 Change-Id: I85adf2439acba271c023c269739f7fa3c6d49f9d
113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
// Copyright (C) 2020 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/private/cfgstruct"
|
|
"storj.io/private/process"
|
|
"storj.io/storj/multinode/nodes"
|
|
"storj.io/storj/storagenode"
|
|
"storj.io/storj/storagenode/apikeys"
|
|
"storj.io/storj/storagenode/storagenodedb"
|
|
)
|
|
|
|
type nodeInfoCfg struct {
|
|
storagenode.Config
|
|
|
|
JSON bool `default:"false" help:"print node info in JSON format"`
|
|
}
|
|
|
|
func newNodeInfoCmd(f *Factory) *cobra.Command {
|
|
var cfg nodeInfoCfg
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "info",
|
|
Short: "Print storage node info",
|
|
Long: `Print storage node info.
|
|
|
|
--json should be specified to print output in JSON format.
|
|
It is expected that the JSON output will mostly be piped to 'multinode add -'.
|
|
|
|
WARNING: The output includes the api secret of the storagenode.
|
|
`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return cmdInfo(cmd, &cfg)
|
|
},
|
|
Example: `
|
|
#=> print node info
|
|
$ storagenode info --config-dir '<path/to/config-dir>' --identity-dir '<path/to/identity-dir>'
|
|
|
|
#=> print output in JSON format
|
|
$ storagenode info --json --config-dir '<path/to/config-dir>' --identity-dir '<path/to/identity-dir>'
|
|
|
|
#=> add node to multinode dashboard
|
|
$ storagenode info --json --config-dir '<path/to/config-dir>' --identity-dir '<path/to/identity-dir>' | multinode add -
|
|
`,
|
|
Args: cobra.ExactArgs(0),
|
|
}
|
|
|
|
process.Bind(cmd, &cfg, f.Defaults, cfgstruct.ConfDir(f.ConfDir), cfgstruct.IdentityDir(f.IdentityDir))
|
|
|
|
return cmd
|
|
}
|
|
|
|
func cmdInfo(cmd *cobra.Command, cfg *nodeInfoCfg) (err error) {
|
|
ctx, _ := process.Ctx(cmd)
|
|
|
|
// TODO(clement): add support for getting info for all available storagenodes
|
|
|
|
identity, err := cfg.Identity.Load()
|
|
if err != nil {
|
|
zap.L().Fatal("Failed to load identity.", zap.Error(err))
|
|
} else {
|
|
zap.L().Info("Identity loaded.", zap.Stringer("Node ID", identity.ID))
|
|
}
|
|
|
|
db, err := storagenodedb.OpenExisting(ctx, zap.L().Named("db"), cfg.DatabaseConfig())
|
|
if err != nil {
|
|
return errs.New("error starting master database on storage node: %v", err)
|
|
}
|
|
defer func() {
|
|
err = errs.Combine(err, db.Close())
|
|
}()
|
|
|
|
service := apikeys.NewService(db.APIKeys())
|
|
|
|
apiKey, err := service.Issue(ctx)
|
|
if err != nil {
|
|
return errs.New("error while trying to issue new api key: %v", err)
|
|
}
|
|
|
|
if cfg.JSON {
|
|
node := nodes.Node{
|
|
ID: identity.ID,
|
|
APISecret: apiKey.Secret,
|
|
PublicAddress: cfg.Contact.ExternalAddress,
|
|
}
|
|
|
|
data, err := json.Marshal(node)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(string(data))
|
|
return nil
|
|
}
|
|
|
|
fmt.Printf(`
|
|
ID: %s
|
|
API Secret: %s
|
|
Public Address: %s
|
|
`, identity.ID, apiKey.Secret, cfg.Contact.ExternalAddress)
|
|
|
|
return nil
|
|
}
|