Added framework to get default query string using dataset and language combination

changelogs/fragments/8896.yml

@@ -0,0 +1,2 @@
+feat:
+- Added framework to get default query string using dataset and language combination ([#8896](https://github.com/opensearch-project/OpenSearch-Dashboards/pull/8896))

src/plugins/data/public/query/query_string/dataset_service/types.ts

@@ -8,6 +8,7 @@ import {
   DatasetField,
   DatasetSearchOptions,
   DataStructure,
+  Query,
   SavedObject,
 } from '../../../../common';
 import { IDataPluginServices } from '../../../types';
@@ -106,4 +107,8 @@ export interface DatasetTypeConfig {
    * Service used for indexedViews related operations
    */
   indexedViewsService?: DatasetIndexedViewsService;
+  /**
+   * Returns the initial query that is added to the query editor when a dataset is selected.
+   */
+  getInitialQueryString?: (query: Query) => string | void;
 }

src/plugins/data/public/query/query_string/query_string_manager.ts

@@ -63,6 +63,16 @@
     return this.storage.get('userQueryString') || '';
   }
 
+  private getInitialDatasetQueryString(query: Query, dataset: Dataset) {
+    const { language } = query;
+
+    const languageConfig = this.languageService.getLanguage(language);
+    const typeConfig = this.datasetService.getType(dataset.type);
+    return (
+      typeConfig?.getInitialQueryString?.(query) ?? (languageConfig?.getQueryString(query) || '')
+    );
+  }
+
   public getDefaultQuery(): Query {
     const defaultLanguageId = this.getDefaultLanguage();
     const defaultQuery = this.getDefaultQueryString();
@@ -79,13 +89,11 @@
       defaultDataset &&
       this.languageService
     ) {
-      const language = this.languageService.getLanguage(defaultLanguageId);
       const newQuery = { ...query, dataset: defaultDataset };
-      const newQueryString = language?.getQueryString(newQuery) || '';
 
       return {
         ...newQuery,
-        query: newQueryString,
+        query: this.getInitialDatasetQueryString(newQuery, defaultDataset),
       };
     }
 
@@ -244,13 +252,12 @@
 
     // Both language and dataset provided - generate fresh query
     if (language && dataset) {
-      const languageService = this.languageService.getLanguage(language);
       const newQuery = {
         language,
         dataset,
         query: '',
       };
-      newQuery.query = languageService?.getQueryString(newQuery) || '';
+      newQuery.query = this.getInitialDatasetQueryString(newQuery, dataset);
       return newQuery;
     }
 
@@ -279,7 +286,11 @@
       ...curQuery,
       language: languageId,
     };
-    const queryString = language?.getQueryString(newQuery) || '';
+    let queryString = language?.getQueryString(newQuery) || '';
+
+    if (curQuery.dataset) {
+      queryString = this.getInitialDatasetQueryString(newQuery, curQuery.dataset) ?? queryString;
+    }
     this.languageService.setUserQueryString(queryString);
 
     return {
@@ -296,17 +307,15 @@
     const curQuery = this.query$.getValue();
     // Use dataset's preferred language or fallback to current language
     const languageId = newDataset.language || curQuery.language;
-    const language = this.languageService.getLanguage(languageId);
     const newQuery = {
       ...curQuery,
       language: languageId,
       dataset: newDataset,
     };
-    const queryString = language?.getQueryString(newQuery) || '';
 
     return {
       ...newQuery,
-      query: queryString,
+      query: this.getInitialDatasetQueryString(newQuery, newDataset),
     };
   };