fix: datapilot code block ui

AltimateAI · Oct 6, 2024 · 49403e9 · 49403e9
1 parent 058fed6
commit 49403e9
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/webview_panels/src/modules/dataPilot/components/queryAnalysis/QueryAnalysis.tsx b/webview_panels/src/modules/dataPilot/components/queryAnalysis/QueryAnalysis.tsx
@@ -10,6 +10,7 @@ import { QueryAnalysisCommands } from "./commands";
 import { AltimateIcon } from "@assets/icons";
 import { QueryAnalysisType } from "./types";
 import { useMemo } from "react";
+import classes from "../../datapilot.module.scss";
 
 const QUERY_HAPPY_LIMIT = 10;
 const DefaultActions = [
@@ -52,7 +53,7 @@ const QueryAnalysis = (): JSX.Element | null => {
     <Stack direction="column">
       <DatapilotHeader />
 
-      <CodeBlock code={chat.query} language="sql" fileName={chat.fileName} />
+      <CodeBlock code={chat.query} language="sql" fileName={chat.fileName} classname={classes.codeblock}/>
       {showLineLimitWarning ? (
         <Card>
           <CardTitle>

diff --git a/webview_panels/src/modules/dataPilot/datapilot.module.scss b/webview_panels/src/modules/dataPilot/datapilot.module.scss
@@ -153,3 +153,7 @@ body {
     }}
   }
 }
+
+.codeblock{
+  --code-bg: transparent;
+}
diff --git a/webview_panels/src/uiCore/components/codeblock/index.tsx b/webview_panels/src/uiCore/components/codeblock/index.tsx
@@ -10,27 +10,32 @@ interface Props {
   fileName?: string;
   showLineNumbers?: boolean;
   titleActions?: ReactNode;
+  classname?: string;
 }
 const CodeBlockComponent = ({
   code,
   language,
   fileName,
   showLineNumbers,
   titleActions,
+  classname
 }: Props): JSX.Element => {
   const {
     state: { theme },
   } = useAppContext();
   const codeBlockTheme = theme === Themes.Dark ? "dark" : "light";
+  const editorTheme = theme === Themes.Dark ? "vsc-dark-plus" : "vs";
   return (
     <div className={classes.codeblock}>
       <CodeblockLib
         showLineNumbers={showLineNumbers}
         code={code}
         fileName={fileName}
         theme={codeBlockTheme}
+        editorTheme={editorTheme}
         language={language}
         titleActions={titleActions}
+        className={classname}
       />
     </div>
   );