i2_bot/lib/config/config.go

// i2_bot – Instinct PokememBro Bot
// Copyright (c) 2017 Vladimir "fat0troll" Hodakov

package config

import (
	// stdlib
	"io/ioutil"
	"log"
	"path/filepath"
	// 3rd-party
	"gopkg.in/yaml.v2"
)

const VERSION = "0.295"

// DatabaseConnection handles database connection settings in config.yaml
type DatabaseConnection struct {
	Host     string `yaml:"host"`
	Port     string `yaml:"port"`
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Database string `yaml:"database"`
}

// TelegramConnection handles settings for Telegram connection in config.yaml
type TelegramConnection struct {
	APIToken string `yaml:"api_token"`
}

// Config is a struct which represents config.yaml structure
type Config struct {
	Telegram TelegramConnection `yaml:"telegram_connection"`
	Database DatabaseConnection `yaml:"database_connection"`
}

// Init is a configuration initializer
func (c *Config) Init() {
	fname, _ := filepath.Abs("./config.yml")
	yamlFile, yerr := ioutil.ReadFile(fname)
	if yerr != nil {
		log.Fatal("Can't read config file")
	}

	yperr := yaml.Unmarshal(yamlFile, c)
	if yperr != nil {
		log.Fatal("Can't parse config file")
	}
}

// New creates new empty Config object
func New() *Config {
	c := &Config{}
	return c
}