From 40d76b22960fcc57e167fb08b97f4ff3aec31090 Mon Sep 17 00:00:00 2001
From: Wim <wim@42.be>
Date: Mon, 11 Feb 2019 01:34:50 +0100
Subject: Fix error handling on bad event queue id (zulip). Closes #694

---
 vendor/github.com/matterbridge/gozulipbot/queue.go | 52 +++++++++++++++++-----
 1 file changed, 42 insertions(+), 10 deletions(-)

(limited to 'vendor/github.com/matterbridge')

diff --git a/vendor/github.com/matterbridge/gozulipbot/queue.go b/vendor/github.com/matterbridge/gozulipbot/queue.go
index 9a37a8e7..af32bb6e 100644
--- a/vendor/github.com/matterbridge/gozulipbot/queue.go
+++ b/vendor/github.com/matterbridge/gozulipbot/queue.go
@@ -13,10 +13,11 @@ import (
 )
 
 var (
-	HeartbeatError    = fmt.Errorf("EventMessage is a heartbeat")
-	UnauthorizedError = fmt.Errorf("Request is unauthorized")
-	BackoffError      = fmt.Errorf("Too many requests")
-	UnknownError      = fmt.Errorf("Error was unknown")
+	HeartbeatError     = fmt.Errorf("EventMessage is a heartbeat")
+	UnauthorizedError  = fmt.Errorf("Request is unauthorized")
+	BackoffError       = fmt.Errorf("Too many requests")
+	BadEventQueueError = fmt.Errorf("BAD_EVENT_QUEUE_ID error")
+	UnknownError       = fmt.Errorf("Error was unknown")
 )
 
 type Queue struct {
@@ -26,6 +27,13 @@ type Queue struct {
 	Bot          *Bot   `json:"-"`
 }
 
+type QueueError struct {
+	Code   string `json:"code"`
+	Msg    string `json:"msg"`
+	ID     string `json:"queue_id"`
+	Result string `json:"result"`
+}
+
 func (q *Queue) EventsChan() (chan EventMessage, func()) {
 	end := false
 	endFunc := func() {
@@ -131,18 +139,22 @@ func (q *Queue) GetEvents() ([]EventMessage, error) {
 	}
 	defer resp.Body.Close()
 
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
 	switch {
 	case resp.StatusCode == 429:
 		return nil, BackoffError
 	case resp.StatusCode == 403:
 		return nil, UnauthorizedError
 	case resp.StatusCode >= 400:
-		return nil, UnknownError
-	}
-
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		return nil, err
+		qErr, err := q.ParseError(body)
+		if err != nil || qErr == nil {
+			return nil, UnknownError
+		}
+		return nil, BadEventQueueError
 	}
 
 	msgs, err := q.ParseEventMessages(body)
@@ -170,6 +182,26 @@ func (q *Queue) RawGetEvents() (*http.Response, error) {
 	return q.Bot.Client.Do(req)
 }
 
+func (q *Queue) ParseError(rawEventResponse []byte) (*QueueError, error) {
+	rawResponse := map[string]json.RawMessage{}
+	err := json.Unmarshal(rawEventResponse, &rawResponse)
+	if err != nil {
+		return nil, err
+	}
+
+	if _, ok := rawResponse["code"]; ok {
+		var qErr QueueError
+		err = json.Unmarshal(rawEventResponse, &qErr)
+		if err != nil {
+			return nil, err
+		}
+		if qErr.Code == "BAD_EVENT_QUEUE_ID" {
+			return &qErr, nil
+		}
+	}
+	return nil, nil
+}
+
 func (q *Queue) ParseEventMessages(rawEventResponse []byte) ([]EventMessage, error) {
 	rawResponse := map[string]json.RawMessage{}
 	err := json.Unmarshal(rawEventResponse, &rawResponse)
-- 
cgit v1.2.3