telegabber/config/config.go

package config

import (
	"fmt"
	"github.com/pkg/errors"
	"io/ioutil"

	"github.com/santhosh-tekuri/jsonschema"
	"gopkg.in/yaml.v2"
)

// Config is for top-level struct for config
type Config struct {
	Telegram TelegramConfig `yaml:":telegram"`
	XMPP     XMPPConfig     `yaml:":xmpp"`
}

// XMPPConfig is for :xmpp: subtree
type XMPPConfig struct {
	Loglevel string `yaml:":loglevel"`
	Jid      string `yaml:":jid"`
	Host     string `yaml:":host"`
	Port     string `yaml:":port"`
	Password string `yaml:":password"`
	Db       string `yaml:":db"`
}

// TelegramConfig is for :telegram: subtree
type TelegramConfig struct {
	Loglevel  string                `yaml:":loglevel"`
	Content   TelegramContentConfig `yaml:":content"`
	Verbosity uint8                 `yaml:":tdlib_verbosity"`
	Tdlib     TelegramTdlibConfig   `yaml:":tdlib"`
}

// TelegramContentConfig is for :content: subtree
type TelegramContentConfig struct {
	Path   string `yaml:":path"`
	Link   string `yaml:":link"`
	Upload string `yaml:":upload"`
	User   string `yaml:":user"`
	Quota  string `yaml:":quota"`
}

// TelegramTdlibConfig is for :tdlib: subtree
type TelegramTdlibConfig struct {
	Datadir string                    `yaml:":datadir"`
	Client  TelegramTdlibClientConfig `yaml:":client"`
}

// TelegramTdlibClientConfig is for :client: subtree
type TelegramTdlibClientConfig struct {
	APIID               string `yaml:":api_id"`
	APIHash             string `yaml:":api_hash"`
	DeviceModel         string `yaml:":device_model"`
	ApplicationVersion  string `yaml:":application_version"`
	UseChatInfoDatabase bool   `yaml:":use_chat_info_database"`
	UseSecretChats      bool   `yaml:":use_secret_chats"`
	CatchTimeout        int64  `yaml:":catch_timeout"`
}

// ReadConfig reads the specified config file, validates it and returns a struct
func ReadConfig(path string, schemaPath string) (Config, error) {
	var config Config

	file, err := ioutil.ReadFile(path)
	if err != nil {
		return config, errors.Wrap(err, "Can't open config file")
	}

	err = yaml.Unmarshal(file, &config)
	if err != nil {
		return config, errors.Wrap(err, "Error parsing config")
	}

	err = validateConfig(file, schemaPath)
	if err != nil {
		return config, errors.Wrap(err, "Validation error")
	}

	return config, nil
}

func validateConfig(file []byte, schemaPath string) error {
	schema, err := jsonschema.Compile(schemaPath)
	if err != nil {
		return errors.Wrap(err, "Corrupted JSON schema")
	}

	var configGeneric interface{}

	err = yaml.Unmarshal(file, &configGeneric)
	if err != nil {
		return errors.Wrap(err, "Error re-parsing config")
	}

	configGeneric, err = convertToStringKeysRecursive(configGeneric, "")
	if err != nil {
		return errors.Wrap(err, "Config conversion error")
	}

	err = schema.ValidateInterface(configGeneric)
	if err != nil {
		return errors.Wrap(err, "Config validation error")
	}

	return nil
}

// copied and adapted from https://github.com/docker/docker-ce/blob/de14285fad39e215ea9763b8b404a37686811b3f/components/cli/cli/compose/loader/loader.go#L330
func convertToStringKeysRecursive(value interface{}, keyPrefix string) (interface{}, error) {
	if mapping, ok := value.(map[interface{}]interface{}); ok {
		dict := make(map[string]interface{})
		for key, entry := range mapping {
			str, ok := key.(string)
			if !ok {
				return nil, formatInvalidKeyError(keyPrefix, key)
			}
			var newKeyPrefix string
			if keyPrefix == "" {
				newKeyPrefix = str
			} else {
				newKeyPrefix = fmt.Sprintf("%s.%s", keyPrefix, str)
			}
			convertedEntry, err := convertToStringKeysRecursive(entry, newKeyPrefix)
			if err != nil {
				return nil, err
			}
			dict[str] = convertedEntry
		}
		return dict, nil
	}
	if list, ok := value.([]interface{}); ok {
		var convertedList []interface{}
		for index, entry := range list {
			newKeyPrefix := fmt.Sprintf("%s[%d]", keyPrefix, index)
			convertedEntry, err := convertToStringKeysRecursive(entry, newKeyPrefix)
			if err != nil {
				return nil, err
			}
			convertedList = append(convertedList, convertedEntry)
		}
		return convertedList, nil
	}

	return value, nil
}

func formatInvalidKeyError(keyPrefix string, key interface{}) error {
	var location string
	if keyPrefix == "" {
		location = "at top level"
	} else {
		location = fmt.Sprintf("in %s", keyPrefix)
	}
	return errors.Errorf("Non-string key %s: %#v", location, key)
}