239 lines
5.0 KiB
Go
239 lines
5.0 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
)
|
|
|
|
// Config holds a single hierarchical structure like JSON
|
|
type Config struct {
|
|
data map[string]any
|
|
}
|
|
|
|
// NewConfig creates a new empty config
|
|
func NewConfig() *Config {
|
|
return &Config{
|
|
data: make(map[string]any),
|
|
}
|
|
}
|
|
|
|
// Get retrieves a value from the config using dot notation
|
|
func (c *Config) Get(key string) (any, error) {
|
|
if key == "" {
|
|
return c.data, nil
|
|
}
|
|
|
|
// Parse the dot-notation path manually
|
|
var start, i int
|
|
var current any = c.data
|
|
|
|
for i = 0; i < len(key); i++ {
|
|
if key[i] == '.' || i == len(key)-1 {
|
|
end := i
|
|
if i == len(key)-1 && key[i] != '.' {
|
|
end = i + 1
|
|
}
|
|
|
|
part := key[start:end]
|
|
|
|
// Handle current node based on its type
|
|
switch node := current.(type) {
|
|
case map[string]any:
|
|
// Simple map lookup
|
|
val, ok := node[part]
|
|
if !ok {
|
|
return nil, fmt.Errorf("key %s not found", part)
|
|
}
|
|
current = val
|
|
|
|
case []any:
|
|
// Must be numeric index
|
|
index, err := strconv.Atoi(part)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid array index: %s", part)
|
|
}
|
|
if index < 0 || index >= len(node) {
|
|
return nil, fmt.Errorf("array index out of bounds: %d", index)
|
|
}
|
|
current = node[index]
|
|
|
|
default:
|
|
return nil, fmt.Errorf("cannot access %s in non-container value", part)
|
|
}
|
|
|
|
// If we've processed the entire key, return the current value
|
|
if i == len(key)-1 || (i < len(key)-1 && key[i] == '.' && end == i) {
|
|
if i == len(key)-1 {
|
|
return current, nil
|
|
}
|
|
}
|
|
|
|
start = i + 1
|
|
}
|
|
}
|
|
|
|
return current, nil
|
|
}
|
|
|
|
// GetOr retrieves a value or returns a default if not found
|
|
func (c *Config) GetOr(key string, defaultValue any) any {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return defaultValue
|
|
}
|
|
return val
|
|
}
|
|
|
|
// GetString gets a value as string
|
|
func (c *Config) GetString(key string) (string, error) {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
switch v := val.(type) {
|
|
case string:
|
|
return v, nil
|
|
case bool:
|
|
return strconv.FormatBool(v), nil
|
|
case int64:
|
|
return strconv.FormatInt(v, 10), nil
|
|
case float64:
|
|
return strconv.FormatFloat(v, 'f', -1, 64), nil
|
|
default:
|
|
return "", fmt.Errorf("value for key %s cannot be converted to string", key)
|
|
}
|
|
}
|
|
|
|
// GetBool gets a value as boolean
|
|
func (c *Config) GetBool(key string) (bool, error) {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
switch v := val.(type) {
|
|
case bool:
|
|
return v, nil
|
|
case string:
|
|
return strconv.ParseBool(v)
|
|
default:
|
|
return false, fmt.Errorf("value for key %s cannot be converted to bool", key)
|
|
}
|
|
}
|
|
|
|
// GetInt gets a value as int64
|
|
func (c *Config) GetInt(key string) (int64, error) {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
switch v := val.(type) {
|
|
case int64:
|
|
return v, nil
|
|
case float64:
|
|
return int64(v), nil
|
|
case string:
|
|
return strconv.ParseInt(v, 10, 64)
|
|
default:
|
|
return 0, fmt.Errorf("value for key %s cannot be converted to int", key)
|
|
}
|
|
}
|
|
|
|
// GetFloat gets a value as float64
|
|
func (c *Config) GetFloat(key string) (float64, error) {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
switch v := val.(type) {
|
|
case float64:
|
|
return v, nil
|
|
case int64:
|
|
return float64(v), nil
|
|
case string:
|
|
return strconv.ParseFloat(v, 64)
|
|
default:
|
|
return 0, fmt.Errorf("value for key %s cannot be converted to float", key)
|
|
}
|
|
}
|
|
|
|
// GetArray gets a value as []any
|
|
func (c *Config) GetArray(key string) ([]any, error) {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if arr, ok := val.([]any); ok {
|
|
return arr, nil
|
|
}
|
|
return nil, fmt.Errorf("value for key %s is not an array", key)
|
|
}
|
|
|
|
// GetMap gets a value as map[string]any
|
|
func (c *Config) GetMap(key string) (map[string]any, error) {
|
|
val, err := c.Get(key)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if m, ok := val.(map[string]any); ok {
|
|
return m, nil
|
|
}
|
|
return nil, fmt.Errorf("value for key %s is not a map", key)
|
|
}
|
|
|
|
// Load parses a config from a reader
|
|
func Load(r io.Reader) (*Config, error) {
|
|
parser := NewParser(r)
|
|
return parser.Parse()
|
|
}
|
|
|
|
// Helpers
|
|
|
|
func isLetter(b byte) bool {
|
|
return (b >= 'a' && b <= 'z') || (b >= 'A' && b <= 'Z')
|
|
}
|
|
|
|
func isDigit(b byte) bool {
|
|
return b >= '0' && b <= '9'
|
|
}
|
|
|
|
// ParseNumber converts a string to a number (int64 or float64)
|
|
func ParseNumber(s string) (any, error) {
|
|
// Check if it has a decimal point
|
|
for i := 0; i < len(s); i++ {
|
|
if s[i] == '.' {
|
|
// It's a float
|
|
return strconv.ParseFloat(s, 64)
|
|
}
|
|
}
|
|
// It's an integer
|
|
return strconv.ParseInt(s, 10, 64)
|
|
}
|
|
|
|
// isDigitOrMinus checks if a string starts with a digit or minus sign
|
|
func isDigitOrMinus(s string) bool {
|
|
if len(s) == 0 {
|
|
return false
|
|
}
|
|
return isDigit(s[0]) || (s[0] == '-' && len(s) > 1 && isDigit(s[1]))
|
|
}
|
|
|
|
// parseStringAsNumber tries to parse a string as a number (float or int)
|
|
func parseStringAsNumber(s string) (any, error) {
|
|
// Check if it has a decimal point
|
|
for i := 0; i < len(s); i++ {
|
|
if s[i] == '.' {
|
|
// It's a float
|
|
return strconv.ParseFloat(s, 64)
|
|
}
|
|
}
|
|
// It's an integer
|
|
return strconv.ParseInt(s, 10, 64)
|
|
}
|