From 50f8f8f4b76fb6303df743a833494b5e73aa7a8a Mon Sep 17 00:00:00 2001
From: Yuuki Wesp <ls-micro@ya.ru>
Date: Sat, 31 Aug 2024 19:52:41 +0300
Subject: [PATCH] update registry

---
 src/controllers/PackageContentController.cs | 10 ++++------
 src/controllers/SearchController.cs         |  2 +-
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/src/controllers/PackageContentController.cs b/src/controllers/PackageContentController.cs
index 3b8a2d7..ffd136b 100644
--- a/src/controllers/PackageContentController.cs
+++ b/src/controllers/PackageContentController.cs
@@ -32,12 +32,10 @@ public async Task<ActionResult<PackageVersionsResponse>> GetPackageVersionsAsync
     [HttpGet("@/packages/{id}/{version}")]
     public async Task<IActionResult> DownloadPackageAsync(string id, string version, CancellationToken cancellationToken)
     {
-        var ver = version switch
-        {
-            "latest" or null => new (0, 0, 0, 0, "", "latest"),
-            "next"           => new (0, 0, 0, 0, "", "next"),
-            not null         => NuGetVersion.Parse(version)
-        };
+        if (version is ("latest" or "next"))
+            return BadRequest(new { message = "not allowed using latest or next version for this endpoint" });
+        if (NuGetVersion.TryParse(version, out var ver))
+            return BadRequest(new { message = "invalid version" });
 
         var packageStream = await content.GetPackageContentStreamOrNullAsync(id, ver, cancellationToken);
         if (packageStream == null)
diff --git a/src/controllers/SearchController.cs b/src/controllers/SearchController.cs
index a7a76d0..4919a68 100644
--- a/src/controllers/SearchController.cs
+++ b/src/controllers/SearchController.cs
@@ -42,7 +42,7 @@ public async Task<ActionResult<Package>> FindByName(string name, string version,
             ? urlGenerator.GetPackageIconDownloadUrl(result.Name, result.Version)
             : result.Icon;
 
-        if (!version!.Equals(Package.LatestTag) && !version.Equals(Package.NextTag))
+        if (version!.Equals(Package.LatestTag) || version.Equals(Package.NextTag))
             return Json(result);
 
         cache.Set((name, version), result, TimeSpan.FromDays(2));