Initial commit

context/context.go

@@ -0,0 +1,64 @@
+// Fantasy World Zookeeper Bot
+// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
+
+package context
+
+import (
+	"fmt"
+	"github.com/rs/zerolog"
+	"gopkg.in/yaml.v2"
+	"io/ioutil"
+	"lab.wtfteam.pro/fat0troll/fw_zookeeper_helper/internal/config"
+	"os"
+	"path/filepath"
+	"runtime"
+)
+
+// getMemoryUsage returns memory usage for logger.
+func (c *Context) getMemoryUsage(e *zerolog.Event, level zerolog.Level, message string) {
+	var m runtime.MemStats
+	runtime.ReadMemStats(&m)
+
+	e.Str("memalloc", fmt.Sprintf("%dMB", m.Alloc/1024/1024))
+	e.Str("memsys", fmt.Sprintf("%dMB", m.Sys/1024/1024))
+	e.Str("numgc", fmt.Sprintf("%d", m.NumGC))
+}
+
+// Init is an initialization function for core context
+// Without these parts of the application we can't start at all
+func (c *Context) Init() {
+	c.Logger = zerolog.New(zerolog.ConsoleWriter{Out: os.Stdout}).With().Timestamp().Logger()
+	c.Logger = c.Logger.Hook(zerolog.HookFunc(c.getMemoryUsage))
+
+	c.Logger.Info().Msgf("fw_zookeeper_helper v. %s is starting...", VERSION)
+}
+
+// InitConfiguration reads configuration from YAML and parses it in
+// config.Struct.
+func (c *Context) InitConfiguration() bool {
+	c.Logger.Info().Msg("Loading configuration files...")
+
+	configPath := os.Getenv("BOT_CONFIG")
+	if configPath == "" {
+		configPath = "./example/fw_zookeeper_helper.yaml"
+	}
+	normalizedConfigPath, _ := filepath.Abs(configPath)
+	c.Logger.Debug().Msgf("Configuration file path: %s", normalizedConfigPath)
+
+	// Read configuration file into []byte.
+	fileData, err := ioutil.ReadFile(normalizedConfigPath)
+	if err != nil {
+		c.Logger.Error().Err(err).Msg("Failed to read configuration file")
+		return false
+	}
+
+	c.Config = &config.Struct{}
+	err = yaml.Unmarshal(fileData, c.Config)
+	if err != nil {
+		c.Logger.Error().Err(err).Msg("Failed to parse configuration file")
+		return false
+	}
+
+	c.Logger.Info().Msg("Configuration file parsed successfully")
+	return true
+}

context/exported.go

@@ -0,0 +1,24 @@
+// Fantasy World Zookeeper Helper Bot
+// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
+
+package context
+
+import (
+	"github.com/rs/zerolog"
+	"lab.wtfteam.pro/fat0troll/fw_zookeeper_helper/internal/config"
+)
+
+// VERSION is the current bot's version
+const VERSION = "0.0.1"
+
+// Context is the main application context.
+type Context struct {
+	Config *config.Struct
+	Logger zerolog.Logger
+}
+
+// NewContext is an initialization function for Context
+func NewContext() *Context {
+	c := &Context{}
+	return c
+}