chirpnest/main.go
2021-10-27 09:44:55 +08:00

149 lines
3.1 KiB
Go

package main
//go:generate go-bindata data/...
import (
"database/sql"
"flag"
"fmt"
"log"
"math/rand"
"net/http"
"sync"
"time"
_ "net/http/pprof"
"github.com/robfig/cron/v3"
_ "github.com/go-sql-driver/mysql"
tb "gopkg.in/tucnak/telebot.v2"
)
var (
config = flag.String("config", "./data/config.json", "config file path")
initdb = flag.Bool("initdb", false, "initialize bot database")
db *sql.DB
bot *tb.Bot
cr *cron.Cron
cfg Config
RndSrc *rand.Rand
RndMux sync.Mutex
MQCWMsgQueue chan ChatWarsMessage
SQLMsgIdentifyQueue chan int64
TGCmdQueue chan TGCommand
MQTGCmdQueue chan TGCommand
JobQueue chan Job
msgParsingRules map[int]MessageParsingRule
clients map[int64]*ChirpClient
muxClients sync.RWMutex
callbacks map[int64]map[int64][]int64
muxCallbacks sync.Mutex
gDepositForwardMux sync.Mutex
gDepositForwardMsg []int64
)
func main() {
log.Println("Starting Chirpnest...")
// Parsing config
flag.Parse()
// randomize
RndSrc = rand.New(rand.NewSource(time.Now().UTC().UnixNano()))
err := ReadConfig(*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()
}
MQCWMsgQueue = make(chan ChatWarsMessage, MQCWMsgQueueSize)
SQLMsgIdentifyQueue = make(chan int64, SQLMsgIdentifyQueueSize)
TGCmdQueue = make(chan TGCommand, TGCmdQueueSize)
MQTGCmdQueue = make(chan TGCommand, MQTGCmdQueueSize)
JobQueue = make(chan Job, JobQueueSize)
clients = make(map[int64]*ChirpClient)
callbacks = make(map[int64]map[int64][]int64)
// Registering bot
bot, 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")
initCache(*initdb)
log.Printf("URL : %s\n", bot.URL)
log.Printf("Token : %s\n", bot.Token)
go BotHandlers(bot)
u := tb.User{
ID: int(cfg.Bot.Admin),
}
bot.Send(&u, fmt.Sprintf("Started (%s-b%s - %s)", githash, commits, buildstamp))
cr = startCron()
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 <= JobWorkers; w++ {
go JobWorker(w, JobQueue)
}
for w := 1; w <= TGCmdWorkers; w++ {
go TGCmdWorker(w, bot, TGCmdQueue)
}
for w := 1; w <= MQTGCmdWorkers; w++ {
go MQTGCmdWorker(w, MQTGCmdQueue)
}
go SQLJobWorker()
go MQKeepAliveWorker()
go MQTidyKeepAliveWorker()
go func() {
log.Println(http.ListenAndServe(":6060", nil))
}()
log.Println("Bot started !")
// Main loop
for {
time.Sleep(1 * time.Second)
}
}