init

initializer-backend/src/main/java/com/schbrain/archetype/initializer/backend/config/WebConfiguration.java

@@ -16,6 +16,8 @@ public class WebConfiguration implements WebMvcConfigurer {
         registry.addMapping("/**")
                 .allowedOrigins("*")
                 .allowedMethods("*")
+                .allowedHeaders("*")
+                .exposedHeaders("*")
                 .maxAge(3600);
     }
 

initializer-backend/src/main/java/com/schbrain/archetype/initializer/backend/controller/InitializerController.java

@@ -2,11 +2,21 @@ package com.schbrain.archetype.initializer.backend.controller;
 
 import com.schbrain.archetype.initializer.backend.maven.MavenUtils;
 import com.schbrain.archetype.initializer.backend.param.ArchetypeGenerateParam;
+import com.schbrain.common.util.ServletUtils;
+import org.springframework.http.ContentDisposition;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.util.StreamUtils;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+
 /**
  * @author liaozan
  * @since 2022/3/20
@@ -15,8 +25,24 @@ import org.springframework.web.bind.annotation.RestController;
 public class InitializerController {
 
     @PostMapping("/archetype/generate")
-    public void generateArchetype(@RequestBody @Validated ArchetypeGenerateParam param) {
-        MavenUtils.generate(param.getGroup(), param.getArtifact());
+    public void generateArchetype(@RequestBody @Validated ArchetypeGenerateParam param) throws IOException {
+        File generatedProject = MavenUtils.generate(param);
+
+        FileInputStream inputStream = new FileInputStream(generatedProject);
+
+        HttpServletResponse response = ServletUtils.getResponse();
+        response.addHeader(HttpHeaders.CONTENT_DISPOSITION, contentDisposition(generatedProject));
+        response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
+
+        StreamUtils.copy(inputStream, response.getOutputStream());
+    }
+
+    private String contentDisposition(File file) {
+        return ContentDisposition
+                .attachment()
+                .filename(file.getName())
+                .build()
+                .toString();
     }
 
 }
\ No newline at end of file

initializer-backend/src/main/java/com/schbrain/archetype/initializer/backend/maven/MavenUtils.java

 package com.schbrain.archetype.initializer.backend.maven;
 
 import cn.hutool.core.io.FileUtil;
+import cn.hutool.core.util.ZipUtil;
+import com.schbrain.archetype.initializer.backend.param.ArchetypeGenerateParam;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.maven.cli.MavenCli;
 
@@ -13,18 +15,26 @@ import java.io.File;
 @Slf4j
 public class MavenUtils {
 
-    public static void generate(String groupId, String artifactId) {
+    public static File generate(ArchetypeGenerateParam param) {
         MavenCli cli = new MavenCli();
         System.getProperties().setProperty("maven.multiModuleProjectDirectory", "$M2_HOME");
-        String tmpDirPath = FileUtil.getTmpDirPath();
-        File archetype = new File(tmpDirPath, artifactId);
-        String[] args = getArgs(groupId, artifactId, archetype.getAbsolutePath());
+        File archetype = getArchetypeDirectory(param.getArtifact());
+        String[] args = getArgs(param, archetype.getAbsolutePath());
         cli.doMain(args, null, System.out, System.err);
         log.info("Generate archetype project at {}", archetype.getAbsolutePath());
+        return ZipUtil.zip(archetype);
+    }
+
+    private static File getArchetypeDirectory(String artifactId) {
+        String tmpDirPath = FileUtil.getTmpDirPath();
+        File archetype = new File(tmpDirPath, artifactId);
+        // clean cache
+        FileUtil.del(archetype);
+        return archetype;
     }
 
     @SuppressWarnings("SpellCheckingInspection")
-    private static String[] getArgs(String groupId, String artifactId, String outputDirectory) {
+    private static String[] getArgs(ArchetypeGenerateParam param, String outputDirectory) {
         return new String[]{
                 "-B",
                 "archetype:generate",
@@ -32,8 +42,10 @@ public class MavenUtils {
                 "-DarchetypeArtifactId=schbrain-archetype",
                 "-DarchetypeVersion=1.0-SNAPSHOT",
                 String.format("-DoutputDirectory=%s", outputDirectory),
-                String.format("-DgroupId=%s", groupId),
-                String.format("-DartifactId=%S", artifactId)
+                String.format("-DgroupId=%s", param.getGroup()),
+                String.format("-DartifactId=%s", param.getArtifact()),
+                String.format("-DpackageName=%s", param.getPackageName()),
+                String.format("-DprojectName=%s", param.getProjectName()),
         };
     }
 

initializer-ui/src/components/BackendStarter.vue

@@ -38,11 +38,11 @@ export default {
   data() {
     return {
       generateForm: {
-        projectName: 'schbrain-alarm',
-        gitlabUrl: 'http://gitlab.schbrain.com/gitlab/tools/schbrain-archetype.git',
-        group: 'com.schbrain.kp',
-        artifact: 'schbrain-alarm',
-        packageName: 'com.schbrain.kp.alarm'
+        projectName: '',
+        gitlabUrl: '',
+        group: '',
+        artifact: '',
+        packageName: ''
       }
     }
   },
@@ -51,16 +51,15 @@ export default {
       axios.post('/archetype/generate', this.generateForm, {
         responseType: 'blob'
       }).then(res => {
-        let fileName = window.decodeURI(res.headers['Content-Disposition'].split('=')[1])
-        let blob = new Blob([res.data])
-
-        let link = document.createElement('a');
+        const {headers, data} = res
+        const fileName = headers['content-disposition'].replace(/\w+;\sfilename="(.*)"/, '$1')
+        const blob = new Blob([data])
+        const link = document.createElement('a');
         link.style.display = "none";
         link.href = window.URL.createObjectURL(blob);
         link.download = fileName;
         document.body.appendChild(link);
         link.click();
-        //释放内存
         window.URL.revokeObjectURL(link.href);
         document.body.removeChild(link);
       })