152 lines
3.2 KiB
Go
152 lines
3.2 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")
|
|
githash string
|
|
buildstamp string
|
|
commits string
|
|
|
|
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("172.16.2.1:6060", nil))
|
|
}()
|
|
|
|
log.Println("Bot started !")
|
|
|
|
// Main loop
|
|
for {
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
|
|
}
|