aboutsummaryrefslogtreecommitdiffstats
path: root/cmd/console/main.go
blob: 00a9ca9c4555c68b5f4353681d1f910355abd597 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
    This file is part of go-ethereum

    go-ethereum is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    go-ethereum is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with go-ethereum.  If not, see <http://www.gnu.org/licenses/>.
*/
/**
 * @authors
 *  Jeffrey Wilcke <i@jev.io>
 */
package main

import (
    "fmt"
    "io"
    "os"

    "github.com/codegangsta/cli"
    "github.com/ethereum/go-ethereum/cmd/utils"
    "github.com/ethereum/go-ethereum/logger"
    "github.com/mattn/go-colorable"
    "github.com/mattn/go-isatty"
)

const (
    ClientIdentifier = "Geth console"
    Version          = "0.9.27"
)

var (
    gitCommit       string // set via linker flag
    nodeNameVersion string
    app             = utils.NewApp(Version, "the geth console")
)

func init() {
    if gitCommit == "" {
        nodeNameVersion = Version
    } else {
        nodeNameVersion = Version + "-" + gitCommit[:8]
    }

    app.Action = run
    app.Flags = []cli.Flag{
        utils.IPCPathFlag,
        utils.VerbosityFlag,
        utils.JSpathFlag,
    }

    app.Before = func(ctx *cli.Context) error {
        utils.SetupLogger(ctx)
        return nil
    }
}

func main() {
    // Wrap the standard output with a colorified stream (windows)
    if isatty.IsTerminal(os.Stdout.Fd()) {
        if pr, pw, err := os.Pipe(); err == nil {
            go io.Copy(colorable.NewColorableStdout(), pr)
            os.Stdout = pw
        }
    }

    var interrupted = false
    utils.RegisterInterrupt(func(os.Signal) {
        interrupted = true
    })
    utils.HandleInterrupt()

    if err := app.Run(os.Args); err != nil {
        fmt.Fprintln(os.Stderr, "Error: ", err)
    }

    // we need to run the interrupt callbacks in case gui is closed
    // this skips if we got here by actual interrupt stopping the GUI
    if !interrupted {
        utils.RunInterruptCallbacks(os.Interrupt)
    }
    logger.Flush()
}

func run(ctx *cli.Context) {
    jspath := ctx.GlobalString(utils.JSpathFlag.Name)
    ipcpath := utils.IpcSocketPath(ctx)
    repl := newJSRE(jspath, ipcpath)

    if ctx.Args().Present() {
        repl.batch(ctx.Args())
    } else {
        repl.interactive(ipcpath)
    }
}