Integration code with front end, #130.

Signed-off-by: Pan Li <[email protected]>

src/main/java/com/microsoft/azure/springcloudplayground/controller/MainController.java

@@ -1,8 +1,10 @@
 package com.microsoft.azure.springcloudplayground.controller;
 
+import com.microsoft.azure.springcloudplayground.exception.GithubProcessException;
 import com.microsoft.azure.springcloudplayground.generator.MicroService;
 import com.microsoft.azure.springcloudplayground.generator.ProjectGenerator;
 import com.microsoft.azure.springcloudplayground.generator.ProjectRequest;
+import com.microsoft.azure.springcloudplayground.github.GithubOperator;
 import com.microsoft.azure.springcloudplayground.metadata.GeneratorMetadataProvider;
 import com.microsoft.azure.springcloudplayground.util.PropertyLoader;
 import com.microsoft.azure.springcloudplayground.util.TelemetryProxy;
@@ -11,6 +13,7 @@
 import org.apache.tools.ant.Project;
 import org.apache.tools.ant.taskdefs.Zip;
 import org.apache.tools.ant.types.ZipFileSet;
+import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.lang.NonNull;
 import org.springframework.security.oauth2.client.authentication.OAuth2AuthenticationToken;
@@ -110,9 +113,19 @@ private void triggerLoginEvent(@NonNull String accountType) {
         this.telemetryProxy.trackEvent(TELEMETRY_EVENT_LOGIN, properties);
     }
 
+    private boolean isValidOAuth2Token(OAuth2AuthenticationToken token) {
+        if (token == null) {
+            return false;
+        } else if (StringUtils.isEmpty(token.getName())) {
+            return false;
+        } else {
+            return true;
+        }
+    }
+
     @RequestMapping(path = "/", produces = "text/html")
     public String home(Map<String, Object> model, OAuth2AuthenticationToken token) {
-        if (token != null && !StringUtils.isEmpty(token.getName())) {
+        if (isValidOAuth2Token(token)) {
             model.put("loggedInUser", token.getPrincipal().getAttributes().get("login"));
         }
 
@@ -123,11 +136,24 @@ public String home(Map<String, Object> model, OAuth2AuthenticationToken token) {
         return "home";
     }
 
-
     @PostMapping("/push-to-github")
-    public String pushToGithub(@RequestBody @Nonnull ProjectRequest request) {
+    public ResponseEntity pushToGithub(@RequestBody @Nonnull ProjectRequest request, OAuth2AuthenticationToken token) {
         log.info("Project request received: " + request);
-        return "push to github not implemented";
+
+        if (isValidOAuth2Token(token)) {
+            String username = token.getPrincipal().getAttributes().get("login").toString();
+            GithubOperator operator = new GithubOperator(username, "");
+            File dir = this.projectGenerator.generate(request);
+
+            try {
+                operator.createRepository(dir, request.getRepoName());
+                return new ResponseEntity(HttpStatus.CREATED);
+            } catch (GithubProcessException e) {
+                return new ResponseEntity(HttpStatus.NOT_ACCEPTABLE);
+            }
+        }
+
+        return new ResponseEntity(HttpStatus.UNAUTHORIZED);
     }
 
     @ResponseBody

src/main/java/com/microsoft/azure/springcloudplayground/exception/GithubFileException.java

@@ -0,0 +1,12 @@
+package com.microsoft.azure.springcloudplayground.exception;
+
+public class GithubFileException extends RuntimeException {
+
+    public GithubFileException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+    public GithubFileException(String message) {
+        super(message);
+    }
+}

src/main/java/com/microsoft/azure/springcloudplayground/exception/GithubProcessException.java

@@ -1,7 +1,7 @@
 package com.microsoft.azure.springcloudplayground.exception;
 
 
-public class GithubProcessException extends RuntimeException {
+public class GithubProcessException extends Exception {
 
     public GithubProcessException(String message, Throwable cause) {
         super(message, cause);

src/main/java/com/microsoft/azure/springcloudplayground/github/GithubOperator.java

@@ -1,6 +1,7 @@
 package com.microsoft.azure.springcloudplayground.github;
 
 import com.google.common.io.Files;
+import com.microsoft.azure.springcloudplayground.exception.GithubFileException;
 import com.microsoft.azure.springcloudplayground.exception.GithubProcessException;
 import com.microsoft.azure.springcloudplayground.github.gitdata.*;
 import com.microsoft.azure.springcloudplayground.github.metadata.Author;
@@ -22,7 +23,7 @@ public GithubOperator(@NonNull String username, @NonNull String token) {
         super(username, token);
     }
 
-    private String getContent(@NonNull HttpResponse response) {
+    private String getContent(@NonNull HttpResponse response) throws GithubProcessException {
         try {
             String line;
             BufferedReader reader = new BufferedReader(new InputStreamReader(response.getEntity().getContent()));
@@ -38,15 +39,15 @@ private String getContent(@NonNull HttpResponse response) {
         }
     }
 
-    private <T> T readValue(@NonNull String json, Class<T> clazz) {
+    private <T> T readValue(@NonNull String json, Class<T> clazz) throws GithubProcessException {
         try {
             return MAPPER.readValue(json, clazz);
         } catch (IOException e) {
             throw new GithubProcessException("Failed to retrieve object from json", e);
         }
     }
 
-    private GithubRepository createRepository(@NonNull String name) {
+    private GithubRepository createRepository(@NonNull String name) throws GithubProcessException {
         GithubRepository repository = GithubRepository.builder(name).build();
         HttpResponse response = super.createRepository(repository);
 
@@ -57,15 +58,15 @@ private GithubRepository createRepository(@NonNull String name) {
         return repository;
     }
 
-    public void deleteRepository(@NonNull GithubRepository repository) {
+    public void deleteRepository(@NonNull GithubRepository repository) throws GithubProcessException {
         HttpResponse response = super.deleteRepository(repository.getName());
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_NO_CONTENT) {
             throw new GithubProcessException("Failed to delete: " + repository.getName());
         }
     }
 
-    private List<GithubCommit> getRepositoryCommits(@NonNull GithubRepository repository) {
+    private List<GithubCommit> getRepositoryCommits(@NonNull GithubRepository repository) throws GithubProcessException {
         HttpResponse response = super.getAllCommits(repository.getName());
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
@@ -79,7 +80,8 @@ private List<GithubCommit> getRepositoryCommits(@NonNull GithubRepository reposi
         return Arrays.asList(commit);
     }
 
-    private GitDataCommit getGitDataCommit(@NonNull GithubRepository repository, @NonNull GithubCommit commit) {
+    private GitDataCommit getGitDataCommit(@NonNull GithubRepository repository, @NonNull GithubCommit commit)
+            throws GithubProcessException {
         HttpResponse response = super.getGitDataCommit(repository.getName(), commit.getSha());
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
@@ -89,7 +91,8 @@ private GitDataCommit getGitDataCommit(@NonNull GithubRepository repository, @No
         return readValue(getContent(response), GitDataCommit.class);
     }
 
-    private GitDataTree getGitDataTree(@NonNull GithubRepository repository, @NonNull GitDataCommit commit) {
+    private GitDataTree getGitDataTree(@NonNull GithubRepository repository, @NonNull GitDataCommit commit)
+            throws GithubProcessException {
         HttpResponse response = super.getGitDataTree(repository.getName(), commit.getTree().getSha());
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
@@ -108,7 +111,8 @@ private GitDataRequestTree getGitDataRequestTree(@NonNull GitDataTree baseTree)
         return requestTree;
     }
 
-    private GitDataBlob createGitDataBlob(@NonNull GithubRepository repository, @NonNull GitDataRequestBlob requestBlob) {
+    private GitDataBlob createGitDataBlob(@NonNull GithubRepository repository, @NonNull GitDataRequestBlob requestBlob)
+            throws GithubProcessException {
         HttpResponse response = super.createGitDataBlob(repository.getName(), requestBlob);
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_CREATED) {
@@ -124,7 +128,7 @@ private GitDataRequestBlob getGitDataRequestBlob(@NonNull String filename) {
 
             return new GitDataRequestBlob(content, "utf-8");
         } catch (IOException e) {
-            throw new GithubProcessException("Failed to read file: " + filename, e);
+            throw new GithubFileException("Failed to read file: " + filename, e);
         }
     }
 
@@ -149,7 +153,8 @@ private GitDataRequestCommit getGitDateRequestCommit(@NonNull GitDataCommit pare
                 .build();
     }
 
-    private GitDataCommit createGitDateCommit(@NonNull GithubRepository repository, @NonNull GitDataRequestCommit commit) {
+    private GitDataCommit createGitDateCommit(@NonNull GithubRepository repository, @NonNull GitDataRequestCommit commit)
+            throws GithubProcessException {
         HttpResponse response = super.createGitDataCommit(repository.getName(), commit);
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_CREATED) {
@@ -159,7 +164,8 @@ private GitDataCommit createGitDateCommit(@NonNull GithubRepository repository,
         return readValue(getContent(response), GitDataCommit.class);
     }
 
-    private GithubTree createGitDataTree(@NonNull GithubRepository repository, @NonNull GitDataRequestTree tree) {
+    private GithubTree createGitDataTree(@NonNull GithubRepository repository, @NonNull GitDataRequestTree tree)
+            throws GithubProcessException {
         HttpResponse response = super.createGitDataTree(repository.getName(), tree);
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_CREATED) {
@@ -174,7 +180,7 @@ private GitDataRequestReference getGitDataRequestReference(@NonNull GitDataCommi
     }
 
     private void updateGithubRepository(@NonNull GithubRepository repository,
-                                        @NonNull GitDataRequestReference reference) {
+                                        @NonNull GitDataRequestReference reference) throws GithubProcessException {
         HttpResponse response = super.updateGitDataReference(repository.getName(), reference);
 
         if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
@@ -219,33 +225,30 @@ private String truncateFileNamePrefix(@NonNull String fileName) {
     }
 
     private GithubTree createGithubTree(@NonNull GithubRepository repository, @NonNull GitDataCommit parentCommit,
-                                        @NonNull File dir) {
+                                        @NonNull File dir) throws GithubProcessException {
         List<String> files = getAllFiles(dir);
         GitDataTree tree = getGitDataTree(repository, parentCommit);
         GitDataRequestTree requestTree = getGitDataRequestTree(tree);
         List<GitDataRequestBlob> requestBlobs = files.stream().map(this::getGitDataRequestBlob).collect(Collectors.toList());
-        List<GitDataBlob> blobs = requestBlobs.stream().map(b -> createGitDataBlob(repository, b)).collect(Collectors.toList());
 
         for (int i = 0; i < files.size(); i++) {
             String filename = files.get(i);
-            String sha = blobs.get(i).getSha();
+            GitDataBlob blob = createGitDataBlob(repository, requestBlobs.get(i));
 
-            requestTree.getTree().add(getRequestTreeNode(truncateFileNamePrefix(filename), sha));
+            requestTree.getTree().add(getRequestTreeNode(truncateFileNamePrefix(filename), blob.getSha()));
         }
 
         return createGitDataTree(repository, requestTree);
     }
 
-    public void createRepository(@NonNull File dir) {
-        GithubRepository repository = createRepository("spring-cloud-azure-demo");
+    public void createRepository(@NonNull File dir, @NonNull String repositoryName) throws GithubProcessException {
+        GithubRepository repository = createRepository(repositoryName);
         GitDataCommit parentCommit = getGitDataCommit(repository, getRepositoryCommits(repository).get(0));
         GithubTree githubTree = createGithubTree(repository, parentCommit, dir);
         GitDataRequestCommit requestCommit = getGitDateRequestCommit(parentCommit, githubTree);
         GitDataCommit commit = createGitDateCommit(repository, requestCommit);
         GitDataRequestReference reference = getGitDataRequestReference(commit);
 
         updateGithubRepository(repository, reference);
-
-        deleteRepository(repository);
     }
 }