From ec66a7b6a763df017f7f74267dd75bf816680995 Mon Sep 17 00:00:00 2001
From: LI JIAHAO <lijiahao99131@gmail.com>
Date: Sun, 7 Oct 2018 22:46:07 +0800
Subject: [PATCH] :bug: use /x/net/websocket due to
 https://github.com/richardchien/coolq-http-api/issues/85

---
 .../websocket-client-cqhttp.go                | 36 ++++++++++---------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/adapters/websocket-client-cqhttp/websocket-client-cqhttp.go b/adapters/websocket-client-cqhttp/websocket-client-cqhttp.go
index 6b84d58..44733db 100644
--- a/adapters/websocket-client-cqhttp/websocket-client-cqhttp.go
+++ b/adapters/websocket-client-cqhttp/websocket-client-cqhttp.go
@@ -3,12 +3,12 @@ package main
 import (
 	"fmt"
 	"github.com/BurntSushi/toml"
-	"github.com/gorilla/websocket"
+	"golang.org/x/net/websocket"
 	"github.com/projectriri/bot-gateway/router"
 	"github.com/projectriri/bot-gateway/types"
 	"github.com/projectriri/bot-gateway/utils"
 	log "github.com/sirupsen/logrus"
-	"net/http"
+	"encoding/json"
 )
 
 var (
@@ -78,13 +78,13 @@ func (p *Plugin) Start() {
 
 	log.Infof("[websocket-client-cqhttp] dialing cqhttp-websocket server")
 	var err error
-	header := http.Header{}
-	header.Add("Authorization", fmt.Sprintf("Token %s", p.config.CQHTTPAccessToken))
 	// Dial /api/ ws
-	p.apiClient, _, err = websocket.DefaultDialer.Dial(
-		p.config.CQHTTPWebSocketAddr+"/api/",
-		header,
-	)
+	apiConfig, err := websocket.NewConfig(p.config.CQHTTPWebSocketAddr+"/api/", "")
+	if err != nil {
+		log.Fatal("[websocket-client-cqhttp] invalid websocket address")
+	}
+	apiConfig.Header.Add("Authorization", fmt.Sprintf("Token %s", p.config.CQHTTPAccessToken))
+	p.apiClient, err = websocket.DialConfig(apiConfig)
 	if err != nil {
 		log.Errorf("[websocket-client-cqhttp] failed to dial cqhttp api websocket (%v)", err)
 	} else {
@@ -92,10 +92,12 @@ func (p *Plugin) Start() {
 	}
 	defer p.apiClient.Close()
 	// Dial /event/ ws
-	p.eventClient, _, err = websocket.DefaultDialer.Dial(
-		p.config.CQHTTPWebSocketAddr+"/event/",
-		header,
-	)
+	eventConfig, err := websocket.NewConfig(p.config.CQHTTPWebSocketAddr+"/event/", "")
+	if err != nil {
+		log.Fatal("[websocket-client-cqhttp] invalid websocket address")
+	}
+	eventConfig.Header.Add("Authorization", fmt.Sprintf("Token %s", p.config.CQHTTPAccessToken))
+	p.eventClient, err = websocket.DialConfig(eventConfig)
 	if err != nil {
 		log.Errorf("[websocket-client-cqhttp] failed to dial cqhttp event websocket (%v)", err)
 	} else {
@@ -106,8 +108,8 @@ func (p *Plugin) Start() {
 	// Start main event update loop
 	go func() {
 		for {
-			_, msg, err := p.eventClient.ReadMessage()
-			if err != nil {
+			msg := json.RawMessage{}
+			if err := websocket.JSON.Receive(p.eventClient, &msg); err != nil {
 				log.Errorf("[websocket-client-cqhttp] failed to read event (%v)", err)
 				continue
 			}
@@ -134,14 +136,14 @@ func (p *Plugin) Start() {
 		for {
 			// send api request
 			apiRequestPkt := cc.Consume()
-			err := p.apiClient.WriteMessage(websocket.TextMessage, apiRequestPkt.Body)
+			err := websocket.JSON.Send(p.apiClient, apiRequestPkt.Body)
 			if err != nil {
 				log.Errorf("[websocket-client-cqhttp] failed to send apirequest (%v)", err)
 				continue
 			}
 			// get api response
-			_, msg, err := p.apiClient.ReadMessage()
-			if err != nil {
+			msg := json.RawMessage{}
+			if err := websocket.JSON.Receive(p.apiClient, &msg); err != nil {
 				log.Errorf("[websocket-client-cqhttp] failed to read apiresponse (%v)", err)
 				continue
 			}