Merge pull request #356 from ANF-Studios/main

Check for zip extension before trimming
CodeMyst · Mar 30, 2021 · f38ddae · f38ddae
2 parents e6e6c35 + eb8b3fe
commit f38ddae
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/source/pastemyst/web/paste.d b/source/pastemyst/web/paste.d
@@ -80,10 +80,13 @@ public class PasteWeb
         import pastemyst.db : tryFindOneById, createZip;
         import std.path : baseName;
 
-        // remove .zip form the id
-        string id = _id[0..($-".zip".length)];
+        // remove .zip form the id if present
+        if (_id.endsWith(".zip"))
+        {
+            _id = _id[0..($-".zip".length)];
+        }
 
-        const auto pasteRes = tryFindOneById!Paste(id);
+        const auto pasteRes = tryFindOneById!Paste(_id);
 
         enforceHTTP(!pasteRes.isNull, HTTPStatus.notFound);