96 lines
1.9 KiB
Go
96 lines
1.9 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"storj.io/storj/pkg/node"
|
|
"storj.io/storj/pkg/overlay"
|
|
"storj.io/storj/pkg/process"
|
|
"storj.io/storj/pkg/provider"
|
|
)
|
|
|
|
var (
|
|
rootCmd = &cobra.Command{
|
|
Use: "inspector",
|
|
Short: "CLI for interacting with Storj Kademlia network",
|
|
}
|
|
getNodeCmd = &cobra.Command{
|
|
Use: "get",
|
|
Short: "get node with `id`",
|
|
RunE: GetNode,
|
|
}
|
|
listNodeCmd = &cobra.Command{
|
|
Use: "list-nodes",
|
|
Short: "get all nodes in cache",
|
|
RunE: ListNodes,
|
|
}
|
|
)
|
|
|
|
// Inspector gives access to kademlia and overlay cache
|
|
type Inspector struct {
|
|
overlay overlay.Client
|
|
identity *provider.FullIdentity
|
|
}
|
|
|
|
// NewInspector returns an Inspector client
|
|
func NewInspector(address string) (*Inspector, error) {
|
|
id, err := node.NewFullIdentity(context.Background(), 12, 4)
|
|
if err != nil {
|
|
return &Inspector{}, nil
|
|
}
|
|
overlay, err := overlay.NewOverlayClient(id, address)
|
|
if err != nil {
|
|
return &Inspector{}, nil
|
|
}
|
|
|
|
return &Inspector{
|
|
overlay: overlay,
|
|
identity: id,
|
|
}, nil
|
|
}
|
|
|
|
// GetNode returns a node with the requested ID or nothing at all
|
|
func GetNode(cmd *cobra.Command, args []string) (err error) {
|
|
i, err := NewInspector("127.0.0.1:7778")
|
|
if err != nil {
|
|
fmt.Printf("error dialing inspector: %+v\n", err)
|
|
return err
|
|
}
|
|
|
|
n := node.IDFromString("testnode")
|
|
found, err := i.overlay.Lookup(context.Background(), n)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("### FOUND: %+v\n", found)
|
|
return nil
|
|
}
|
|
|
|
// ListNodes returns the nodes in the cache
|
|
func ListNodes(cmd *cobra.Command, args []string) (err error) {
|
|
i, err := NewInspector("127.0.0.1:7778")
|
|
if err != nil {
|
|
fmt.Printf("error dialing inspector: %+v\n", err)
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("Inspector: %+v\n", i)
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(getNodeCmd)
|
|
rootCmd.AddCommand(listNodeCmd)
|
|
}
|
|
|
|
func main() {
|
|
process.Exec(rootCmd)
|
|
}
|