More error handling.

Author: Layoric

src/ServiceStackIDEA/.idea/workspace.xml

@@ -1,4 +1,8 @@
+import com.google.gson.Gson;
 import com.intellij.codeInsight.intention.impl.QuickEditAction;
+import com.intellij.notification.Notification;
+import com.intellij.notification.NotificationType;
+import com.intellij.notification.Notifications;
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.command.CommandProcessor;
 import com.intellij.openapi.command.UndoConfirmationPolicy;
@@ -78,6 +82,75 @@ private boolean containsOptionsHeader(PsiJavaFile psiJavaFile) {
         return false;
     }
 
+    private boolean validateEndPoint(String url) {
+
+        URL metadataUrl = null;
+        URIBuilder builder = null;
+        try {
+            builder = new URIBuilder(url.replace("types/java","types/metadata"));
+            builder.addParameter("format","json");
+            metadataUrl = new URL(builder.build().toString());
+        } catch (URISyntaxException e) {
+            //Log error to IDEA warning bubble/window.
+            e.printStackTrace();
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "Invalid BaseUrl provided", NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
+            return false;
+        } catch (MalformedURLException e) {
+            e.printStackTrace();
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "Invalid BaseUrl provided", NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
+            return false;
+        }
+
+        URLConnection metadataConnection = null;
+        String errorMessage = null;
+        try {
+            metadataConnection = metadataUrl.openConnection();
+        } catch (IOException e) {
+            errorMessage = "Problem connecting to BaseUrl - " + e.getMessage();
+        }
+        metadataConnection.setRequestProperty("content-type", "application/json; charset=utf-8");
+        BufferedReader metadataBufferReader = null;
+        try {
+            metadataBufferReader = new BufferedReader(
+                    new InputStreamReader(
+                            metadataConnection.getInputStream()));
+        } catch (IOException e) {
+            errorMessage = "Problem connecting to BaseUrl - " + e.getMessage();
+        }
+        String metadataInputLine;
+        StringBuilder metadataResponse = new StringBuilder();
+        try {
+            while ((metadataInputLine = metadataBufferReader.readLine()) != null)
+                metadataResponse.append(metadataInputLine);
+
+            metadataBufferReader.close();
+        } catch (IOException e) {
+            errorMessage = "Invalid response, check the BaseUrl is a valid ServiceStack endpoint - " + e.getMessage();
+        }
+
+
+        String metadataJson = metadataResponse.toString();
+        Gson gson = new Gson();
+        try {
+            ServiceStackMetadata metadata = gson.fromJson(metadataJson, ServiceStackMetadata.class);
+            if (metadata == null || metadata.getConfig() == null || metadata.getConfig().getBaseUrl() == null) {
+                errorMessage = "The address url is not a valid ServiceStack endpoint.";
+            }
+        } catch (Exception e) {
+            errorMessage = "The address url is not a valid ServiceStack endpoint.";
+        }
+
+        if(errorMessage != null) {
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", errorMessage, NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
+            return false;
+        }
+
+        return true;
+    }
+
     @Override
     public void invoke(@NotNull Project project, Editor editor, final PsiFile psiFile) throws IncorrectOperationException {
         String code = psiFile.getText();
@@ -100,7 +173,8 @@ public void invoke(@NotNull Project project, Editor editor, final PsiFile psiFil
             }
         }
         if(baseUrl == null) {
-            //throw error
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "BaseUrl property not found.", NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
             return;
         }
         if(!baseUrl.endsWith("/")) {
@@ -112,9 +186,30 @@ public void invoke(@NotNull Project project, Editor editor, final PsiFile psiFil
             builder = new URIBuilder(baseUrl);
         } catch (URISyntaxException e) {
             //Log error to IDEA warning bubble/window.
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "Invalid BaseUrl provided", NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
+            return;
+        }
+
+        String existingPath = builder.getPath();
+        if(existingPath == null || existingPath.equals("/")) {
+            builder.setPath("/types/java");
+        } else {
+            builder.setPath(existingPath + "/types/java");
+        }
+
+
+        try {
+            if(!validateEndPoint(builder.build().toString())) {
+                return;
+            }
+        } catch (URISyntaxException e) {
+            e.printStackTrace();
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "Unable to parse BaseUrl", NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
             return;
         }
-        builder.setPath("/types/java");
+
         for(int i = startParamsIndex; i < linesOfCode.size(); i++) {
             String configLine = linesOfCode.get(i);
             if(!configLine.startsWith("//") && configLine.contains(":")) {
@@ -123,7 +218,6 @@ public void invoke(@NotNull Project project, Editor editor, final PsiFile psiFil
             }
         }
 
-
         try {
             String serverUrl = builder.build().toString();
             URL javaCodeUrl = new URL(serverUrl);
@@ -147,9 +241,12 @@ public void invoke(@NotNull Project project, Editor editor, final PsiFile psiFil
                 document.setText(javaCodeResponse);
             } else {
                 //Show error
+                Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "DTO file not found.", NotificationType.ERROR);
+                Notifications.Bus.notify(notification);
             }
         } catch (Exception e) {
-            //Log with IDEA bubble
+            Notification notification = new Notification("ServiceStackIDEA", "Error Updating Reference", "Invalid BaseUrl provided", NotificationType.ERROR);
+            Notifications.Bus.notify(notification);
             e.printStackTrace();
         }
     }