path: root/cmd/puppeth/wizard_ethstats.go

// Copyright 2017 The go-ethereum Authors
// 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/>.

package main

import (
    "fmt"
    "sort"

    "github.com/ethereum/go-ethereum/log"
)

// deployEthstats queries the user for various input on deploying an ethstats
// monitoring server, after which it executes it.
func (w *wizard) deployEthstats() {
    // Select the server to interact with
    server := w.selectServer()
    if server == "" {
        return
    }
    client := w.servers[server]

    // Retrieve any active ethstats configurations from the server
    infos, err := checkEthstats(client, w.network)
    if err != nil {
        infos = &ethstatsInfos{
            port:   80,
            host:   client.server,
            secret: "",
        }
    }
    existed := err == nil

    // Figure out which port to listen on
    fmt.Println()
    fmt.Printf("Which port should ethstats listen on? (default = %d)\n", infos.port)
    infos.port = w.readDefaultInt(infos.port)

    // Figure which virtual-host to deploy ethstats on
    if infos.host, err = w.ensureVirtualHost(client, infos.port, infos.host); err != nil {
        log.Error("Failed to decide on ethstats host", "err", err)
        return
    }
    // Port and proxy settings retrieved, figure out the secret and boot ethstats
    fmt.Println()
    if infos.secret == "" {
        fmt.Printf("What should be the secret password for the API? (must not be empty)\n")
        infos.secret = w.readString()
    } else {
        fmt.Printf("What should be the secret password for the API? (default = %s)\n", infos.secret)
        infos.secret = w.readDefaultString(infos.secret)
    }
    // Gather any blacklists to ban from reporting
    if existed {
        fmt.Println()
        fmt.Printf("Keep existing IP %v blacklist (y/n)? (default = yes)\n", infos.banned)
        if w.readDefaultString("y") != "y" {
            // The user might want to clear the entire list, although generally probably not
            fmt.Println()
            fmt.Printf("Clear out blacklist and start over (y/n)? (default = no)\n")
            if w.readDefaultString("n") != "n" {
                infos.banned = nil
            }
            // Offer the user to explicitly add/remove certain IP addresses
            fmt.Println()
            fmt.Println("Which additional IP addresses should be blacklisted?")
            for {
                if ip := w.readIPAddress(); ip != "" {
                    infos.banned = append(infos.banned, ip)
                    continue
                }
                break
            }
            fmt.Println()
            fmt.Println("Which IP addresses should not be blacklisted?")
            for {
                if ip := w.readIPAddress(); ip != "" {
                    for i, addr := range infos.banned {
                        if ip == addr {
                            infos.banned = append(infos.banned[:i], infos.banned[i+1:]...)
                            break
                        }
                    }
                    continue
                }
                break
            }
            sort.Strings(infos.banned)
        }
    }
    // Try to deploy the ethstats server on the host
    nocache := false
    if existed {
        fmt.Println()
        fmt.Printf("Should the ethstats be built from scratch (y/n)? (default = no)\n")
        nocache = w.readDefaultString("n") != "n"
    }
    trusted := make([]string, 0, len(w.servers))
    for _, client := range w.servers {
        if client != nil {
            trusted = append(trusted, client.address)
        }
    }
    if out, err := deployEthstats(client, w.network, infos.port, infos.secret, infos.host, trusted, infos.banned, nocache); err != nil {
        log.Error("Failed to deploy ethstats container", "err", err)
        if len(out) > 0 {
            fmt.Printf("%s\n", out)
        }
        return
    }
    // All ok, run a network scan to pick any changes up
    w.networkStats()
}