More robust port check for vite dev mode

categolj · Dec 6, 2024 · 167a0dd · 167a0dd
1 parent f9b2f67
commit 167a0dd
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/main/java/am/ik/blog/admin/web/AdminController.java b/src/main/java/am/ik/blog/admin/web/AdminController.java
@@ -54,11 +54,10 @@ public ResponseEntity<Void> login(@RequestParam(name = "redirect_path") String r
 		}
 		UriComponents uriComponents = uriComponentsBuilder.path(redirectPath).build();
 		String location;
-		if (referer.getPort() == VITE_DEV_PORT) {
+		int port = referer.getPort();
+		if (isViteDevPort(port)) {
 			// behind the vite proxy in the dev-mode
-			location = UriComponentsBuilder.fromUriString(uriComponents.toUriString())
-				.port(referer.getPort())
-				.toUriString();
+			location = UriComponentsBuilder.fromUriString(uriComponents.toUriString()).port(port).toUriString();
 		}
 		else {
 			location = uriComponents.toUriString();
@@ -91,4 +90,8 @@ public void deleteComment(@PathVariable Long commentId) {
 	public record CommentUpdateRequest(Comment.Status status) {
 	}
 
+	static boolean isViteDevPort(int port) {
+		return port >= 5170 && port <= 5179;
+	}
+
 }