Add basic context with config creation

internal/context/context.go

@@ -0,0 +1,97 @@
+// NNM-Club torrent filess mass downloader
+// Created for Uploaders group
+// Copyright (c) 2012-2019 Vladimir "fat0troll" Hodakov
+
+package context
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/kirsle/configdir"
+	"github.com/rs/zerolog"
+	"gitlab.com/pztrn/flagger"
+	"log"
+	"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))
+}
+
+// initFlagger initializes flags parser
+func (c *Context) initFlagger() {
+	c.Flagger = flagger.New("NNM-Club mass torrent files downloader", flagger.LoggerInterface(log.New(os.Stdout, "", log.Lshortfile)))
+	c.Flagger.Initialize()
+}
+
+func (c *Context) readConfig() {
+	fh, err := os.Open(c.configFilePath)
+	if err != nil {
+		dlog.Fatal().Err(err).Msg("Failed to read config file")
+	}
+
+	defer fh.Close()
+
+	decoder := json.NewDecoder(fh)
+	err = decoder.Decode(&c.Config)
+	if err != nil {
+		dlog.Fatal().Err(err).Msg("Failed to decode config")
+	}
+}
+
+// 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.initFlagger()
+
+	c.Logger = zerolog.New(zerolog.ConsoleWriter{Out: os.Stdout}).With().Timestamp().Logger()
+	c.Logger = c.Logger.Hook(zerolog.HookFunc(c.getMemoryUsage))
+
+	dlog = c.Logger.With().Str("domain", "context").Logger()
+}
+
+func (c *Context) InitConfig() {
+	configPath := configdir.LocalConfig("uploader_tools")
+	err := configdir.MakePath(configPath)
+	if err != nil {
+		dlog.Fatal().Err(err).Str("config directory", configPath).Msg("Failed to obtain config path")
+	}
+
+	dlog.Debug().Str("config directory", configPath).Msg("Got config directory")
+
+	configFile := filepath.Join(configPath, "settings.json")
+	c.configFilePath = configFile
+
+	if _, err = os.Stat(configFile); os.IsNotExist(err) {
+		// Generating new config on first run
+		dlog.Debug().Msg("Generating new config")
+
+		c.SaveConfig()
+	} else {
+		dlog.Debug().Msg("Using existing config")
+
+		c.readConfig()
+	}
+}
+
+func (c *Context) SaveConfig() {
+	fh, err := os.Create(c.configFilePath)
+	if err != nil {
+		dlog.Fatal().Err(err).Msg("Failed to create config file")
+	}
+	defer fh.Close()
+
+	encoder := json.NewEncoder(fh)
+	err = encoder.Encode(&c.Config)
+	if err != nil {
+		dlog.Fatal().Err(err).Msg("Failed to encode config")
+	}
+}

internal/context/exported.go

@@ -0,0 +1,30 @@
+// NNM-Club torrent filess mass downloader
+// Created for Uploaders group
+// Copyright (c) 2012-2019 Vladimir "fat0troll" Hodakov
+
+package context
+
+import (
+	"github.com/rs/zerolog"
+	"gitlab.com/fat0troll/uploader_tools/internal/config"
+	"gitlab.com/pztrn/flagger"
+)
+
+var (
+	dlog zerolog.Logger
+)
+
+// Context is the main application context.
+type Context struct {
+	configFilePath string
+
+	Config  config.Config
+	Flagger *flagger.Flagger
+	Logger  zerolog.Logger
+}
+
+// New creates new Context
+func New() *Context {
+	c := &Context{}
+	return c
+}