TheAnachronism/docspell
1
0
Fork 0
mirror of https://github.com/TheAnachronism/docspell.git synced 2025-06-22 02:18:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix url parsing with trailing slash

Browse Source 
Refs: #1545
This commit is contained in:
eikek
2022-07-07 11:31:36 +02:00
parent e9cb3d2b42
commit 2c9e012c96
5 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
modules/common/src/main/scala/docspell/common/LenientUri.scala
View File

@ -121,7 +121,7 @@ object LenientUri {
val isRoot = true
val isEmpty = false
def /(seg: String): Path =
NonEmptyPath(NonEmptyList.of(seg))
NonEmptyPath(NonEmptyList.of(seg), false)
def asString = "/"
}
case object EmptyPath extends Path {
@ -129,20 +129,22 @@ object LenientUri {
val isRoot = false
val isEmpty = true
def /(seg: String): Path =
NonEmptyPath(NonEmptyList.of(seg))
NonEmptyPath(NonEmptyList.of(seg), false)
def asString = ""
}
case class NonEmptyPath(segs: NonEmptyList[String]) extends Path {
case class NonEmptyPath(segs: NonEmptyList[String], trailingSlash: Boolean)
extends Path {
def segments = segs.toList
val isEmpty = false
val isRoot = false
private val slashSuffix = if (trailingSlash) "/" else ""
def /(seg: String): Path =
copy(segs = segs.append(seg))
def asString =
segs.head match {
case "." => segments.map(percentEncode).mkString("/")
case ".." => segments.map(percentEncode).mkString("/")
case _ => "/" + segments.map(percentEncode).mkString("/")
case "." => segments.map(percentEncode).mkString("/") + slashSuffix
case ".." => segments.map(percentEncode).mkString("/") + slashSuffix
case _ => "/" + segments.map(percentEncode).mkString("/") + slashSuffix
}
}
@ -157,14 +159,14 @@ object LenientUri {
str.trim match {
case "/" => Right(RootPath)
case "" => Right(EmptyPath)
case _ =>
case uriStr =>
Either.fromOption(
stripLeading(str, '/')
stripLeading(uriStr, '/')
.split('/')
.toList
.traverse(percentDecode)
.flatMap(NonEmptyList.fromList)
.map(NonEmptyPath.apply),
.map(NonEmptyPath(_, uriStr.endsWith("/"))),
s"Invalid path: $str"
)
}

2
modules/common/src/main/scala/docspell/common/UrlMatcher.scala
View File

@ -62,7 +62,7 @@ object UrlMatcher {
// strip path to only match prefixes
val mPath: LenientUri.Path =
NonEmptyList.fromList(url.path.segments.take(pathSegmentCount)) match {
case Some(nel) => LenientUri.NonEmptyPath(nel)
case Some(nel) => LenientUri.NonEmptyPath(nel, false)
case None => LenientUri.RootPath
}

7
modules/common/src/test/scala/docspell/common/LenientUriTest.scala
View File

@ -29,4 +29,11 @@ class LenientUriTest extends FunSuite {
)
assertEquals(LenientUri.percentDecode("a%25b%5Cc%7Cd%23e"), "a%b\\c|d#e".some)
}
test("parse with trailing slash") {
assertEquals(LenientUri.unsafe("http://a.com/").asString, "http://a.com/")
assertEquals(LenientUri.unsafe("http://a.com").asString, "http://a.com")
assertEquals(LenientUri.unsafe("http://a.com/path").asString, "http://a.com/path")
assertEquals(LenientUri.unsafe("http://a.com/path/").asString, "http://a.com/path/")
}
}