From 3a162be265af6a983115e30aa191019d8af327d3 Mon Sep 17 00:00:00 2001 From: Taloth Saldono Date: Mon, 4 Apr 2016 21:28:59 +0200 Subject: [PATCH] CombinePath now simple, uri resolve done via operator and CombineRelativePath. --- .../Http/HttpUriFixture.cs | 71 ++++++++++++++++--- src/NzbDrone.Common/Http/HttpUri.cs | 31 ++++++-- 2 files changed, 90 insertions(+), 12 deletions(-) diff --git a/src/NzbDrone.Common.Test/Http/HttpUriFixture.cs b/src/NzbDrone.Common.Test/Http/HttpUriFixture.cs index 09cda2be7..099ab990f 100644 --- a/src/NzbDrone.Common.Test/Http/HttpUriFixture.cs +++ b/src/NzbDrone.Common.Test/Http/HttpUriFixture.cs @@ -7,23 +7,78 @@ namespace NzbDrone.Common.Test.Http { public class HttpUriFixture : TestBase { - private HttpUri GivenHttpUri(string basePath) + [TestCase("", "", "")] + [TestCase("/", "", "/")] + [TestCase("base", "", "base")] + [TestCase("/base", "", "/base")] + [TestCase("/base/", "", "/base/")] + [TestCase("", "relative", "relative")] + [TestCase("", "/relative", "/relative")] + [TestCase("/", "relative", "/relative")] + [TestCase("/", "/relative", "/relative")] + [TestCase("base", "relative", "relative")] + [TestCase("base", "/relative", "/relative")] + [TestCase("/base", "relative", "/relative")] + [TestCase("/base", "/relative", "/relative")] + [TestCase("/base/", "relative", "/base/relative")] + [TestCase("/base/", "/relative", "/relative")] + [TestCase("base/sub", "relative", "base/relative")] + [TestCase("base/sub", "/relative", "/relative")] + [TestCase("/base/sub", "relative", "/base/relative")] + [TestCase("/base/sub", "/relative", "/relative")] + [TestCase("/base/sub/", "relative", "/base/sub/relative")] + [TestCase("/base/sub/", "/relative", "/relative")] + [TestCase("abc://host.com:8080/root/file.xml", "relative/path", "abc://host.com:8080/root/relative/path")] + [TestCase("abc://host.com:8080/root/file.xml", "/relative/path", "abc://host.com:8080/relative/path")] + [TestCase("abc://host.com:8080/root/file.xml?query=1#fragment", "relative/path", "abc://host.com:8080/root/relative/path")] + [TestCase("abc://host.com:8080/root/file.xml?query=1#fragment", "/relative/path", "abc://host.com:8080/relative/path")] + [TestCase("abc://host.com:8080/root/api", "relative/path", "abc://host.com:8080/root/relative/path")] + [TestCase("abc://host.com:8080/root/api", "/relative/path", "abc://host.com:8080/relative/path")] + [TestCase("abc://host.com:8080/root/api/", "relative/path", "abc://host.com:8080/root/api/relative/path")] + [TestCase("abc://host.com:8080/root/api/", "/relative/path", "abc://host.com:8080/relative/path")] + [TestCase("abc://host.com:8080/root/api/", "//otherhost.com/path", "abc://otherhost.com/path")] + public void should_combine_uri(string basePath, string relativePath, string expected) { - return new HttpUri("http", "localhost", 8989, basePath, null, null); + var newUri = new HttpUri(basePath) + new HttpUri(relativePath); + newUri.FullUri.Should().Be(expected); } [TestCase("", "", "")] - [TestCase("base", "", "/base")] + [TestCase("/", "", "/")] + [TestCase("base", "", "base")] [TestCase("/base", "", "/base")] - [TestCase("", "relative", "/relative")] + [TestCase("/base/", "", "/base/")] + [TestCase("", "relative", "relative")] [TestCase("", "/relative", "/relative")] - [TestCase("base", "relative", "/base/relative")] - [TestCase("base", "/relative", "/base/relative")] + [TestCase("/", "relative", "/relative")] + [TestCase("/", "/relative", "/relative")] + [TestCase("base", "relative", "base/relative")] + [TestCase("base", "/relative", "base/relative")] [TestCase("/base", "relative", "/base/relative")] [TestCase("/base", "/relative", "/base/relative")] - public void should_combine_base_path_and_relative_path(string basePath, string relativePath, string expected) + [TestCase("/base/", "relative", "/base/relative")] + [TestCase("/base/", "/relative", "/base/relative")] + [TestCase("base/sub", "relative", "base/sub/relative")] + [TestCase("base/sub", "/relative", "base/sub/relative")] + [TestCase("/base/sub", "relative", "/base/sub/relative")] + [TestCase("/base/sub", "/relative", "/base/sub/relative")] + [TestCase("/base/sub/", "relative", "/base/sub/relative")] + [TestCase("/base/sub/", "/relative", "/base/sub/relative")] + [TestCase("/base/sub/", "relative/", "/base/sub/relative/")] + [TestCase("/base/sub/", "/relative/", "/base/sub/relative/")] + [TestCase("abc://host.com:8080/root/file.xml", "relative/path", "abc://host.com:8080/root/file.xml/relative/path")] + [TestCase("abc://host.com:8080/root/file.xml", "/relative/path", "abc://host.com:8080/root/file.xml/relative/path")] + [TestCase("abc://host.com:8080/root/file.xml?query=1#fragment", "relative/path", "abc://host.com:8080/root/file.xml/relative/path?query=1#fragment")] + [TestCase("abc://host.com:8080/root/file.xml?query=1#fragment", "/relative/path", "abc://host.com:8080/root/file.xml/relative/path?query=1#fragment")] + [TestCase("abc://host.com:8080/root/api", "relative/path", "abc://host.com:8080/root/api/relative/path")] + [TestCase("abc://host.com:8080/root/api", "/relative/path", "abc://host.com:8080/root/api/relative/path")] + [TestCase("abc://host.com:8080/root/api/", "relative/path", "abc://host.com:8080/root/api/relative/path")] + [TestCase("abc://host.com:8080/root/api/", "/relative/path", "abc://host.com:8080/root/api/relative/path")] + public void should_combine_relative_path(string basePath, string relativePath, string expected) { - GivenHttpUri(basePath).CombinePath(relativePath).Path.Should().Be(expected); + var newUri = new HttpUri(basePath).CombinePath(relativePath); + + newUri.FullUri.Should().Be(expected); } } } diff --git a/src/NzbDrone.Common/Http/HttpUri.cs b/src/NzbDrone.Common/Http/HttpUri.cs index 05d12e7d8..a52a06137 100644 --- a/src/NzbDrone.Common/Http/HttpUri.cs +++ b/src/NzbDrone.Common/Http/HttpUri.cs @@ -131,7 +131,7 @@ namespace NzbDrone.Common.Http return _queryParams; } } - + public HttpUri CombinePath(string path) { return new HttpUri(Scheme, Host, Port, CombinePath(Path, path), Query, Fragment); @@ -146,10 +146,32 @@ namespace NzbDrone.Common.Http if (basePath.IsNullOrWhiteSpace()) { - return "/" + relativePath.TrimStart('/'); + return relativePath; } - return basePath.TrimEnd("/") + "/" + relativePath.TrimStart('/'); + return basePath.TrimEnd('/') + "/" + relativePath.TrimStart('/'); + } + + private static string CombineRelativePath(string basePath, string relativePath) + { + if (relativePath.IsNullOrWhiteSpace()) + { + return basePath; + } + + if (relativePath.StartsWith("/")) + { + return relativePath; + } + + var baseSlashIndex = basePath.LastIndexOf('/'); + + if (baseSlashIndex >= 0) + { + return basePath.Substring(0, baseSlashIndex) + "/" + relativePath; + } + + return relativePath; } public HttpUri SetQuery(string query) @@ -188,6 +210,7 @@ namespace NzbDrone.Common.Http return SetQuery(builder.ToString()); } + public override int GetHashCode() { return _uri.GetHashCode(); @@ -240,7 +263,7 @@ namespace NzbDrone.Common.Http if (relativeUrl.Path.IsNotNullOrWhiteSpace()) { - return new HttpUri(baseUrl.Scheme, baseUrl.Host, baseUrl.Port, HttpUri.CombinePath(baseUrl.Path, relativeUrl.Path), relativeUrl.Query, relativeUrl.Fragment); + return new HttpUri(baseUrl.Scheme, baseUrl.Host, baseUrl.Port, CombineRelativePath(baseUrl.Path, relativeUrl.Path), relativeUrl.Query, relativeUrl.Fragment); } return new HttpUri(baseUrl.Scheme, baseUrl.Host, baseUrl.Port, baseUrl.Path, relativeUrl.Query, relativeUrl.Fragment);