src/server/handlers/catch-all-handler.tsx

   1 import { initializeSite, isAuthPath } from "@utils/app";
   2 import { ErrorPageData } from "@utils/types";
   3 import fetch from "cross-fetch";
   4 import type { Request, Response } from "express";
   5 import { StaticRouter, matchPath } from "inferno-router";
   6 import { renderToString } from "inferno-server";
   7 import IsomorphicCookie from "isomorphic-cookie";
   8 import { GetSite, GetSiteResponse, LemmyHttp } from "lemmy-js-client";
   9 import { App } from "../../shared/components/app/app";
  10 import { getHttpBaseInternal } from "../../shared/env";
  11 import {
  12   InitialFetchRequest,
  13   IsoDataOptionalSite,
  14   RouteData,
  15 } from "../../shared/interfaces";
  16 import { routes } from "../../shared/routes";
  17 import {
  18   FailedRequestState,
  19   wrapClient,
  20 } from "../../shared/services/HttpService";
  21 import { createSsrHtml } from "../utils/create-ssr-html";
  22 import { getErrorPageData } from "../utils/get-error-page-data";
  23 import { setForwardedHeaders } from "../utils/set-forwarded-headers";
  24
  25 export default async (req: Request, res: Response) => {
  26   try {
  27     const activeRoute = routes.find(route => matchPath(req.path, route));
  28     let auth: string | undefined = IsomorphicCookie.load("jwt", req);
  29
  30     const getSiteForm: GetSite = { auth };
  31
  32     const headers = setForwardedHeaders(req.headers);
  33     const client = wrapClient(
  34       new LemmyHttp(getHttpBaseInternal(), { fetchFunction: fetch, headers })
  35     );
  36
  37     const { path, url, query } = req;
  38
  39     // Get site data first
  40     // This bypasses errors, so that the client can hit the error on its own,
  41     // in order to remove the jwt on the browser. Necessary for wrong jwts
  42     let site: GetSiteResponse | undefined = undefined;
  43     let routeData: RouteData = {};
  44     let errorPageData: ErrorPageData | undefined = undefined;
  45     let try_site = await client.getSite(getSiteForm);
  46     if (try_site.state === "failed" && try_site.msg == "not_logged_in") {
  47       console.error(
  48         "Incorrect JWT token, skipping auth so frontend can remove jwt cookie"
  49       );
  50       getSiteForm.auth = undefined;
  51       auth = undefined;
  52       try_site = await client.getSite(getSiteForm);
  53     }
  54
  55     if (!auth && isAuthPath(path)) {
  56       return res.redirect("/login");
  57     }
  58
  59     if (try_site.state === "success") {
  60       site = try_site.data;
  61       initializeSite(site);
  62
  63       if (path !== "/setup" && !site.site_view.local_site.site_setup) {
  64         return res.redirect("/setup");
  65       }
  66
  67       if (site && activeRoute?.fetchInitialData) {
  68         const initialFetchReq: InitialFetchRequest = {
  69           client,
  70           auth,
  71           path,
  72           query,
  73           site,
  74         };
  75
  76         routeData = await activeRoute.fetchInitialData(initialFetchReq);
  77       }
  78     } else if (try_site.state === "failed") {
  79       errorPageData = getErrorPageData(new Error(try_site.msg), site);
  80     }
  81
  82     const error = Object.values(routeData).find(
  83       res => res.state === "failed"
  84     ) as FailedRequestState | undefined;
  85
  86     // Redirect to the 404 if there's an API error
  87     if (error) {
  88       console.error(error.msg);
  89       if (error.msg === "instance_is_private") {
  90         return res.redirect(`/signup`);
  91       } else {
  92         errorPageData = getErrorPageData(new Error(error.msg), site);
  93       }
  94     }
  95
  96     const isoData: IsoDataOptionalSite = {
  97       path,
  98       site_res: site,
  99       routeData,
 100       errorPageData,
 101     };
 102
 103     const wrapper = (
 104       <StaticRouter location={url} context={isoData}>
 105         <App />
 106       </StaticRouter>
 107     );
 108
 109     const root = renderToString(wrapper);
 110
 111     res.send(await createSsrHtml(root, isoData));
 112   } catch (err) {
 113     // If an error is caught here, the error page couldn't even be rendered
 114     console.error(err);
 115     res.statusCode = 500;
 116     return res.send(
 117       process.env.NODE_ENV === "development" ? err.message : "Server error"
 118     );
 119   }
 120 };