istanbul-tools/container/options.go

// Copyright 2017 AMIS Technologies
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package container

import (
	"crypto/ecdsa"
	"fmt"
	"net"
	"path/filepath"

	"github.com/ethereum/go-ethereum/cmd/utils"
	"github.com/ethereum/go-ethereum/common"
)

type Option func(*ethereum)

func ImageRepository(repository string) Option {
	return func(eth *ethereum) {
		eth.imageRepository = repository
	}
}

func ImageTag(tag string) Option {
	return func(eth *ethereum) {
		eth.imageTag = tag
	}
}

func DockerNetworkName(dockerNetworkName string) Option {
	return func(eth *ethereum) {
		eth.dockerNetworkName = dockerNetworkName
	}
}

func HostName(hostName string) Option {
	return func(eth *ethereum) {
		eth.hostName = hostName
	}
}

func HostDataDir(path string) Option {
	return func(eth *ethereum) {
		eth.dataDir = path
	}
}

func HostIP(ip net.IP) Option {
	return func(eth *ethereum) {
		eth.ip = ip.String()
	}
}

func HostPort(port int) Option {
	return func(eth *ethereum) {
		eth.port = fmt.Sprintf("%d", port)
	}
}

func HostRPCPort(port int) Option {
	return func(eth *ethereum) {
		eth.rpcPort = fmt.Sprintf("%d", port)
	}
}

func HostWebSocketPort(port int) Option {
	return func(eth *ethereum) {
		eth.wsPort = fmt.Sprintf("%d", port)
	}
}

func Logging(enabled bool) Option {
	return func(eth *ethereum) {
		eth.logging = enabled
	}
}

func IsQuorum(isQuorum bool) Option {
	return func(eth *ethereum) {
		eth.isQuorum = isQuorum
	}
}

func DockerEnv(env []string) Option {
	return func(eth *ethereum) {
		eth.dockerEnv = env
	}
}

func DockerBinds(binds []string) Option {
	return func(eth *ethereum) {
		eth.dockerBinds = binds
	}
}

// ----------------------------------------------------------------------------

func Key(key *ecdsa.PrivateKey) Option {
	return func(eth *ethereum) {
		eth.key = key
	}
}

func DataDir(dir string) Option {
	return func(eth *ethereum) {
		utils.DataDirFlag.Value = utils.DirectoryString{
			Value: dir,
		}
		eth.flags = append(eth.flags, "--"+utils.DataDirFlag.Name)
		eth.flags = append(eth.flags, dir)

	}
}

func Etherbase(etherbase string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.MinerEtherbaseFlag.Name)
		eth.flags = append(eth.flags, etherbase)
	}
}

func Identity(id string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.IdentityFlag.Name)
		eth.flags = append(eth.flags, id)
	}
}

func IPC(enabled bool) Option {
	return func(eth *ethereum) {
		if !enabled {
			eth.flags = append(eth.flags, "--"+utils.IPCDisabledFlag.Name)
		}
	}
}

func KeyStore(dir string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.KeyStoreDirFlag.Name)
		eth.flags = append(eth.flags, dir)
	}
}

func NetworkID(networkID string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.NetworkIdFlag.Name)
		eth.flags = append(eth.flags, networkID)
	}
}

func Mine() Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.MiningEnabledFlag.Name)
	}
}

func NAT(nat string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.NATFlag.Name)
		eth.flags = append(eth.flags, nat)
	}
}

func NodeKey(nodekey string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.NodeKeyFileFlag.Name)
		eth.flags = append(eth.flags, nodekey)
	}
}

func NodeKeyHex(hex string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.NodeKeyHexFlag.Name)
		eth.flags = append(eth.flags, hex)
	}
}

func NoDiscover() Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.NoDiscoverFlag.Name)
	}
}

func Port(port int) Option {
	return func(eth *ethereum) {
		utils.ListenPortFlag.Value = port
		eth.flags = append(eth.flags, "--"+utils.ListenPortFlag.Name)
		eth.flags = append(eth.flags, fmt.Sprintf("%d", port))
	}
}

func RPC() Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.RPCEnabledFlag.Name)
	}
}

func RPCAddress(address string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.RPCListenAddrFlag.Name)
		eth.flags = append(eth.flags, address)
	}
}

func RPCAPI(apis string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.RPCApiFlag.Name)
		eth.flags = append(eth.flags, apis)
	}
}

func RPCPort(port int) Option {
	return func(eth *ethereum) {
		utils.RPCPortFlag.Value = port
		eth.flags = append(eth.flags, "--"+utils.RPCPortFlag.Name)
		eth.flags = append(eth.flags, fmt.Sprintf("%d", port))
	}
}

func WebSocket() Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.WSEnabledFlag.Name)
	}
}

func WebSocketAddress(address string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.WSListenAddrFlag.Name)
		eth.flags = append(eth.flags, address)
	}
}

func WebSocketAPI(apis string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.WSApiFlag.Name)
		eth.flags = append(eth.flags, apis)
	}
}

func WebSocketPort(port int) Option {
	return func(eth *ethereum) {
		utils.WSPortFlag.Value = port
		eth.flags = append(eth.flags, "--"+utils.WSPortFlag.Name)
		eth.flags = append(eth.flags, fmt.Sprintf("%d", port))
	}
}

func WebSocketOrigin(origins string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.WSAllowedOriginsFlag.Name)
		eth.flags = append(eth.flags, origins)
	}
}

func Verbosity(verbosity int) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--verbosity")
		eth.flags = append(eth.flags, fmt.Sprintf("%d", verbosity))
	}
}

func FaultyMode(mode int) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--istanbul.faultymode")
		eth.flags = append(eth.flags, fmt.Sprintf("%d", mode))
	}
}

func SyncMode(mode string) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.SyncModeFlag.Name)
		eth.flags = append(eth.flags, mode)

	}
}

func NoUSB() Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.NoUSBFlag.Name)
	}
}

func Accounts(accounts []common.Address) Option {
	return func(eth *ethereum) {
		eth.accounts = accounts
	}
}

func Unlock(index int) Option {
	return func(eth *ethereum) {
		eth.flags = append(eth.flags, "--"+utils.UnlockedAccountFlag.Name)
		eth.flags = append(eth.flags, fmt.Sprintf("%d", index))
	}
}

func Password(password string) Option {
	return func(eth *ethereum) {
		eth.password = password
		eth.flags = append(eth.flags, "--"+utils.PasswordFileFlag.Name)
		eth.flags = append(eth.flags, filepath.Join(utils.DataDirFlag.Value.Value, password))
	}
}