gocw2/td.go

package main

import (
	"encoding/json"
	"fmt"
	"log"
	"math"
	"time"

	"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)
	var msgCount int32 = 0

	fmt.Printf("Retrieving chat.\n")

	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, 50, false)
		if err != nil {
			if err.Error() == "timeout" {
				logOnError(err, "Waiting...")
				fmt.Printf("Waiting (%d messages retrieved).\n", msgCount)
			} else {
				logOnError(err, "Cannot get history")
				lastMsgID64 = -1
			}
		} else if msgs.TotalCount > 0 {
			for _, msg := range msgs.Messages {
				switch msg.Content.GetMessageContentEnum() {
				case tdlib.MessageTextType:
					m := ChatWarsMessage{
						UserID64: int64(msg.SenderUserID),
						MsgDate:  msg.Date,
						MsgText:  msg.Content.(*tdlib.MessageText).Text.Text,
						MsgID64:  msg.ID,
					}
					if msg.SenderUserID == ownUserID32 {
						m.Direction = Outgoing
					} else {
						m.Direction = Incoming
					}

					b, _ := 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
					}
					msgCount = msgCount + 1
				default:
					log.Printf("getHistory : no handler for %s\n", msg.Content.GetMessageContentEnum())
				}
			}
		} else {
			lastMsgID64 = -1
		}
	}
	log.Printf("Exported %d messages.\n", msgCount)
	fmt.Printf("Exported %d messages.\n", msgCount)
}

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
}