clessy/tg/client.go

package tg

import (
	"bufio"
	"encoding/json"
	"io"
	"log"
)

// UpdateHandler is an update handler for webhook updates
type UpdateHandler func(broker *Broker, message APIMessage)

// BrokerCallback is a callback for broker responses to client requests
type BrokerCallback func(broker *Broker, update BrokerUpdate)

// CreateBrokerClient creates a connection to a broker and sends all webhook updates to a given function
// This is the intended way to create clients, please refer to examples for how to make a simple client
func CreateBrokerClient(brokerAddr string, updateFn UpdateHandler) error {
	broker, err := ConnectToBroker(brokerAddr)
	if err != nil {
		return err
	}
	defer broker.Close()

	in := bufio.NewReader(broker.Socket)
	for {
		bytes, _, err := in.ReadLine()
		if err != nil {
			break
		}

		var update BrokerUpdate
		err = json.Unmarshal(bytes, &update)
		if err != nil {
			log.Printf("[tg - CreateBrokerClient] ERROR reading JSON: %s\r\n", err.Error())
			log.Printf("%s\n", string(bytes))
			continue
		}

		if update.Callback == nil {
			// It's a generic message: dispatch to UpdateHandler
			go updateFn(broker, *(update.Message))
		} else {
			// It's a response to a request: retrieve callback and call it
			go broker.SpliceCallback(*(update.Callback))(broker, update)
		}
	}
	return io.EOF
}