package main import ( "database/sql" "flag" "fmt" _ "github.com/go-sql-driver/mysql" "gopkg.in/gcfg.v1" tb "gopkg.in/tucnak/telebot.v2" "log" "time" ) type Config struct { Telegram struct { URL string Token string } SQL struct { Driver string Type string Address string Username string Password string Database string } Rabbit struct { User string Password string Host string Queue string } Bot struct { Admin uint64 Guildname string Guild string } } var ( config = flag.String("config", "chirpnest.cfg", "config file path") initdb = flag.Bool("initdb", false, "initialize bot database") db *sql.DB b *tb.Bot cfg Config MQCWMsgQueue chan ChatWarsMessage SQLMsgIdentifyQueue chan int64 msgParsingRules map[int]MessageParsingRule ) func PrintText(m *tb.Message) { fmt.Printf("[%d] %s(%d) | %s(%d) : %s\n", m.ID, m.Chat.Title, m.Chat.ID, m.Sender.Username, m.Sender.ID, m.Text) return } func main() { log.Println("Starting Chirpnest...") // Parsing config flag.Parse() err := gcfg.ReadFileInto(&cfg, *config) failOnError(err, "Parsing config") // Connecting to DB switch cfg.SQL.Driver { case "mysql": db, err = sql.Open("mysql", cfg.SQL.Username+":"+cfg.SQL.Password+"@"+cfg.SQL.Type+"("+cfg.SQL.Address+")/"+cfg.SQL.Database+"?parseTime=true") failOnError(err, "Opening MySQL connection") defer db.Close() } // Check if any issue err = db.Ping() failOnError(err, "Checking MySQL connection") log.Println("SQL connection initialized") if *initdb { initDB() } resetMsgParsingRules() msgParsingRules, err = loadMsgParsingRules() logOnError(err, "Message parsing rules") // Registering bot b, err := tb.NewBot(tb.Settings{ Token: cfg.Telegram.Token, URL: cfg.Telegram.URL, Poller: &tb.LongPoller{Timeout: 10 * time.Second}, }) failOnError(err, "Registering bot") log.Println("Bot registered") go BotHandlers(b) u := tb.User{ ID: int(cfg.Bot.Admin), } b.Send(&u, "Bot restarted") MQCWMsgQueue = make(chan ChatWarsMessage, 100) SQLMsgIdentifyQueue = make(chan int64, 100) for w := 1; w <= MQGetMsgWorkers; w++ { go MQGetMsgWorker(w, MQCWMsgQueue) } for w := 1; w <= SQLCWMsgWorkers; w++ { go SQLCWMsgWorker(w, MQCWMsgQueue, SQLMsgIdentifyQueue) } for w := 1; w <= SQLIdentifyMsgWorkers; w++ { go SQLIdentifyMsgWorker(w, SQLMsgIdentifyQueue) } for w := 1; w <= SQLJobWorkers; w++ { go SQLJobWorker(w) } log.Println("Bot started !") // Main loop for { time.Sleep(1 * time.Second) } }