baqend/lib/query/Node.ts

Baqend JavaScript SDK

import { Entity } from '../binding';

import { JsonMap } from '../util';
import {
  CountCallback,
  FailCallback,
  Query,
  ResultListCallback,
  ResultOptions,
  SingleResultCallback,
} from './Query';
import * as message from '../message';
import type { FilterObject } from './Filter';
import { Metadata } from '../intersection';

/**
 * A Query Node saves the state of the query being built
 */
export class Node<T extends Entity> extends Query<T> {
  /**
   * The offset how many results should be skipped
   */
  public firstResult: number = 0;

  /**
   * The limit how many objects should be returned
   * @type number
   * @readonly
   */
  public maxResults: number = -1;

  /**
   * The properties which should be used sort the result
   */
  public order: { [field: string]: 1 | -1 } = {};

  /**
   * @inheritDoc
   */
  resultList(options?: ResultOptions | ResultListCallback<T>, doneCallback?: ResultListCallback<T> | FailCallback,
    failCallback?: FailCallback): Promise<T[]> {
    if (options instanceof Function) {
      return this.resultList({}, options as ResultListCallback<T>, doneCallback as FailCallback);
    }

    const type = this.resultClass ? this.entityManager.metamodel.entity(this.resultClass) : null;

    if (!type) {
      throw new Error('Only typed queries can be executed.');
    }

    const query = this.serializeQuery();
    const sort = this.serializeSort();

    const uriSize = (this.entityManager.connection?.host.length || 0) + query.length + sort.length;
    let msg;

    if (uriSize > Query.MAX_URI_SIZE) {
      msg = new message.AdhocQueryPOST(type.name, this.firstResult, this.maxResults, sort)
        .entity(query, 'text');
    } else {
      msg = new message.AdhocQuery(type.name, query, this.firstResult, this.maxResults, sort);
    }

    return this.entityManager.send(msg)
      .then((response) => this.createResultList(response.entity, options as ResultOptions))
      .then(doneCallback as ResultListCallback<T>, failCallback);
  }

  /**
   * @inheritDoc
   */
  singleResult(options?: ResultOptions | SingleResultCallback<T>, doneCallback?: SingleResultCallback<T> | FailCallback,
    failCallback?: FailCallback): Promise<T | null> {
    if (options instanceof Function) {
      return this.singleResult({} as ResultOptions, options as SingleResultCallback<T>, doneCallback as FailCallback);
    }

    const type = this.resultClass ? this.entityManager.metamodel.entity(this.resultClass) : null;

    if (!type) {
      throw new Error('Only typed queries can be executed.');
    }

    const query = this.serializeQuery();
    const sort = this.serializeSort();

    const uriSize = (this.entityManager.connection?.host.length || 0) + query.length;
    let msg;
    if (uriSize > Query.MAX_URI_SIZE) {
      msg = new message.AdhocQueryPOST(type.name, this.firstResult, 1, sort)
        .entity(query, 'text');
    } else {
      msg = new message.AdhocQuery(type.name, query, this.firstResult, 1, sort);
    }

    return this.entityManager.send(msg)
      .then((response) => this.createResultList(response.entity, options as ResultOptions))
      .then((list) => (list.length ? list[0] : null))
      .then(doneCallback as SingleResultCallback<T>, failCallback);
  }

  /**
   * @inheritDoc
   */
  count(doneCallback?: CountCallback, failCallback?: FailCallback) {
    const type = this.resultClass ? this.entityManager.metamodel.entity(this.resultClass) : null;

    if (!type) {
      throw new Error('Only typed queries can be executed.');
    }

    const query = this.serializeQuery();

    const uriSize = (this.entityManager.connection?.host.length || 0) + query.length;
    let msg;
    if (uriSize > Query.MAX_URI_SIZE) {
      msg = new message.AdhocCountQueryPOST(type.name)
        .entity(query, 'text');
    } else {
      msg = new message.AdhocCountQuery(type.name, query);
    }

    return this.entityManager.send(msg)
      .then((response) => response.entity.count)
      .then(doneCallback, failCallback);
  }

  private serializeQuery() {
    return JSON.stringify(this, function argSerializer(this: FilterObject, k, v) {
      // this referees here to the object which owns the key k
      const typedValue = this[k];
      if (typedValue instanceof Date) {
        return { $date: v };
      } if (typedValue instanceof Entity) {
        return typedValue.id;
      }
      return v;
    });
  }

  private serializeSort() {
    return JSON.stringify(this.order);
  }

  private createResultList(result: JsonMap[], options: ResultOptions): Promise<T[]> {
    if (result.length) {
      return Promise.all<T | null>(result.map((el: JsonMap) => {
        if (el.id) {
          const entity: T = this.entityManager.getReference(this.resultClass, el.id as string);
          const metadata = Metadata.get(entity);
          metadata.type.fromJsonValue(metadata, el, entity, { persisting: true });
          return this.entityManager.resolveDepth(entity, options);
        }

        return this.entityManager.load<T>(Object.keys(el)[0]);
      }))
        .then((objects) => objects.filter((val: T | null) => !!val) as T[]);
    }

    return Promise.resolve([]);
  }

  addOrder(fieldOrSort: string | { [field: string]: 1 | -1 }, order?: 1 | -1): this {
    if (typeof fieldOrSort === 'string') {
      this.order[fieldOrSort] = order!;
    } else {
      this.order = fieldOrSort;
    }
    return this;
  }

  addOffset(offset: number): this {
    this.firstResult = offset;
    return this;
  }

  addLimit(limit: number): this {
    this.maxResults = limit;
    return this;
  }
}