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

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