prologic-saltyim/readmsgs.go

package saltyim

import (
	"fmt"
	"os"

	"git.mills.io/prologic/msgbus"
	"git.mills.io/prologic/msgbus/client"
	"github.com/keys-pub/keys"

	"go.mills.io/salty"
)

func handleMessage(key *keys.EdX25519Key, msgs chan string) msgbus.HandlerFunc {
	return func(msg *msgbus.Message) error {
		data, _, err := salty.Decrypt(key, msg.Payload)
		if err != nil {
			fmt.Fprintf(os.Stderr, "error decrypting message")
			return err
		}

		msgs <- string(data)

		return nil
	}
}

// Read ...
func Read(key *keys.EdX25519Key, uri, inbox string, stop chan struct{}) chan string {
	client := client.NewClient(uri, nil)

	msgs := make(chan string)

	s := client.Subscribe(inbox, handleMessage(key, msgs))
	s.Start()

	go func() {
		<-stop
		s.Stop()
		close(msgs)
	}()

	return msgs
}