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(new LemmyHttp(getHttpBaseInternal(), headers));
  32
  33     const { path, url, query } = req;
  34
  35     // Get site data first
  36     // This bypasses errors, so that the client can hit the error on its own,
  37     // in order to remove the jwt on the browser. Necessary for wrong jwts
  38     let site: GetSiteResponse | undefined = undefined;
  39     let routeData: RouteData = {};
  40     let errorPageData: ErrorPageData | undefined = undefined;
  41     let try_site = await client.getSite(getSiteForm);
  42     if (try_site.state === "failed" && try_site.msg == "not_logged_in") {
  43       console.error(
  44         "Incorrect JWT token, skipping auth so frontend can remove jwt cookie"
  45       );
  46       getSiteForm.auth = undefined;
  47       auth = undefined;
  48       try_site = await client.getSite(getSiteForm);
  49     }
  50
  51     if (!auth && isAuthPath(path)) {
  52       return res.redirect("/login");
  53     }
  54
  55     if (try_site.state === "success") {
  56       site = try_site.data;
  57       initializeSite(site);
  58
  59       if (path !== "/setup" && !site.site_view.local_site.site_setup) {
  60         return res.redirect("/setup");
  61       }
  62
  63       if (site && activeRoute?.fetchInitialData) {
  64         const initialFetchReq: InitialFetchRequest = {
  65           client,
  66           auth,
  67           path,
  68           query,
  69           site,
  70         };
  71
  72         routeData = await activeRoute.fetchInitialData(initialFetchReq);
  73       }
  74     } else if (try_site.state === "failed") {
  75       errorPageData = getErrorPageData(new Error(try_site.msg), site);
  76     }
  77
  78     const error = Object.values(routeData).find(
  79       res => res.state === "failed"
  80     ) as FailedRequestState | undefined;
  81
  82     // Redirect to the 404 if there's an API error
  83     if (error) {
  84       console.error(error.msg);
  85       if (error.msg === "instance_is_private") {
  86         return res.redirect(`/signup`);
  87       } else {
  88         errorPageData = getErrorPageData(new Error(error.msg), site);
  89       }
  90     }
  91
  92     const isoData: IsoDataOptionalSite = {
  93       path,
  94       site_res: site,
  95       routeData,
  96       errorPageData,
  97     };
  98
  99     const wrapper = (
 100       <StaticRouter location={url} context={isoData}>
 101         <App />
 102       </StaticRouter>
 103     );
 104
 105     const root = renderToString(wrapper);
 106
 107     res.send(await createSsrHtml(root, isoData));
 108   } catch (err) {
 109     // If an error is caught here, the error page couldn't even be rendered
 110     console.error(err);
 111     res.statusCode = 500;
 112     return res.send(
 113       process.env.NODE_ENV === "development" ? err.message : "Server error"
 114     );
 115   }
 116 };