datahub/datahub-web/packages/data-portal/app/components/search/containers/search-box.ts

import Component from '@ember/component';
import { inject as service } from '@ember/service';
import SearchService from '@datahub/shared/services/search';
import { alias } from '@ember/object/computed';
import { grammarProcessingSteps, typeaheadQueryProcessor } from 'datahub-web/utils/parsers/autocomplete';
import { ISuggestionGroup } from 'datahub-web/utils/parsers/autocomplete/types';
import { DatasetEntity } from '@datahub/data-models/entity/dataset/dataset-entity';
import { tagName } from '@ember-decorators/component';
import { computed } from '@ember/object';
import { DataModelEntity, DataModelName } from '@datahub/data-models/constants/entity';
import {
  getFacetDefaultValueForEntity,
  transformDefaultsIntoSelections
} from '@datahub/data-models/entity/utils/facets';
import RouterService from '@ember/routing/router-service';
import { task } from 'ember-concurrency';
import { ETaskPromise } from '@datahub/utils/types/concurrency';
import DataModelsService from '@datahub/data-models/services/data-models';
import { facetToParamUrl } from '@datahub/shared/utils/search/search';

/**
 * Search box container that handle all the data part for
 * the search box
 */
@tagName('')
export default class SearchBoxContainer extends Component {
  @service
  router: RouterService;

  @service
  search: SearchService;

  @service('data-models')
  dataModels!: DataModelsService;

  @alias('search.entity')
  entity?: DataModelName;

  /**
   * Search route will update the service with current keywords
   * since url will contain keyword
   */
  @alias('search.keyword')
  keyword: string;

  /**
   * Entity definition for current entity
   */
  @computed('entity')
  get dataModelEntity(): DataModelEntity | undefined {
    const { entity, dataModels } = this;
    if (entity && dataModels) {
      return dataModels.getModel(entity);
    }
    return;
  }

  /**
   * Invokes an async function to process the typeahead query and related parameters
   * @param {string} [query=''] user input for to query typeahead endpoint with
   * @param {Entity} [entity]
   * @returns {(IterableIterator<Promise<Array<ISuggestionGroup>>>)}
   * @memberof SearchBoxContainer
   */
  @task(function*(
    this: SearchBoxContainer,
    query = '',
    entity?: DataModelName
  ): IterableIterator<Promise<Array<ISuggestionGroup>>> {
    const { dataModels } = this;
    if (entity) {
      return yield typeaheadQueryProcessor(query as string, dataModels.getModel(entity), grammarProcessingSteps);
    }
  })
  onSearchInputTask: ETaskPromise<Array<ISuggestionGroup>>;

  /**
   * When search actually happens, then we transition to a new route.
   * @param {string} [text=''] user input as search term
   * @param {Entity} [entity]
   * @memberof SearchBoxContainer
   */
  onSearch(text = '', entity: DataModelName = DatasetEntity.displayName): void {
    // entity (dropdown value) might be different than this.entity (Page that you are in)
    const { dataModels } = this;
    if (entity) {
      const dataModelEntity = dataModels.getModel(entity);
      const { attributes } = dataModelEntity.renderProps.search;
      const defaultFacets = facetToParamUrl(transformDefaultsIntoSelections(getFacetDefaultValueForEntity(attributes)));

      this.router.transitionTo('search', {
        queryParams: {
          entity,
          keyword: text,
          page: 1,
          facets: defaultFacets
        }
      });
    }
  }
}