src/server/index.tsx

   1 import serialize from 'serialize-javascript';
   2 import express from 'express';
   3 import { StaticRouter } from 'inferno-router';
   4 import { renderToString } from 'inferno-server';
   5 import { matchPath } from 'inferno-router';
   6 import path from 'path';
   7 import { App } from '../shared/components/app';
   8 import {
   9   ILemmyConfig,
  10   InitialFetchRequest,
  11   IsoData,
  12 } from '../shared/interfaces';
  13 import { routes } from '../shared/routes';
  14 import IsomorphicCookie from 'isomorphic-cookie';
  15 import { GetSite, LemmyHttp } from 'lemmy-js-client';
  16 import process from 'process';
  17 import { Helmet } from 'inferno-helmet';
  18 import { initializeSite } from '../shared/initialize';
  19 import { httpUri } from '../shared/env';
  20 import { IncomingHttpHeaders } from 'http';
  21 import { setOptionalAuth } from '../shared/utils';
  22
  23 const server = express();
  24 const port = 1234;
  25
  26 server.use(express.json());
  27 server.use(express.urlencoded({ extended: false }));
  28 server.use('/static', express.static(path.resolve('./dist')));
  29
  30 // server.use(cookieParser());
  31
  32 server.get('/*', async (req, res) => {
  33   const activeRoute = routes.find(route => matchPath(req.path, route)) || {};
  34   const context = {} as any;
  35   let auth: string = IsomorphicCookie.load('jwt', req);
  36
  37   let getSiteForm: GetSite = {};
  38   setOptionalAuth(getSiteForm, auth);
  39
  40   let promises: Promise<any>[] = [];
  41
  42   let headers = setForwardedHeaders(req.headers);
  43
  44   let initialFetchReq: InitialFetchRequest = {
  45     client: new LemmyHttp(httpUri, headers),
  46     auth,
  47     path: req.path,
  48   };
  49
  50   // Get site data first
  51   let site = await initialFetchReq.client.getSite(getSiteForm);
  52   initializeSite(site);
  53
  54   if (activeRoute.fetchInitialData) {
  55     promises.push(...activeRoute.fetchInitialData(initialFetchReq));
  56   }
  57
  58   let routeData = await Promise.all(promises);
  59
  60   // Redirect to the 404 if there's an API error
  61   if (routeData[0] && routeData[0].error) {
  62     let errCode = routeData[0].error;
  63     return res.redirect(`/404?err=${errCode}`);
  64   }
  65
  66   let acceptLang = req.headers['accept-language']
  67     ? req.headers['accept-language'].split(',')[0]
  68     : 'en';
  69   let lang = site.my_user
  70     ? site.my_user.lang == 'browser'
  71       ? acceptLang
  72       : 'en'
  73     : acceptLang;
  74
  75   let isoData: IsoData = {
  76     path: req.path,
  77     site_res: site,
  78     routeData,
  79     lang,
  80   };
  81
  82   const wrapper = (
  83     <StaticRouter location={req.url} context={isoData}>
  84       <App siteRes={isoData.site_res} />
  85     </StaticRouter>
  86   );
  87   if (context.url) {
  88     return res.redirect(context.url);
  89   }
  90
  91   const cspHtml = (
  92     <meta
  93       http-equiv="Content-Security-Policy"
  94       content="default-src data: 'self'; connect-src * ws: wss:; frame-src *; img-src * data:; script-src 'self'; style-src 'self' 'unsafe-inline'; manifest-src 'self'"
  95     />
  96   );
  97
  98   const root = renderToString(wrapper);
  99   const cspStr = process.env.LEMMY_EXTERNAL_HOST ? renderToString(cspHtml) : '';
 100   const helmet = Helmet.renderStatic();
 101
 102   const config: ILemmyConfig = { wsHost: process.env.LEMMY_WS_HOST };
 103
 104   res.send(`
 105            <!DOCTYPE html>
 106            <html ${helmet.htmlAttributes.toString()} lang="en">
 107            <head>
 108            <script>window.isoData = ${serialize(isoData)}</script>
 109            <script>window.lemmyConfig = ${serialize(config)}</script>
 110
 111            ${helmet.title.toString()}
 112            ${helmet.meta.toString()}
 113
 114            <!-- Required meta tags -->
 115            <meta name="Description" content="Lemmy">
 116            <meta charset="utf-8">
 117            <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
 118
 119            <!-- Content Security Policy -->
 120            ${cspStr}
 121
 122            <!-- Web app manifest -->
 123            <link rel="manifest" href="/static/assets/manifest.webmanifest">
 124
 125            <!-- Icons -->
 126            <link rel="shortcut icon" type="image/svg+xml" href="/static/assets/icons/favicon.svg" />
 127            <link rel="apple-touch-icon" href="/static/assets/icons/apple-touch-icon.png" />
 128
 129            <!-- Styles -->
 130            <link rel="stylesheet" type="text/css" href="/static/styles/styles.css" />
 131
 132            <!-- Current theme and more -->
 133            ${helmet.link.toString()}
 134            </head>
 135
 136            <body ${helmet.bodyAttributes.toString()}>
 137              <noscript>
 138                <div class="alert alert-danger rounded-0" role="alert">
 139                  <b>Javascript is disabled. Actions will not work.</b>
 140                </div>
 141              </noscript>
 142
 143              <div id='root'>${root}</div>
 144              <script defer src='/static/js/client.js'></script>
 145            </body>
 146          </html>
 147 `);
 148 });
 149
 150 server.listen(port, () => {
 151   console.log(`http://localhost:${port}`);
 152 });
 153
 154 function setForwardedHeaders(
 155   headers: IncomingHttpHeaders
 156 ): { [key: string]: string } {
 157   let out = {
 158     host: headers.host,
 159   };
 160   if (headers['x-real-ip']) {
 161     out['x-real-ip'] = headers['x-real-ip'];
 162   }
 163   if (headers['x-forwarded-for']) {
 164     out['x-forwarded-for'] = headers['x-forwarded-for'];
 165   }
 166
 167   return out;
 168 }
 169
 170 process.on('SIGINT', () => {
 171   console.info('Interrupted');
 172   process.exit(0);
 173 });