upd

main.go

@@ -40,8 +40,8 @@ type Config struct {
 }
 
 var (
-	config = flag.String("config", "gocw2.cfg", "config file path")
+	config  = flag.String("config", "gocw2.cfg", "config file path")
-	initdb = flag.Bool("initdb", false, "initialize tracker database")
+	history = flag.Bool("history", false, "initialize chat history")
 
 	cfg Config
 
@@ -113,6 +113,10 @@ func main() {
 	ownUserID32 = OwnUserID(client)
 	ownUserID64 = int64(OwnUserID(client))
 
+	if *history {
+		getHistory(client)
+	}
+
 	go ListenCW(client)
 
 	fmt.Println("Started !")
@@ -123,122 +127,3 @@ func main() {
 	}
 
 }
-
-func ListenCW(c *tdlib.Client) {
-	eventFilter := func(msg *tdlib.TdMessage) bool {
-		updateMsg := (*msg).(*tdlib.UpdateNewMessage)
-		chatID := updateMsg.Message.ChatID
-		if chatID == user_chtwrsbot {
-			return true
-		}
-		return false
-	}
-
-	conn, err := amqp.Dial("amqp://" + cfg.Rabbit.User + ":" + cfg.Rabbit.Password + "@" + cfg.Rabbit.Host + "/" + cfg.Rabbit.Queue)
-	if err != nil {
-		log.Fatal(err)
-	}
-	defer conn.Close()
-
-	ch, err := conn.Channel()
-	if err != nil {
-		log.Fatal(err)
-	}
-	defer ch.Close()
-
-	q, err := ch.QueueDeclare(
-		"msg", // name
-		false, // durable
-		false, // delete when unused
-		false, // exclusive
-		false, // no-wait
-		nil,   // arguments
-	)
-	if err != nil {
-		log.Fatal(err)
-	}
-
-	receiver := c.AddEventReceiver(&tdlib.UpdateNewMessage{}, eventFilter, 100)
-	for newMsg := range receiver.Chan {
-		updateMsg := (newMsg).(*tdlib.UpdateNewMessage)
-		senderUserID := updateMsg.Message.SenderUserID
-		mType := updateMsg.Message.Content.GetMessageContentEnum()
-		if mType == "messageText" {
-			user, err := c.GetUser(senderUserID)
-			if err != nil {
-				fmt.Println("ListenCW:", err.Error())
-				continue
-			}
-
-			txt := updateMsg.Message.Content.(*tdlib.MessageText).Text.Text
-			t := time.Now()
-
-			m := ChatWarsMessage{
-				UserID64: ownUserID64,
-				MsgDate:  updateMsg.Message.Date,
-				MsgText:  string(txt),
-				MsgID64:  updateMsg.Message.ID,
-			}
-			if senderUserID == ownUserID32 {
-				m.Direction = Outgoing
-			} else {
-				m.Direction = Incoming
-			}
-
-			b, err := json.Marshal(m)
-
-			_ = ch.Publish(
-				"",     // exchange
-				q.Name, // routing key
-				false,  // mandatory
-				false,  // immediate
-				amqp.Publishing{
-					ContentType: "application/json",
-					Body:        []byte(b),
-				})
-
-			fmt.Printf("[%d-%02d-%02d %02d:%02d:%02d-00:00]", t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute(), t.Second())
-			fmt.Println(" === CHATWARS (", user.Username, ") =====================================================================")
-			fmt.Println(txt, "\n")
-			fmt.Println("************ DETAILS ************")
-			fmt.Println("ID : ", updateMsg.Message.ID)
-			fmt.Println("Date : ", updateMsg.Message.Date)
-			fmt.Println("SenderUserID : ", updateMsg.Message.SenderUserID)
-			fmt.Println("ChatID : ", updateMsg.Message.ChatID)
-			fmt.Println("SendingState : ", updateMsg.Message.SendingState)
-			fmt.Println("IsOutgoing : ", updateMsg.Message.IsOutgoing)
-			fmt.Println("CanBeEdited : ", updateMsg.Message.CanBeEdited)
-			fmt.Println("CanBeForwarded : ", updateMsg.Message.CanBeForwarded)
-			fmt.Println("IsChannelPost : ", updateMsg.Message.IsChannelPost)
-			fmt.Println("ContainsUnreadMention : ", updateMsg.Message.ContainsUnreadMention)
-			fmt.Println("ForwardInfo : ", updateMsg.Message.ForwardInfo)
-			fmt.Println("ReplyToMessageID : ", updateMsg.Message.ReplyToMessageID)
-			fmt.Println("ViaBotUserID : ", updateMsg.Message.ViaBotUserID)
-			fmt.Println("================================================================================================================")
-
-		}
-	}
-}
-
-func ForwardMsg(c *tdlib.Client, msgID int64, fromChatID int64, toChatID int64) int64 {
-	msgIDs := make([]int64, 1)
-	msgIDs[0] = msgID
-	msgs, _ := c.ForwardMessages(toChatID, fromChatID, msgIDs, false, false, false)
-	if msgs != nil {
-		return msgs.Messages[0].ID
-	} else {
-		return 0
-	}
-}
-
-func DeleteMsg(c *tdlib.Client, chatID int64, msgID int64) {
-	msgIDs := make([]int64, 1)
-	msgIDs[0] = msgID
-	c.DeleteMessages(chatID, msgIDs, false)
-	fmt.Println("Deleting message ", msgID)
-}
-
-func OwnUserID(c *tdlib.Client) int32 {
-	user, _ := c.GetMe()
-	return user.ID
-}

td.go

@@ -0,0 +1,199 @@
+package main
+
+import (
+	"encoding/json"
+	"flag"
+	"fmt"
+	"log"
+	"time"
+
+	"gopkg.in/gcfg.v1"
+
+	"github.com/Arman92/go-tdlib"
+	"github.com/streadway/amqp"
+)
+
+func ListenCW(c *tdlib.Client) {
+	eventFilter := func(msg *tdlib.TdMessage) bool {
+		updateMsg := (*msg).(*tdlib.UpdateNewMessage)
+		chatID := updateMsg.Message.ChatID
+		if chatID == user_chtwrsbot {
+			return true
+		}
+		return false
+	}
+
+	conn, err := amqp.Dial("amqp://" + cfg.Rabbit.User + ":" + cfg.Rabbit.Password + "@" + cfg.Rabbit.Host + "/" + cfg.Rabbit.Queue)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer conn.Close()
+
+	ch, err := conn.Channel()
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer ch.Close()
+
+	q, err := ch.QueueDeclare(
+		"msg", // name
+		false, // durable
+		false, // delete when unused
+		false, // exclusive
+		false, // no-wait
+		nil,   // arguments
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	receiver := c.AddEventReceiver(&tdlib.UpdateNewMessage{}, eventFilter, 100)
+	for newMsg := range receiver.Chan {
+		updateMsg := (newMsg).(*tdlib.UpdateNewMessage)
+		senderUserID := updateMsg.Message.SenderUserID
+		mType := updateMsg.Message.Content.GetMessageContentEnum()
+		if mType == "messageText" {
+			user, err := c.GetUser(senderUserID)
+			if err != nil {
+				fmt.Println("ListenCW:", err.Error())
+				continue
+			}
+
+			txt := updateMsg.Message.Content.(*tdlib.MessageText).Text.Text
+			t := time.Now()
+
+			m := ChatWarsMessage{
+				UserID64: ownUserID64,
+				MsgDate:  updateMsg.Message.Date,
+				MsgText:  string(txt),
+				MsgID64:  updateMsg.Message.ID,
+			}
+			if senderUserID == ownUserID32 {
+				m.Direction = Outgoing
+			} else {
+				m.Direction = Incoming
+			}
+
+			b, err := json.Marshal(m)
+
+			_ = ch.Publish(
+				"",     // exchange
+				q.Name, // routing key
+				false,  // mandatory
+				false,  // immediate
+				amqp.Publishing{
+					ContentType: "application/json",
+					Body:        []byte(b),
+				})
+
+			fmt.Printf("[%d-%02d-%02d %02d:%02d:%02d-00:00]", t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute(), t.Second())
+			fmt.Println(" === CHATWARS (", user.Username, ") =====================================================================")
+			fmt.Println(txt, "\n")
+			fmt.Println("************ DETAILS ************")
+			fmt.Println("ID : ", updateMsg.Message.ID)
+			fmt.Println("Date : ", updateMsg.Message.Date)
+			fmt.Println("SenderUserID : ", updateMsg.Message.SenderUserID)
+			fmt.Println("ChatID : ", updateMsg.Message.ChatID)
+			fmt.Println("SendingState : ", updateMsg.Message.SendingState)
+			fmt.Println("IsOutgoing : ", updateMsg.Message.IsOutgoing)
+			fmt.Println("CanBeEdited : ", updateMsg.Message.CanBeEdited)
+			fmt.Println("CanBeForwarded : ", updateMsg.Message.CanBeForwarded)
+			fmt.Println("IsChannelPost : ", updateMsg.Message.IsChannelPost)
+			fmt.Println("ContainsUnreadMention : ", updateMsg.Message.ContainsUnreadMention)
+			fmt.Println("ForwardInfo : ", updateMsg.Message.ForwardInfo)
+			fmt.Println("ReplyToMessageID : ", updateMsg.Message.ReplyToMessageID)
+			fmt.Println("ViaBotUserID : ", updateMsg.Message.ViaBotUserID)
+			fmt.Println("================================================================================================================")
+
+		}
+	}
+}
+
+func getHistory(c *tdlib.Client) {
+	var lastMsgID64 int64 = int64(math.MaxInt64)
+
+	conn, err := amqp.Dial("amqp://" + cfg.Rabbit.User + ":" + cfg.Rabbit.Password + "@" + cfg.Rabbit.Host + "/" + cfg.Rabbit.Queue)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer conn.Close()
+
+	ch, err := conn.Channel()
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer ch.Close()
+
+	q, err := ch.QueueDeclare(
+		"msg", // name
+		false, // durable
+		false, // delete when unused
+		false, // exclusive
+		false, // no-wait
+		nil,   // arguments
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for lastMsgID64 >= 0 {
+		msgs, err := c.GetChatHistory(user_chtwrsbot, lastMsgID64, 0, 100, false)
+		if err != nil {
+			logOnError(err, "Cannot get history")
+			lastMsgID64 = -1
+		} else {
+			for msg := range msgs.Messages {
+
+				m := ChatWarsMessage{
+					UserID64: int64(msg.SenderUserID),
+					MsgDate:  msg.Date,
+					MsgText:  msg.Content.(*tdlib.MessageText).Text.Text,
+					MsgID64:  msg.ID,
+				}
+				if senderUserID == ownUserID32 {
+					m.Direction = Outgoing
+				} else {
+					m.Direction = Incoming
+				}
+
+				b, err := json.Marshal(m)
+
+				_ = ch.Publish(
+					"",     // exchange
+					q.Name, // routing key
+					false,  // mandatory
+					false,  // immediate
+					amqp.Publishing{
+						ContentType: "application/json",
+						Body:        []byte(b),
+					})
+				if msg.ID < lastMsgID64 {
+					lastMsgID64 = msg.ID
+				}
+			}
+		}
+	}
+}
+
+func ForwardMsg(c *tdlib.Client, msgID int64, fromChatID int64, toChatID int64) int64 {
+	msgIDs := make([]int64, 1)
+	msgIDs[0] = msgID
+	msgs, _ := c.ForwardMessages(toChatID, fromChatID, msgIDs, false, false, false)
+	if msgs != nil {
+		return msgs.Messages[0].ID
+	} else {
+		return 0
+	}
+}
+
+func DeleteMsg(c *tdlib.Client, chatID int64, msgID int64) {
+	msgIDs := make([]int64, 1)
+	msgIDs[0] = msgID
+	c.DeleteMessages(chatID, msgIDs, false)
+	fmt.Println("Deleting message ", msgID)
+}
+
+func OwnUserID(c *tdlib.Client) int32 {
+	user, _ := c.GetMe()
+	return user.ID
+}

utils.go

@@ -0,0 +1,17 @@
+package main
+
+import (
+	"log"
+)
+
+func failOnError(err error, msg string) {
+	if err != nil {
+		log.Fatalf("%s: %s", msg, err)
+	}
+}
+
+func logOnError(err error, msg string) {
+	if err != nil {
+		log.Printf("%s: %s", msg, err)
+	}
+}