mirror of
https://gitee.com/dcren/initializr.git
synced 2025-05-04 04:38:02 +08:00
175 lines
4.7 KiB
Groovy
175 lines
4.7 KiB
Groovy
package app
|
|
|
|
@Grab("org.springframework.boot:spring-boot-starter-actuator:0.5.0.BUILD-SNAPSHOT")
|
|
@Grab("org.codehaus.groovy:groovy-ant:2.1.6")
|
|
|
|
@Controller
|
|
@Log
|
|
class MainController {
|
|
|
|
@Value('${info.home:http://localhost:8080/}')
|
|
private String home
|
|
|
|
@Value('${TMPDIR:.}')
|
|
private String tmpdir
|
|
|
|
@Autowired
|
|
private Reactor reactor
|
|
|
|
@RequestMapping("/")
|
|
@ResponseBody
|
|
String home() {
|
|
def model = [:]
|
|
model["styles"] = [[name:"Standard", value:""]]
|
|
model["styles"] << [name:"Web", value:"web"]
|
|
model["styles"] << [name:"Actuator", value:"actuator"]
|
|
model["styles"] << [name:"Batch", value:"batch"]
|
|
model["styles"] << [name:"JPA", value:"jpa"]
|
|
model["types"] = [[name:"Maven POM", value:"pom", selected: true], [name:"Maven Project", value:"pomproject", selected: false]]
|
|
template "home.html", model
|
|
}
|
|
|
|
@RequestMapping("/spring")
|
|
@ResponseBody
|
|
ResponseEntity<byte[]> spring() {
|
|
File download = new File(tmpdir, "spring.zip")
|
|
if (!download.exists()) {
|
|
log.info("Creating: " + download)
|
|
new AntBuilder().zip(destfile: download) {
|
|
zipfileset(dir:".", includes:"spring/bin/**", filemode:"775")
|
|
zipfileset(dir:".", includes:"spring/**", excludes:"spring/bin/**")
|
|
}
|
|
}
|
|
log.info("Downloading: " + download)
|
|
new ResponseEntity<byte[]>(download.bytes, ["Content-Type":"application/zip"] as HttpHeaders, HttpStatus.OK)
|
|
}
|
|
|
|
@RequestMapping("/starter")
|
|
@ResponseBody
|
|
ResponseEntity<byte[]> spring(PomRequest request) {
|
|
|
|
def tempFiles = []
|
|
|
|
def model = [:]
|
|
String pom = new String(pom(request, model).body)
|
|
File dir = File.createTempFile("tmp","",new File(tmpdir));
|
|
tempFiles << dir
|
|
dir.delete()
|
|
dir.mkdirs()
|
|
new File(dir, "pom.xml").write(pom)
|
|
|
|
File src = new File(new File(dir, "src/main/java"),request.packageName.replace(".", "/"))
|
|
src.mkdirs()
|
|
|
|
def body = template "Application.java", model
|
|
log.info("Creating: " + src + "/Application.java")
|
|
new File(src, "Application.java").write(body)
|
|
|
|
File download = new File(tmpdir, dir.name + ".zip")
|
|
log.info("Creating: " + download)
|
|
tempFiles << download
|
|
|
|
new AntBuilder().zip(destfile: download) {
|
|
zipfileset(dir:dir, includes:"**")
|
|
}
|
|
log.info("Downloading: " + download)
|
|
def result = new ResponseEntity<byte[]>(download.bytes, ["Content-Type":"application/zip"] as HttpHeaders, HttpStatus.OK)
|
|
|
|
reactor.notify("tempfiles", Event.wrap(tempFiles))
|
|
|
|
result
|
|
}
|
|
|
|
@RequestMapping("/pom")
|
|
@ResponseBody
|
|
ResponseEntity<byte[]> pom(PomRequest request, Map model) {
|
|
|
|
def style = request.style
|
|
log.info("Styles requested: " + style)
|
|
|
|
model.groupId = request.groupId
|
|
model.artifactId = request.artifactId
|
|
model.version = request.version
|
|
model.name = request.name
|
|
model.description = request.description
|
|
model.packageName = request.packageName
|
|
|
|
if (style==null || style.size()==0) {
|
|
style = [""]
|
|
}
|
|
if (!style.class.isArray() && !(style instanceof Collection)) {
|
|
style = [style]
|
|
}
|
|
model["styles"] = style.collect{ it=="" ? "" : "-" + it }
|
|
|
|
log.info("Model: " + model)
|
|
|
|
def body = template "starter-pom.xml", model
|
|
new ResponseEntity<byte[]>(body, ["Content-Type":"application/octet-stream"] as HttpHeaders, HttpStatus.OK)
|
|
}
|
|
|
|
}
|
|
|
|
import reactor.spring.context.ConsumerBeanPostProcessor;
|
|
@Configuration
|
|
@EnableReactor
|
|
class ReactorConfiguration {
|
|
|
|
@Bean
|
|
public reactor.core.Environment reactorEnvironment() {
|
|
return new reactor.core.Environment(); // TODO: use Spring Environment to configure?
|
|
}
|
|
|
|
@Bean
|
|
public Reactor rootReactor() {
|
|
return reactorEnvironment().getRootReactor();
|
|
}
|
|
|
|
}
|
|
|
|
@Component
|
|
@Log
|
|
class TemporaryFileCleaner {
|
|
|
|
@Autowired
|
|
Reactor reactor
|
|
|
|
@PostConstruct
|
|
void init() {
|
|
reactor.on(Selectors.$("tempfiles"), [
|
|
accept: { event ->
|
|
def tempFiles = event.data
|
|
log.info "Tempfiles: " + tempFiles
|
|
if (tempFiles) {
|
|
tempFiles.each {
|
|
File file = it as File
|
|
if (file.directory) {
|
|
file.deleteDir()
|
|
} else {
|
|
file.delete()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
] as Consumer)
|
|
}
|
|
|
|
}
|
|
|
|
class PomRequest {
|
|
def style = []
|
|
|
|
String name = "demo"
|
|
String description = "Demo project for Spring Boot"
|
|
String groupId = "org.test"
|
|
String artifactId
|
|
String version = "0.0.1.SNAPSHOT"
|
|
String packageName
|
|
String getArtifactId() {
|
|
artifactId == null ? name : artifactId
|
|
}
|
|
String getPackageName() {
|
|
packageName == null ? name.replace('-', '.') : packageName
|
|
}
|
|
}
|