feat: Redirection feature using .location method on response object

Author: arpan404

src/core/response.ts

@@ -2,12 +2,26 @@ import { ServerResponse } from "node:http";
 import Jrror, { JoorError } from "@/core/error";
 import { HeaderSent } from "@/core/error/response";
 import logger from '@/helpers/joorLogger';
+import Response, {RESPONSE_LOCATION_STATUS} from "@/types/response";
 
-const response = ServerResponse.prototype;
-response.status = function (this: ServerResponse, status: number) {
+const response = ServerResponse.prototype as Response;
+
+/**
+ * Sets the HTTP status code for the response.
+ * 
+ * @param {number} status - The HTTP status code to set.
+ * 
+ * @returns {Response} The response object for chaining.
+ * 
+ * @example
+ * ```typescript
+ * response.status(200);
+ * ```
+*/
+response.status = function (this: ServerResponse, status: number): Response {
     try {
         if (this.headersSent) {
-            throw HeaderSent;
+            throw HeaderSent; // Headers have already been sent
         }
         if (!Number.isInteger(status)) {
             throw new Jrror({
@@ -40,4 +54,35 @@ response.status = function (this: ServerResponse, status: number) {
         return this;
     }
 
+}
+
+/**
+Sets a location header for the response. Used for redirection.
+
+@param {string} location - The URL to redirect to.
+@param {RESPONSE_LOCATION_STATUS} [status=301] - The HTTP status code for the redirection (default is 301).
+
+example
+```typescript
+response.location('https://example.com', 302);
+```
+For more information, see the [HTTP/1.1 RFC](https://datatracker.ietf.org/doc/html/rfc7231#section-6.4).
+*/
+response.location = function (this: ServerResponse, location: string, status: RESPONSE_LOCATION_STATUS = 301): void {
+    try {
+        if (this.headersSent) {
+            throw HeaderSent; // Headers have already been sent
+        }
+        this.setHeader("Location", location);
+        this.status(status);
+    }
+    catch (error: unknown) {
+
+        if (error instanceof Jrror || error instanceof JoorError) {
+            error.handle();
+        }
+        else {
+            logger.error(error);
+        }
+    }
 }

src/packages/vary/index.ts

@@ -1,7 +1,7 @@
 import Jrror from '@/core/error';
 import Response from '@/types/response';
 
-const FIEL_NAME_REGEX = /^[!#$%&'*+\-.^_`|~0-9A-Za-z]+$/;
+const FIELD_NAME_REGEX = /^[!#$%&'*+\-.^_`|~0-9A-Za-z]+$/;
 
 const parse = (field: string) => {
   if (!field) return [];
@@ -50,7 +50,7 @@ const append = (header: string, field: unknown): string => {
   const fields = Array.isArray(field) ? field : parse(String(field));
 
   for (let i = 0; i < fields.length; i++) {
-    if (!FIEL_NAME_REGEX.test(fields[i])) {
+    if (!FIELD_NAME_REGEX.test(fields[i])) {
       throw new Jrror({
         code: 'vary-field-invalid',
         message: `Field name "${fields[i]}" contains invalid characters.`,

src/types/response.ts

@@ -7,6 +7,8 @@ type RESPONSE_MESSAGE = string;
 
 type RESPONSE_DATA = unknown;
 
+type RESPONSE_LOCATION_STATUS = 301 | 302 | 303 | 307 | 308;
+
 // Interface for response cookies
 interface RESPONSE_COOKIES {
   [key: string]: {
@@ -61,4 +63,5 @@ export {
   RESPONSE_DATA,
   RESPONSE_COOKIES,
   RESPONSE_HEADERS,
+  RESPONSE_LOCATION_STATUS,
 };