68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// Config represents the application configuration
|
|
type Config struct {
|
|
Arbitrum ArbitrumConfig `yaml:"arbitrum"`
|
|
Bot BotConfig `yaml:"bot"`
|
|
Uniswap UniswapConfig `yaml:"uniswap"`
|
|
Log LogConfig `yaml:"log"`
|
|
Database DatabaseConfig `yaml:"database"`
|
|
}
|
|
|
|
// ArbitrumConfig represents the Arbitrum node configuration
|
|
type ArbitrumConfig struct {
|
|
RPCEndpoint string `yaml:"rpc_endpoint"`
|
|
WSEndpoint string `yaml:"ws_endpoint"`
|
|
ChainID int64 `yaml:"chain_id"`
|
|
}
|
|
|
|
// BotConfig represents the bot configuration
|
|
type BotConfig struct {
|
|
Enabled bool `yaml:"enabled"`
|
|
PollingInterval int `yaml:"polling_interval"`
|
|
MinProfitThreshold float64 `yaml:"min_profit_threshold"`
|
|
GasPriceMultiplier float64 `yaml:"gas_price_multiplier"`
|
|
}
|
|
|
|
// UniswapConfig represents the Uniswap configuration
|
|
type UniswapConfig struct {
|
|
FactoryAddress string `yaml:"factory_address"`
|
|
PositionManagerAddress string `yaml:"position_manager_address"`
|
|
FeeTiers []int64 `yaml:"fee_tiers"`
|
|
}
|
|
|
|
// LogConfig represents the logging configuration
|
|
type LogConfig struct {
|
|
Level string `yaml:"level"`
|
|
Format string `yaml:"format"`
|
|
File string `yaml:"file"`
|
|
}
|
|
|
|
// DatabaseConfig represents the database configuration
|
|
type DatabaseConfig struct {
|
|
File string `yaml:"file"`
|
|
}
|
|
|
|
// Load loads the configuration from a file
|
|
func Load(filename string) (*Config, error) {
|
|
// Read the config file
|
|
data, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to read config file: %w", err)
|
|
}
|
|
|
|
// Parse the YAML
|
|
var config Config
|
|
if err := yaml.Unmarshal(data, &config); err != nil {
|
|
return nil, fmt.Errorf("failed to parse config file: %w", err)
|
|
}
|
|
|
|
return &config, nil
|
|
} |