// VulcanizeDB
// Copyright © 2019 Vulcanize

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

// This program 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 Affero General Public License for more details.

// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package cmd

import (
	"github.com/spf13/viper"

	pg "github.com/ethereum/go-ethereum/statediff/indexer/postgres"
)

const (
	ETH_NODE_ID       = "ETH_NODE_ID"
	ETH_CLIENT_NAME   = "ETH_CLIENT_NAME"
	ETH_GENESIS_BLOCK = "ETH_GENESIS_BLOCK"
	ETH_NETWORK_ID    = "ETH_NETWORK_ID"
	ETH_CHAIN_ID      = "ETH_CHAIN_ID"

	DB_CACHE_SIZE_MB            = "DB_CACHE_SIZE_MB"
	TRIE_CACHE_SIZE_MB          = "TRIE_CACHE_SIZE_MB"
	LVLDB_PATH                  = "LVLDB_PATH"
	LVLDB_ANCIENT               = "LVLDB_ANCIENT"
	STATEDIFF_TRIE_WORKERS      = "STATEDIFF_TRIE_WORKERS"
	STATEDIFF_SERVICE_WORKERS   = "STATEDIFF_SERVICE_WORKERS"
	STATEDIFF_WORKER_QUEUE_SIZE = "STATEDIFF_WORKER_QUEUE_SIZE"

	SERVICE_IPC_PATH  = "SERVICE_IPC_PATH"
	SERVICE_HTTP_PATH = "SERVICE_HTTP_PATH"

	PROM_METRICS   = "PROM_METRICS"
	PROM_HTTP      = "PROM_HTTP"
	PROM_HTTP_ADDR = "PROM_HTTP_ADDR"
	PROM_HTTP_PORT = "PROM_HTTP_PORT"
	PROM_DB_STATS  = "PROM_DB_STATS"

	PRERUN_ONLY        = "PRERUN_ONLY"
	PRERUN_RANGE_START = "PRERUN_RANGE_START"
	PRERUN_RANGE_STOP  = "PRERUN_RANGE_STOP"
)

// Bind env vars for eth node and DB configuration
func init() {
	viper.BindEnv("server.ipcPath", SERVICE_IPC_PATH)
	viper.BindEnv("server.httpPath", SERVICE_HTTP_PATH)

	viper.BindEnv("ethereum.nodeID", ETH_NODE_ID)
	viper.BindEnv("ethereum.clientName", ETH_CLIENT_NAME)
	viper.BindEnv("ethereum.genesisBlock", ETH_GENESIS_BLOCK)
	viper.BindEnv("ethereum.networkID", ETH_NETWORK_ID)
	viper.BindEnv("ethereum.chainID", ETH_CHAIN_ID)

	viper.BindEnv("database.name", pg.DATABASE_NAME)
	viper.BindEnv("database.hostname", pg.DATABASE_HOSTNAME)
	viper.BindEnv("database.port", pg.DATABASE_PORT)
	viper.BindEnv("database.user", pg.DATABASE_USER)
	viper.BindEnv("database.password", pg.DATABASE_PASSWORD)
	viper.BindEnv("database.maxIdle", pg.DATABASE_MAX_IDLE_CONNECTIONS)
	viper.BindEnv("database.maxOpen", pg.DATABASE_MAX_OPEN_CONNECTIONS)
	viper.BindEnv("database.maxLifetime", pg.DATABASE_MAX_CONN_LIFETIME)

	viper.BindEnv("cache.database", DB_CACHE_SIZE_MB)
	viper.BindEnv("cache.trie", TRIE_CACHE_SIZE_MB)

	viper.BindEnv("leveldb.path", LVLDB_PATH)
	viper.BindEnv("leveldb.ancient", LVLDB_ANCIENT)

	viper.BindEnv("prom.metrics", PROM_METRICS)
	viper.BindEnv("prom.http", PROM_HTTP)
	viper.BindEnv("prom.httpAddr", PROM_HTTP_ADDR)
	viper.BindEnv("prom.httpPort", PROM_HTTP_PORT)
	viper.BindEnv("prom.dbStats", PROM_DB_STATS)

	viper.BindEnv("statediff.serviceWorkers", STATEDIFF_SERVICE_WORKERS)
	viper.BindEnv("statediff.trieWorkers", STATEDIFF_TRIE_WORKERS)
	viper.BindEnv("statediff.workerQueueSize", STATEDIFF_WORKER_QUEUE_SIZE)

	viper.BindEnv("prerun.only", PRERUN_ONLY)
	viper.BindEnv("prerun.start", PRERUN_RANGE_START)
	viper.BindEnv("prerun.stop", PRERUN_RANGE_STOP)
}