130 lines
4.2 KiB
Go
130 lines
4.2 KiB
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()
|
||
}
|