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/broadcaster/broadcasterinterface"
	"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/migrations/migrationsinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/pinner/pinnerinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/pokedexer/pokedexerinterface"
	"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/statistics/statisticsinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/talkers/talkersinterface"
	"lab.pztrn.name/fat0troll/i2_bot/lib/users/usersinterface"
	"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
	Pokedexer   pokedexerinterface.PokedexerInterface
	Db          *sqlx.DB
	Talkers     talkersinterface.TalkersInterface
	Broadcaster broadcasterinterface.BroadcasterInterface
	Welcomer    welcomerinterface.WelcomerInterface
	Pinner      pinnerinterface.PinnerInterface
	Chatter     chatterinterface.ChatterInterface
	Squader     squaderinterface.SquaderInterface
	Users       usersinterface.UsersInterface
	Statistics  statisticsinterface.StatisticsInterface
}

// 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()
}

// RegisterPokedexerInterface registering parsers interface in application
func (c *Context) RegisterPokedexerInterface(pi pokedexerinterface.PokedexerInterface) {
	c.Pokedexer = pi
}

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

// RegisterBroadcasterInterface registering broadcaster interface in application
func (c *Context) RegisterBroadcasterInterface(bi broadcasterinterface.BroadcasterInterface) {
	c.Broadcaster = bi
	c.Broadcaster.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()
}

// RegisterUsersInterface registers users interface in application
func (c *Context) RegisterUsersInterface(ui usersinterface.UsersInterface) {
	c.Users = ui
	c.Users.Init()
}

// RegisterStatisticsInterface registers statistics interface in application
func (c *Context) RegisterStatisticsInterface(si statisticsinterface.StatisticsInterface) {
	c.Statistics = si
	c.Statistics.Init()
}

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