73 lines
1.4 KiB
Go
73 lines
1.4 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/pelletier/go-toml"
|
|
)
|
|
|
|
type Config struct {
|
|
Store StoreConfig `toml:"Store"`
|
|
Honeypot HoneypotConfig `toml:"Honeypot"`
|
|
Frontend FrontendConfig `toml:"Frontend"`
|
|
}
|
|
type StoreConfig struct {
|
|
Type string `toml:"Type"`
|
|
Postgres PostgresStoreConfig `toml:"Postgres"`
|
|
}
|
|
|
|
type PostgresStoreConfig struct {
|
|
DSN string `toml:"DSN"`
|
|
}
|
|
|
|
type HoneypotConfig struct {
|
|
ListenAddr string `toml:"ListenAddr"`
|
|
LogLevel string `toml:"LogLevel"`
|
|
HostKeyPath string `toml:"HostKeyPath"`
|
|
}
|
|
|
|
type FrontendConfig struct {
|
|
ListenAddr string `toml:"ListenAddr"`
|
|
LogLevel string `toml:"LogLevel"`
|
|
AccessLogEnable bool `toml:"AccessLogEnable"`
|
|
}
|
|
|
|
func FromReader(r io.Reader) (Config, error) {
|
|
var c Config
|
|
|
|
// Set some defaults
|
|
c.Honeypot.ListenAddr = ":2222"
|
|
c.Honeypot.LogLevel = "INFO"
|
|
c.Frontend.ListenAddr = ":8080"
|
|
c.Frontend.LogLevel = "INFO"
|
|
c.Frontend.AccessLogEnable = true
|
|
|
|
// Read from config
|
|
decoder := toml.NewDecoder(r)
|
|
if err := decoder.Decode(&c); err != nil {
|
|
return c, fmt.Errorf("unable to parse config: %w", err)
|
|
}
|
|
|
|
//c.readEnv()
|
|
|
|
return c, nil
|
|
}
|
|
|
|
func FromString(s string) (Config, error) {
|
|
r := strings.NewReader(s)
|
|
return FromReader(r)
|
|
}
|
|
|
|
func FromFile(path string) (Config, error) {
|
|
f, err := os.Open(path)
|
|
if err != nil {
|
|
return Config{}, fmt.Errorf("error opening config: %w", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
return FromReader(f)
|
|
}
|