subscription.go

package kahoot

import "errors"

func (c *Connection) Subscribe(channels ...string) error {
	for _, ch := range channels {
		content := map[string]interface{}{"subscription": ch,
			"clientId": c.clientId}
		pack := c.Packet("/meta/subscribe", content)
		if err := c.Write(pack); err != nil {
			return err
		}
		res, err := c.ReadId(pack.Id)
		if err != nil {
			return err
		}
		// Check the 'success' field
		success, ok := res.Content["successful"].(bool)
		if !ok || !success {
			return errors.New("Negative 'successful' field for channel: " + ch)
		}
	}
	return nil
}

func (c *Connection) Unsubscribe(channels ...string) error {
	for _, ch := range channels {
		content := map[string]interface{}{"subscription": ch,
			"clientId": c.clientId}
		pack := c.Packet("/meta/unsubscribe", content)
		if err := c.Write(pack); err != nil {
			return err
		}
		res, err := c.ReadId(pack.Id)
		if err != nil {
			return err
		}
		// Check the 'success' field
		success, ok := res.Content["successful"].(bool)
		if !ok || !success {
			return errors.New("Negative 'successful' field for channel: " + ch)
		}
	}
	return nil
}