Initial commit

config/config.go

@@ -0,0 +1,72 @@
+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)
+}