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 } } type MsgDirection string const ( MsgIncoming string = `incoming` MsgOutgoing string = `outgoing` ) type ChatWarsMessage struct { MsgID64 int64 `json:"msg_id"` UserID64 int64 `json:"user_id"` Direction string `json:"direction"` MsgText string `json:"msg"` MsgDate int32 `json:"date"` } var ( config = flag.String("config", "chirpnest.cfg", "config file path") initdb = flag.Bool("initdb", false, "initialize bot database") db *sql.DB cfg Config ) 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) if err != nil { log.Fatalf("Failed to parse gcfg data: %s", err) } // 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) if err != nil { log.Fatal(err) } defer db.Close() } // Check if any issue err = db.Ping() if err != nil { log.Fatal(err) } else { log.Println("SQL connection initialized") } if *initdb { initDB() } go StartBot() go MQMainReceive() fmt.Println("Started !") // Main loop for { time.Sleep(1 * time.Second) } }