Add url to server response

src/functions/server/server-params-gen.ts

@@ -7,6 +7,7 @@ import handleRoutes from "./handle-routes";
 import isDevelopment from "../../utils/is-development";
 import grabConstants from "../../utils/grab-constants";
 import { AppData } from "../../data/app-data";
+import { existsSync } from "fs";
 
 type Params = {
     dev?: boolean;
@@ -91,28 +92,52 @@ export default async function (params?: Params): Promise<ServeOptions> {
                 }
 
                 if (url.pathname.startsWith("/public/")) {
-                    const file = Bun.file(
-                        path.join(
+                    try {
+                        const file_path = path.join(
                             PUBLIC_DIR,
                             url.pathname.replace(/^\/public/, ""),
-                        ),
-                    );
+                        );
 
-                    let res_opts: ResponseInit = {};
+                        if (!existsSync(file_path)) {
+                            return new Response(`Public File Doesn't Exist`, {
+                                status: 404,
+                            });
+                        }
 
-                    if (!is_dev && url.pathname.match(/__bunext/)) {
-                        res_opts.headers = {
-                            "Cache-Control": `public, max-age=${AppData["BunextStaticFilesCacheExpiry"]}, must-revalidate`,
-                        };
+                        const file = Bun.file(file_path);
+
+                        let res_opts: ResponseInit = {};
+
+                        if (!is_dev && url.pathname.match(/__bunext/)) {
+                            res_opts.headers = {
+                                "Cache-Control": `public, max-age=${AppData["BunextStaticFilesCacheExpiry"]}, must-revalidate`,
+                            };
+                        }
+
+                        return new Response(file, res_opts);
+                    } catch (error) {
+                        return new Response(`Public File Not Found`, {
+                            status: 404,
+                        });
                     }
-
-                    return new Response(file, res_opts);
                 }
 
-                if (url.pathname.startsWith("/favicon.")) {
-                    const file = Bun.file(path.join(PUBLIC_DIR, url.pathname));
+                // if (url.pathname.startsWith("/favicon.") ) {
+                if (url.pathname.match(/\..*$/)) {
+                    try {
+                        const file_path = path.join(PUBLIC_DIR, url.pathname);
 
-                    return new Response(file);
+                        if (!existsSync(file_path)) {
+                            return new Response(`File Doesn't Exist`, {
+                                status: 404,
+                            });
+                        }
+
+                        const file = Bun.file(file_path);
+                        return new Response(file);
+                    } catch (error) {
+                        return new Response(`File Not Found`, { status: 404 });
+                    }
                 }
 
                 return await handleWebPages({ req });

src/functions/server/web-pages/grab-page-component.tsx

@@ -12,6 +12,7 @@ import AppNames from "../../../utils/grab-app-names";
 import { existsSync } from "fs";
 import grabPageErrorComponent from "./grab-page-error-component";
 import grabPageBundledReactComponent from "./grab-page-bundled-react-component";
+import _ from "lodash";
 
 class NotFoundError extends Error {}
 
@@ -84,20 +85,40 @@ export default async function grabPageComponent({
         const module: BunextPageModule = await import(file_path);
 
         const serverRes: BunextPageModuleServerReturn = await (async () => {
+            const default_props: BunextPageModuleServerReturn = {
+                url: {
+                    ...(_.pick<URL, keyof URL>(url!, [
+                        "host",
+                        "hostname",
+                        "pathname",
+                        "origin",
+                        "port",
+                        "search",
+                        "searchParams",
+                        "hash",
+                        "href",
+                        "password",
+                        "protocol",
+                        "username",
+                    ]) as any),
+                },
+                query: match?.query,
+            };
+
             try {
                 if (routeParams) {
                     const serverData = await module["server"]?.(routeParams);
                     return {
                         ...serverData,
-                        query: match?.query,
+                        ...default_props,
                     };
                 }
                 return {
-                    query: match?.query,
+                    ...default_props,
                 };
             } catch (error) {
                 return {
-                    query: match?.query,
+                    ...default_props,
                 };
             }
         })();

src/types/index.ts

@@ -220,8 +220,11 @@ export type BunextPageModuleServerReturn<
      * Expiry time of the cache in seconds
      */
     cacheExpiry?: number;
+    url: BunextPageModuleServerReturnURLObject;
 };
 
+export type BunextPageModuleServerReturnURLObject = URL & {};
+
 export type BunextPageModuleServerRedirect = {
     destination: string;
     permanent?: boolean;