Merge pull request skygragon#98 from EINDEX/master

skygragon · web-flow · commit bb09530b15f4 · 2018-05-23T14:54:26.000+08:00
support translatedContent test on leetcode-cn.com
diff --git a/lib/plugins/leetcode.js b/lib/plugins/leetcode.js
@@ -137,6 +137,7 @@ plugin.getProblem = function(problem, cb) {
       '    sampleTestCase',
       '    enableRunCode',
       '    metaData',
+      '    translatedContent',
       '    discussCategoryId',
       '  }',
       '}'
@@ -156,7 +157,11 @@ plugin.getProblem = function(problem, cb) {
 
     problem.totalAC = JSON.parse(q.stats).totalAccepted;
     problem.totalSubmit = JSON.parse(q.stats).totalSubmission;
-    problem.desc = he.decode(cheerio.load(q.content).root().text());
+    if (!q.translatedContent) {
+        problem.desc = he.decode(cheerio.load(q.content).root().text());
+    }else{
+        problem.desc = he.decode(cheerio.load(q.translatedContent).root().text());
+    }
     problem.templates = JSON.parse(q.codeDefinition);
     problem.testcase = q.sampleTestCase;
     problem.testable = q.enableRunCode;
