1 import express from "express";
2 import { existsSync } from "fs";
3 import { readdir, readFile } from "fs/promises";
4 import { IncomingHttpHeaders } from "http";
5 import { Helmet } from "inferno-helmet";
6 import { matchPath, StaticRouter } from "inferno-router";
7 import { renderToString } from "inferno-server";
8 import IsomorphicCookie from "isomorphic-cookie";
9 import { GetSite, GetSiteResponse, LemmyHttp, Site } from "lemmy-js-client";
10 import path from "path";
11 import process from "process";
12 import serialize from "serialize-javascript";
13 import sharp from "sharp";
14 import { App } from "../shared/components/app/app";
15 import { getHttpBase, getHttpBaseInternal } from "../shared/env";
20 } from "../shared/interfaces";
21 import { routes } from "../shared/routes";
28 } from "../shared/utils";
30 const server = express();
31 const [hostname, port] = process.env["LEMMY_UI_HOST"]
32 ? process.env["LEMMY_UI_HOST"].split(":")
33 : ["0.0.0.0", "1234"];
34 const extraThemesFolder =
35 process.env["LEMMY_UI_EXTRA_THEMES_FOLDER"] || "./extra_themes";
37 if (!process.env["LEMMY_UI_DISABLE_CSP"] && !process.env["LEMMY_UI_DEBUG"]) {
38 server.use(function (_req, res, next) {
40 "Content-Security-Policy",
41 `default-src 'self'; manifest-src *; connect-src *; img-src * data:; script-src 'self' 'unsafe-inline' 'unsafe-eval'; style-src 'self' 'unsafe-inline'; form-action 'self'; base-uri 'self'; frame-src *`
46 const customHtmlHeader = process.env["LEMMY_UI_CUSTOM_HTML_HEADER"] || "";
48 server.use(express.json());
49 server.use(express.urlencoded({ extended: false }));
50 server.use("/static", express.static(path.resolve("./dist")));
52 const robotstxt = `User-Agent: *
55 Disallow: /create_community
56 Disallow: /create_post
57 Disallow: /create_private_message
61 Disallow: /password_change
65 server.get("/service-worker.js", async (_req, res) => {
66 res.setHeader("Content-Type", "application/javascript");
67 res.sendFile(path.resolve("./dist/service-worker.js"));
70 server.get("/robots.txt", async (_req, res) => {
71 res.setHeader("content-type", "text/plain; charset=utf-8");
75 server.get("/css/themes/:name", async (req, res) => {
76 res.contentType("text/css");
77 const theme = req.params.name;
78 if (!theme.endsWith(".css")) {
80 res.send("Theme must be a css file");
83 const customTheme = path.resolve(`./${extraThemesFolder}/${theme}`);
84 if (existsSync(customTheme)) {
85 res.sendFile(customTheme);
87 const internalTheme = path.resolve(`./dist/assets/css/themes/${theme}`);
89 // If the theme doesn't exist, just send litely
90 if (existsSync(internalTheme)) {
91 res.sendFile(internalTheme);
93 res.sendFile(path.resolve("./dist/assets/css/themes/litely.css"));
98 async function buildThemeList(): Promise<string[]> {
99 const themes = ["darkly", "darkly-red", "litely", "litely-red"];
100 if (existsSync(extraThemesFolder)) {
101 const dirThemes = await readdir(extraThemesFolder);
102 const cssThemes = dirThemes
103 .filter(d => d.endsWith(".css"))
104 .map(d => d.replace(".css", ""));
105 themes.push(...cssThemes);
110 server.get("/css/themelist", async (_req, res) => {
112 res.send(JSON.stringify(await buildThemeList()));
115 // server.use(cookieParser());
116 server.get("/*", async (req, res) => {
118 const activeRoute = routes.find(route => matchPath(req.path, route));
119 let auth: string | undefined = IsomorphicCookie.load("jwt", req);
121 const getSiteForm: GetSite = { auth };
123 const headers = setForwardedHeaders(req.headers);
124 const client = new LemmyHttp(getHttpBaseInternal(), headers);
126 const { path, url, query } = req;
128 // Get site data first
129 // This bypasses errors, so that the client can hit the error on its own,
130 // in order to remove the jwt on the browser. Necessary for wrong jwts
131 let site: GetSiteResponse | undefined = undefined;
132 let routeData: Record<string, any> = {};
133 let errorPageData: ErrorPageData | undefined;
135 let try_site: any = await client.getSite(getSiteForm);
136 if (try_site.error == "not_logged_in") {
138 "Incorrect JWT token, skipping auth so frontend can remove jwt cookie"
140 getSiteForm.auth = undefined;
142 try_site = await client.getSite(getSiteForm);
145 if (!auth && isAuthPath(path)) {
146 res.redirect("/login");
151 initializeSite(site);
154 const initialFetchReq: InitialFetchRequest = {
162 if (activeRoute?.fetchInitialData) {
163 const routeDataKeysAndVals = await Promise.all(
164 Object.entries(activeRoute.fetchInitialData(initialFetchReq)).map(
165 async ([key, val]) => [key, await val]
169 routeData = routeDataKeysAndVals.reduce((acc, [key, val]) => {
177 errorPageData = getErrorPageData(error, site);
180 const error = Object.values(routeData).find(val => val?.error)?.error;
182 // Redirect to the 404 if there's an API error
184 console.error(error);
185 if (error === "instance_is_private") {
186 return res.redirect(`/signup`);
188 errorPageData = getErrorPageData(error, site);
192 const isoData: IsoDataOptionalSite = {
200 <StaticRouter location={url} context={isoData}>
205 const root = renderToString(wrapper);
207 res.send(await createSsrHtml(root, isoData));
209 // If an error is caught here, the error page couldn't even be rendered
211 res.statusCode = 500;
213 process.env.NODE_ENV === "development" ? err.message : "Server error"
218 server.listen(Number(port), hostname, () => {
219 console.log(`http://${hostname}:${port}`);
222 function setForwardedHeaders(headers: IncomingHttpHeaders): {
223 [key: string]: string;
225 let out: { [key: string]: string } = {};
227 out.host = headers.host;
229 let realIp = headers["x-real-ip"];
231 out["x-real-ip"] = realIp as string;
233 let forwardedFor = headers["x-forwarded-for"];
235 out["x-forwarded-for"] = forwardedFor as string;
241 process.on("SIGINT", () => {
242 console.info("Interrupted");
246 const iconSizes = [72, 96, 128, 144, 152, 192, 384, 512];
247 const defaultLogoPathDirectory = path.join(
254 export async function generateManifestBase64(site: Site) {
256 process.env.NODE_ENV === "development"
257 ? "http://localhost:1236/"
259 ).replace(/\/$/g, "");
260 const icon = site.icon ? await fetchIconPng(site.icon) : null;
264 description: site.description ?? "A link aggregator for the fediverse",
267 display: "standalone",
269 background_color: "#222222",
270 theme_color: "#222222",
271 icons: await Promise.all(
272 iconSizes.map(async size => {
273 let src = await readFile(
274 path.join(defaultLogoPathDirectory, `icon-${size}x${size}.png`)
275 ).then(buf => buf.toString("base64"));
278 src = await sharp(icon)
282 .then(buf => buf.toString("base64"));
286 sizes: `${size}x${size}`,
288 src: `data:image/png;base64,${src}`,
289 purpose: "any maskable",
295 return Buffer.from(JSON.stringify(manifest)).toString("base64");
298 async function fetchIconPng(iconUrl: string) {
300 iconUrl.replace(/https?:\/\/[^\/]+/g, getHttpBaseInternal())
302 .then(res => res.blob())
303 .then(blob => blob.arrayBuffer());
306 function getErrorPageData(error: Error, site?: GetSiteResponse) {
307 const errorPageData: ErrorPageData = {};
310 errorPageData.error = error.message;
313 const adminMatrixIds = site?.admins
314 .map(({ person: { matrix_user_id } }) => matrix_user_id)
315 .filter(id => id) as string[] | undefined;
316 if (adminMatrixIds && adminMatrixIds.length > 0) {
317 errorPageData.adminMatrixIds = adminMatrixIds;
320 return errorPageData;
323 async function createSsrHtml(root: string, isoData: IsoDataOptionalSite) {
324 const site = isoData.site_res;
325 const appleTouchIcon = site?.site_view.site.icon
326 ? `data:image/png;base64,${sharp(
327 await fetchIconPng(site.site_view.site.icon)
335 background: "#222222",
339 .then(buf => buf.toString("base64"))}`
344 <script src="//cdn.jsdelivr.net/npm/eruda"></script>
345 <script>eruda.init();</script>
349 const erudaStr = process.env["LEMMY_UI_DEBUG"] ? renderToString(eruda) : "";
351 const helmet = Helmet.renderStatic();
353 const config: ILemmyConfig = { wsHost: process.env.LEMMY_UI_LEMMY_WS_HOST };
357 <html ${helmet.htmlAttributes.toString()} lang="en">
359 <script>window.isoData = ${JSON.stringify(isoData)}</script>
360 <script>window.lemmyConfig = ${serialize(config)}</script>
362 <!-- A remote debugging utility for mobile -->
365 <!-- Custom injected script -->
368 ${helmet.title.toString()}
369 ${helmet.meta.toString()}
371 <!-- Required meta tags -->
372 <meta name="Description" content="Lemmy">
373 <meta charset="utf-8">
374 <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
379 href=${site?.site_view.site.icon ?? favIconUrl}
382 <!-- Web app manifest -->
387 href={${`data:application/manifest+json;base64,${await generateManifestBase64(
392 <link rel="apple-touch-icon" href=${appleTouchIcon} />
393 <link rel="apple-touch-startup-image" href=${appleTouchIcon} />
396 <link rel="stylesheet" type="text/css" href="/static/styles/styles.css" />
398 <!-- Current theme and more -->
399 ${helmet.link.toString()}
403 <body ${helmet.bodyAttributes.toString()}>
405 <div class="alert alert-danger rounded-0" role="alert">
406 <b>Javascript is disabled. Actions will not work.</b>
410 <div id='root'>${root}</div>
411 <script defer src='/static/js/client.js'></script>