@finos/legend-application-studio 22.6.0 → 23.0.1

package/src/stores/editor-state/element-editor-state/RuntimeEditorState.ts

@@ -1,935 +0,0 @@
-/**
- * Copyright (c) 2020-present, Goldman Sachs
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-import { computed, action, observable, makeObservable } from 'mobx';
-import type { EditorStore } from '../../EditorStore.js';
-import {
-  guaranteeType,
-  uuid,
-  isNonNullable,
-  UnsupportedOperationError,
-  uniq,
-  addUniqueEntry,
-  assertErrorThrown,
-  filterByType,
-} from '@finos/legend-shared';
-import { ElementEditorState } from './ElementEditorState.js';
-import type { RuntimeExplorerTreeNodeData } from '../../shared/TreeUtils.js';
-import type { TreeData } from '@finos/legend-art';
-import { ConnectionEditorState } from './connection/ConnectionEditorState.js';
-import { getMappingElementSource } from './mapping/MappingEditorState.js';
-import {
-  type PackageableElement,
-  type Mapping,
-  type Connection,
-  type PackageableConnection,
-  type SetImplementation,
-  type PackageableElementReference,
-  getAllClassMappings,
-  PackageableRuntime,
-  Runtime,
-  EngineRuntime,
-  IdentifiedConnection,
-  RuntimePointer,
-  ConnectionPointer,
-  Store,
-  ModelStore,
-  PureModelConnection,
-  JsonModelConnection,
-  XmlModelConnection,
-  Class,
-  RootFlatDataRecordType,
-  FlatData,
-  FlatDataConnection,
-  PackageableElementExplicitReference,
-  Database,
-  TableAlias,
-  DatabaseType,
-  RelationalDatabaseConnection,
-  StaticDatasourceSpecification,
-  DefaultH2AuthenticationStrategy,
-  ModelChainConnection,
-  isStubbed_StoreConnections,
-  getAllIdentifiedConnections,
-  generateIdentifiedConnectionId,
-} from '@finos/legend-graph';
-import type { DSL_Mapping_LegendStudioApplicationPlugin_Extension } from '../../DSL_Mapping_LegendStudioApplicationPlugin_Extension.js';
-import { packageableElementReference_setValue } from '../../shared/modifier/DomainGraphModifierHelper.js';
-import {
-  runtime_addIdentifiedConnection,
-  runtime_addMapping,
-  runtime_addUniqueStoreConnectionsForStore,
-  runtime_deleteIdentifiedConnection,
-  runtime_deleteMapping,
-} from '../../shared/modifier/DSL_Mapping_GraphModifierHelper.js';
-
-export const getClassMappingStore = (
-  setImplementation: SetImplementation,
-  editorStore: EditorStore,
-): Store | undefined => {
-  const sourceElement = getMappingElementSource(
-    setImplementation,
-    editorStore.pluginManager.getApplicationPlugins(),
-  );
-  if (sourceElement instanceof Class) {
-    return ModelStore.INSTANCE;
-  } else if (sourceElement instanceof RootFlatDataRecordType) {
-    return sourceElement._OWNER._OWNER;
-  } else if (sourceElement instanceof TableAlias) {
-    return sourceElement.relation.ownerReference.value;
-  }
-  if (sourceElement) {
-    const extraInstanceSetImplementationStoreExtractors =
-      editorStore.pluginManager
-        .getApplicationPlugins()
-        .flatMap(
-          (plugin) =>
-            (
-              plugin as DSL_Mapping_LegendStudioApplicationPlugin_Extension
-            ).getExtraInstanceSetImplementationStoreExtractors?.() ?? [],
-        );
-    for (const extractor of extraInstanceSetImplementationStoreExtractors) {
-      const instanceSetImplementationStore = extractor(sourceElement);
-      if (instanceSetImplementationStore) {
-        return instanceSetImplementationStore;
-      }
-    }
-    throw new UnsupportedOperationError(
-      `Can't extract store for class mapping: no compatible extractor available from plugins`,
-      setImplementation,
-    );
-  }
-  return undefined;
-};
-
-const getStoresFromMappings = (
-  mappings: Mapping[],
-  editorStore: EditorStore,
-): Store[] =>
-  uniq(
-    mappings.flatMap((mapping) =>
-      getAllClassMappings(mapping)
-        .map((setImplementation) =>
-          getClassMappingStore(setImplementation, editorStore),
-        )
-        .filter(isNonNullable),
-    ),
-  );
-
-/**
- * Since model connection are pretty tedious to add, we automatically create new connections for mapped classes
- * as we add/change mapping for the runtime
- *
- * NOTE: as of now, to be safe and simple, we will not remove the connections as we remove the mapping from the runtime
- */
-export const decorateRuntimeWithNewMapping = (
-  runtime: Runtime,
-  mapping: Mapping,
-  editorStore: EditorStore,
-): void => {
-  const runtimeValue =
-    runtime instanceof RuntimePointer
-      ? runtime.packageableRuntime.value.runtimeValue
-      : guaranteeType(runtime, EngineRuntime);
-  getStoresFromMappings([mapping], editorStore).forEach((store) =>
-    runtime_addUniqueStoreConnectionsForStore(
-      runtimeValue,
-      store,
-      editorStore.changeDetectionState.observerContext,
-    ),
-  );
-  const sourceClasses: Class[] = [];
-  mapping.classMappings.forEach((classMapping) => {
-    const mappingSource = getMappingElementSource(
-      classMapping,
-      editorStore.pluginManager.getApplicationPlugins(),
-    );
-    if (mappingSource instanceof Class) {
-      addUniqueEntry(sourceClasses, mappingSource);
-    }
-  });
-  let classesSpecifiedInModelConnections: Class[] = [];
-  runtimeValue.connections.forEach((storeConnections) => {
-    if (storeConnections.store.value instanceof ModelStore) {
-      classesSpecifiedInModelConnections = storeConnections.storeConnections
-        .filter(
-          (identifiedConnection) =>
-            identifiedConnection.connection instanceof JsonModelConnection ||
-            identifiedConnection.connection instanceof XmlModelConnection,
-        )
-        .map(
-          (identifiedConnection) =>
-            (
-              identifiedConnection.connection as
-                | JsonModelConnection
-                | XmlModelConnection
-            ).class.value,
-        );
-    }
-  });
-  sourceClasses
-    .filter((_class) => !classesSpecifiedInModelConnections.includes(_class))
-    .forEach((_class) =>
-      runtime_addIdentifiedConnection(
-        runtimeValue,
-        new IdentifiedConnection(
-          generateIdentifiedConnectionId(runtimeValue),
-          new JsonModelConnection(
-            PackageableElementExplicitReference.create(ModelStore.INSTANCE),
-            PackageableElementExplicitReference.create(_class),
-          ),
-        ),
-        editorStore.changeDetectionState.observerContext,
-      ),
-    );
-};
-
-export const isConnectionForStore = (
-  connection: Connection,
-  store: Store,
-): boolean => {
-  const connectionValue =
-    connection instanceof ConnectionPointer
-      ? connection.packageableConnection.value.connectionValue
-      : connection;
-  if (connectionValue instanceof PureModelConnection) {
-    return store instanceof ModelStore;
-  }
-  return connectionValue.store.value === store;
-};
-
-export const isConnectionForModelStoreWithClass = (
-  connection: Connection,
-  _class: Class,
-): boolean => {
-  const connectionValue =
-    connection instanceof ConnectionPointer
-      ? connection.packageableConnection.value.connectionValue
-      : connection;
-  if (
-    connectionValue instanceof JsonModelConnection ||
-    connectionValue instanceof XmlModelConnection
-  ) {
-    return connectionValue.class.value === _class;
-  } else if (connectionValue instanceof ModelChainConnection) {
-    return connectionValue.mappings.some((mapping) =>
-      mapping.value.classMappings.some(
-        (classMapping) => classMapping.class.value === _class,
-      ),
-    );
-  }
-  return false;
-};
-
-export const getConnectionsForModelStoreWithClass = (
-  connections: Connection[],
-  _class: Class,
-): Connection[] =>
-  uniq(
-    connections.filter((connection) =>
-      isConnectionForModelStoreWithClass(connection, _class),
-    ),
-  );
-
-/**
- * Derive the stores from the runtime's mappings and then partition the list of runtime's connections based on the store
- */
-export const getRuntimeExplorerTreeData = (
-  runtime: Runtime,
-  editorStore: EditorStore,
-): TreeData<RuntimeExplorerTreeNodeData> => {
-  const runtimeValue =
-    runtime instanceof RuntimePointer
-      ? runtime.packageableRuntime.value.runtimeValue
-      : guaranteeType(runtime, EngineRuntime);
-  const rootIds: string[] = [];
-  const nodes = new Map<string, RuntimeExplorerTreeNodeData>();
-  const allSourceClassesFromMappings = uniq(
-    runtimeValue.mappings.flatMap((mapping) =>
-      getAllClassMappings(mapping.value)
-        .map((setImplementation) =>
-          getMappingElementSource(
-            setImplementation,
-            editorStore.pluginManager.getApplicationPlugins(),
-          ),
-        )
-        .filter(filterByType(Class)),
-    ),
-  );
-  // runtime (root)
-  const runtimeNode = {
-    data: runtimeValue,
-    id: 'runtime',
-    label:
-      runtime instanceof RuntimePointer
-        ? runtime.packageableRuntime.value.name
-        : '(custom)',
-    isOpen: true,
-    childrenIds: [] as string[],
-  };
-  nodes.set(runtimeNode.id, runtimeNode);
-  addUniqueEntry(rootIds, runtimeNode.id);
-  // stores (1st level)
-  runtimeValue.connections
-    .map((storeConnections) => storeConnections.store.value)
-    .sort((a, b) => a.name.localeCompare(b.name))
-    .forEach((store) => {
-      const childrenIds: string[] = [];
-      if (store instanceof ModelStore) {
-        // expand ModelStore to show source classes
-        const classes = uniq(
-          getAllIdentifiedConnections(runtimeValue)
-            .filter(
-              (identifiedConnection) =>
-                identifiedConnection.connection.store.value instanceof
-                ModelStore,
-            )
-            .map((identifiedConnection) => {
-              const connectionValue =
-                identifiedConnection.connection instanceof ConnectionPointer
-                  ? identifiedConnection.connection.packageableConnection.value
-                      .connectionValue
-                  : identifiedConnection.connection;
-              if (
-                connectionValue instanceof JsonModelConnection ||
-                connectionValue instanceof XmlModelConnection
-              ) {
-                return connectionValue.class.value;
-              }
-              return undefined;
-            })
-            .concat(allSourceClassesFromMappings),
-        ); // make sure we add classes (from mappings) that we expect to have connections for
-        // classes (2nd level) - only for `ModelStore`
-        classes.filter(isNonNullable).forEach((_class) => {
-          const classNode = {
-            data: _class,
-            id: _class.path,
-            label: _class.name,
-          };
-          nodes.set(classNode.id, classNode);
-          addUniqueEntry(childrenIds, classNode.id);
-        });
-      }
-      const storeNode = {
-        data: store,
-        id: store.path,
-        label: store.name,
-        isOpen: true,
-        childrenIds,
-      };
-      addUniqueEntry(runtimeNode.childrenIds, storeNode.id);
-      nodes.set(storeNode.id, storeNode);
-    });
-  return { rootIds, nodes };
-};
-
-export abstract class RuntimeEditorTabState {
-  readonly uuid = uuid();
-  editorStore: EditorStore;
-  runtimeEditorState: RuntimeEditorState;
-
-  constructor(
-    editorStore: EditorStore,
-    runtimeEditorState: RuntimeEditorState,
-  ) {
-    this.editorStore = editorStore;
-    this.runtimeEditorState = runtimeEditorState;
-  }
-}
-
-export class IdentifiedConnectionEditorState {
-  editorStore: EditorStore;
-  idenfitiedConnection: IdentifiedConnection;
-  connectionEditorState: ConnectionEditorState;
-
-  constructor(
-    editorStore: EditorStore,
-    idenfitiedConnection: IdentifiedConnection,
-  ) {
-    this.editorStore = editorStore;
-    this.idenfitiedConnection = idenfitiedConnection;
-    this.connectionEditorState = new ConnectionEditorState(
-      this.editorStore,
-      idenfitiedConnection.connection,
-    );
-  }
-}
-
-export abstract class IdentifiedConnectionsEditorTabState extends RuntimeEditorTabState {
-  identifiedConnectionEditorState?: IdentifiedConnectionEditorState | undefined;
-
-  constructor(
-    editorStore: EditorStore,
-    runtimeEditorState: RuntimeEditorState,
-  ) {
-    super(editorStore, runtimeEditorState);
-
-    makeObservable(this, {
-      identifiedConnectionEditorState: observable,
-      openIdentifiedConnection: action,
-      addIdentifiedConnection: action,
-    });
-  }
-
-  openIdentifiedConnection(identifiedConnection: IdentifiedConnection): void {
-    this.identifiedConnectionEditorState = new IdentifiedConnectionEditorState(
-      this.editorStore,
-      identifiedConnection,
-    );
-  }
-
-  addIdentifiedConnection(packageableConnection?: PackageableConnection): void {
-    let newConnection: Connection;
-    if (
-      packageableConnection &&
-      this.packageableConnections.includes(packageableConnection)
-    ) {
-      newConnection = new ConnectionPointer(
-        PackageableElementExplicitReference.create(packageableConnection),
-      );
-    } else if (this.packageableConnections.length) {
-      newConnection = new ConnectionPointer(
-        PackageableElementExplicitReference.create(
-          this.packageableConnections[0] as PackageableConnection,
-        ),
-      );
-    } else {
-      try {
-        newConnection = this.createDefaultConnection();
-      } catch (error) {
-        assertErrorThrown(error);
-        this.editorStore.applicationStore.notificationService.notifyWarning(
-          error.message,
-        );
-        return;
-      }
-    }
-    const newIdentifiedConnection = new IdentifiedConnection(
-      generateIdentifiedConnectionId(this.runtimeEditorState.runtimeValue),
-      newConnection,
-    );
-    runtime_addIdentifiedConnection(
-      this.runtimeEditorState.runtimeValue,
-      newIdentifiedConnection,
-      this.editorStore.changeDetectionState.observerContext,
-    );
-    this.openIdentifiedConnection(newIdentifiedConnection);
-  }
-
-  getConnectionEditorState(): ConnectionEditorState | undefined {
-    return this.identifiedConnectionEditorState?.connectionEditorState
-      .connection instanceof ConnectionPointer
-      ? new ConnectionEditorState(
-          this.editorStore,
-          this.identifiedConnectionEditorState.connectionEditorState.connection.packageableConnection.value.connectionValue,
-        )
-      : this.identifiedConnectionEditorState?.connectionEditorState;
-  }
-
-  abstract get identifiedConnections(): IdentifiedConnection[];
-  abstract get packageableConnections(): PackageableConnection[];
-  abstract createDefaultConnection(): Connection;
-  abstract deleteIdentifiedConnection(
-    identifiedConnection: IdentifiedConnection,
-  ): void;
-}
-
-export class IdentifiedConnectionsPerStoreEditorTabState extends IdentifiedConnectionsEditorTabState {
-  store: Store;
-
-  constructor(
-    editorStore: EditorStore,
-    runtimeEditorState: RuntimeEditorState,
-    store: Store,
-  ) {
-    super(editorStore, runtimeEditorState);
-
-    makeObservable(this, {
-      store: observable,
-      identifiedConnections: computed,
-      packageableConnections: computed,
-      deleteIdentifiedConnection: action,
-    });
-
-    this.store = store;
-    if (this.identifiedConnections.length) {
-      this.openIdentifiedConnection(
-        this.identifiedConnections[0] as IdentifiedConnection,
-      );
-    }
-  }
-
-  get identifiedConnections(): IdentifiedConnection[] {
-    return (
-      this.runtimeEditorState.runtimeValue.connections.find(
-        (storeConnections) => storeConnections.store.value === this.store,
-      )?.storeConnections ?? []
-    );
-  }
-
-  get packageableConnections(): PackageableConnection[] {
-    return this.editorStore.graphManagerState.graph.ownConnections.filter(
-      (connection) =>
-        isConnectionForStore(connection.connectionValue, this.store),
-    );
-  }
-
-  createDefaultConnection(): Connection {
-    if (this.store instanceof FlatData) {
-      return new FlatDataConnection(
-        PackageableElementExplicitReference.create(this.store),
-      );
-    } else if (this.store instanceof Database) {
-      return new RelationalDatabaseConnection(
-        PackageableElementExplicitReference.create(this.store),
-        DatabaseType.H2,
-        new StaticDatasourceSpecification('host', 80, 'db'),
-        new DefaultH2AuthenticationStrategy(),
-      );
-    }
-    const extraDefaultConnectionValueBuilders = this.editorStore.pluginManager
-      .getApplicationPlugins()
-      .flatMap(
-        (plugin) =>
-          (
-            plugin as DSL_Mapping_LegendStudioApplicationPlugin_Extension
-          ).getExtraDefaultConnectionValueBuilders?.() ?? [],
-      );
-
-    for (const builder of extraDefaultConnectionValueBuilders) {
-      const defaultConnection = builder(this.store);
-      if (defaultConnection) {
-        return defaultConnection;
-      }
-    }
-
-    throw new UnsupportedOperationError(
-      `Can't build default connection for the specified store: no compatible builder available from plugins`,
-      this.store,
-    );
-  }
-
-  deleteIdentifiedConnection(identifiedConnection: IdentifiedConnection): void {
-    runtime_deleteIdentifiedConnection(
-      this.runtimeEditorState.runtimeValue,
-      identifiedConnection,
-    );
-    if (
-      identifiedConnection.connection ===
-      this.identifiedConnectionEditorState?.connectionEditorState.connection
-    ) {
-      this.identifiedConnectionEditorState = undefined;
-    }
-    this.runtimeEditorState.reprocessRuntimeExplorerTree();
-    if (!this.identifiedConnections.length) {
-      const stores = getStoresFromMappings(
-        this.runtimeEditorState.runtimeValue.mappings.map(
-          (mapping) => mapping.value,
-        ),
-        this.editorStore,
-      );
-      if (!stores.includes(this.store)) {
-        this.runtimeEditorState.openTabFor(
-          this.runtimeEditorState.runtimeValue,
-        );
-      }
-    }
-  }
-}
-
-export class IdentifiedConnectionsPerClassEditorTabState extends IdentifiedConnectionsEditorTabState {
-  class: Class;
-
-  constructor(
-    editorStore: EditorStore,
-    runtimeEditorState: RuntimeEditorState,
-    _class: Class,
-  ) {
-    super(editorStore, runtimeEditorState);
-
-    makeObservable(this, {
-      class: observable,
-      identifiedConnections: computed,
-      packageableConnections: computed,
-      deleteIdentifiedConnection: action,
-    });
-
-    this.class = _class;
-    if (this.identifiedConnections.length) {
-      this.openIdentifiedConnection(
-        this.identifiedConnections[0] as IdentifiedConnection,
-      );
-    }
-  }
-
-  get identifiedConnections(): IdentifiedConnection[] {
-    return getAllIdentifiedConnections(
-      this.runtimeEditorState.runtimeValue,
-    ).filter((identifiedConnection) =>
-      isConnectionForModelStoreWithClass(
-        identifiedConnection.connection,
-        this.class,
-      ),
-    );
-  }
-
-  get packageableConnections(): PackageableConnection[] {
-    return this.editorStore.graphManagerState.graph.ownConnections.filter(
-      (connection) =>
-        isConnectionForModelStoreWithClass(
-          connection.connectionValue,
-          this.class,
-        ),
-    );
-  }
-
-  createDefaultConnection(): Connection {
-    return new JsonModelConnection(
-      PackageableElementExplicitReference.create(ModelStore.INSTANCE),
-      PackageableElementExplicitReference.create(this.class),
-    );
-  }
-
-  deleteIdentifiedConnection(identifiedConnection: IdentifiedConnection): void {
-    runtime_deleteIdentifiedConnection(
-      this.runtimeEditorState.runtimeValue,
-      identifiedConnection,
-    );
-    if (
-      identifiedConnection.connection ===
-      this.identifiedConnectionEditorState?.connectionEditorState.connection
-    ) {
-      this.identifiedConnectionEditorState = undefined;
-    }
-    this.runtimeEditorState.reprocessRuntimeExplorerTree();
-    if (!this.identifiedConnections.length) {
-      const allSourceClassesFromMappings = uniq(
-        this.runtimeEditorState.runtimeValue.mappings.flatMap((mapping) =>
-          getAllClassMappings(mapping.value)
-            .map((setImplementation) =>
-              getMappingElementSource(
-                setImplementation,
-                this.editorStore.pluginManager.getApplicationPlugins(),
-              ),
-            )
-            .filter(filterByType(Class)),
-        ),
-      );
-      if (!allSourceClassesFromMappings.includes(this.class)) {
-        this.runtimeEditorState.openTabFor(
-          this.runtimeEditorState.runtimeValue,
-        );
-      }
-    }
-  }
-}
-
-export class RuntimeEditorRuntimeTabState extends RuntimeEditorTabState {}
-
-export class RuntimeEditorState {
-  /**
-   * NOTE: used to force component remount on state change
-   */
-  readonly uuid = uuid();
-  editorStore: EditorStore;
-  runtime: Runtime;
-  runtimeValue: EngineRuntime;
-  isEmbeddedRuntime: boolean;
-  explorerTreeData: TreeData<RuntimeExplorerTreeNodeData>;
-  currentTabState?: RuntimeEditorTabState | undefined;
-
-  constructor(
-    editorStore: EditorStore,
-    runtime: Runtime,
-    isEmbeddedRuntime: boolean,
-  ) {
-    makeObservable(this, {
-      explorerTreeData: observable.ref,
-      currentTabState: observable,
-      setExplorerTreeData: action,
-      addMapping: action,
-      deleteMapping: action,
-      changeMapping: action,
-      addIdentifiedConnection: action,
-      openTabFor: action,
-      decorateRuntimeConnections: action,
-      cleanUpDecoration: action,
-      reprocessRuntimeExplorerTree: action,
-      reprocessCurrentTabState: action,
-    });
-
-    this.editorStore = editorStore;
-    this.runtime = runtime;
-    this.isEmbeddedRuntime = isEmbeddedRuntime;
-    this.runtimeValue =
-      runtime instanceof RuntimePointer
-        ? runtime.packageableRuntime.value.runtimeValue
-        : guaranteeType(runtime, EngineRuntime);
-    this.explorerTreeData = getRuntimeExplorerTreeData(
-      this.runtime,
-      this.editorStore,
-    );
-    this.openTabFor(this.runtimeValue); // open runtime tab on init
-  }
-
-  setExplorerTreeData(treeData: TreeData<RuntimeExplorerTreeNodeData>): void {
-    this.explorerTreeData = treeData;
-  }
-
-  addMapping(mapping: Mapping): void {
-    if (!this.runtimeValue.mappings.map((m) => m.value).includes(mapping)) {
-      runtime_addMapping(
-        this.runtimeValue,
-        PackageableElementExplicitReference.create(mapping),
-      );
-      decorateRuntimeWithNewMapping(
-        this.runtimeValue,
-        mapping,
-        this.editorStore,
-      );
-      this.reprocessRuntimeExplorerTree();
-    }
-  }
-
-  deleteMapping(mappingRef: PackageableElementReference<Mapping>): void {
-    runtime_deleteMapping(this.runtimeValue, mappingRef);
-    this.reprocessRuntimeExplorerTree();
-  }
-
-  changeMapping(
-    mappingRef: PackageableElementReference<Mapping>,
-    newVal: Mapping,
-  ): void {
-    packageableElementReference_setValue(mappingRef, newVal);
-    decorateRuntimeWithNewMapping(this.runtimeValue, newVal, this.editorStore);
-    this.reprocessRuntimeExplorerTree();
-  }
-
-  addIdentifiedConnection(identifiedConnection: IdentifiedConnection): void {
-    runtime_addIdentifiedConnection(
-      this.runtimeValue,
-      identifiedConnection,
-      this.editorStore.changeDetectionState.observerContext,
-    );
-    const connectionValue =
-      identifiedConnection.connection instanceof ConnectionPointer
-        ? identifiedConnection.connection.packageableConnection.value
-            .connectionValue
-        : identifiedConnection.connection;
-    this.openTabFor(
-      connectionValue instanceof JsonModelConnection ||
-        connectionValue instanceof XmlModelConnection
-        ? connectionValue.class.value
-        : connectionValue.store.value,
-    );
-    if (this.currentTabState instanceof IdentifiedConnectionsEditorTabState) {
-      this.currentTabState.openIdentifiedConnection(identifiedConnection);
-    }
-    this.reprocessRuntimeExplorerTree();
-  }
-
-  openTabFor(tabElement: Runtime | Store | Class): void {
-    if (tabElement instanceof Runtime) {
-      if (!(this.currentTabState instanceof RuntimeEditorRuntimeTabState)) {
-        this.currentTabState = new RuntimeEditorRuntimeTabState(
-          this.editorStore,
-          this,
-        );
-      }
-    } else if (tabElement instanceof ModelStore) {
-      return;
-    } else if (tabElement instanceof Class) {
-      if (
-        !(
-          this.currentTabState instanceof
-          IdentifiedConnectionsPerClassEditorTabState
-        ) ||
-        this.currentTabState.class !== tabElement
-      ) {
-        this.currentTabState = new IdentifiedConnectionsPerClassEditorTabState(
-          this.editorStore,
-          this,
-          tabElement,
-        );
-      }
-    } else if (tabElement instanceof Store) {
-      if (
-        !(
-          this.currentTabState instanceof
-          IdentifiedConnectionsPerStoreEditorTabState
-        ) ||
-        this.currentTabState.store !== tabElement
-      ) {
-        this.currentTabState = new IdentifiedConnectionsPerStoreEditorTabState(
-          this.editorStore,
-          this,
-          tabElement,
-        );
-      }
-    }
-  }
-
-  decorateRuntimeConnections(): void {
-    getStoresFromMappings(
-      this.runtimeValue.mappings.map((mapping) => mapping.value),
-      this.editorStore,
-    ).forEach((store) =>
-      runtime_addUniqueStoreConnectionsForStore(
-        this.runtimeValue,
-        store,
-        this.editorStore.changeDetectionState.observerContext,
-      ),
-    );
-  }
-
-  cleanUpDecoration(): void {
-    this.runtimeValue.connections = this.runtimeValue.connections.filter(
-      (storeConnections) => !isStubbed_StoreConnections(storeConnections),
-    );
-  }
-
-  onExplorerTreeNodeExpand = (node: RuntimeExplorerTreeNodeData): void => {
-    if (node.childrenIds?.length) {
-      node.isOpen = !node.isOpen;
-    }
-    this.setExplorerTreeData({ ...this.explorerTreeData });
-  };
-
-  onExplorerTreeNodeSelect = (node: RuntimeExplorerTreeNodeData): void => {
-    this.openTabFor(node.data);
-    this.setExplorerTreeData({ ...this.explorerTreeData });
-  };
-
-  isTreeNodeSelected = (node: RuntimeExplorerTreeNodeData): boolean => {
-    if (node.data instanceof Runtime) {
-      return this.currentTabState instanceof RuntimeEditorRuntimeTabState;
-    } else if (node.data instanceof Class) {
-      return (
-        this.currentTabState instanceof
-          IdentifiedConnectionsPerClassEditorTabState &&
-        this.currentTabState.class === node.data
-      );
-    } else if (
-      node.data instanceof Store &&
-      !(node.data instanceof ModelStore)
-    ) {
-      return (
-        this.currentTabState instanceof
-          IdentifiedConnectionsPerStoreEditorTabState &&
-        this.currentTabState.store === node.data
-      );
-    }
-    return false;
-  };
-
-  getExplorerTreeChildNodes = (
-    node: RuntimeExplorerTreeNodeData,
-  ): RuntimeExplorerTreeNodeData[] => {
-    if (!node.childrenIds) {
-      return [];
-    }
-    return node.childrenIds
-      .map((id) => this.explorerTreeData.nodes.get(id))
-      .filter(isNonNullable);
-  };
-
-  reprocessRuntimeExplorerTree(): void {
-    const openedTreeNodeIds = Array.from(this.explorerTreeData.nodes.values())
-      .filter((node) => node.isOpen)
-      .map((node) => node.id);
-    const treeData = getRuntimeExplorerTreeData(this.runtime, this.editorStore);
-    openedTreeNodeIds.forEach((nodeId) => {
-      const node = treeData.nodes.get(nodeId);
-      if (node && !node.isOpen) {
-        node.isOpen = true;
-      }
-    });
-    this.setExplorerTreeData(treeData);
-  }
-
-  /**
-   * If the currently opened connection tab is a connection pointer whose store/source class has been changed, we will
-   * remove this tab and switch to default runtime tab
-   */
-  reprocessCurrentTabState(): void {
-    if (this.currentTabState instanceof IdentifiedConnectionsEditorTabState) {
-      const connection =
-        this.currentTabState.identifiedConnectionEditorState
-          ?.connectionEditorState.connection;
-      const connectionValue =
-        connection instanceof ConnectionPointer
-          ? connection.packageableConnection.value.connectionValue
-          : connection;
-      if (
-        this.currentTabState instanceof
-        IdentifiedConnectionsPerClassEditorTabState
-      ) {
-        if (
-          ((connectionValue instanceof JsonModelConnection ||
-            connectionValue instanceof XmlModelConnection) &&
-            connectionValue.class.value !== this.currentTabState.class) ||
-          (this.currentTabState instanceof
-            IdentifiedConnectionsPerStoreEditorTabState &&
-            this.currentTabState.store !== connectionValue?.store.value)
-        ) {
-          this.openTabFor(this.runtimeValue);
-        }
-      }
-    }
-  }
-}
-
-export class PackageableRuntimeEditorState extends ElementEditorState {
-  runtimeEditorState: RuntimeEditorState;
-
-  constructor(
-    editorStore: EditorStore,
-    packageableRuntime: PackageableElement,
-  ) {
-    super(editorStore, packageableRuntime);
-
-    makeObservable(this, {
-      runtime: computed,
-      reprocess: action,
-    });
-
-    this.runtimeEditorState = new RuntimeEditorState(
-      editorStore,
-      this.runtime.runtimeValue,
-      false,
-    );
-  }
-
-  get runtime(): PackageableRuntime {
-    return guaranteeType(
-      this.element,
-      PackageableRuntime,
-      'Element inside runtime editor state must be a packageable runtime',
-    );
-  }
-
-  reprocess(
-    newElement: PackageableElement,
-    editorStore: EditorStore,
-  ): ElementEditorState {
-    const editorState = new PackageableRuntimeEditorState(
-      editorStore,
-      newElement,
-    );
-    return editorState;
-  }
-}