From 63a5f94a673a242e06a69918be1ced443ec172b9 Mon Sep 17 00:00:00 2001
From: lyswhut <lyswhut@qq.com>
Date: Mon, 28 Oct 2019 22:42:01 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=85=B7=E6=88=91=E6=BA=90?=
 =?UTF-8?q?=E6=90=9C=E7=B4=A2=E6=8F=90=E7=A4=BA=E3=80=81=E6=8E=92=E8=A1=8C?=
 =?UTF-8?q?=E6=A6=9C=E6=8E=A5=E5=8F=A3=E6=8C=82=E6=8E=89=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 publish/changeLog.md                       |  4 ++++
 src/renderer/utils/music/kw/leaderboard.js | 23 +++++++++++++++++----
 src/renderer/utils/music/kw/tempSearch.js  | 24 +++++++++++++++++-----
 src/renderer/utils/music/kw/util.js        | 12 +++++++++++
 4 files changed, 54 insertions(+), 9 deletions(-)

diff --git a/publish/changeLog.md b/publish/changeLog.md
index 1c6d6a07..f1e7efbe 100644
--- a/publish/changeLog.md
+++ b/publish/changeLog.md
@@ -1,3 +1,7 @@
 #### 优化
 
 - 大幅减少程序播放时对CPU与GPU的使用
+
+#### 修复
+
+- 修复酷我源**搜索提示、排行榜**接口挂掉的问题
diff --git a/src/renderer/utils/music/kw/leaderboard.js b/src/renderer/utils/music/kw/leaderboard.js
index 118e48ec..ebf0832d 100644
--- a/src/renderer/utils/music/kw/leaderboard.js
+++ b/src/renderer/utils/music/kw/leaderboard.js
@@ -1,7 +1,6 @@
 import { httpGet, cancelHttp } from '../../request'
 import { formatPlayTime, decodeName } from '../../index'
-import { formatSinger } from './util'
-
+import { formatSinger, getToken, matchToken } from './util'
 
 export default {
   list: [
@@ -67,6 +66,8 @@ export default {
 
   },
   limit: 30,
+  token: null,
+  isGetingToken: false,
   _cancelRequestObj: null,
   _cancelPromiseCancelFn: null,
   _cancelRequestObj2: null,
@@ -89,20 +90,34 @@ export default {
       })
     })
   },
-  getData2(url) {
+  async getData2(url) {
     if (this._cancelRequestObj2 != null) {
       cancelHttp(this._cancelRequestObj2)
       this._cancelPromiseCancelFn2(new Error('取消http请求'))
     }
+    if (this.isGetingToken) return Promise.reject(new Error('正在获取token'))
+    let token = this.token
+    if (!token) {
+      this.isGetingToken = true
+      token = await getToken()
+      this.isGetingToken = false
+    }
     return new Promise((resolve, reject) => {
       this._cancelPromiseCancelFn2 = reject
-      this._cancelRequestObj2 = httpGet(url, (err, resp, body) => {
+      this._cancelRequestObj2 = httpGet(url, {
+        headers: {
+          Referer: 'http://www.kuwo.cn/',
+          csrf: token,
+          cookie: 'kw_token=' + token,
+        },
+      }, (err, resp, body) => {
         this._cancelRequestObj2 = null
         this._cancelPromiseCancelFn2 = null
         if (err) {
           console.log(err)
           reject(err)
         }
+        this.token = matchToken(resp.headers)
         resolve(body)
       })
     })
diff --git a/src/renderer/utils/music/kw/tempSearch.js b/src/renderer/utils/music/kw/tempSearch.js
index d484b29c..9e6fa9bb 100644
--- a/src/renderer/utils/music/kw/tempSearch.js
+++ b/src/renderer/utils/music/kw/tempSearch.js
@@ -1,20 +1,27 @@
 import { httpFetch } from '../../request'
 import { decodeName } from '../../index'
+import { getToken, matchToken } from './util'
 
 export default {
   regExps: {
     relWord: /RELWORD=(.+)/,
   },
+  token: null,
+  isGetingToken: false,
   requestObj: null,
-  tempSearch(str) {
+  tempSearch(str, token) {
     this.cancelTempSearch()
     this.requestObj = httpFetch(`http://www.kuwo.cn/api/www/search/searchKey?key=${encodeURIComponent(str)}`, {
       headers: {
         Referer: 'http://www.kuwo.cn/',
+        csrf: token,
+        cookie: 'kw_token=' + token,
       },
     })
-    return this.requestObj.promise.then(({ statusCode, body }) => {
-      if (statusCode != 200 || body.code !== 200) return Promise.reject(new Error('请求失败'))
+    return this.requestObj.promise.then(({ statusCode, body, headers }) => {
+      if (statusCode != 200) return Promise.reject(new Error('请求失败'))
+      this.token = matchToken(headers)
+      if (body.code !== 200) return Promise.reject(new Error('请求失败'))
       return body
     })
   },
@@ -27,7 +34,14 @@ export default {
   cancelTempSearch() {
     if (this.requestObj && this.requestObj.cancelHttp) this.requestObj.cancelHttp()
   },
-  search(str) {
-    return this.tempSearch(str).then(result => this.handleResult(result.data))
+  async search(str) {
+    let token = this.token
+    if (this.isGetingToken) return Promise.reject(new Error('正在获取token'))
+    if (!this.token) {
+      this.isGetingToken = true
+      token = await getToken()
+      this.isGetingToken = false
+    }
+    return this.tempSearch(str, token).then(result => this.handleResult(result.data))
   },
 }
diff --git a/src/renderer/utils/music/kw/util.js b/src/renderer/utils/music/kw/util.js
index 86eedb1b..7c5fb247 100644
--- a/src/renderer/utils/music/kw/util.js
+++ b/src/renderer/utils/music/kw/util.js
@@ -1,2 +1,14 @@
+import { httpGet } from '../../request'
 
 export const formatSinger = rawData => rawData.replace(/&/g, '、')
+
+export const matchToken = headers => headers['set-cookie'][0].match(/kw_token=(\w+)/)[1]
+
+export const getToken = () => new Promise((resolve, reject) => {
+  httpGet('http://www.kuwo.cn', (err, resp) => {
+    if (err) return reject(err)
+    if (resp.statusCode != 200) return reject(new Error('获取失败'))
+    const token = matchToken(resp.headers)
+    resolve(token)
+  })
+})