202 lines
6.8 KiB
Go
202 lines
6.8 KiB
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 (
|
|
"bytes"
|
|
"fmt"
|
|
"html/template"
|
|
"math/rand"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
)
|
|
|
|
// walletDockerfile is the Dockerfile required to run a web wallet.
|
|
var walletDockerfile = `
|
|
FROM puppeth/wallet:latest
|
|
|
|
ADD genesis.json /genesis.json
|
|
|
|
RUN \
|
|
echo 'node server.js &' > wallet.sh && \
|
|
echo 'geth --cache 512 init /genesis.json' >> wallet.sh && \
|
|
echo $'exec geth --networkid {{.NetworkID}} --port {{.NodePort}} --bootnodes {{.Bootnodes}} --ethstats \'{{.Ethstats}}\' --cache=512 --rpc --rpcaddr=0.0.0.0 --rpccorsdomain "*" --rpcvhosts "*"' >> wallet.sh
|
|
|
|
RUN \
|
|
sed -i 's/PuppethNetworkID/{{.NetworkID}}/g' dist/js/etherwallet-master.js && \
|
|
sed -i 's/PuppethNetwork/{{.Network}}/g' dist/js/etherwallet-master.js && \
|
|
sed -i 's/PuppethDenom/{{.Denom}}/g' dist/js/etherwallet-master.js && \
|
|
sed -i 's/PuppethHost/{{.Host}}/g' dist/js/etherwallet-master.js && \
|
|
sed -i 's/PuppethRPCPort/{{.RPCPort}}/g' dist/js/etherwallet-master.js
|
|
|
|
ENTRYPOINT ["/bin/sh", "wallet.sh"]
|
|
`
|
|
|
|
// walletComposefile is the docker-compose.yml file required to deploy and
|
|
// maintain a web wallet.
|
|
var walletComposefile = `
|
|
version: '2'
|
|
services:
|
|
wallet:
|
|
build: .
|
|
image: {{.Network}}/wallet
|
|
container_name: {{.Network}}_wallet_1
|
|
ports:
|
|
- "{{.NodePort}}:{{.NodePort}}"
|
|
- "{{.NodePort}}:{{.NodePort}}/udp"
|
|
- "{{.RPCPort}}:8545"{{if not .VHost}}
|
|
- "{{.WebPort}}:80"{{end}}
|
|
volumes:
|
|
- {{.Datadir}}:/root/.ethereum
|
|
environment:
|
|
- NODE_PORT={{.NodePort}}/tcp
|
|
- STATS={{.Ethstats}}{{if .VHost}}
|
|
- VIRTUAL_HOST={{.VHost}}
|
|
- VIRTUAL_PORT=80{{end}}
|
|
logging:
|
|
driver: "json-file"
|
|
options:
|
|
max-size: "1m"
|
|
max-file: "10"
|
|
restart: always
|
|
`
|
|
|
|
// deployWallet deploys a new web wallet container to a remote machine via SSH,
|
|
// docker and docker-compose. If an instance with the specified network name
|
|
// already exists there, it will be overwritten!
|
|
func deployWallet(client *sshClient, network string, bootnodes []string, config *walletInfos, nocache bool) ([]byte, error) {
|
|
// Generate the content to upload to the server
|
|
workdir := fmt.Sprintf("%d", rand.Int63())
|
|
files := make(map[string][]byte)
|
|
|
|
dockerfile := new(bytes.Buffer)
|
|
template.Must(template.New("").Parse(walletDockerfile)).Execute(dockerfile, map[string]interface{}{
|
|
"Network": strings.ToTitle(network),
|
|
"Denom": strings.ToUpper(network),
|
|
"NetworkID": config.network,
|
|
"NodePort": config.nodePort,
|
|
"RPCPort": config.rpcPort,
|
|
"Bootnodes": strings.Join(bootnodes, ","),
|
|
"Ethstats": config.ethstats,
|
|
"Host": client.address,
|
|
})
|
|
files[filepath.Join(workdir, "Dockerfile")] = dockerfile.Bytes()
|
|
|
|
composefile := new(bytes.Buffer)
|
|
template.Must(template.New("").Parse(walletComposefile)).Execute(composefile, map[string]interface{}{
|
|
"Datadir": config.datadir,
|
|
"Network": network,
|
|
"NodePort": config.nodePort,
|
|
"RPCPort": config.rpcPort,
|
|
"VHost": config.webHost,
|
|
"WebPort": config.webPort,
|
|
"Ethstats": config.ethstats[:strings.Index(config.ethstats, ":")],
|
|
})
|
|
files[filepath.Join(workdir, "docker-compose.yaml")] = composefile.Bytes()
|
|
|
|
files[filepath.Join(workdir, "genesis.json")] = config.genesis
|
|
|
|
// Upload the deployment files to the remote server (and clean up afterwards)
|
|
if out, err := client.Upload(files); err != nil {
|
|
return out, err
|
|
}
|
|
defer client.Run("rm -rf " + workdir)
|
|
|
|
// Build and deploy the boot or seal node service
|
|
if nocache {
|
|
return nil, client.Stream(fmt.Sprintf("cd %s && docker-compose -p %s build --pull --no-cache && docker-compose -p %s up -d --force-recreate --timeout 60", workdir, network, network))
|
|
}
|
|
return nil, client.Stream(fmt.Sprintf("cd %s && docker-compose -p %s up -d --build --force-recreate --timeout 60", workdir, network))
|
|
}
|
|
|
|
// walletInfos is returned from a web wallet status check to allow reporting
|
|
// various configuration parameters.
|
|
type walletInfos struct {
|
|
genesis []byte
|
|
network int64
|
|
datadir string
|
|
ethstats string
|
|
nodePort int
|
|
rpcPort int
|
|
webHost string
|
|
webPort int
|
|
}
|
|
|
|
// Report converts the typed struct into a plain string->string map, containing
|
|
// most - but not all - fields for reporting to the user.
|
|
func (info *walletInfos) Report() map[string]string {
|
|
report := map[string]string{
|
|
"Data directory": info.datadir,
|
|
"Ethstats username": info.ethstats,
|
|
"Node listener port ": strconv.Itoa(info.nodePort),
|
|
"RPC listener port ": strconv.Itoa(info.rpcPort),
|
|
"Website address ": info.webHost,
|
|
"Website listener port ": strconv.Itoa(info.webPort),
|
|
}
|
|
return report
|
|
}
|
|
|
|
// checkWallet does a health-check against web wallet server to verify whether
|
|
// it's running, and if yes, whether it's responsive.
|
|
func checkWallet(client *sshClient, network string) (*walletInfos, error) {
|
|
// Inspect a possible web wallet container on the host
|
|
infos, err := inspectContainer(client, fmt.Sprintf("%s_wallet_1", network))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !infos.running {
|
|
return nil, ErrServiceOffline
|
|
}
|
|
// Resolve the port from the host, or the reverse proxy
|
|
webPort := infos.portmap["80/tcp"]
|
|
if webPort == 0 {
|
|
if proxy, _ := checkNginx(client, network); proxy != nil {
|
|
webPort = proxy.port
|
|
}
|
|
}
|
|
if webPort == 0 {
|
|
return nil, ErrNotExposed
|
|
}
|
|
// Resolve the host from the reverse-proxy and the config values
|
|
host := infos.envvars["VIRTUAL_HOST"]
|
|
if host == "" {
|
|
host = client.server
|
|
}
|
|
// Run a sanity check to see if the devp2p and RPC ports are reachable
|
|
nodePort := infos.portmap[infos.envvars["NODE_PORT"]]
|
|
if err = checkPort(client.server, nodePort); err != nil {
|
|
log.Warn("Wallet devp2p port seems unreachable", "server", client.server, "port", nodePort, "err", err)
|
|
}
|
|
rpcPort := infos.portmap["8545/tcp"]
|
|
if err = checkPort(client.server, rpcPort); err != nil {
|
|
log.Warn("Wallet RPC port seems unreachable", "server", client.server, "port", rpcPort, "err", err)
|
|
}
|
|
// Assemble and return the useful infos
|
|
stats := &walletInfos{
|
|
datadir: infos.volumes["/root/.ethereum"],
|
|
nodePort: nodePort,
|
|
rpcPort: rpcPort,
|
|
webHost: host,
|
|
webPort: webPort,
|
|
ethstats: infos.envvars["STATS"],
|
|
}
|
|
return stats, nil
|
|
}
|