Merge pull request #1189 from matthieulapatate/AddDownloadAPI

Download API add to init context
jhipster · Mar 30, 2022 · 0da1948 · 0da1948
2 parents b98fa43 + afc29db
commit 0da1948
Show file tree

Hide file tree

Showing 9 changed files with 74 additions and 2 deletions.
diff --git a/src/main/java/tech/jhipster/lite/common/domain/FileUtils.java b/src/main/java/tech/jhipster/lite/common/domain/FileUtils.java
@@ -229,7 +229,7 @@ public static void rename(String source, String sourceFilename, String destinati
 
   public static byte[] convertFileToByte(String path) throws IOException {
     try (InputStream inputStream = new FileInputStream(path);) {
-      return new byte[inputStream.available()];
+      return inputStream.readAllBytes();
     }
   }
 }
diff --git a/src/main/java/tech/jhipster/lite/generator/init/application/InitApplicationService.java b/src/main/java/tech/jhipster/lite/generator/init/application/InitApplicationService.java
@@ -40,4 +40,8 @@ public void addPrettier(Project project) {
   public void gitInit(Project project) {
     initService.gitInit(project);
   }
+
+  public byte[] download(Project project) {
+    return initService.download(project);
+  }
 }
diff --git a/src/main/java/tech/jhipster/lite/generator/init/domain/InitDomainService.java b/src/main/java/tech/jhipster/lite/generator/init/domain/InitDomainService.java
@@ -116,4 +116,9 @@ public void addPrettier(Project project) {
   public void gitInit(Project project) {
     projectRepository.gitInit(project);
   }
+
+  @Override
+  public byte[] download(Project project) {
+    return projectRepository.download(project);
+  }
 }
diff --git a/src/main/java/tech/jhipster/lite/generator/init/domain/InitService.java b/src/main/java/tech/jhipster/lite/generator/init/domain/InitService.java
@@ -11,4 +11,5 @@ public interface InitService {
   void addEditorConfiguration(Project project);
   void addPrettier(Project project);
   void gitInit(Project project);
+  byte[] download(Project project);
 }
diff --git a/...main/java/tech/jhipster/lite/generator/init/infrastructure/primary/rest/InitResource.java b/...main/java/tech/jhipster/lite/generator/init/infrastructure/primary/rest/InitResource.java
@@ -3,6 +3,11 @@
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.tags.Tag;
+import org.springframework.core.io.ByteArrayResource;
+import org.springframework.core.io.Resource;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -31,4 +36,21 @@ public void init(@RequestBody ProjectDTO projectDTO) {
     Project project = ProjectDTO.toProject(projectDTO);
     initApplicationService.init(project);
   }
+
+  @Operation(summary = "Download project")
+  @ApiResponse(responseCode = "500", description = "An error occurred while downloading project")
+  @PostMapping("/download")
+  @GeneratorStep(id = "download")
+  public ResponseEntity<Resource> download(@RequestBody ProjectDTO projectDTO) {
+    Project project = ProjectDTO.toProject(projectDTO);
+    byte[] out = initApplicationService.download(project);
+    ByteArrayResource resource = new ByteArrayResource(out);
+    return ResponseEntity
+      .ok()
+      .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + project.getBaseName().orElse("application") + ".zip")
+      .contentType(MediaType.parseMediaType("application/octet-stream"))
+      .contentLength(out.length)
+      .header("X-Suggested-Filename", project.getBaseName().orElse("application") + ".zip")
+      .body(resource);
+  }
 }
diff --git a/src/test/java/tech/jhipster/lite/common/domain/FileUtilsTest.java b/src/test/java/tech/jhipster/lite/common/domain/FileUtilsTest.java
@@ -751,4 +751,13 @@ void shouldNotRename() {
 
     assertThatThrownBy(() -> FileUtils.rename(folder, "hello.world", "hello.beer")).isInstanceOf(IOException.class);
   }
+
+  @Test
+  void shouldConvertFileToByte() throws IOException {
+    String folder = tmpDirForTest();
+    createFolder(folder);
+    Files.createFile(Paths.get(folder, "hello.world"));
+
+    assertThat(FileUtils.convertFileToByte(folder + "/hello.world")).isNotNull();
+  }
 }
diff --git a/src/test/java/tech/jhipster/lite/generator/init/application/InitApplicationServiceIT.java b/src/test/java/tech/jhipster/lite/generator/init/application/InitApplicationServiceIT.java
@@ -1,5 +1,6 @@
 package tech.jhipster.lite.generator.init.application;
 
+import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
 import static tech.jhipster.lite.TestUtils.*;
 import static tech.jhipster.lite.common.domain.WordUtils.CRLF;
 import static tech.jhipster.lite.generator.init.application.InitAssertFiles.*;
@@ -138,4 +139,11 @@ void shouldGitInit() {
 
     assertFileGitInit(project);
   }
+
+  @Test
+  void shouldDownloadProject() {
+    Project project = tmpProjectWithPomXml();
+
+    assertThat(initApplicationService.download(project)).isNotNull();
+  }
 }
diff --git a/src/test/java/tech/jhipster/lite/generator/init/domain/InitDomainServiceTest.java b/src/test/java/tech/jhipster/lite/generator/init/domain/InitDomainServiceTest.java
@@ -8,6 +8,7 @@
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 import static tech.jhipster.lite.TestUtils.tmpProject;
+import static tech.jhipster.lite.TestUtils.tmpProjectWithPomXml;
 
 import java.util.Optional;
 import org.junit.jupiter.api.Test;
@@ -100,4 +101,11 @@ void shouldAddPrettier() {
     verify(projectRepository).template(any(Project.class), anyString(), anyString());
     verify(projectRepository).setExecutable(any(Project.class), anyString(), anyString());
   }
+
+  @Test
+  void shouldDownload() {
+    Project project = tmpProjectWithPomXml();
+
+    assertThatCode(() -> initDomainService.download(project)).doesNotThrowAnyException();
+  }
 }
diff --git a/...st/java/tech/jhipster/lite/generator/init/infrastructure/primary/rest/InitResourceIT.java b/...st/java/tech/jhipster/lite/generator/init/infrastructure/primary/rest/InitResourceIT.java
@@ -1,7 +1,7 @@
 package tech.jhipster.lite.generator.init.infrastructure.primary.rest;
 
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
-import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
 import static tech.jhipster.lite.TestUtils.*;
 import static tech.jhipster.lite.generator.init.application.InitAssertFiles.*;
 import static tech.jhipster.lite.generator.project.domain.Constants.PACKAGE_JSON;
@@ -46,4 +46,19 @@ void shouldInit() throws Exception {
     assertFileContent(project, ".prettierrc", "tabWidth: 2");
     assertFileContent(project, PACKAGE_JSON, "chips");
   }
+
+  @Test
+  void shouldDownload() throws Exception {
+    ProjectDTO projectDTO = readFileToObject("json/chips.json", ProjectDTO.class).folder(FileUtils.tmpDirForTest());
+    Project project = ProjectDTO.toProject(projectDTO);
+
+    initApplicationService.init(project);
+
+    mockMvc
+      .perform(post("/api/projects/download").contentType(MediaType.APPLICATION_JSON).content(convertObjectToJsonBytes(projectDTO)))
+      .andExpect(status().isOk())
+      .andExpect(header().string("Content-Disposition", "attachment; filename=" + project.getBaseName().orElse("application") + ".zip"))
+      .andExpect(header().string("X-Suggested-Filename", project.getBaseName().orElse("application") + ".zip"))
+      .andExpect(content().contentType("application/octet-stream"));
+  }
 }