From a73720411cd16376e01dac4fc27fb8d29b057045 Mon Sep 17 00:00:00 2001
From: moonrailgun <moonrailgun@gmail.com>
Date: Thu, 18 Apr 2024 00:20:01 +0800
Subject: [PATCH] feat: add custom domain support for status page

---
 .../monitor/StatusPage/EditForm.tsx           | 22 ++++-
 .../components/monitor/StatusPage/index.tsx   | 15 +++-
 src/client/utils/validator.ts                 |  9 ++
 src/server/app.ts                             | 22 ++++-
 src/server/model/monitor/page/manager.ts      | 84 +++++++++++++++++++
 .../migration.sql                             |  2 +
 src/server/prisma/schema.prisma               |  1 +
 src/server/prisma/zod/monitorstatuspage.ts    |  1 +
 src/server/trpc/routers/monitor.ts            | 43 ++++++++--
 src/server/ws/index.ts                        |  3 +-
 10 files changed, 192 insertions(+), 10 deletions(-)
 create mode 100644 src/server/model/monitor/page/manager.ts
 create mode 100644 src/server/prisma/migrations/20240417150948_add_status_page_domain/migration.sql

diff --git a/src/client/components/monitor/StatusPage/EditForm.tsx b/src/client/components/monitor/StatusPage/EditForm.tsx
index eff9612..0665d12 100644
--- a/src/client/components/monitor/StatusPage/EditForm.tsx
+++ b/src/client/components/monitor/StatusPage/EditForm.tsx
@@ -1,7 +1,7 @@
 import { Switch, Divider, Form, Input, Typography } from 'antd';
 import React from 'react';
 import { MonitorPicker } from '../MonitorPicker';
-import { urlSlugValidator } from '../../../utils/validator';
+import { domainValidator, urlSlugValidator } from '../../../utils/validator';
 import { useTranslation } from '@i18next-toolkit/react';
 import { Button } from '@/components/ui/button';
 import { LuMinusCircle, LuPlus } from 'react-icons/lu';
@@ -12,6 +12,7 @@ export interface MonitorStatusPageEditFormValues {
   title: string;
   slug: string;
   monitorList: PrismaJson.MonitorStatusPageList;
+  domain: string;
 }
 
 interface MonitorStatusPageEditFormProps {
@@ -71,6 +72,25 @@ export const MonitorStatusPageEditForm: React.FC<MonitorStatusPageEditFormProps>
             <Input addonBefore={`${window.origin}/status/`} />
           </Form.Item>
 
+          <Form.Item
+            label={t('Custom Domain')}
+            name="domain"
+            extra={
+              <div>
+                {t(
+                  'You can config your status page in your own domain, for example: status.example.com'
+                )}
+              </div>
+            }
+            rules={[
+              {
+                validator: domainValidator,
+              },
+            ]}
+          >
+            <Input />
+          </Form.Item>
+
           <Form.List name="monitorList">
             {(fields, { add, remove }, { errors }) => {
               return (
diff --git a/src/client/components/monitor/StatusPage/index.tsx b/src/client/components/monitor/StatusPage/index.tsx
index 868cc6d..df2f4f7 100644
--- a/src/client/components/monitor/StatusPage/index.tsx
+++ b/src/client/components/monitor/StatusPage/index.tsx
@@ -1,4 +1,4 @@
-import React, { useState } from 'react';
+import React, { useMemo, useState } from 'react';
 import { trpc } from '../../../api/trpc';
 import { useAllowEdit } from './useAllowEdit';
 import {
@@ -38,6 +38,17 @@ export const MonitorStatusPage: React.FC<MonitorStatusPageProps> = React.memo(
 
     const monitorList = info?.monitorList ?? [];
 
+    const initialValues = useMemo(() => {
+      if (!info) {
+        return {};
+      }
+
+      return {
+        ...info,
+        domain: info.domain ?? '',
+      };
+    }, [info]);
+
     const [{ loading }, handleSave] = useRequest(
       async (values: MonitorStatusPageEditFormValues) => {
         if (!info) {
@@ -84,7 +95,7 @@ export const MonitorStatusPage: React.FC<MonitorStatusPageProps> = React.memo(
           <div className="w-1/3 overflow-auto border-r border-gray-300 px-4 py-8 dark:border-gray-600">
             <MonitorStatusPageEditForm
               isLoading={loading}
-              initialValues={info ?? {}}
+              initialValues={initialValues}
               onFinish={handleSave}
               onCancel={() => setEditMode(false)}
             />
diff --git a/src/client/utils/validator.ts b/src/client/utils/validator.ts
index e676f9c..3348f80 100644
--- a/src/client/utils/validator.ts
+++ b/src/client/utils/validator.ts
@@ -17,6 +17,15 @@ export const hostnameValidator: Validator = (rule, value, callback) => {
   }
 };
 
+export const domainValidator: Validator = (rule, value, callback) => {
+  try {
+    z.string().regex(hostnameRegex).parse(value);
+    callback();
+  } catch (err) {
+    callback('Not valid, it should be domain, for example: example.com');
+  }
+};
+
 export const urlSlugValidator: Validator = (rule, value, callback) => {
   try {
     z.string().regex(slugRegex).parse(value);
diff --git a/src/server/app.ts b/src/server/app.ts
index b0434c9..2a89761 100644
--- a/src/server/app.ts
+++ b/src/server/app.ts
@@ -19,6 +19,7 @@ import { logger } from './utils/logger';
 import { monitorRouter } from './router/monitor';
 import { healthRouter } from './router/health';
 import path from 'path';
+import { monitorPageManager } from './model/monitor/page/manager';
 
 const app = express();
 
@@ -27,7 +28,6 @@ app.use(express.json());
 app.use(passport.initialize());
 app.use(morgan('tiny'));
 app.use(cors());
-app.use(express.static('public'));
 
 // http://expressjs.com/en/advanced/best-practice-security.html#at-a-minimum-disable-x-powered-by-header
 app.disable('x-powered-by');
@@ -60,6 +60,26 @@ if (env.allowOpenapi) {
   app.use('/open', trpcOpenapiHttpHandler);
 }
 
+// Custom Status Page
+app.use('/*', (req, res, next) => {
+  if (req.baseUrl === '/' || req.baseUrl === '') {
+    const customDomain = monitorPageManager.findPageDomain(req.hostname);
+    if (customDomain) {
+      res
+        .status(200)
+        .send(
+          `<body style="padding: 0; margin: 0;"><iframe style="border:none; width: 100%; height: 100%;" title="" src="/status/${customDomain.slug}" /></body>`
+        );
+      return;
+    }
+  }
+
+  next();
+});
+
+// Static
+app.use(express.static('public'));
+
 // fallback
 app.use('/*', (req, res) => {
   if (req.method === 'GET' && req.accepts('html')) {
diff --git a/src/server/model/monitor/page/manager.ts b/src/server/model/monitor/page/manager.ts
new file mode 100644
index 0000000..62bb7c9
--- /dev/null
+++ b/src/server/model/monitor/page/manager.ts
@@ -0,0 +1,84 @@
+import { logger } from '../../../utils/logger';
+import { prisma } from '../../_client';
+
+class MonitorPageManager {
+  private customDomainPage: Record<
+    string,
+    {
+      workspaceId: string;
+      pageId: string;
+      slug: string;
+    }
+  > = {}; // key: domain
+
+  constructor() {
+    prisma.monitorStatusPage
+      .findMany({
+        where: {
+          domain: {
+            not: null,
+          },
+        },
+        select: {
+          id: true,
+          domain: true,
+          workspaceId: true,
+          slug: true,
+        },
+      })
+      .then((res) => {
+        res.forEach((item) => {
+          if (!item.domain) {
+            return;
+          }
+
+          this.customDomainPage[item.domain] = {
+            pageId: item.id,
+            workspaceId: item.workspaceId,
+            slug: item.slug,
+          };
+        });
+
+        logger.info(`Loaded ${res.length} custom domain for status page`);
+      })
+      .catch((err) => {
+        logger.error('Cannot load monitor page domain list:', err);
+      });
+  }
+
+  /**
+   * check domain existed
+   * if domain not been used, return true
+   */
+  async checkDomain(domain: string) {
+    const res = await prisma.monitorStatusPage.findFirst({
+      where: {
+        domain,
+      },
+    });
+
+    return !res;
+  }
+
+  async updatePageDomain(
+    domain: string,
+    info: {
+      workspaceId: string;
+      pageId: string;
+      slug: string;
+    }
+  ) {
+    this.customDomainPage[domain] = info;
+    logger.info(`Update page domain: ${domain} to page ${info.pageId}`);
+  }
+
+  findPageDomain(domain: string) {
+    if (!this.customDomainPage[domain]) {
+      return null;
+    }
+
+    return this.customDomainPage[domain];
+  }
+}
+
+export const monitorPageManager = new MonitorPageManager();
diff --git a/src/server/prisma/migrations/20240417150948_add_status_page_domain/migration.sql b/src/server/prisma/migrations/20240417150948_add_status_page_domain/migration.sql
new file mode 100644
index 0000000..fa9e734
--- /dev/null
+++ b/src/server/prisma/migrations/20240417150948_add_status_page_domain/migration.sql
@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "MonitorStatusPage" ADD COLUMN     "domain" TEXT;
diff --git a/src/server/prisma/schema.prisma b/src/server/prisma/schema.prisma
index f6a338b..4973e6a 100644
--- a/src/server/prisma/schema.prisma
+++ b/src/server/prisma/schema.prisma
@@ -355,6 +355,7 @@ model MonitorStatusPage {
   /// [MonitorStatusPageList]
   /// @zod.custom(imports.MonitorStatusPageListSchema)
   monitorList Json     @default("[]") // monitor list
+  domain      String? // custom domain which can add cname record
   createdAt   DateTime @default(now()) @db.Timestamptz(6)
   updatedAt   DateTime @updatedAt @db.Timestamptz(6)
 
diff --git a/src/server/prisma/zod/monitorstatuspage.ts b/src/server/prisma/zod/monitorstatuspage.ts
index 696ece9..22f26b8 100644
--- a/src/server/prisma/zod/monitorstatuspage.ts
+++ b/src/server/prisma/zod/monitorstatuspage.ts
@@ -18,6 +18,7 @@ export const MonitorStatusPageModelSchema = z.object({
    * [MonitorStatusPageList]
    */
   monitorList: imports.MonitorStatusPageListSchema,
+  domain: z.string().nullish(),
   createdAt: z.date(),
   updatedAt: z.date(),
 })
diff --git a/src/server/trpc/routers/monitor.ts b/src/server/trpc/routers/monitor.ts
index 6ab2b4f..ae89a74 100644
--- a/src/server/trpc/routers/monitor.ts
+++ b/src/server/trpc/routers/monitor.ts
@@ -31,6 +31,7 @@ import {
   MonitorInfoWithNotificationIds,
   MonitorPublicInfoSchema,
 } from '../../model/_schema/monitor';
+import { monitorPageManager } from '../../model/monitor/page/manager';
 
 export const monitorRouter = router({
   all: workspaceProcedure
@@ -599,12 +600,14 @@ export const monitorRouter = router({
           MonitorStatusPageModelSchema.pick({
             description: true,
             monitorList: true,
+            domain: true,
           }).partial()
         )
     )
     .output(MonitorStatusPageModelSchema)
     .mutation(async ({ input }) => {
-      const { workspaceId, slug, title, description, monitorList } = input;
+      const { workspaceId, slug, title, description, monitorList, domain } =
+        input;
 
       const existSlugCount = await prisma.monitorStatusPage.count({
         where: {
@@ -616,17 +619,30 @@ export const monitorRouter = router({
         throw new Error('This slug has been existed');
       }
 
-      const res = await prisma.monitorStatusPage.create({
+      if (domain && !(await monitorPageManager.checkDomain(domain))) {
+        throw new Error('This domain has been used');
+      }
+
+      const page = await prisma.monitorStatusPage.create({
         data: {
           workspaceId,
           slug,
           title,
           description,
           monitorList,
+          domain: domain || null, // make sure not ''
         },
       });
 
-      return res;
+      if (page.domain) {
+        monitorPageManager.updatePageDomain(page.domain, {
+          workspaceId: page.workspaceId,
+          pageId: page.id,
+          slug: page.slug,
+        });
+      }
+
+      return page;
     }),
   editPage: workspaceOwnerProcedure
     .meta(
@@ -644,12 +660,14 @@ export const monitorRouter = router({
           title: true,
           description: true,
           monitorList: true,
+          domain: true,
         }).partial()
       )
     )
     .output(MonitorStatusPageModelSchema)
     .mutation(async ({ input }) => {
-      const { id, workspaceId, slug, title, description, monitorList } = input;
+      const { id, workspaceId, slug, title, description, monitorList, domain } =
+        input;
 
       if (slug) {
         const existSlugCount = await prisma.monitorStatusPage.count({
@@ -666,7 +684,11 @@ export const monitorRouter = router({
         }
       }
 
-      return prisma.monitorStatusPage.update({
+      if (domain && !(await monitorPageManager.checkDomain(domain))) {
+        throw new Error('This domain has been used');
+      }
+
+      const page = await prisma.monitorStatusPage.update({
         where: {
           id,
           workspaceId,
@@ -676,8 +698,19 @@ export const monitorRouter = router({
           title,
           description,
           monitorList,
+          domain: domain || null,
         },
       });
+
+      if (page.domain) {
+        monitorPageManager.updatePageDomain(page.domain, {
+          workspaceId: page.workspaceId,
+          pageId: page.id,
+          slug: page.slug,
+        });
+      }
+
+      return page;
     }),
   deletePage: workspaceOwnerProcedure
     .meta(
diff --git a/src/server/ws/index.ts b/src/server/ws/index.ts
index 263a989..4db434c 100644
--- a/src/server/ws/index.ts
+++ b/src/server/ws/index.ts
@@ -3,6 +3,7 @@ import { Server as HTTPServer } from 'http';
 import { jwtVerify } from '../middleware/auth';
 import { socketEventBus } from './shared';
 import { isCuid } from '../utils/common';
+import { logger } from '../utils/logger';
 
 export function initSocketio(httpServer: HTTPServer) {
   const io = new SocketIOServer(httpServer, {
@@ -30,7 +31,7 @@ export function initSocketio(httpServer: HTTPServer) {
         try {
           const user = jwtVerify(token);
 
-          console.log('[Socket] Authenticated via JWT:', user.username);
+          logger.info('[Socket] Authenticated via JWT:', user.username);
 
           socket.data.user = user;
           socket.data.token = token;