refactor: server ts (#179)

* refactor: server ts

* refactor: add more tests

deploy/server.ts

@@ -1,85 +1,15 @@
-import * as fs from 'fs';
-import path from 'path';
+import { baseURL } from './config';
+import { logger } from './logger';
+import { routes } from './routes';
 
-// @ts-expect-error no bun
-// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-import { fetch } from 'bun';
-import { type CheerioAPI, load } from 'cheerio';
-import pino from 'pino';
-
-const distDir = path.join(__dirname, 'dist');
-const indexPath = path.join(distDir, 'index.html');
-const baseURL = process.env.APPFLOWY_BASE_URL as string;
-const defaultSite = 'https://appflowy.com';
-
-const setOrUpdateMetaTag = ($: CheerioAPI, selector: string, attribute: string, content: string) => {
-  if ($(selector).length === 0) {
-    $('head').append(`<meta ${attribute}="${selector.match(/\[(.*?)\]/)?.[1]}" content="${content}">`);
-  } else {
-    $(selector).attr('content', content);
-  }
+export type RequestContext = {
+  req: Request;
+  url: URL;
+  hostname: string | null;
+  logger: typeof logger;
 };
 
-const prettyTransport = {
-  target: 'pino-pretty',
-  options: {
-    colorize: true,
-    translateTime: 'SYS:standard',
-  },
-};
-
-const logger = pino({
-  transport: process.env.NODE_ENV === 'production' ? undefined : prettyTransport,
-  level: process.env.LOG_LEVEL || 'info',
-});
-
-const logRequestTimer = (req: Request) => {
-  const start = Date.now();
-  const pathname = new URL(req.url).pathname;
-
-  if (!pathname.startsWith('/health')) {
-    logger.debug(`Incoming request: ${pathname}`);
-  }
-
-  return () => {
-    const duration = Date.now() - start;
-
-    if (!pathname.startsWith('/health')) {
-      logger.debug(`Request for ${pathname} took ${duration}ms`);
-    }
-  };
-};
-
-const fetchMetaData = async (namespace: string, publishName?: string) => {
-  let url = `${baseURL}/api/workspace/published/${namespace}`;
-
-  if (publishName) {
-    url = `${baseURL}/api/workspace/v1/published/${namespace}/${publishName}`;
-  }
-
-  logger.debug(`Fetching meta data from ${url}`);
-  try {
-    const response = await fetch(url, {
-      verbose: false,
-    });
-
-    if (!response.ok) {
-      throw new Error(`HTTP error! Status: ${response.status}`);
-    }
-
-    const data = await response.json();
-
-    logger.debug(`Fetched meta data from ${url}: ${JSON.stringify(data)}`);
-
-    return data;
-  } catch (error) {
-    logger.error(`Failed to fetch meta data from ${url}: ${error}`);
-    return null;
-  }
-};
-
-const createServer = async (req: Request) => {
-  const timer = logRequestTimer(req);
+export const createServer = async (req: Request) => {
   const reqUrl = new URL(req.url);
   const hostname = req.headers.get('host');
 
@@ -88,7 +18,6 @@ const createServer = async (req: Request) => {
   }
 
   if (reqUrl.pathname === '/') {
-    timer();
     return new Response(null, {
       status: 302,
       headers: {
@@ -97,179 +26,29 @@ const createServer = async (req: Request) => {
     });
   }
 
-  if (['/after-payment', '/login', '/as-template', '/app', '/accept-invitation', '/import'].some(item => reqUrl.pathname.startsWith(item))) {
-    timer();
-    const htmlData = fs.readFileSync(indexPath, 'utf8');
-    const $ = load(htmlData);
+  const context: RequestContext = {
+    req,
+    url: reqUrl,
+    hostname,
+    logger,
+  };
 
-    let title, description;
+  for (const route of routes) {
+    const response = await route(context);
 
-    if (reqUrl.pathname === '/after-payment') {
-      title = 'Payment Success | AppFlowy';
-      description = 'Payment success on AppFlowy';
+    if (response) {
+      return response;
     }
-
-    if (reqUrl.pathname === '/login') {
-      title = 'Login | AppFlowy';
-      description = 'Login to AppFlowy';
-    }
-
-    if (title) $('title').text(title);
-    if (description) setOrUpdateMetaTag($, 'meta[name="description"]', 'name', description);
-
-    return new Response($.html(), {
-      headers: { 'Content-Type': 'text/html' },
-    });
   }
 
-  const [namespace, publishName] = reqUrl.pathname.slice(1).split('/');
-
-  logger.debug(`Namespace: ${namespace}, Publish Name: ${publishName}`);
-
-  if (req.method === 'GET') {
-    if (namespace === '') {
-      timer();
-      return new Response(null, {
-        status: 302,
-        headers: {
-          Location: defaultSite,
-        },
-      });
-    }
-
-    let metaData;
-    let redirectAttempted = false;
-
-    try {
-      const data = await fetchMetaData(namespace, publishName);
-
-      if (publishName) {
-        if (data && data.code === 0) {
-          metaData = data.data;
-        } else {
-          logger.error(
-            `Publish view lookup failed for namespace="${namespace}" publishName="${publishName}" response=${JSON.stringify(data)}`
-          );
-        }
-      } else {
-        const publishInfo = data?.data?.info;
-
-        if (publishInfo?.namespace && publishInfo?.publish_name) {
-          const newURL = `/${encodeURIComponent(publishInfo.namespace)}/${encodeURIComponent(publishInfo.publish_name)}`;
-
-          logger.debug(`Redirecting to default page in: ${JSON.stringify(publishInfo)}`);
-          redirectAttempted = true;
-          timer();
-          return new Response(null, {
-            status: 302,
-            headers: {
-              Location: newURL,
-            },
-          });
-        } else {
-          logger.warn(`Namespace "${namespace}" has no default publish page. response=${JSON.stringify(data)}`);
-        }
-      }
-    } catch (error) {
-      logger.error(`Error fetching meta data: ${error}`);
-    }
-
-    const htmlData = fs.readFileSync(indexPath, 'utf8');
-    const $ = load(htmlData);
-
-    const description = 'Write, share, and publish docs quickly on AppFlowy.\nGet started for free.';
-    let title = 'AppFlowy';
-    const url = `https://${hostname}${reqUrl.pathname}`;
-    let image = '/og-image.png';
-    let favicon = '/appflowy.ico';
-
-    try {
-      if (metaData && metaData.view) {
-        const view = metaData.view;
-        const emoji = view.icon?.ty === 0 && view.icon?.value;
-        const icon = view.icon?.ty === 2 && view.icon?.value;
-        const titleList = [];
-
-        if (emoji) {
-          const emojiCode = emoji.codePointAt(0).toString(16); // Convert emoji to hex code
-          const baseUrl = 'https://raw.githubusercontent.com/googlefonts/noto-emoji/main/svg/emoji_u';
-
-          favicon = `${baseUrl}${emojiCode}.svg`;
-        } else if (icon) {
-          try {
-            const { iconContent, color } = JSON.parse(view.icon?.value);
-
-            favicon = getIconBase64(iconContent, color);
-            $('link[rel="icon"]').attr('type', 'image/svg+xml');
-          } catch (_) {
-            // Do nothing
-          }
-        }
-
-        if (view.name) {
-          titleList.push(view.name);
-          titleList.push('|');
-        }
-
-        titleList.push('AppFlowy');
-        title = titleList.join(' ');
-
-        try {
-          const cover = view.extra ? JSON.parse(view.extra)?.cover : null;
-
-          if (cover) {
-            if (['unsplash', 'custom'].includes(cover.type)) {
-              image = cover.value;
-            } else if (cover.type === 'built_in') {
-              image = `/covers/m_cover_image_${cover.value}.png`;
-            }
-          }
-        } catch (_) {
-          // Do nothing
-        }
-      }
-    } catch (error) {
-      logger.error(`Error injecting meta data: ${error}`);
-    }
-
-    if (!metaData) {
-      logger.warn(
-        `Serving fallback landing page for namespace="${namespace}" publishName="${publishName ?? ''}". redirectAttempted=${redirectAttempted}`
-      );
-    }
-
-    $('title').text(title);
-    $('link[rel="icon"]').attr('href', favicon);
-    $('link[rel="canonical"]').attr('href', url);
-    setOrUpdateMetaTag($, 'meta[name="description"]', 'name', description);
-    setOrUpdateMetaTag($, 'meta[property="og:title"]', 'property', title);
-    setOrUpdateMetaTag($, 'meta[property="og:description"]', 'property', description);
-    setOrUpdateMetaTag($, 'meta[property="og:image"]', 'property', image);
-    setOrUpdateMetaTag($, 'meta[property="og:url"]', 'property', url);
-    setOrUpdateMetaTag($, 'meta[property="og:site_name"]', 'property', 'AppFlowy');
-    setOrUpdateMetaTag($, 'meta[property="og:type"]', 'property', 'website');
-    setOrUpdateMetaTag($, 'meta[name="twitter:card"]', 'name', 'summary_large_image');
-    setOrUpdateMetaTag($, 'meta[name="twitter:title"]', 'name', title);
-    setOrUpdateMetaTag($, 'meta[name="twitter:description"]', 'name', description);
-    setOrUpdateMetaTag($, 'meta[name="twitter:image"]', 'name', image);
-    setOrUpdateMetaTag($, 'meta[name="twitter:site"]', 'name', '@appflowy');
-
-    timer();
-    return new Response($.html(), {
-      headers: { 'Content-Type': 'text/html' },
-    });
-  } else {
-    timer();
-    logger.error({ message: 'Method not allowed', method: req.method });
-    return new Response('Method not allowed', { status: 405 });
-  }
+  return new Response('Not Found', { status: 404 });
 };
 
 declare const Bun: {
   serve: (options: { port: number; fetch: typeof createServer; error: (err: Error) => Response }) => void;
 };
 
-const start = () => {
+export const start = () => {
   try {
     // eslint-disable-next-line @typescript-eslint/no-var-requires
     Bun.serve({
@@ -288,33 +67,6 @@ const start = () => {
   }
 };
 
-start();
-
-export { };
-
-function getIconBase64(svgText: string, color: string) {
-  let newSvgText = svgText.replace(/fill="[^"]*"/g, ``);
-
-  newSvgText = newSvgText.replace('<svg', `<svg fill="${argbToRgba(color)}"`);
-
-  const base64String = btoa(newSvgText);
-
-  return `data:image/svg+xml;base64,${base64String}`;
-}
-
-function argbToRgba(color: string): string {
-  const hex = color.replace(/^#|0x/, '');
-
-  const hasAlpha = hex.length === 8;
-
-  if (!hasAlpha) {
-    return color.replace('0x', '#');
-  }
-
-  const r = parseInt(hex.slice(2, 4), 16);
-  const g = parseInt(hex.slice(4, 6), 16);
-  const b = parseInt(hex.slice(6, 8), 16);
-  const a = hasAlpha ? parseInt(hex.slice(0, 2), 16) / 255 : 1;
-
-  return `rgba(${r}, ${g}, ${b}, ${a})`;
+if (process.env.NODE_ENV !== 'test') {
+  start();
 }