clessy/tg/broker.go

package tg

import (
	"encoding/json"
	"fmt"
	"log"
	"net"
)

type Broker struct {
	Socket net.Conn
}

func ConnectToBroker(brokerAddr string) (*Broker, error) {
	sock, err := net.Dial("tcp", brokerAddr)
	if err != nil {
		return nil, err
	}

	broker := new(Broker)
	broker.Socket = sock
	return broker, nil
}

func (b *Broker) Close() {
	b.Socket.Close()
}

func (b *Broker) SendTextMessage(chat *APIChat, text string, original *int) {
	cmd := ClientCommand{
		Type: CmdSendTextMessage,
		TextMessageData: &ClientTextMessageData{
			Text:    text,
			ChatID:  chat.ChatID,
			ReplyID: original,
		},
	}
	// Encode command and send to broker
	data, err := json.Marshal(cmd)
	if err != nil {
		log.Printf("[SendTextMessage] JSON Encode error: %s\n", err.Error())
	}
	fmt.Fprintln(b.Socket, string(data))
}