fix #566

Author: shuzijun

src/main/java/com/shuzijun/leetcode/plugin/manager/QuestionManager.java

@@ -295,12 +295,17 @@ public static Question pick(Project project, PageInfo<?> pageInfo) {
     }
 
     public static User getUser() {
-        HttpResponse response = Graphql.builder().cn(URLUtils.isCn()).operationName("userStatus").request();
+        HttpResponse response = Graphql.builder().cn(URLUtils.isCn()).operationName("userStatus","globalData").request();
         if (response.getStatusCode() == 200) {
             JSONObject userObject = JSONObject.parseObject(response.getBody()).getJSONObject("data").getJSONObject("userStatus");
             User user = new User();
             user.setPremium(userObject.getBoolean("isPremium"));
             user.setUsername(userObject.getString("username"));
+            if (userObject.containsValue("userSlug")){
+                user.setUserSlug(userObject.getString("userSlug"));
+            }else {
+                user.setUserSlug(user.getUsername());
+            }
             user.setSignedIn(userObject.getBoolean("isSignedIn"));
             user.setVerified(userObject.getBoolean("isVerified"));
             user.setPhoneVerified(userObject.getBoolean("isPhoneVerified"));

src/main/java/com/shuzijun/leetcode/plugin/manager/SessionManager.java

@@ -4,9 +4,11 @@
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.intellij.openapi.project.Project;
+import com.shuzijun.leetcode.plugin.model.Graphql;
 import com.shuzijun.leetcode.plugin.model.HttpRequest;
 import com.shuzijun.leetcode.plugin.model.Session;
 import com.shuzijun.leetcode.plugin.utils.*;
+import com.shuzijun.leetcode.plugin.window.WindowFactory;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -37,6 +39,39 @@ public static List<Session> getSession(Project project, boolean cache) {
             defSession.setEasy(jsonObject.getJSONObject("solvedPerDifficulty").getInteger("Easy"));
             defSession.setMedium(jsonObject.getJSONObject("solvedPerDifficulty").getInteger("Medium"));
             defSession.setHard(jsonObject.getJSONObject("solvedPerDifficulty").getInteger("Hard"));
+            if (URLUtils.isCn()){
+                HttpResponse sessionHttpResponse = Graphql.builder().cn(URLUtils.isCn()).operationName("userSessionProgress")
+                        .variables("userSlug", WindowFactory.getDataContext(project).getData(DataKeys.LEETCODE_PROJECTS_TABS).getUser().getUserSlug())
+                        .cacheParam(defSession.getName())
+                        .cache(cache)
+                        .request();
+                if (sessionHttpResponse.getStatusCode() == 200){
+                    JSONObject questionProgressObject = JSON.parseObject(sessionHttpResponse.getBody()).getJSONObject("data").getJSONObject("userProfileUserQuestionProgress");
+                    JSONArray numAcceptedQuestions = questionProgressObject.getJSONArray("numAcceptedQuestions");
+                    for (int i = 0; i < numAcceptedQuestions.size(); i++) {
+                        JSONObject acceptedQuestions = numAcceptedQuestions.getJSONObject(i);
+                        if ("EASY".equals(acceptedQuestions.getString("difficulty"))){
+                            defSession.setEasy(acceptedQuestions.getInteger("count"));
+                        }else if ("MEDIUM".equals(acceptedQuestions.getString("difficulty"))){
+                            defSession.setMedium(acceptedQuestions.getInteger("count"));
+                        }else if ("HARD".equals(acceptedQuestions.getString("difficulty"))){
+                            defSession.setHard(acceptedQuestions.getInteger("count"));
+                        }
+                    }
+                    Integer attempted = 0;
+                    JSONArray numFailedQuestions = questionProgressObject.getJSONArray("numFailedQuestions");
+                    for (int i = 0; i < numFailedQuestions.size(); i++) {
+                        attempted = attempted + numFailedQuestions.getJSONObject(i).getInteger("count");
+                    }
+                    defSession.setAttempted(attempted);
+                    defSession.setSolvedTotal(defSession.getEasy()+defSession.getMedium()+defSession.getHard());
+                    defSession.setUnsolved(defSession.getQuestionTotal() - defSession.getSolvedTotal());
+
+                }
+            }
+
+
+
             sessionList.add(defSession);
 
             JSONArray jsonArray = jsonObject.getJSONArray("sessionList");

src/main/java/com/shuzijun/leetcode/plugin/model/User.java

@@ -9,6 +9,9 @@ public class User {
 
     private String username;
 
+
+    private String userSlug;
+
     private boolean isPremium;
 
     private boolean isSignedIn = Boolean.FALSE;
@@ -25,6 +28,14 @@ public void setUsername(String username) {
         this.username = username;
     }
 
+    public String getUserSlug() {
+        return userSlug;
+    }
+
+    public void setUserSlug(String userSlug) {
+        this.userSlug = userSlug;
+    }
+
     public boolean isPremium() {
         return isPremium;
     }

src/main/resources/graphql/userSessionProgress_cn.graphql

@@ -0,0 +1,26 @@
+query userSessionProgress($userSlug: String!) {
+    userProfileUserQuestionSubmitStats(userSlug: $userSlug) {
+        acSubmissionNum {
+            difficulty
+            count
+        }
+        totalSubmissionNum {
+            difficulty
+            count
+        }
+    }
+    userProfileUserQuestionProgress(userSlug: $userSlug) {
+        numAcceptedQuestions {
+            difficulty
+            count
+        }
+        numFailedQuestions {
+            difficulty
+            count
+        }
+        numUntouchedQuestions {
+            difficulty
+            count
+        }
+    }
+}

src/main/resources/graphql/userStatus.graphql

@@ -1,12 +1,20 @@
-query userStatus {
+query globalData {
     userStatus {
+        userId
         isSignedIn
+        isMockUser
         isPremium
+        isVerified
         username
         avatar
         isAdmin
         isSuperuser
+        permissions
         isTranslator
-        isVerified
+        activeSessionId
+        notificationStatus {
+            lastModified
+            numUnread
+        }
     }
 }

src/main/resources/graphql/userStatus_cn.graphql

@@ -1,13 +1,21 @@
-query userStatus {
+query globalData {
     userStatus {
         isSignedIn
         isPremium
         username
+        realName
         avatar
+        userSlug
         isAdmin
-        isSuperuser
+        useTranslation
+        premiumExpiredAt
         isTranslator
-        isVerified
+        isSuperuser
         isPhoneVerified
+        isVerified
+    }
+    jobsMyCompany {
+        nameSlug
     }
-}
+    commonNojPermissionTypes
+}