From 7a71e614be82016dd5d95c956e4c7d85c0a15fd3 Mon Sep 17 00:00:00 2001
From: Evan Simkowitz <esimkowitz@users.noreply.github.com>
Date: Wed, 6 Nov 2024 11:47:01 -0800
Subject: [PATCH] Set base url for storybook using env var

---
 .github/workflows/deploy-docsite.yml |  2 +-
 .storybook/main.ts                   | 12 ++++++++++++
 .storybook/manager-head.html         |  8 --------
 3 files changed, 13 insertions(+), 9 deletions(-)
 delete mode 100644 .storybook/manager-head.html

diff --git a/.github/workflows/deploy-docsite.yml b/.github/workflows/deploy-docsite.yml
index 72c6973c08..0a38973593 100644
--- a/.github/workflows/deploy-docsite.yml
+++ b/.github/workflows/deploy-docsite.yml
@@ -41,7 +41,7 @@ jobs:
               working-directory: docs/
 
             - name: Build Storybook
-              run: yarn build-storybook
+              run: IS_OFFICIAL=1 yarn build-storybook
             # Copy the storybook static site into the docsite output and copy the assets from the main repo into the assets dir of the docsite output.
             # It needs to be in assets/assets because assets/ gets served from the / root of the site and storybook expects to find assets in /assets/
             - name: Copy Storybook to docs
diff --git a/.storybook/main.ts b/.storybook/main.ts
index 5679562282..68810e004f 100644
--- a/.storybook/main.ts
+++ b/.storybook/main.ts
@@ -38,5 +38,17 @@ const config: StorybookConfig = {
     },
 
     staticDirs: [{ from: "../assets", to: "/assets" }],
+    managerHead: (head) => `
+        ${head}
+        <link rel="shortcut icon" href="/assets/waveterm-logo-with-bg.ico" />
+        <link rel="icon" type="image/png" href="/assets/waveterm-logo-with-bg.png" sizes="250x250" />
+        <style>
+        .sidebar-header img {
+            max-width: 150px !important;
+            max-height: 100px !important;
+        }
+        </style>
+        <base href="${process.env.IS_OFFICIAL ? "/storybook" : "/"}" />
+`,
 };
 export default config;
diff --git a/.storybook/manager-head.html b/.storybook/manager-head.html
deleted file mode 100644
index a041edc4aa..0000000000
--- a/.storybook/manager-head.html
+++ /dev/null
@@ -1,8 +0,0 @@
-<link rel="shortcut icon" href="/assets/waveterm-logo-with-bg.ico" />
-<link rel="icon" type="image/png" href="/assets/waveterm-logo-with-bg.png" sizes="250x250" />
-<style>
-  .sidebar-header img {
-    max-width: 150px !important;
-    max-height: 100px !important;
-  }
-</style>