Merge pull request #225 from shuzijun/gradle

fix bugs

Author: shuzijun

src/main/java/com/shuzijun/leetcode/plugin/actions/editor/OpenSolutionAction.java

@@ -23,6 +23,9 @@ public class OpenSolutionAction extends AbstractEditAction {
     @Override
     public void update(@NotNull AnActionEvent anActionEvent) {
         VirtualFile vf = anActionEvent.getData(PlatformDataKeys.VIRTUAL_FILE);
+        if (vf == null) {
+            return;
+        }
         LeetcodeEditor leetcodeEditor = ProjectConfig.getInstance(anActionEvent.getProject()).getEditor(vf.getPath());
         if (leetcodeEditor == null) {
             return;

src/main/java/com/shuzijun/leetcode/plugin/actions/editor/PositionAction.java

@@ -25,6 +25,9 @@ public class PositionAction extends AbstractEditAction {
 
     @Override
     public void update(@NotNull AnActionEvent e) {
+        if (e.getProject() == null) {
+            return;
+        }
         VirtualFile vf = ArrayUtil.getFirstElement(FileEditorManager.getInstance(e.getProject()).getSelectedFiles());
         if (vf == null) {
             e.getPresentation().setEnabled(false);

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

@@ -120,8 +120,9 @@ public static void openSubmission(Submission submission, Question question, Proj
                                 sb.append(codeTypeEnum.getComment()).append("expected_output:").append(submissionData.getString("expected_output")).append("\n");
                                 sb.append(codeTypeEnum.getComment()).append("code_output:").append(submissionData.getString("code_output")).append("\n");
                                 sb.append(codeTypeEnum.getComment()).append("runtime_error:").append(submissionData.getString("runtime_error")).append("\n");
-                                sb.append(codeTypeEnum.getComment()).append("last_testcase:").append(submissionData.getString("last_testcase").replaceAll("(\\r|\\r\\n|\\n\\r|\\n)", " ")).append("\n");
-
+                                if(submissionData.containsKey("last_testcase")) {
+                                    sb.append(codeTypeEnum.getComment()).append("last_testcase:").append(submissionData.getString("last_testcase").replaceAll("(\\r|\\r\\n|\\n\\r|\\n)", " ")).append("\n");
+                                }
                             }
                             FileUtils.saveFile(file, sb.toString());
                             FileUtils.openFileEditor(file,project);

src/main/java/com/shuzijun/leetcode/plugin/utils/FileUtils.java

@@ -138,7 +138,7 @@ public Boolean compute() throws Throwable {
                     codeList.add(line);
                     lineCount++;
                 }
-                if (codeBegins.size() == codeBegins.size() && codeBegins.size() > 0) {
+                if (codeBegins.size() == codeEnds.size() && codeBegins.size() > 0) {
                     for (int s = 0; s < codeBegins.size(); s++) {
                         for (int i = codeBegins.get(s) + 1; i < codeEnds.get(s); i++) {
                             code.append(codeList.get(i)).append("\n");

src/main/java/com/shuzijun/leetcode/plugin/window/LoginFrame.java

@@ -175,10 +175,14 @@ public void actionPerformed(ActionEvent e) {
                 for (String cookieString : cookies) {
                     String[] cookie = cookieString.trim().split("=");
                     if (cookie.length >= 2) {
-                        HttpCookie basicClientCookie = new HttpCookie(cookie[0], cookie[1]);
-                        basicClientCookie.setDomain("." + URLUtils.getLeetcodeHost());
-                        basicClientCookie.setPath("/");
-                        cookieList.add(basicClientCookie);
+                        try {
+                            HttpCookie basicClientCookie = new HttpCookie(cookie[0], cookie[1]);
+                            basicClientCookie.setDomain("." + URLUtils.getLeetcodeHost());
+                            basicClientCookie.setPath("/");
+                            cookieList.add(basicClientCookie);
+                        }catch (IllegalArgumentException ignore){
+
+                        }
                     }
                 }
                 HttpRequestUtils.setCookie(cookieList);