mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-22 14:02:32 +00:00
2429b623fc
* [NOD-327] Add --migrate cli flag to API server * [NOD-327] Change log messages * [NOD-327] Remove `required` flag from API server RPC CLI arguments * [NOD-327] Add database version in migrations logs
79 lines
2.6 KiB
Go
79 lines
2.6 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/daglabs/btcd/apiserver/logger"
|
|
"github.com/daglabs/btcd/util"
|
|
"github.com/jessevdk/go-flags"
|
|
"path/filepath"
|
|
)
|
|
|
|
const (
|
|
defaultLogFilename = "apiserver.log"
|
|
defaultErrLogFilename = "apiserver_err.log"
|
|
)
|
|
|
|
var (
|
|
// Default configuration options
|
|
defaultLogDir = util.AppDataDir("apiserver", false)
|
|
defaultDBAddr = "localhost:3306"
|
|
defaultHTTPListen = "0.0.0.0:8080"
|
|
)
|
|
|
|
// Config defines the configuration options for the API server.
|
|
type Config struct {
|
|
LogDir string `long:"logdir" description:"Directory to log output."`
|
|
RPCUser string `short:"u" long:"rpcuser" description:"RPC username"`
|
|
RPCPassword string `short:"P" long:"rpcpass" default-mask:"-" description:"RPC password"`
|
|
RPCServer string `short:"s" long:"rpcserver" description:"RPC server to connect to"`
|
|
RPCCert string `short:"c" long:"rpccert" description:"RPC server certificate chain for validation"`
|
|
DisableTLS bool `long:"notls" description:"Disable TLS"`
|
|
DBAddress string `long:"dbaddress" description:"Database address"`
|
|
DBUser string `long:"dbuser" description:"Database user" required:"true"`
|
|
DBPassword string `long:"dbpass" description:"Database password" required:"true"`
|
|
DBName string `long:"dbname" description:"Database name" required:"true"`
|
|
HTTPListen string `long:"listen" description:"HTTP address to listen on (default: 0.0.0.0:8080)"`
|
|
Migrate bool `long:"migrate" description:"Migrate the database to the latest version. The server will not start when using this flag."`
|
|
}
|
|
|
|
// Parse parses the CLI arguments and returns a config struct.
|
|
func Parse() (*Config, error) {
|
|
cfg := &Config{
|
|
LogDir: defaultLogDir,
|
|
DBAddress: defaultDBAddr,
|
|
HTTPListen: defaultHTTPListen,
|
|
}
|
|
parser := flags.NewParser(cfg, flags.PrintErrors|flags.HelpFlag)
|
|
_, err := parser.Parse()
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !cfg.Migrate {
|
|
if cfg.RPCUser == "" {
|
|
return nil, errors.New("--rpcuser is required if --migrate flag is not used")
|
|
}
|
|
if cfg.RPCPassword == "" {
|
|
return nil, errors.New("--rpcpass is required if --migrate flag is not used")
|
|
}
|
|
if cfg.RPCServer == "" {
|
|
return nil, errors.New("--rpcserver is required if --migrate flag is not used")
|
|
}
|
|
}
|
|
|
|
if cfg.RPCCert == "" && !cfg.DisableTLS {
|
|
return nil, errors.New("--notls has to be disabled if --cert is used")
|
|
}
|
|
|
|
if cfg.RPCCert != "" && cfg.DisableTLS {
|
|
return nil, errors.New("--cert should be omitted if --notls is used")
|
|
}
|
|
|
|
logFile := filepath.Join(cfg.LogDir, defaultLogFilename)
|
|
errLogFile := filepath.Join(cfg.LogDir, defaultErrLogFilename)
|
|
logger.InitLog(logFile, errLogFile)
|
|
|
|
return cfg, nil
|
|
}
|