Initial Work Setup

internal/cli/attach.go

@@ -0,0 +1,103 @@
+package cli
+
+import (
+ "fmt"
+ "os"
+ "path/filepath"
+ "strings"
+
+ "github.com/ethereum/go-ethereum/rpc"
+
+ "github.com/ethereum/go-ethereum/cmd/utils"
+ "github.com/ethereum/go-ethereum/console"
+ "github.com/ethereum/go-ethereum/node"
+ "github.com/mitchellh/cli"
+)
+
+// VersionCommand is the command to show the version of the agent
+type AttachCommand struct {
+ UI cli.Ui
+ Meta *Meta
+ Meta2 *Meta2
+}
+
+// Help implements the cli.Command interface
+func (c *AttachCommand) Help() string {
+ return `Usage: bor attach <IPC FILE>
+
+ Connect to Bor IPC console.`
+}
+
+// Synopsis implements the cli.Command interface
+func (c *AttachCommand) Synopsis() string {
+ return "Connect to Bor via IPC"
+}
+
+// Run implements the cli.Command interface
+func (c *AttachCommand) Run(args []string) int {
+
+ c.remoteConsole(args)
+
+ return 0
+}
+
+// remoteConsole will connect to a remote geth instance, attaching a JavaScript
+// console to it.
+func (c *AttachCommand) remoteConsole(args []string) error {
+ // Attach to a remotely running geth instance and start the JavaScript console
+ endpoint := args[0]
+ path := node.DefaultDataDir()
+
+ if endpoint == "" {
+ if c.Meta.dataDir != "" {
+ path = c.Meta.dataDir
+ }
+ if path != "" {
+ homeDir, _ := os.UserHomeDir()
+ path = filepath.Join(homeDir, "/.bor/data")
+ }
+ endpoint = fmt.Sprintf("%s/geth.ipc", path)
+ }
+ client, err := dialRPC(endpoint)
+ if err != nil {
+ utils.Fatalf("Unable to attach to remote geth: %v", err)
+ }
+ config := console.Config{
+ DataDir: path,
+ DocRoot: utils.JSpathFlag.Name,
+ Client: client,
+ }
+
+ console, err := console.New(config)
+ if err != nil {
+ utils.Fatalf("Failed to start the JavaScript console: %v", err)
+ }
+ defer console.Stop(false)
+
+ if len(args) > 1 {
+ if script := args[1]; script == "exec" {
+ console.Evaluate(args[2])
+ return nil
+ }
+ }
+
+ // Otherwise print the welcome screen and enter interactive mode
+ console.Welcome()
+ console.Interactive()
+
+ return nil
+}
+
+// dialRPC returns a RPC client which connects to the given endpoint.
+// The check for empty endpoint implements the defaulting logic
+// for "geth attach" with no argument.
+func dialRPC(endpoint string) (*rpc.Client, error) {
+ if endpoint == "" {
+ endpoint = node.DefaultIPCEndpoint("bor")
+ } else if strings.HasPrefix(endpoint, "rpc:") || strings.HasPrefix(endpoint, "ipc:") {
+ // Backwards compatibility with geth < 1.5 which required
+ // these prefixes.
+ endpoint = endpoint[4:]
+ }
+ return rpc.Dial(endpoint)
+}

internal/cli/command.go

@@ -130,6 +130,13 @@ func commands() map[string]cli.CommandFactory {
  UI: ui,
  }, nil
  },
+ "attach": func() (cli.Command, error) {
+ return &AttachCommand{
+ UI: ui,
+ Meta: meta,
+ Meta2: meta2,
+ }, nil
+ },
  }
 }