@finos/legend-application-studio 23.1.7 → 23.1.8

package/src/components/editor/edit-panel/mapping-editor/MappingExecutionBuilder.tsx

@@ -65,12 +65,10 @@ import {
   MappingExecutionRelationalInputDataState,
 } from '../../../../stores/editor/editor-state/element-editor-state/mapping/MappingExecutionState.js';
 import {
-  EDITOR_LANGUAGE,
+  CODE_EDITOR_LANGUAGE,
   ActionAlertActionType,
   ActionAlertType,
   useApplicationStore,
-  ExecutionPlanViewer,
-  TextInputEditor,
 } from '@finos/legend-application';
 import { useEditorStore } from '../../EditorStoreProvider.js';
 import {
@@ -89,7 +87,11 @@ import {
   relationalInputData_setInputType,
 } from '../../../../stores/editor/shared/modifier/STO_Relational_GraphModifierHelper.js';
 import { MappingExecutionQueryBuilderState } from '../../../../stores/editor/editor-state/element-editor-state/mapping/MappingExecutionQueryBuilderState.js';
-import type { QueryBuilderState } from '@finos/legend-query-builder';
+import {
+  ExecutionPlanViewer,
+  type QueryBuilderState,
+} from '@finos/legend-query-builder';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 interface ClassMappingSelectOption {
   label: string;
@@ -167,12 +169,12 @@ export const ClassMappingSelectorModal = observer(
 
 export const getRelationalInputTestDataEditorLanguage = (
   type: RelationalInputType,
-): EDITOR_LANGUAGE => {
+): CODE_EDITOR_LANGUAGE => {
   switch (type) {
     case RelationalInputType.SQL:
-      return EDITOR_LANGUAGE.SQL;
+      return CODE_EDITOR_LANGUAGE.SQL;
     default:
-      return EDITOR_LANGUAGE.TEXT;
+      return CODE_EDITOR_LANGUAGE.TEXT;
   }
 };
 
@@ -385,10 +387,10 @@ const MappingExecutionQueryEditor = observer(
         {!isStubbed_RawLambda(queryState.query) && (
           <PanelContent>
             <div className="mapping-execution-builder__query-panel__query">
-              <TextInputEditor
+              <CodeEditor
                 inputValue={queryState.lambdaString}
                 isReadOnly={true}
-                language={EDITOR_LANGUAGE.PURE}
+                language={CODE_EDITOR_LANGUAGE.PURE}
                 showMiniMap={false}
               />
             </div>
@@ -434,8 +436,8 @@ export const MappingExecutionObjectInputDataBuilder = observer(
 
     return (
       <div className="panel__content mapping-execution-builder__input-data-panel__content">
-        <TextInputEditor
-          language={EDITOR_LANGUAGE.JSON}
+        <CodeEditor
+          language={CODE_EDITOR_LANGUAGE.JSON}
           inputValue={inputDataState.inputData.data}
           updateInput={updateInput}
         />
@@ -454,8 +456,8 @@ export const MappingExecutionFlatDataInputDataBuilder = observer(
 
     return (
       <div className="panel__content mapping-execution-builder__input-data-panel__content">
-        <TextInputEditor
-          language={EDITOR_LANGUAGE.TEXT}
+        <CodeEditor
+          language={CODE_EDITOR_LANGUAGE.TEXT}
           inputValue={inputDataState.inputData.data}
           updateInput={updateInput}
         />
@@ -477,7 +479,7 @@ export const MappingExecutionRelationalInputDataBuilder = observer(
 
     return (
       <div className="panel__content mapping-execution-builder__input-data-panel__content">
-        <TextInputEditor
+        <CodeEditor
           language={getRelationalInputTestDataEditorLanguage(
             inputDataState.inputData.inputType,
           )}
@@ -853,10 +855,10 @@ export const MappingExecutionBuilder = observer(
                   </div>
                 </div>
                 <div className="panel__content mapping-execution-builder__result-panel__content">
-                  <TextInputEditor
+                  <CodeEditor
                     inputValue={executionResultText ?? ''}
                     isReadOnly={true}
-                    language={EDITOR_LANGUAGE.JSON}
+                    language={CODE_EDITOR_LANGUAGE.JSON}
                   />
                 </div>
               </div>

package/src/components/editor/edit-panel/mapping-editor/MappingTestEditor.tsx

@@ -17,7 +17,7 @@
 import { useState, useEffect, useCallback } from 'react';
 import { observer } from 'mobx-react-lite';
 import {
-  type DEPRECATED_MappingTestState,
+  type DEPRECATED__MappingTestState as DEPRECATED__MappingTestState,
   MAPPING_TEST_EDITOR_TAB_TYPE,
   TEST_RESULT,
   MappingTestObjectInputDataState,
@@ -59,12 +59,10 @@ import {
   assertErrorThrown,
 } from '@finos/legend-shared';
 import {
-  EDITOR_LANGUAGE,
+  CODE_EDITOR_LANGUAGE,
   useApplicationStore,
   ActionAlertActionType,
   ActionAlertType,
-  ExecutionPlanViewer,
-  TextInputEditor,
 } from '@finos/legend-application';
 import {
   ClassMappingSelectorModal,
@@ -94,11 +92,13 @@ import {
 import {
   type QueryBuilderState,
   QueryBuilderTextEditorMode,
+  ExecutionPlanViewer,
 } from '@finos/legend-query-builder';
 import { MappingExecutionQueryBuilderState } from '../../../../stores/editor/editor-state/element-editor-state/mapping/MappingExecutionQueryBuilderState.js';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const MappingTestQueryEditor = observer(
-  (props: { testState: DEPRECATED_MappingTestState; isReadOnly: boolean }) => {
+  (props: { testState: DEPRECATED__MappingTestState; isReadOnly: boolean }) => {
     const { testState, isReadOnly } = props;
     const queryState = testState.queryState;
     const editorStore = useEditorStore();
@@ -309,10 +309,10 @@ const MappingTestQueryEditor = observer(
                 editWithQueryBuilder()();
               }}
             >
-              <TextInputEditor
+              <CodeEditor
                 inputValue={queryState.lambdaString}
                 isReadOnly={true}
-                language={EDITOR_LANGUAGE.PURE}
+                language={CODE_EDITOR_LANGUAGE.PURE}
                 showMiniMap={false}
               />
             </div>
@@ -361,8 +361,8 @@ export const MappingTestObjectInputDataBuilder = observer(
 
     return (
       <div className="panel__content mapping-test-editor__input-data-panel__content">
-        <TextInputEditor
-          language={EDITOR_LANGUAGE.JSON}
+        <CodeEditor
+          language={CODE_EDITOR_LANGUAGE.JSON}
           inputValue={inputDataState.data}
           isReadOnly={isReadOnly}
           updateInput={updateInput}
@@ -385,8 +385,8 @@ export const MappingTestFlatDataInputDataBuilder = observer(
 
     return (
       <div className="panel__content mapping-test-editor__input-data-panel__content">
-        <TextInputEditor
-          language={EDITOR_LANGUAGE.TEXT}
+        <CodeEditor
+          language={CODE_EDITOR_LANGUAGE.TEXT}
           inputValue={inputDataState.inputData.data}
           isReadOnly={isReadOnly}
           updateInput={updateInput}
@@ -412,7 +412,7 @@ export const MappingTestRelationalInputDataBuilder = observer(
 
     return (
       <div className="panel__content mapping-test-editor__input-data-panel__content">
-        <TextInputEditor
+        <CodeEditor
           language={getRelationalInputTestDataEditorLanguage(
             inputDataState.inputData.inputType,
           )}
@@ -468,7 +468,7 @@ const RelationalMappingTestInputDataTypeSelector = observer(
 );
 
 export const MappingTestInputDataBuilder = observer(
-  (props: { testState: DEPRECATED_MappingTestState; isReadOnly: boolean }) => {
+  (props: { testState: DEPRECATED__MappingTestState; isReadOnly: boolean }) => {
     const { testState, isReadOnly } = props;
     const inputDataState = testState.inputDataState;
     const editorStore = useEditorStore();
@@ -573,7 +573,7 @@ export const MappingTestInputDataBuilder = observer(
 
 export const MappingTestExpectedOutputAssertionBuilder = observer(
   (props: {
-    testState: DEPRECATED_MappingTestState;
+    testState: DEPRECATED__MappingTestState;
     assertionState: MappingTestExpectedOutputAssertionState;
     isReadOnly: boolean;
   }) => {
@@ -638,11 +638,11 @@ export const MappingTestExpectedOutputAssertionBuilder = observer(
               <ErrorIcon />
             </div>
           )}
-          <TextInputEditor
+          <CodeEditor
             inputValue={assertionState.expectedResult}
             updateInput={updateExpectedResult}
             isReadOnly={isReadOnly}
-            language={EDITOR_LANGUAGE.JSON}
+            language={CODE_EDITOR_LANGUAGE.JSON}
           />
         </div>
       </div>
@@ -651,7 +651,7 @@ export const MappingTestExpectedOutputAssertionBuilder = observer(
 );
 
 export const MappingTestAssertionBuilder = observer(
-  (props: { testState: DEPRECATED_MappingTestState; isReadOnly: boolean }) => {
+  (props: { testState: DEPRECATED__MappingTestState; isReadOnly: boolean }) => {
     const { testState, isReadOnly } = props;
     const assertionState = testState.assertionState;
 
@@ -669,7 +669,7 @@ export const MappingTestAssertionBuilder = observer(
 );
 
 export const MappingTestBuilder = observer(
-  (props: { testState: DEPRECATED_MappingTestState; isReadOnly: boolean }) => {
+  (props: { testState: DEPRECATED__MappingTestState; isReadOnly: boolean }) => {
     const { testState, isReadOnly } = props;
     const applicationStore = useApplicationStore();
 
@@ -720,7 +720,7 @@ export const MappingTestBuilder = observer(
 );
 
 export const MappingTestEditor = observer(
-  (props: { testState: DEPRECATED_MappingTestState; isReadOnly: boolean }) => {
+  (props: { testState: DEPRECATED__MappingTestState; isReadOnly: boolean }) => {
     const { testState, isReadOnly } = props;
     const applicationStore = useApplicationStore();
     const selectedTab = testState.selectedTab;

package/src/components/editor/edit-panel/mapping-editor/MappingTestsExplorer.tsx

@@ -18,7 +18,7 @@ import { useState, useCallback, forwardRef } from 'react';
 import { observer } from 'mobx-react-lite';
 import { useDrop } from 'react-dnd';
 import {
-  type DEPRECATED_MappingTestState,
+  type DEPRECATED__MappingTestState,
   MAPPING_TEST_EDITOR_TAB_TYPE,
   TEST_RESULT,
 } from '../../../../stores/editor/editor-state/element-editor-state/mapping/MappingTestState.js';
@@ -61,7 +61,7 @@ export const MappingTestExplorerContextMenu = observer(
   forwardRef<
     HTMLDivElement,
     {
-      mappingTestState?: DEPRECATED_MappingTestState;
+      mappingTestState?: DEPRECATED__MappingTestState;
       showCreateNewTestModal?: () => void;
       isReadOnly: boolean;
     }
@@ -132,7 +132,7 @@ export const MappingTestExplorerContextMenu = observer(
 );
 
 export const MappingTestStatusIndicator: React.FC<{
-  testState: DEPRECATED_MappingTestState;
+  testState: DEPRECATED__MappingTestState;
 }> = (props) => {
   const { testState } = props;
   if (testState.isSkipped) {
@@ -194,7 +194,7 @@ export const MappingTestStatusIndicator: React.FC<{
 };
 
 export const MappingTestExplorer = observer(
-  (props: { testState: DEPRECATED_MappingTestState; isReadOnly: boolean }) => {
+  (props: { testState: DEPRECATED__MappingTestState; isReadOnly: boolean }) => {
     const { isReadOnly, testState } = props;
     const editorStore = useEditorStore();
     const applicationStore = useApplicationStore();

package/src/components/editor/edit-panel/project-configuration-editor/ProjectConfigurationEditor.tsx

@@ -45,15 +45,16 @@ import { useEditorStore } from '../../EditorStoreProvider.js';
 import {
   ActionAlertActionType,
   ActionAlertType,
-  DocumentationLink,
-  DocumentationPreview,
-  LEGEND_APPLICATION_DOCUMENTATION_KEY,
   shouldDisplayVirtualAssistantDocumentationEntry,
   useApplicationStore,
 } from '@finos/legend-application';
 import { ProjectDependencyEditor } from './ProjectDependencyEditor.js';
 import type { ProjectData } from '@finos/legend-server-depot';
 import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../../../application/LegendStudioDocumentation.js';
+import {
+  DocumentationLink,
+  DocumentationPreview,
+} from '@finos/legend-application/components';
 
 const ProjectStructureEditor = observer(
   (props: { projectConfig: ProjectConfiguration; isReadOnly: boolean }) => {
@@ -309,7 +310,7 @@ const ProjectPlatformVersionEditor = observer(
               artifacts; in this case, you can freeze the platform dependencies'
               versions"
             documentationKey={
-              LEGEND_APPLICATION_DOCUMENTATION_KEY.QUESTION_WHEN_TO_CONFIGURE_PLATFORM_VERSIONS
+              LEGEND_STUDIO_DOCUMENTATION_KEY.QUESTION_WHEN_TO_CONFIGURE_PLATFORM_VERSIONS
             }
           />
           <div className="platform-configurations-editor__dependencies">

package/src/components/editor/edit-panel/service-editor/ServiceEditor.tsx

@@ -48,7 +48,6 @@ import {
   service_updateOwner,
 } from '../../../../stores/editor/shared/modifier/DSL_Service_GraphModifierHelper.js';
 import {
-  LEGEND_APPLICATION_DOCUMENTATION_KEY,
   useApplicationNavigationContext,
   useApplicationStore,
 } from '@finos/legend-application';
@@ -56,6 +55,7 @@ import { validate_ServicePattern } from '@finos/legend-graph';
 import { LEGEND_STUDIO_APPLICATION_NAVIGATION_CONTEXT_KEY } from '../../../../application/LegendStudioApplicationNavigationContext.js';
 import { ServiceTestableEditor } from './testable/ServiceTestableEditor.js';
 import { flowResult } from 'mobx';
+import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../../../application/LegendStudioDocumentation.js';
 
 type UserOption = { label: string; value: string };
 
@@ -481,7 +481,7 @@ export const ServiceEditor = observer(() => {
   );
   const seeDocumentation = (): void =>
     applicationStore.assistantService.openDocumentationEntry(
-      LEGEND_APPLICATION_DOCUMENTATION_KEY.QUESTION_HOW_TO_WRITE_A_SERVICE_TEST,
+      LEGEND_STUDIO_DOCUMENTATION_KEY.QUESTION_HOW_TO_WRITE_A_SERVICE_TEST,
     );
 
   useApplicationNavigationContext(

package/src/components/editor/edit-panel/service-editor/ServiceExecutionQueryEditor.tsx

@@ -49,9 +49,7 @@ import {
 import { flowResult } from 'mobx';
 import { useEditorStore } from '../../EditorStoreProvider.js';
 import {
-  EDITOR_LANGUAGE,
-  ExecutionPlanViewer,
-  TextInputEditor,
+  CODE_EDITOR_LANGUAGE,
   useApplicationStore,
 } from '@finos/legend-application';
 import {
@@ -65,12 +63,14 @@ import {
   ServiceQueryBuilderState,
   LambdaParameterValuesEditor,
   QueryBuilderTextEditorMode,
+  ExecutionPlanViewer,
 } from '@finos/legend-query-builder';
 import { ProjectViewerEditorMode } from '../../../../stores/project-view/ProjectViewerEditorMode.js';
 import { useLegendStudioApplicationStore } from '../../../LegendStudioBaseStoreProvider.js';
 import { WorkspaceType } from '@finos/legend-server-sdlc';
 import { SNAPSHOT_VERSION_ALIAS } from '@finos/legend-server-depot';
 import type { ProjectGAVCoordinates } from '@finos/legend-storage';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const ServiceExecutionResultViewer = observer(
   (props: { executionState: ServicePureExecutionState }) => {
@@ -93,10 +93,10 @@ const ServiceExecutionResultViewer = observer(
         <Modal darkMode={true} className="editor-modal">
           <ModalHeader title="Execution Result" />
           <ModalBody>
-            <TextInputEditor
+            <CodeEditor
               inputValue={executionResultText ?? ''}
               isReadOnly={true}
-              language={EDITOR_LANGUAGE.JSON}
+              language={CODE_EDITOR_LANGUAGE.JSON}
               showMiniMap={true}
             />
           </ModalBody>
@@ -222,10 +222,10 @@ const ServiceExecutionQueryImporter = observer(
               isLoading={queryState.loadQueryInfoState.isInProgress}
             />
             {queryState.selectedQueryInfo && (
-              <TextInputEditor
+              <CodeEditor
                 inputValue={queryState.selectedQueryInfo.content}
                 isReadOnly={true}
-                language={EDITOR_LANGUAGE.PURE}
+                language={CODE_EDITOR_LANGUAGE.PURE}
                 showMiniMap={false}
                 hideGutter={true}
               />
@@ -601,10 +601,10 @@ export const ServiceExecutionQueryEditor = observer(
               editWithQueryBuilder()();
             }}
           >
-            <TextInputEditor
+            <CodeEditor
               inputValue={queryState.lambdaString}
               isReadOnly={true}
-              language={EDITOR_LANGUAGE.PURE}
+              language={CODE_EDITOR_LANGUAGE.PURE}
               showMiniMap={true}
             />
           </div>

package/src/components/editor/edit-panel/service-editor/testable/ServiceTestDataEditor.tsx

@@ -55,12 +55,12 @@ import {
   ActionAlertActionType,
   ActionAlertType,
   buildElementOption,
-  LEGEND_APPLICATION_DOCUMENTATION_KEY,
   useApplicationStore,
 } from '@finos/legend-application';
 import { prettyCONSTName } from '@finos/legend-shared';
 import type { DSL_Data_LegendStudioApplicationPlugin_Extension } from '../../../../../stores/extensions/DSL_Data_LegendStudioApplicationPlugin_Extension.js';
 import { useEditorStore } from '../../../EditorStoreProvider.js';
+import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../../../../application/LegendStudioDocumentation.js';
 
 export const ConnectionTestDataEditor = observer(
   (props: { connectionTestDataState: ConnectionTestDataState }) => {
@@ -431,7 +431,7 @@ export const ServiceTestDataEditor = observer(
     };
     const seeDocumentation = (): void =>
       applicationStore.assistantService.openDocumentationEntry(
-        LEGEND_APPLICATION_DOCUMENTATION_KEY.QUESTION_HOW_TO_WRITE_SERVICE_CONNECTION_TEST_DATA,
+        LEGEND_STUDIO_DOCUMENTATION_KEY.QUESTION_HOW_TO_WRITE_SERVICE_CONNECTION_TEST_DATA,
       );
     return (
       <div className="service-test-data-editor panel">

package/src/components/editor/edit-panel/testable/TestAssertionEditor.tsx

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { EDITOR_LANGUAGE, TextInputEditor } from '@finos/legend-application';
+import { CODE_EDITOR_LANGUAGE } from '@finos/legend-application';
 import {
   clsx,
   CompareIcon,
@@ -49,6 +49,7 @@ import { externalFormatData_setData } from '../../../../stores/editor/shared/mod
 import { TESTABLE_RESULT } from '../../../../stores/editor/sidebar-state/testable/GlobalTestRunnerState.js';
 import { JsonDiffView } from '../../../shared/DiffView.js';
 import { UnsupportedEditorPanel } from '../UnsupportedElementEditor.js';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const EqualToJsonAsssertionEditor = observer(
   (props: {
@@ -84,9 +85,9 @@ const EqualToJsonAsssertionEditor = observer(
         </div>
         <div className="equal-to-json-editor__content panel__content">
           <div className="equal-to-json-editor__content__data">
-            <TextInputEditor
+            <CodeEditor
               inputValue={assertion.expected.data}
-              language={EDITOR_LANGUAGE.JSON}
+              language={CODE_EDITOR_LANGUAGE.JSON}
               updateInput={(val: string): void => {
                 equalToJsonAssertionState.setExpectedValue(val);
               }}

package/src/components/editor/edit-panel/uml-editor/AssociationEditor.tsx

@@ -93,9 +93,9 @@ import {
   buildElementOption,
   useApplicationNavigationContext,
   type PackageableElementOption,
-  LEGEND_APPLICATION_DOCUMENTATION_KEY,
 } from '@finos/legend-application';
 import { LEGEND_STUDIO_APPLICATION_NAVIGATION_CONTEXT_KEY } from '../../../../application/LegendStudioApplicationNavigationContext.js';
+import { LEGEND_STUDIO_DOCUMENTATION_KEY } from '../../../../application/LegendStudioDocumentation.js';
 
 const AssociationPropertyBasicEditor = observer(
   (props: {
@@ -212,7 +212,7 @@ const AssociationPropertyBasicEditor = observer(
     };
     const seeErrorDocumentation = (): void =>
       editorStore.applicationStore.assistantService.openDocumentationEntry(
-        LEGEND_APPLICATION_DOCUMENTATION_KEY.QUESTION_WHY_DO_I_SEE_ERROR_WITH_ASSOCIATION_PROPERTY_TYPE,
+        LEGEND_STUDIO_DOCUMENTATION_KEY.QUESTION_WHY_DO_I_SEE_ERROR_WITH_ASSOCIATION_PROPERTY_TYPE,
       );
 
     return (

package/src/components/editor/side-bar/Explorer.tsx

@@ -106,10 +106,9 @@ import {
   getElementRootPackage,
 } from '@finos/legend-graph';
 import {
-  EDITOR_LANGUAGE,
+  CODE_EDITOR_LANGUAGE,
   getPackageableElementOptionFormatter,
   type PackageableElementOption,
-  TextInputEditor,
   useApplicationStore,
 } from '@finos/legend-application';
 import { PACKAGEABLE_ELEMENT_TYPE } from '../../../stores/editor/shared/ModelClassifierUtils.js';
@@ -124,6 +123,7 @@ import {
   CLASS_MOCK_DATA_GENERATION_FORMAT,
   createMockDataForClassWithFormat,
 } from '../../../stores/editor/shared/MockDataUtils.js';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const ElementRenamer = observer(() => {
   const editorStore = useEditorStore();
@@ -266,16 +266,16 @@ const GENERATION_DEFAULT_DEPTH = 100;
 const GENERATION_DEFAULT_FORMAT = CLASS_MOCK_DATA_GENERATION_FORMAT.JSON;
 const getMockDataEditorLanguage = (
   format: CLASS_MOCK_DATA_GENERATION_FORMAT,
-): EDITOR_LANGUAGE => {
+): CODE_EDITOR_LANGUAGE => {
   switch (format) {
     case CLASS_MOCK_DATA_GENERATION_FORMAT.JSON:
-      return EDITOR_LANGUAGE.JSON;
+      return CODE_EDITOR_LANGUAGE.JSON;
     case CLASS_MOCK_DATA_GENERATION_FORMAT.XML:
-      return EDITOR_LANGUAGE.XML;
+      return CODE_EDITOR_LANGUAGE.XML;
     case CLASS_MOCK_DATA_GENERATION_FORMAT.YAML:
-      return EDITOR_LANGUAGE.YAML;
+      return CODE_EDITOR_LANGUAGE.YAML;
     default:
-      return EDITOR_LANGUAGE.TEXT;
+      return CODE_EDITOR_LANGUAGE.TEXT;
   }
 };
 
@@ -416,7 +416,7 @@ const SampleDataGenerator = observer(() => {
           />
         </div>
         <div className="sample-data-generator__result">
-          <TextInputEditor
+          <CodeEditor
             inputValue={resultText}
             isReadOnly={true}
             hideGutter={true}

package/src/components/editor/side-bar/WorkflowManager.tsx

@@ -48,8 +48,7 @@ import { LEGEND_STUDIO_TEST_ID } from '../../../application/LegendStudioTesting.
 import { flowResult } from 'mobx';
 import { WorkflowJobStatus, WorkflowStatus } from '@finos/legend-server-sdlc';
 import {
-  EDITOR_LANGUAGE,
-  TextInputEditor,
+  CODE_EDITOR_LANGUAGE,
   useApplicationStore,
 } from '@finos/legend-application';
 import {
@@ -66,6 +65,7 @@ import {
   guaranteeType,
   isNonNullable,
 } from '@finos/legend-shared';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 const getWorkflowStatusIcon = (
   workflowStatus: WorkflowStatus,
@@ -238,10 +238,10 @@ const WorkflowJobLogsViewer = observer(
             </ModalHeaderActions>
           </ModalHeader>
           <ModalBody>
-            <TextInputEditor
+            <CodeEditor
               inputValue={logs}
               isReadOnly={true}
-              language={EDITOR_LANGUAGE.TEXT}
+              language={CODE_EDITOR_LANGUAGE.TEXT}
               showMiniMap={true}
             />
           </ModalBody>

package/src/components/editor/side-bar/testable/GlobalTestRunner.tsx

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { EDITOR_LANGUAGE, TextInputEditor } from '@finos/legend-application';
+import { CODE_EDITOR_LANGUAGE } from '@finos/legend-application';
 import {
   type TreeData,
   type TreeNodeContainerProps,
@@ -78,6 +78,7 @@ import { TextDiffView } from '../../../shared/DiffView.js';
 import { getElementTypeIcon } from '../../../shared/ElementIconUtils.js';
 import { UnsupportedEditorPanel } from '../../edit-panel/UnsupportedElementEditor.js';
 import { useEditorStore } from '../../EditorStoreProvider.js';
+import { CodeEditor } from '@finos/legend-lego/code-editor';
 
 export const getTestableResultIcon = (
   testableResult: TESTABLE_RESULT,
@@ -189,26 +190,26 @@ const TestFailViewer = observer(
           <ModalHeader title={id} />
           <ModalBody>
             {failure instanceof TestError && (
-              <TextInputEditor
+              <CodeEditor
                 inputValue={failure.error}
                 isReadOnly={true}
-                language={EDITOR_LANGUAGE.TEXT}
+                language={CODE_EDITOR_LANGUAGE.TEXT}
                 showMiniMap={true}
               />
             )}
             {failure instanceof EqualToJsonAssertFail && (
               <TextDiffView
-                language={EDITOR_LANGUAGE.JSON}
+                language={CODE_EDITOR_LANGUAGE.JSON}
                 from={failure.expected}
                 to={failure.actual}
               />
             )}
             {failure instanceof AssertFail &&
               !(failure instanceof EqualToJsonAssertFail) && (
-                <TextInputEditor
+                <CodeEditor
                   inputValue={failure.message ?? ''}
                   isReadOnly={true}
-                  language={EDITOR_LANGUAGE.TEXT}
+                  language={CODE_EDITOR_LANGUAGE.TEXT}
                 />
               )}
           </ModalBody>

package/src/components/extensions/Core_LegendStudioApplicationPlugin.tsx

@@ -21,6 +21,8 @@ import {
   type ContextualDocumentationEntry,
   type SettingConfigurationEntry,
   collectSettingConfigurationEntriesFromConfig,
+  type LegendApplicationSetup,
+  setupPureLanguageService,
 } from '@finos/legend-application';
 import packageJson from '../../../package.json';
 import { LEGEND_STUDIO_APPLICATION_NAVIGATION_CONTEXT_KEY } from '../../application/LegendStudioApplicationNavigationContext.js';
@@ -31,6 +33,7 @@ import {
 import { LegendStudioApplicationPlugin } from '../../stores/LegendStudioApplicationPlugin.js';
 import { LEGEND_STUDIO_COMMAND_CONFIG } from '../../application/LegendStudioCommand.js';
 import { LEGEND_STUDIO_SETTING_CONFIG } from '../../application/LegendStudioSetting.js';
+import type { LegendStudioApplicationStore } from '../../stores/LegendStudioBaseStore.js';
 
 export class Core_LegendStudioApplicationPlugin extends LegendStudioApplicationPlugin {
   static NAME = packageJson.extensions.applicationStudioPlugin;
@@ -39,6 +42,24 @@ export class Core_LegendStudioApplicationPlugin extends LegendStudioApplicationP
     super(Core_LegendStudioApplicationPlugin.NAME, packageJson.version);
   }
 
+  override getExtraApplicationSetups(): LegendApplicationSetup[] {
+    return [
+      async (applicationStore) => {
+        setupPureLanguageService(
+          // TODO: move these inside of `DSL_LegendStudioApplicationPlugin_Extension`
+          (applicationStore as LegendStudioApplicationStore).pluginManager
+            .getPureGraphManagerPlugins()
+            .flatMap((plugin) => plugin.getExtraPureGrammarKeywords?.() ?? []),
+          (applicationStore as LegendStudioApplicationStore).pluginManager
+            .getPureGraphManagerPlugins()
+            .flatMap(
+              (plugin) => plugin.getExtraPureGrammarParserNames?.() ?? [],
+            ),
+        );
+      },
+    ];
+  }
+
   override getExtraKeyedCommandConfigEntries(): KeyedCommandConfigEntry[] {
     return collectKeyedCommandConfigEntriesFromConfig(
       LEGEND_STUDIO_COMMAND_CONFIG,