Fix ranking

tsuwatch · Aug 11, 2019 · 9c884b1 · 9c884b1
1 parent 16dc9d3
commit 9c884b1
Showing 1 changed file with 12 additions and 20 deletions.
diff --git a/src/renderer/use-cases/ViewRankingUseCase.js b/src/renderer/use-cases/ViewRankingUseCase.js
@@ -61,35 +61,27 @@ export default class ViewRankingUseCase extends UseCase {
   _extractRanking(mode, body) {
     const $ = cheerio.load(body);
     if (mode === 'user') {
-      const rankingVideo = $('.ranking_video');
+      const userRanking = $('.rk-RankingList-User');
+      const rankingVideo = userRanking.find('.rk-ProgramCard');
       return rankingVideo.map((i, el) => {
-        return LiveFactory.createWithRankingData(this._extractUserRanking($(el)))
+        return LiveFactory.createWithRankingData(this._extractData($(el)))
       });
     } else if (mode === 'official') {
-      const rank = $('#rank > .active');
-      return rank.children().map((i, el) => {
-        return LiveFactory.createWithRankingData(this._extractOfficialRanking($(el)))
+      const officialAndChannelRanking = $('.rk-RankingList-OfficialAndChannel');
+      const rankingVideo = officialAndChannelRanking.find('.rk-ProgramCard');
+      return rankingVideo.map((i, el) => {
+        return LiveFactory.createWithRankingData(this._extractData($(el)))
       });
     }
   }
 
-  _extractUserRanking(el) {
-    const data = {};
-    data['id'] = el.find('.title').attr('href').match(/lv[0-9]*/)[0];
-    data['title'] = el.find('.video_title').text();
-    data['communityIcon'] = el.find('a > img').attr('src');
-    data['commentCounter'] = el.find('.coment').text();
-    data['viewCounter'] = el.find('.audience').text();
-    return data;
-  }
-
-  _extractOfficialRanking(el) {
+  _extractData(el) {
     const data = {};
-    data['id'] = el.find('p > a').attr('href').match(/lv[0-9]*/)[0];
-    data['title'] = el.find('.detail > p > a').text();
+    data['id'] = el.find('.rk-ProgramCard_DetailTitle').attr('href').match(/lv[0-9]*/)[0];
+    data['title'] = el.find('.rk-ProgramCard_DetailTitle').text();
     data['communityIcon'] = el.find('a > img').attr('src');
-    data['commentCounter'] = el.find('.coment').text();
-    data['viewCounter'] = el.find('.audience').text();
+    data['commentCounter'] = el.find('.rk-ProgramCard_DetailDataCommentCount').text();
+    data['viewCounter'] = el.find('.rk-ProgramCard_DetailDataViewCount').text();
     return data;
   }
 }