i2_bot/lib/appcontext/appcontext.go

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

package appcontext

import (
	"github.com/go-telegram-bot-api/telegram-bot-api"
	"github.com/jmoiron/sqlx"
	"lab.pztrn.name/fat0troll/i2_bot/lib/chatter/chatterinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/config"
	"lab.pztrn.name/fat0troll/i2_bot/lib/connections"
	"lab.pztrn.name/fat0troll/i2_bot/lib/forwarder/forwarderinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/getters/gettersinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/migrations/migrationsinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/parsers/parsersinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/pinner/pinnerinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/router/routerinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/squader/squaderinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/talkers/talkersinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/welcomer/welcomerinterface"
	"lab.pztrn.name/golibs/mogrus"
	"os"
)

// Context is an application context struct
type Context struct {
	Cfg        *config.Config
	Log        *mogrus.LoggerHandler
	Bot        *tgbotapi.BotAPI
	Forwarder  forwarderinterface.ForwarderInterface
	Migrations migrationsinterface.MigrationsInterface
	Router     routerinterface.RouterInterface
	Parsers    parsersinterface.ParsersInterface
	Db         *sqlx.DB
	Talkers    talkersinterface.TalkersInterface
	Getters    gettersinterface.GettersInterface
	Welcomer   welcomerinterface.WelcomerInterface
	Pinner     pinnerinterface.PinnerInterface
	Chatter    chatterinterface.ChatterInterface
	Squader    squaderinterface.SquaderInterface
}

// Init is a initialization function for context
func (c *Context) Init() {
	c.Cfg = config.New()
	c.Cfg.Init()

	l := mogrus.New()
	l.Initialize()

	log := l.CreateLogger("i2_bot")
	log.CreateOutput("stdout", os.Stdout, true, "debug")

	logFile, err := os.OpenFile(c.Cfg.Logs.LogPath, os.O_RDWR|os.O_APPEND|os.O_CREATE, 0660)
	if err != nil {
		log.Fatalln(err)
	}
	log.CreateOutput("file="+c.Cfg.Logs.LogPath, logFile, true, "debug")

	c.Log = log

	c.Bot = connections.BotInit(c.Cfg, c.Log)
	c.Db = connections.DBInit(c.Cfg, c.Log)
}

// RegisterRouterInterface registering router interface in application
func (c *Context) RegisterRouterInterface(ri routerinterface.RouterInterface) {
	c.Router = ri
	c.Router.Init()
}

// RegisterMigrationsInterface registering migrations interface in application
func (c *Context) RegisterMigrationsInterface(mi migrationsinterface.MigrationsInterface) {
	c.Migrations = mi
	c.Migrations.Init()
}

// RegisterParsersInterface registering parsers interface in application
func (c *Context) RegisterParsersInterface(pi parsersinterface.ParsersInterface) {
	c.Parsers = pi
}

// RegisterTalkersInterface registering talkers interface in application
func (c *Context) RegisterTalkersInterface(ti talkersinterface.TalkersInterface) {
	c.Talkers = ti
	c.Talkers.Init()
}

// RegisterGettersInterface registering getters interface in application
func (c *Context) RegisterGettersInterface(gi gettersinterface.GettersInterface) {
	c.Getters = gi
	c.Getters.Init()
}

// RegisterWelcomerInterface registering welcomer interface in application
func (c *Context) RegisterWelcomerInterface(wi welcomerinterface.WelcomerInterface) {
	c.Welcomer = wi
	c.Welcomer.Init()
}

// RegisterPinnerInterface registering pinner interface in application
func (c *Context) RegisterPinnerInterface(pi pinnerinterface.PinnerInterface) {
	c.Pinner = pi
	c.Pinner.Init()
}

// RegisterForwarderInterface registers forwarder interface in application
func (c *Context) RegisterForwarderInterface(fi forwarderinterface.ForwarderInterface) {
	c.Forwarder = fi
	c.Forwarder.Init()
}

// RegisterChatterInterface registers chatter interface in application
func (c *Context) RegisterChatterInterface(ci chatterinterface.ChatterInterface) {
	c.Chatter = ci
	c.Chatter.Init()
}

// RegisterSquaderInterface registers squader interface in application
func (c *Context) RegisterSquaderInterface(si squaderinterface.SquaderInterface) {
	c.Squader = si
	c.Squader.Init()
}

// RunDatabaseMigrations applies migrations on bot's startup
func (c *Context) RunDatabaseMigrations() {
	c.Migrations.SetDialect("mysql")
	c.Migrations.Migrate()
}