1 files changed, 61 insertions, 2 deletions
diff --git a/tools/gui/src/Main.scala b/tools/gui/src/Main.scala
index d7a9f7d..7bb299c 100644
--- a/tools/gui/src/Main.scala
+++ b/tools/gui/src/Main.scala
@@ -1,5 +1,7 @@
 import java.io.{File, IOException}
 import java.net.MalformedURLException
+import java.nio.file.attribute.BasicFileAttributes
+import java.nio.file.{FileVisitResult, Files, Path, SimpleFileVisitor}
 
 import scala.io.Source
 import scala.util.{Failure, Success, Try}
@@ -21,10 +23,13 @@ object Main {
   def launchUi(projectDirectory: File, scalaMajorVersion: String): Unit = {
     val staticBase = new File(cbt_home / "tools" / "gui" / "resources" / "web").toURI.toURL.toExternalForm
     val server = new JettyServer(uiPort, staticBase) {
-      override def route(method: String, path: String, param: String => String) = (method, path) match {
+      override def route(method: String,
+                         path: String,
+                         param: String => String,
+                         setContentType: String => Unit) = (method, path) match {
         case ("GET", "/cwd") =>
           Success(s"""["$projectDirectory"]""")
-        case ("POST", "/project") =>
+        case ("POST", "/project/new") =>
           val name = param("name")
           val defaultPackage = param("pack")
           val dependencies = param("dependencies")
@@ -33,6 +38,14 @@ object Main {
             new ProjectBuilder(name, defaultPackage, dependencies, flags, projectDirectory, scalaMajorVersion).build()
             Success("[]")
           }
+        case ("POST", "/project/copy") =>
+          val name = param("name")
+          val source = new File(cbt_home / "examples" / name)
+          val target = new File(projectDirectory.getAbsolutePath / name)
+          handleIoException {
+            new FileCopier(source, target).copy()
+            Success("[]")
+          }
         case ("GET", "/dependency") =>
           val query = param("query")
           handleIoException(handleMavenBadResponse(searchDependency(query)))
@@ -40,6 +53,28 @@ object Main {
           val group = param("group")
           val artifact = param("artifact")
           handleIoException(handleMavenBadResponse(searchDependencyVersion(group, artifact)))
+        case ("GET", "/examples") =>
+          handleIoException {
+            val names = new File(cbt_home / "examples").listFiles().filter(_.isDirectory).sortBy(_.getName)
+                .map('"' + _.getName + '"').mkString(",")
+            Success(s"[$names]")
+          }
+        case ("GET", "/example/files") =>
+          val name = param("name")
+          handleIoException {
+            val dir = new File(cbt_home / "examples" / name)
+            if (dir.exists())
+              Success(serializeTree(dir))
+            else
+              Failure(new IllegalArgumentException(s"Incorrect example name: $name"))
+          }
+        case ("GET", "/example/file") =>
+          setContentType("text/plain")
+          val path = param("path")
+          handleIoException {
+            val file = new File(path)
+            Success(Source.fromFile(file).mkString)
+          }
         case _ =>
           Failure(new MalformedURLException(s"Incorrect path: $path"))
       }
@@ -82,4 +117,28 @@ object Main {
       Failure(new Exception(s"Bad response from $maven_host: $result"))
   }
 
+  private def serializeTree(file: File): String = {
+    val data = if (file.isDirectory)
+      s""","children":[${file.listFiles().sortBy(_.getName).map(serializeTree).mkString(",")}]"""
+    else
+      ""
+    s"""{"name":"${file.getName}","path":"${file.getAbsolutePath}"$data}"""
+  }
+
+  private class FileCopier(source: File, target: File) extends SimpleFileVisitor[Path] {
+
+    def copy() = Files.walkFileTree(source.toPath, this)
+
+    override def preVisitDirectory(dir: Path, attrs: BasicFileAttributes) = {
+      Files.createDirectories(target.toPath.resolve(source.toPath.relativize(dir)))
+      FileVisitResult.CONTINUE
+    }
+
+    override def visitFile(file: Path, attrs: BasicFileAttributes) = {
+      Files.copy(file, target.toPath.resolve(source.toPath.relativize(file)))
+      FileVisitResult.CONTINUE
+    }
+
+  }
+
 }