0.1.5

com-lihaoyi · Aug 25, 2018 · ece16bb · ece16bb
1 parent 8304925
commit ece16bb
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/cask/src/cask/endpoints/StaticEndpoints.scala b/cask/src/cask/endpoints/StaticEndpoints.scala
@@ -10,7 +10,9 @@ class staticFiles(val path: String) extends Endpoint{
   type InputParser[T] = QueryParamReader[T]
   override def subpath = true
   def wrapFunction(ctx: Request, delegate: Delegate): Returned = {
-    delegate(Map()).map(t => cask.model.StaticFile(t + "/" + ctx.remainingPathSegments.mkString("/")))
+    delegate(Map()).map(t =>
+      cask.model.StaticFile(t + "/" + ctx.remainingPathSegments.filter(s => s != "." && s != "..").mkString("/"))
+    )
   }
 
   def wrapPathSegment(s: String): Input = Seq(s)
@@ -24,7 +26,7 @@ class staticResources(val path: String, resourceRoot: ClassLoader = getClass.get
   override def subpath = true
   def wrapFunction(ctx: Request, delegate: Delegate): Returned = {
     delegate(Map()).map(t =>
-      cask.model.StaticResource(t + "/" + ctx.remainingPathSegments.mkString("/"), resourceRoot)
+      cask.model.StaticResource(t + "/" + ctx.remainingPathSegments.filter(s => s != "." && s != "..").mkString("/"), resourceRoot)
     )
   }
 

diff --git a/example/staticFiles/app/src/StaticFiles.scala b/example/staticFiles/app/src/StaticFiles.scala
@@ -11,5 +11,8 @@ object StaticFiles extends cask.MainRoutes{
   @cask.staticResources("/static/resource")
   def staticResourceRoutes() = "cask"
 
+  @cask.staticResources("/static/resource2")
+  def staticResourceRoutes2() = "."
+
   initialize()
 }
diff --git a/example/staticFiles/app/test/src/ExampleTests.scala b/example/staticFiles/app/test/src/ExampleTests.scala
@@ -24,6 +24,11 @@ object ExampleTests extends TestSuite{
 
       requests.get(s"$host/static/resource/example.txt").text() ==>
         "the quick brown fox jumps over the lazy dog"
+
+      requests.get(s"$host/static/resource2/cask/example.txt").text() ==>
+        "the quick brown fox jumps over the lazy dog"
+
+      requests.get(s"$host/static/file/../../../build.sc").statusCode ==> 404
     }
 
   }