@finos/legend-application-query 5.0.0

package/package.json

@@ -0,0 +1,99 @@
+{
+  "name": "@finos/legend-application-query",
+  "version": "5.0.0",
+  "description": "Legend Query application core",
+  "keywords": [
+    "legend",
+    "legend-application",
+    "legend-query",
+    "query builder"
+  ],
+  "homepage": "https://github.com/finos/legend-studio/tree/master/packages/legend-application-query",
+  "bugs": {
+    "url": "https://github.com/finos/legend-studio/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/finos/legend-studio.git",
+    "directory": "packages/legend-application-query"
+  },
+  "license": "Apache-2.0",
+  "sideEffects": false,
+  "type": "module",
+  "exports": {
+    ".": "./lib/index.js",
+    "./lib/index.css": "./lib/index.css"
+  },
+  "module": "lib/index.js",
+  "types": "lib/index.d.ts",
+  "scripts": {
+    "build": "yarn clean && yarn build:sass && yarn build:ts",
+    "build:sass": "cross-env INIT_CWD=$INIT_CWD node ../../scripts/workflow/buildSass.js",
+    "build:ts": "tsc --project ./tsconfig.build.json",
+    "clean": "npm-run-all clean:cache clean:lib",
+    "clean:cache": "rimraf \"build\"",
+    "clean:lib": "rimraf \"lib\"",
+    "dev": "npm-run-all --parallel dev:sass dev:ts",
+    "dev:sass": "sass style:lib --watch --load-path=../../node_modules/@finos/legend-art/scss",
+    "dev:ts": "tsc --watch --preserveWatchOutput",
+    "lint:js": "cross-env NODE_ENV=production eslint --cache --cache-location ./build/.eslintcache --report-unused-disable-directives --parser-options=project:\"./tsconfig.json\" \"./src/**/*.{js,ts,tsx}\"",
+    "publish:prepare": "node ../../scripts/release/preparePublishContent.js",
+    "publish:snapshot": "node ../../scripts/release/publishDevSnapshot.js",
+    "test": "jest",
+    "test:watch": "jest --watch"
+  },
+  "dependencies": {
+    "@ag-grid-community/client-side-row-model": "28.0.2",
+    "@ag-grid-community/core": "28.0.2",
+    "@ag-grid-community/react": "28.0.0",
+    "@finos/legend-application": "7.0.1",
+    "@finos/legend-art": "2.0.10",
+    "@finos/legend-graph": "12.0.0",
+    "@finos/legend-server-depot": "3.0.1",
+    "@finos/legend-shared": "5.0.1",
+    "@finos/legend-storage": "3.0.0",
+    "@testing-library/react": "13.3.0",
+    "@types/react": "18.0.15",
+    "@types/react-dom": "18.0.6",
+    "@types/react-router-dom": "5.3.3",
+    "date-fns": "2.29.1",
+    "history": "5.3.0",
+    "mobx": "6.6.1",
+    "mobx-react-lite": "3.4.0",
+    "monaco-editor": "0.33.0",
+    "react": "18.2.0",
+    "react-dnd": "16.0.1",
+    "react-dnd-html5-backend": "16.0.1",
+    "react-dom": "18.2.0",
+    "react-hotkeys": "2.0.0",
+    "react-router": "5.3.3",
+    "react-router-dom": "5.3.3"
+  },
+  "devDependencies": {
+    "@finos/legend-dev-utils": "2.0.6",
+    "@jest/globals": "28.1.3",
+    "@testing-library/dom": "8.16.0",
+    "cross-env": "7.0.3",
+    "eslint": "8.20.0",
+    "jest": "28.1.3",
+    "npm-run-all": "4.1.5",
+    "rimraf": "3.0.2",
+    "sass": "1.54.0",
+    "typescript": "4.7.4"
+  },
+  "peerDependencies": {
+    "react": "^18.0.0"
+  },
+  "publishConfig": {
+    "directory": "build/publishContent"
+  },
+  "extensions": {
+    "graphManagerPreset": "@finos/legend-graph-manager-preset-query-builder",
+    "pureProtocolProcessorPlugin": "@finos/legend-pure-protocol-processor-plugin-query-builder",
+    "pureGraphManagerPlugin": "@finos/legend-pure-graph-manager-plugin-query-builder"
+  },
+  "typedoc": {
+    "entryPoint": "./src/index.ts",
+    "readmeFile": "./README.md"
+  }
+}

package/src/LegendQueryAppEvent.ts

@@ -0,0 +1,20 @@
+/**
+ * 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.
+ */
+
+export enum LEGEND_QUERY_APP_EVENT {
+  // TODO: split this into specific events
+  QUERY_PROBLEM = 'application.failure.generic',
+}

package/src/QueryBuilder_Const.ts

@@ -0,0 +1,88 @@
+/**
+ * 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.
+ */
+export enum QUERY_BUILDER_SOURCE_ID_LABEL {
+  QUERY_BUILDER = 'query-builder',
+  PROJECTION = 'projection',
+}
+
+export const DEFAULT_LAMBDA_VARIABLE_NAME = 'x';
+export const DEFAULT_VARIABLE_NAME = 'var';
+
+export enum QUERY_BUILDER_SUPPORTED_FUNCTIONS {
+  AND = 'meta::pure::functions::boolean::and',
+  EXISTS = 'meta::pure::functions::collection::exists',
+  FILTER = 'meta::pure::functions::collection::filter',
+  GET_ALL = 'meta::pure::functions::collection::getAll',
+  NOT = 'meta::pure::functions::boolean::not',
+  OR = 'meta::pure::functions::boolean::or',
+  TAKE = 'meta::pure::functions::collection::take',
+  SUBTYPE = 'meta::pure::functions::lang::subType',
+
+  // graph-fetch
+  GRAPH_FETCH = 'meta::pure::graphFetch::execution::graphFetch',
+  GRAPH_FETCH_CHECKED = 'meta::pure::graphFetch::execution::graphFetchChecked',
+  SERIALIZE = 'meta::pure::graphFetch::execution::serialize',
+
+  // TDS
+  TDS_FILTER = 'meta::pure::tds::filter',
+  TDS_ASC = 'meta::pure::tds::asc',
+  TDS_DESC = 'meta::pure::tds::desc',
+  TDS_AGG = 'meta::pure::tds::agg',
+  TDS_DISTINCT = 'meta::pure::tds::distinct',
+  TDS_PROJECT = 'meta::pure::tds::project',
+  TDS_GROUP_BY = 'meta::pure::tds::groupBy',
+  TDS_SORT = 'meta::pure::tds::sort',
+  TDS_TAKE = 'meta::pure::tds::take',
+
+  // filter
+  CONTAINS = 'meta::pure::functions::string::contains',
+  ENDS_WITH = 'meta::pure::functions::string::endsWith',
+  EQUAL = 'meta::pure::functions::boolean::equal',
+  GREATER_THAN = 'meta::pure::functions::lang::tests::greaterThan',
+  GREATER_THAN_EQUAL = 'meta::pure::functions::lang::tests::greaterThanEqual',
+  IN = 'meta::pure::functions::collection::in',
+  IS_EMPTY = 'meta::pure::functions::collection::isEmpty',
+  LESS_THAN = 'meta::pure::functions::lang::tests::lessThan',
+  LESS_THAN_EQUAL = 'meta::pure::functions::lang::tests::lessThanEqual',
+  STARTS_WITH = 'meta::pure::functions::string::startsWith',
+  // aggregation
+  AVERAGE = 'meta::pure::functions::math::average',
+  COUNT = 'meta::pure::functions::collection::count',
+  JOIN_STRINGS = 'meta::pure::functions::string::joinStrings',
+  DISTINCT = 'meta::pure::functions::collection::distinct',
+  DATE_MAX = 'meta::pure::functions::date::max',
+  DATE_MIN = 'meta::pure::functions::date::min',
+  MAX = 'meta::pure::functions::math::max',
+  MIN = 'meta::pure::functions::math::min',
+  STD_DEV_POPULATION = 'meta::pure::functions::math::stdDevPopulation',
+  STD_DEV_SAMPLE = 'meta::pure::functions::math::stdDevSample',
+  SUM = 'meta::pure::functions::math::sum',
+  UNIQUE_VALUE_ONLY = 'meta::pure::functions::collection::uniqueValueOnly',
+}
+
+// TDS
+export const DEFAULT_POST_FILTER_LAMBDA_VARIABLE_NAME = 'row';
+export const TDS_ROW = 'meta::pure::tds::TDSRow';
+
+export const QUERY_BUILDER_PROPERTY_SEARCH_MAX_DEPTH = 5;
+export const QUERY_BUILDER_PROPERTY_SEARCH_RESULTS_LIMIT = 100;
+export enum QUERY_BUILDER_PROPERTY_SEARCH_TYPE {
+  CLASS = 'CLASS',
+  STRING = 'STRING',
+  BOOLEAN = 'BOOLEAN',
+  NUMBER = 'NUMBER',
+  DATE = 'DATE',
+}

package/src/application/LegendQuery.tsx

@@ -0,0 +1,92 @@
+/**
+ * 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 { createRoot } from 'react-dom/client';
+import {
+  type LegendApplicationConfig,
+  ApplicationStoreProvider,
+  LegendApplication,
+  setupLegendApplicationUILibrary,
+  WebApplicationNavigatorProvider,
+  type LegendApplicationConfigurationInput,
+} from '@finos/legend-application';
+import { configure as configureReactHotkeys } from 'react-hotkeys';
+import { ModuleRegistry as agGrid_ModuleRegistry } from '@ag-grid-community/core';
+import { ClientSideRowModelModule } from '@ag-grid-community/client-side-row-model';
+import { BrowserRouter } from 'react-router-dom';
+import { LegendQueryApplication } from '../components/LegendQueryApplication.js';
+import { LegendQueryPluginManager } from './LegendQueryPluginManager.js';
+import { QueryBuilder_GraphManagerPreset } from '../graphManager/QueryBuilder_GraphManagerPreset.js';
+import { getRootElement } from '@finos/legend-art';
+import { CorePureGraphManagerPlugin } from '@finos/legend-graph';
+import {
+  type LegendQueryApplicationConfigurationData,
+  LegendQueryApplicationConfig,
+} from './LegendQueryApplicationConfig.js';
+
+export const setupLegendQueryUILibrary = async (): Promise<void> => {
+  // Register module extensions for `ag-grid`
+  agGrid_ModuleRegistry.registerModules([ClientSideRowModelModule]);
+
+  configureReactHotkeys({
+    // By default, `react-hotkeys` will avoid capturing keys from input tags like <input>, <textarea>, <select>
+    // We want to listen to hotkey from every where in the app so we disable that
+    // See https://github.com/greena13/react-hotkeys#ignoring-events
+    ignoreTags: [],
+  });
+};
+
+export class LegendQuery extends LegendApplication {
+  declare config: LegendQueryApplicationConfig;
+  declare pluginManager: LegendQueryPluginManager;
+
+  static create(): LegendQuery {
+    const application = new LegendQuery(LegendQueryPluginManager.create());
+    application.withBasePlugins([new CorePureGraphManagerPlugin()]);
+    application.withBasePresets([new QueryBuilder_GraphManagerPreset()]);
+    return application;
+  }
+
+  async configureApplication(
+    input: LegendApplicationConfigurationInput<LegendQueryApplicationConfigurationData>,
+  ): Promise<LegendApplicationConfig> {
+    return new LegendQueryApplicationConfig(input);
+  }
+
+  async loadApplication(): Promise<void> {
+    // Setup React application libraries
+    await setupLegendApplicationUILibrary(this.pluginManager, this.logger);
+    await setupLegendQueryUILibrary();
+
+    // Render React application
+    const rootElement = createRoot(getRootElement());
+    rootElement.render(
+      <BrowserRouter basename={this.baseUrl}>
+        <WebApplicationNavigatorProvider>
+          <ApplicationStoreProvider
+            config={this.config}
+            pluginManager={this.pluginManager}
+          >
+            <LegendQueryApplication
+              config={this.config}
+              pluginManager={this.pluginManager}
+            />
+          </ApplicationStoreProvider>
+        </WebApplicationNavigatorProvider>
+      </BrowserRouter>,
+    );
+  }
+}

package/src/application/LegendQueryApplicationConfig.ts

@@ -0,0 +1,77 @@
+/**
+ * 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 {
+  assertNonNullable,
+  guaranteeNonEmptyString,
+} from '@finos/legend-shared';
+import {
+  LegendApplicationConfig,
+  type LegendApplicationConfigurationInput,
+  type LegendApplicationConfigurationData,
+} from '@finos/legend-application';
+
+export interface LegendQueryApplicationConfigurationData
+  extends LegendApplicationConfigurationData {
+  appName: string;
+  env: string;
+  depot: {
+    url: string;
+    /**
+     * This is needed since some of our legacy infrastructure does not yet support
+     * the new API calls, we need to update them to use the latest version of
+     * finos/legend-depot though
+     */
+    TEMPORARY__useLegacyDepotServerAPIRoutes?: boolean;
+  };
+  engine: { url: string; queryUrl?: string };
+  studio: { url: string };
+  extensions?: Record<PropertyKey, unknown>;
+}
+
+export class LegendQueryApplicationConfig extends LegendApplicationConfig {
+  readonly engineServerUrl: string;
+  readonly engineQueryServerUrl?: string | undefined;
+  readonly depotServerUrl: string;
+  readonly studioUrl: string;
+  readonly TEMPORARY__useLegacyDepotServerAPIRoutes?: boolean | undefined;
+
+  constructor(
+    input: LegendApplicationConfigurationInput<LegendQueryApplicationConfigurationData>,
+  ) {
+    super(input);
+
+    assertNonNullable(
+      input.configData.engine,
+      `Can't configure application: 'engine' field is missing`,
+    );
+    this.engineServerUrl = guaranteeNonEmptyString(
+      input.configData.engine.url,
+      `Can't configure application: 'engine.url' field is missing or empty`,
+    );
+    this.engineQueryServerUrl = input.configData.engine.queryUrl;
+    this.depotServerUrl = guaranteeNonEmptyString(
+      input.configData.depot.url,
+      `Can't configure application: 'depot.url' field is missing or empty`,
+    );
+    this.studioUrl = guaranteeNonEmptyString(
+      input.configData.studio.url,
+      `Can't configure application: 'studio.url' field is missing or empty`,
+    );
+    this.TEMPORARY__useLegacyDepotServerAPIRoutes =
+      input.configData.depot.TEMPORARY__useLegacyDepotServerAPIRoutes;
+  }
+}

package/src/application/LegendQueryPluginManager.ts

@@ -0,0 +1,67 @@
+/**
+ * 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 { LegendApplicationPluginManager } from '@finos/legend-application';
+import type {
+  GraphManagerPluginManager,
+  PureGraphManagerPlugin,
+  PureGraphPlugin,
+  PureProtocolProcessorPlugin,
+} from '@finos/legend-graph';
+import type { LegendQueryApplicationPlugin } from '../stores/LegendQueryApplicationPlugin.js';
+
+export class LegendQueryPluginManager
+  extends LegendApplicationPluginManager<LegendQueryApplicationPlugin>
+  implements GraphManagerPluginManager
+{
+  private pureProtocolProcessorPlugins: PureProtocolProcessorPlugin[] = [];
+  private pureGraphManagerPlugins: PureGraphManagerPlugin[] = [];
+  private pureGraphPlugins: PureGraphPlugin[] = [];
+
+  private constructor() {
+    super();
+  }
+
+  static create(): LegendQueryPluginManager {
+    return new LegendQueryPluginManager();
+  }
+
+  registerPureProtocolProcessorPlugin(
+    plugin: PureProtocolProcessorPlugin,
+  ): void {
+    this.pureProtocolProcessorPlugins.push(plugin);
+  }
+
+  registerPureGraphManagerPlugin(plugin: PureGraphManagerPlugin): void {
+    this.pureGraphManagerPlugins.push(plugin);
+  }
+
+  registerPureGraphPlugin(plugin: PureGraphPlugin): void {
+    this.pureGraphPlugins.push(plugin);
+  }
+
+  getPureGraphManagerPlugins(): PureGraphManagerPlugin[] {
+    return [...this.pureGraphManagerPlugins];
+  }
+
+  getPureProtocolProcessorPlugins(): PureProtocolProcessorPlugin[] {
+    return [...this.pureProtocolProcessorPlugins];
+  }
+
+  getPureGraphPlugins(): PureGraphPlugin[] {
+    return [...this.pureGraphPlugins];
+  }
+}

package/src/components/LegendQueryApplication.tsx

@@ -0,0 +1,104 @@
+/**
+ * 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 { Redirect, Route, Switch } from 'react-router';
+import { observer } from 'mobx-react-lite';
+import { LEGEND_QUERY_ROUTE_PATTERN } from '../stores/LegendQueryRouter.js';
+import { QuerySetup } from './QuerySetup.js';
+import {
+  CreateQueryEditor,
+  ExistingQueryEditor,
+  ServiceQueryEditor,
+} from './QueryEditor.js';
+import { DepotServerClientProvider } from '@finos/legend-server-depot';
+import {
+  generateExtensionUrlPattern,
+  LegendApplicationComponentFrameworkProvider,
+} from '@finos/legend-application';
+import type { LegendQueryPluginManager } from '../application/LegendQueryPluginManager.js';
+import type { LegendQueryApplicationConfig } from '../application/LegendQueryApplicationConfig.js';
+import {
+  LegendQueryBaseStoreProvider,
+  useLegendQueryApplicationStore,
+} from './LegendQueryBaseStoreProvider.js';
+
+const LegendQueryApplicationRoot = observer(() => {
+  const applicationStore = useLegendQueryApplicationStore();
+  const extraApplicationPageEntries = applicationStore.pluginManager
+    .getApplicationPlugins()
+    .flatMap((plugin) => plugin.getExtraApplicationPageEntries?.() ?? []);
+
+  return (
+    <div className="app">
+      <Switch>
+        <Route
+          exact={true}
+          path={LEGEND_QUERY_ROUTE_PATTERN.SETUP}
+          component={QuerySetup}
+        />
+        <Route
+          exact={true}
+          path={LEGEND_QUERY_ROUTE_PATTERN.EXISTING_QUERY}
+          component={ExistingQueryEditor}
+        />
+        <Route
+          exact={true}
+          path={LEGEND_QUERY_ROUTE_PATTERN.SERVICE_QUERY}
+          component={ServiceQueryEditor}
+        />
+        <Route
+          exact={true}
+          path={LEGEND_QUERY_ROUTE_PATTERN.CREATE_QUERY}
+          component={CreateQueryEditor}
+        />
+        {extraApplicationPageEntries.map((entry) => (
+          <Route
+            key={entry.key}
+            exact={true}
+            path={entry.urlPatterns.map(generateExtensionUrlPattern)}
+            component={entry.renderer as React.ComponentType<unknown>}
+          />
+        ))}
+        <Redirect to={LEGEND_QUERY_ROUTE_PATTERN.SETUP} />
+      </Switch>
+    </div>
+  );
+});
+
+export const LegendQueryApplication = observer(
+  (props: {
+    config: LegendQueryApplicationConfig;
+    pluginManager: LegendQueryPluginManager;
+  }) => {
+    const { config, pluginManager } = props;
+
+    return (
+      <DepotServerClientProvider
+        config={{
+          serverUrl: config.depotServerUrl,
+          TEMPORARY__useLegacyDepotServerAPIRoutes:
+            config.TEMPORARY__useLegacyDepotServerAPIRoutes,
+        }}
+      >
+        <LegendQueryBaseStoreProvider pluginManager={pluginManager}>
+          <LegendApplicationComponentFrameworkProvider>
+            <LegendQueryApplicationRoot />
+          </LegendApplicationComponentFrameworkProvider>
+        </LegendQueryBaseStoreProvider>
+      </DepotServerClientProvider>
+    );
+  },
+);

package/src/components/LegendQueryBaseStoreProvider.tsx

@@ -0,0 +1,68 @@
+/**
+ * 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 { createContext, useContext } from 'react';
+import { useLocalObservable } from 'mobx-react-lite';
+import { LegendQueryBaseStore } from '../stores/LegendQueryBaseStore.js';
+import { guaranteeNonNullable } from '@finos/legend-shared';
+import { useDepotServerClient } from '@finos/legend-server-depot';
+import type { LegendQueryPluginManager } from '../application/LegendQueryPluginManager.js';
+import type { LegendQueryApplicationConfig } from '../application/LegendQueryApplicationConfig.js';
+import type { LegendQueryApplicationPlugin } from '../stores/LegendQueryApplicationPlugin.js';
+import {
+  type ApplicationStore,
+  useApplicationStore,
+} from '@finos/legend-application';
+
+export const useLegendQueryApplicationStore = (): ApplicationStore<
+  LegendQueryApplicationConfig,
+  LegendQueryApplicationPlugin
+> =>
+  useApplicationStore<
+    LegendQueryApplicationConfig,
+    LegendQueryApplicationPlugin
+  >();
+
+const LegendQueryBaseStoreContext = createContext<
+  LegendQueryBaseStore | undefined
+>(undefined);
+
+export const LegendQueryBaseStoreProvider: React.FC<{
+  children: React.ReactNode;
+  pluginManager: LegendQueryPluginManager;
+}> = ({ children, pluginManager }) => {
+  const applicationStore = useLegendQueryApplicationStore();
+  const depotServerClient = useDepotServerClient();
+  const store = useLocalObservable(
+    () =>
+      new LegendQueryBaseStore(
+        applicationStore,
+        depotServerClient,
+        pluginManager,
+      ),
+  );
+  return (
+    <LegendQueryBaseStoreContext.Provider value={store}>
+      {children}
+    </LegendQueryBaseStoreContext.Provider>
+  );
+};
+
+export const useLegendQueryBaseStore = (): LegendQueryBaseStore =>
+  guaranteeNonNullable(
+    useContext(LegendQueryBaseStoreContext),
+    `Can't find Legend Query base store in context`,
+  );