ark/server/internal/config/config.go

package config

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/ark-network/ark/common"
	"github.com/spf13/viper"
)

type Config struct {
	Datadir             string
	WalletAddr          string
	RoundInterval       int64
	Port                uint32
	EventDbType         string
	DbType              string
	DbDir               string
	DbMigrationPath     string
	SchedulerType       string
	TxBuilderType       string
	NoTLS               bool
	NoMacaroons         bool
	Network             common.Network
	LogLevel            int
	RoundLifetime       int64
	UnilateralExitDelay int64
	BoardingExitDelay   int64
	EsploraURL          string
	NeutrinoPeer        string
	BitcoindRpcUser     string
	BitcoindRpcPass     string
	BitcoindRpcHost     string
	TLSExtraIPs         []string
	TLSExtraDomains     []string
	UnlockerType        string
	UnlockerFilePath    string
	UnlockerPassword    string
	NostrDefaultRelays  []string
	NoteUriPrefix       string
}

var (
	Datadir             = "DATADIR"
	WalletAddr          = "WALLET_ADDR"
	RoundInterval       = "ROUND_INTERVAL"
	Port                = "PORT"
	EventDbType         = "EVENT_DB_TYPE"
	DbType              = "DB_TYPE"
	DbMigrationPath     = "DB_MIGRATION_PATH"
	SchedulerType       = "SCHEDULER_TYPE"
	TxBuilderType       = "TX_BUILDER_TYPE"
	LogLevel            = "LOG_LEVEL"
	Network             = "NETWORK"
	RoundLifetime       = "ROUND_LIFETIME"
	UnilateralExitDelay = "UNILATERAL_EXIT_DELAY"
	BoardingExitDelay   = "BOARDING_EXIT_DELAY"
	EsploraURL          = "ESPLORA_URL"
	NeutrinoPeer        = "NEUTRINO_PEER"
	NostrDefaultRelays  = "NOSTR_DEFAULT_RELAYS"
	// #nosec G101
	BitcoindRpcUser = "BITCOIND_RPC_USER"
	// #nosec G101
	BitcoindRpcPass  = "BITCOIND_RPC_PASS"
	BitcoindRpcHost  = "BITCOIND_RPC_HOST"
	NoMacaroons      = "NO_MACAROONS"
	NoTLS            = "NO_TLS"
	TLSExtraIP       = "TLS_EXTRA_IP"
	TLSExtraDomain   = "TLS_EXTRA_DOMAIN"
	UnlockerType     = "UNLOCKER_TYPE"
	UnlockerFilePath = "UNLOCKER_FILE_PATH"
	UnlockerPassword = "UNLOCKER_PASSWORD"
	NoteUriPrefix    = "NOTE_URI_PREFIX"

	defaultDatadir             = common.AppDataDir("arkd", false)
	defaultRoundInterval       = 15
	DefaultPort                = 7070
	defaultDbType              = "sqlite"
	defaultEventDbType         = "badger"
	defaultDbMigrationPath     = "file://internal/infrastructure/db/sqlite/migration"
	defaultSchedulerType       = "gocron"
	defaultTxBuilderType       = "covenantless"
	defaultNetwork             = "bitcoin"
	defaultEsploraURL          = "https://blockstream.info/api"
	defaultLogLevel            = 5
	defaultRoundLifetime       = 604672
	defaultUnilateralExitDelay = 1024
	defaultBoardingExitDelay   = 604672
	defaultNoMacaroons         = false
	defaultNoTLS               = true
	defaultNostrDefaultRelays  = []string{"wss://relay.primal.net", "wss://relay.damus.io"}
)

func LoadConfig() (*Config, error) {
	viper.SetEnvPrefix("ARK")
	viper.AutomaticEnv()

	viper.SetDefault(Datadir, defaultDatadir)
	viper.SetDefault(Port, DefaultPort)
	viper.SetDefault(DbType, defaultDbType)
	viper.SetDefault(DbMigrationPath, defaultDbMigrationPath)
	viper.SetDefault(NoTLS, defaultNoTLS)
	viper.SetDefault(LogLevel, defaultLogLevel)
	viper.SetDefault(Network, defaultNetwork)
	viper.SetDefault(RoundInterval, defaultRoundInterval)
	viper.SetDefault(RoundLifetime, defaultRoundLifetime)
	viper.SetDefault(SchedulerType, defaultSchedulerType)
	viper.SetDefault(EventDbType, defaultEventDbType)
	viper.SetDefault(TxBuilderType, defaultTxBuilderType)
	viper.SetDefault(UnilateralExitDelay, defaultUnilateralExitDelay)
	viper.SetDefault(EsploraURL, defaultEsploraURL)
	viper.SetDefault(NoMacaroons, defaultNoMacaroons)
	viper.SetDefault(BoardingExitDelay, defaultBoardingExitDelay)
	viper.SetDefault(NostrDefaultRelays, defaultNostrDefaultRelays)
	net, err := getNetwork()
	if err != nil {
		return nil, fmt.Errorf("error while getting network: %s", err)
	}

	if err := initDatadir(); err != nil {
		return nil, fmt.Errorf("error while creating datadir: %s", err)
	}

	return &Config{
		Datadir:             viper.GetString(Datadir),
		WalletAddr:          viper.GetString(WalletAddr),
		RoundInterval:       viper.GetInt64(RoundInterval),
		Port:                viper.GetUint32(Port),
		EventDbType:         viper.GetString(EventDbType),
		DbType:              viper.GetString(DbType),
		DbMigrationPath:     viper.GetString(DbMigrationPath),
		SchedulerType:       viper.GetString(SchedulerType),
		TxBuilderType:       viper.GetString(TxBuilderType),
		NoTLS:               viper.GetBool(NoTLS),
		DbDir:               filepath.Join(viper.GetString(Datadir), "db"),
		LogLevel:            viper.GetInt(LogLevel),
		Network:             net,
		RoundLifetime:       viper.GetInt64(RoundLifetime),
		UnilateralExitDelay: viper.GetInt64(UnilateralExitDelay),
		BoardingExitDelay:   viper.GetInt64(BoardingExitDelay),
		EsploraURL:          viper.GetString(EsploraURL),
		NeutrinoPeer:        viper.GetString(NeutrinoPeer),
		BitcoindRpcUser:     viper.GetString(BitcoindRpcUser),
		BitcoindRpcPass:     viper.GetString(BitcoindRpcPass),
		BitcoindRpcHost:     viper.GetString(BitcoindRpcHost),
		NoMacaroons:         viper.GetBool(NoMacaroons),
		TLSExtraIPs:         viper.GetStringSlice(TLSExtraIP),
		TLSExtraDomains:     viper.GetStringSlice(TLSExtraDomain),
		UnlockerType:        viper.GetString(UnlockerType),
		UnlockerFilePath:    viper.GetString(UnlockerFilePath),
		UnlockerPassword:    viper.GetString(UnlockerPassword),
		NostrDefaultRelays:  viper.GetStringSlice(NostrDefaultRelays),
		NoteUriPrefix:       viper.GetString(NoteUriPrefix),
	}, nil
}

func initDatadir() error {
	datadir := viper.GetString(Datadir)
	return makeDirectoryIfNotExists(datadir)
}

func makeDirectoryIfNotExists(path string) error {
	if _, err := os.Stat(path); os.IsNotExist(err) {
		return os.MkdirAll(path, os.ModeDir|0755)
	}
	return nil
}

func getNetwork() (common.Network, error) {
	switch strings.ToLower(viper.GetString(Network)) {
	case common.Liquid.Name:
		return common.Liquid, nil
	case common.LiquidTestNet.Name:
		return common.LiquidTestNet, nil
	case common.LiquidRegTest.Name:
		return common.LiquidRegTest, nil
	case common.Bitcoin.Name:
		return common.Bitcoin, nil
	case common.BitcoinTestNet.Name:
		return common.BitcoinTestNet, nil
	case common.BitcoinRegTest.Name:
		return common.BitcoinRegTest, nil
	case common.BitcoinSigNet.Name:
		return common.BitcoinSigNet, nil
	default:
		return common.Network{}, fmt.Errorf("unknown network %s", viper.GetString(Network))
	}
}