From de21303fb141f0e6495131475e47ad8da8424ba1 Mon Sep 17 00:00:00 2001
From: Red Adaya <red.adaya@gmail.com>
Date: Wed, 3 Apr 2024 22:12:48 +0800
Subject: [PATCH] root level

---
 src/app/app.tsx                        | 11 ++++-
 src/app/common/elements/styleblock.tsx |  4 +-
 src/app/workspace/workspaceview.tsx    | 62 ++------------------------
 3 files changed, 15 insertions(+), 62 deletions(-)

diff --git a/src/app/app.tsx b/src/app/app.tsx
index 684d04f61..692d2be0b 100644
--- a/src/app/app.tsx
+++ b/src/app/app.tsx
@@ -20,7 +20,7 @@ import { MainSideBar } from "./sidebar/main";
 import { RightSideBar } from "./sidebar/right";
 import { DisconnectedModal, ClientStopModal } from "@/modals";
 import { ModalsProvider } from "@/modals/provider";
-import { Button } from "@/elements";
+import { Button, StyleBlock } from "@/elements";
 import { ErrorBoundary } from "@/common/error/errorboundary";
 import cn from "classnames";
 import "./app.less";
@@ -124,9 +124,10 @@ class App extends React.Component<{}, {}> {
         const rightSidebarCollapsed = GlobalModel.rightSidebarModel.getCollapsed();
         const activeMainView = GlobalModel.activeMainView.get();
         const lightDarkClass = GlobalModel.isDarkTheme.get() ? "is-dark" : "is-light";
+        const termTheme = GlobalModel.getTermTheme();
         return (
             <div
-                key={"version-" + renderVersion}
+                key={`version- + ${renderVersion}`}
                 id="main"
                 className={cn(
                     "platform-" + platform,
@@ -135,6 +136,12 @@ class App extends React.Component<{}, {}> {
                 )}
                 onContextMenu={this.handleContextMenu}
             >
+                <StyleBlock
+                    termTheme={termTheme}
+                    themeSrcEl={document.documentElement}
+                    themeKey="global"
+                    className=":root"
+                />
                 <If condition={mainSidebarCollapsed}>
                     <div key="logo-button" className="logo-button-container">
                         <div className="logo-button-spacer" />
diff --git a/src/app/common/elements/styleblock.tsx b/src/app/common/elements/styleblock.tsx
index 91eb58d67..630963f28 100644
--- a/src/app/common/elements/styleblock.tsx
+++ b/src/app/common/elements/styleblock.tsx
@@ -43,7 +43,7 @@ class StyleBlock extends React.Component<
                         .map(([key, value]) => `--term-${key}: ${value};`)
                         .join(" ");
 
-                    this.styleRules.set(`.${className} { ${styleProperties} }`);
+                    this.styleRules.set(`${className} { ${styleProperties} }`);
                     GlobalModel.bumpTermRenderVersion();
                     this.theme = currTheme;
                 } else {
@@ -56,6 +56,8 @@ class StyleBlock extends React.Component<
     }
 
     render() {
+        console.log("this.styleRules", this.styleRules.get());
+
         if (isBlank(this.styleRules.get())) {
             return null;
         }
diff --git a/src/app/workspace/workspaceview.tsx b/src/app/workspace/workspaceview.tsx
index 08c43cf4c..5dcb6c7fc 100644
--- a/src/app/workspace/workspaceview.tsx
+++ b/src/app/workspace/workspaceview.tsx
@@ -16,8 +16,8 @@ import { ErrorBoundary } from "@/common/error/errorboundary";
 import * as textmeasure from "@/util/textmeasure";
 import { boundMethod } from "autobind-decorator";
 import type { Screen } from "@/models";
-import { Button } from "@/elements";
-import { commandRtnHandler, isBlank } from "@/util/util";
+import { Button, StyleBlock } from "@/elements";
+import { commandRtnHandler } from "@/util/util";
 import { getTermThemes } from "@/util/themeutil";
 import { Dropdown } from "@/elements/dropdown";
 import { getRemoteStrWithAlias } from "@/common/prompt/prompt";
@@ -206,62 +206,6 @@ class TabSettings extends React.Component<{ screen: Screen }, {}> {
     }
 }
 
-@mobxReact.observer
-class StyleBlock extends React.Component<
-    { themeSrcEl: HTMLElement; themeKey: string; className: string; termTheme: TermThemeType },
-    { styleRules: string }
-> {
-    styleRules: OV<string> = mobx.observable.box("", { name: "StyleBlock-styleRules" });
-    theme: string;
-
-    componentDidMount(): void {
-        GlobalModel.termThemeSrcEl.set(this.props.themeSrcEl);
-        this.loadThemeStyles();
-    }
-
-    componentDidUpdate(prevProps): void {
-        const { themeKey, termTheme } = this.props;
-        const currTheme = termTheme[themeKey];
-        if (themeKey !== prevProps.themeKey || currTheme !== this.theme) {
-            this.loadThemeStyles();
-        }
-        if (this.props.themeSrcEl !== prevProps.themeSrcEl) {
-            GlobalModel.termThemeSrcEl.set(this.props.themeSrcEl);
-        }
-    }
-
-    async loadThemeStyles() {
-        const { themeKey, className, termTheme } = this.props;
-        const currTheme = termTheme[themeKey];
-
-        if (currTheme !== this.theme && currTheme) {
-            const rtn = GlobalModel.getTermThemeJson(currTheme);
-            rtn.then((termThemeJson) => {
-                if (termThemeJson && typeof termThemeJson === "object") {
-                    const styleProperties = Object.entries(termThemeJson)
-                        .map(([key, value]) => `--term-${key}: ${value};`)
-                        .join(" ");
-
-                    this.styleRules.set(`.${className} { ${styleProperties} }`);
-                    GlobalModel.bumpTermRenderVersion();
-                    this.theme = currTheme;
-                } else {
-                    console.error("termThemeJson is not an object:", termThemeJson);
-                }
-            }).catch((error) => {
-                console.error("error loading theme styles:", error);
-            });
-        }
-    }
-
-    render() {
-        if (isBlank(this.styleRules.get())) {
-            return null;
-        }
-        return <style>{this.styleRules.get()}</style>;
-    }
-}
-
 @mobxReact.observer
 class WorkspaceView extends React.Component<{}, {}> {
     sessionRef = React.createRef<HTMLDivElement>();
@@ -359,7 +303,7 @@ class WorkspaceView extends React.Component<{}, {}> {
                         themeSrcEl={this.sessionRef.current}
                         themeKey={session.sessionId}
                         termTheme={termTheme}
-                        className="session-view"
+                        className=".session-view"
                     />
                 </If>
                 <If condition={!isHidden}>