<commit_before>import { Client, Tracking } from "sajari";

import { Listener, ClickTracking, Analytics } from "./";

export const searchSentEvent = "search-sent";
export const responseUpdatedEvent = "response-updated";
export const resultClickedEvent = "result-clicked";

const events = [searchSentEvent, responseUpdatedEvent, resultClickedEvent];

class Pipeline {
  /**
   * Constructs a Pipeline object.
   * @param {string} project Name of the project
   * @param {string} collection Name of the collection
   * @param {string} name Name of the pipeline.
   * @param {Tracking|undefined} [tracking=ClickTracking()] Default tracking to use in searches.
   */
  constructor(project, collection, name, tracking = new ClickTracking()) {
    /** @private */
    this.client = new Client(project, collection);
    /** @private */
    this.name = name;
    /** @private */
    this.tracking = tracking;
    /** @private */
    this.listeners = {
      [searchSentEvent]: new Listener(),
      [responseUpdatedEvent]: new Listener(),
      [resultClickedEvent]: new Listener()
    };
    /** @private */
    this.searchCount = 0;
    /** @private */
    this.response = new Response();
    /** @private */
    this.analytics = new Analytics(this, this.tracking);
  }

  /**
   * Register a listener for a specific event.
   * @param {string} event Event to listen for
   * @param {function()} callback Callback to run when the event happens.
   * @return {function()} The unregister function to remove the callback from the listener.
   */
  listen(event, callback) {
    if (events.indexOf(event) === -1) {
      throw new Error(`unknown event type "${event}"`);
    }
    return this.listeners[event].listen(callback);
  }

  /**
   * Emits a search event to the search event listener.
   * @private
   */
  _emitSearchSent(values) {
    this.listeners[searchSentEvent].notify(listener => {
      listener(values);
    });
  }

  /**
   * Emits a results event to the results event listener.
   * @private
   */
  _emitResponseUpdated(response) {
    this.listeners[responseUpdatedEvent].notify(listener => {
      listener(response);
    });
  }

  /**
   * Emits a result clicked event to the results clicked event listeners.
   * @param {*} value Value to send to the listeners.
   */
  emitResultClicked(value) {
    this.listeners[resultClickedEvent].notify(listener => {
      listener(value);
    });
  }

  /**
   * Perform a search.
   * @param {Object} values Key-value pair parameters to run the search with.
   */
  search(values) {
    const tracking = this.tracking.tracking(values);

    this.searchCount++;
    const currentSearch = this.searchCount;

    this.client.searchPipeline(
      this.name,
      values,
      tracking,
      (error, response = {}) => {
        if (currentSearch < this.searchCount) {
          return;
        }
        this.response = new Response(
          error ? error : undefined,
          values,
          response ? response.searchResponse : undefined,
          response ? response.values : undefined
        );
        this._emitResponseUpdated(this.response);
      }
    );
    this._emitSearchSent(values);
  }

  /**
   * Clears the error, response, and response values from this object.
   * @param {Object} values Key-value pair parameters.
   */
  clearResponse(values) {
    this.tracking.tracking(values);

    this.response = new Response();
    this._emitResponseUpdated(this.response);
  }

  /**
   * The current response.
   * @return {Response}
   */
  getResponse() {
    return this.response;
  }

  /**
   * The analytics adaptor connected to this pipeline.
   * @return {Analytics}
   */
  getAnalytics() {
    return this.analytics;
  }
}

class Response {
  /**
   * Constructs a Response object.
   * @param {*} error
   * @param {*} queryValues
   * @param {*} response
   * @param {*} values
   */
  constructor(error, queryValues, response, values) {
    /** @private */
    this.error = error;
    /** @private */
    this.queryValues = queryValues;
    /** @private */
    this.response = response;
    /** @private */
    this.values = values;
  }

  /**
   * Is this response empty?
   * @return {boolean}
   */
  isEmpty() {
    return (
      this.error === undefined &&
      this.response === undefined &&
      this.values === undefined &&
      this.queryValues === undefined
    );
  }

  /**
   * Is this response an error?
   * @return {boolean}
   */
  isError() {
    return this.error !== undefined;
  }

  /**
   * The error associated with this response.
   * @return {string|undefined}
   */
  getError() {
    return this.error;
  }

  /**
   * Return the query values used in the search which created this response.
   * @return {Object|undefined}
   */
  getQueryValues() {
    return this.queryValues;
  }

  /**
   * Returns the response, which includes results and aggregates etc.
   * @return {Object|undefined}
   */
  getResponse() {
    return this.response;
  }

  /**
   * Return the pipeline values returned by the search.
   * @return {Object|undefined}
   */
  getValues() {
    return this.values;
  }

  /**
   * Return results from the response.
   * @return {Object|undefined}
   */
  getResults() {
    return this.response !== undefined ? this.response.results : undefined;
  }

  /**
   * Return the total number of results.
   * @return {number|undefined}
   */
  getTotalResults() {
    return this.response !== undefined
      ? parseInt(this.response.totalResults, 10)
      : undefined;
  }

  /**
   * Return time from the response.
   * @return {string|undefined}
   */
  getTime() {
    return this.response !== undefined ? this.response.time : undefined;
  }

  /**
   * Return the aggregates in the response.
   * @return {Object|undefined}
   */
  getAggregates() {
    if (this.response === undefined || this.response.aggregates === undefined) {
      return undefined;
    }
    return this.response.aggregates;
  }
}

export { Pipeline, Response };
<commit_msg>pipeline: add analyticsAdapter argument to pipeline, ga by default<commit_after>import { Client, Tracking } from "sajari";

import { GoogleAnalytics } from "../analytics";
import { Listener, ClickTracking, Analytics } from "./";

export const searchSentEvent = "search-sent";
export const responseUpdatedEvent = "response-updated";
export const resultClickedEvent = "result-clicked";

const events = [searchSentEvent, responseUpdatedEvent, resultClickedEvent];

class Pipeline {
  /**
   * Constructs a Pipeline object.
   * @param {string} project Name of the project
   * @param {string} collection Name of the collection
   * @param {string} name Name of the pipeline.
   * @param {Tracking|undefined} [tracking=ClickTracking()] Default tracking to use in searches.
   * @param {Object[]} [analyticsAdapters=GoogleAnalytics]
   */
  constructor(
    project,
    collection,
    name,
    tracking = new ClickTracking(),
    analyticsAdapters = [GoogleAnalytics]
  ) {
    /** @private */
    this.client = new Client(project, collection);
    /** @private */
    this.name = name;
    /** @private */
    this.tracking = tracking;
    /** @private */
    this.listeners = {
      [searchSentEvent]: new Listener(),
      [responseUpdatedEvent]: new Listener(),
      [resultClickedEvent]: new Listener()
    };
    /** @private */
    this.searchCount = 0;
    /** @private */
    this.response = new Response();
    /** @private */
    this.analytics = new Analytics(this, this.tracking);
    analyticsAdapters.forEach(adapter => {
      new adapter(this.analytics);
    });
  }

  /**
   * Register a listener for a specific event.
   * @param {string} event Event to listen for
   * @param {function()} callback Callback to run when the event happens.
   * @return {function()} The unregister function to remove the callback from the listener.
   */
  listen(event, callback) {
    if (events.indexOf(event) === -1) {
      throw new Error(`unknown event type "${event}"`);
    }
    return this.listeners[event].listen(callback);
  }

  /**
   * Emits a search event to the search event listener.
   * @private
   */
  _emitSearchSent(values) {
    this.listeners[searchSentEvent].notify(listener => {
      listener(values);
    });
  }

  /**
   * Emits a results event to the results event listener.
   * @private
   */
  _emitResponseUpdated(response) {
    this.listeners[responseUpdatedEvent].notify(listener => {
      listener(response);
    });
  }

  /**
   * Emits a result clicked event to the results clicked event listeners.
   * @param {*} value Value to send to the listeners.
   */
  emitResultClicked(value) {
    this.listeners[resultClickedEvent].notify(listener => {
      listener(value);
    });
  }

  /**
   * Perform a search.
   * @param {Object} values Key-value pair parameters to run the search with.
   */
  search(values) {
    const tracking = this.tracking.tracking(values);

    this.searchCount++;
    const currentSearch = this.searchCount;

    this.client.searchPipeline(
      this.name,
      values,
      tracking,
      (error, response = {}) => {
        if (currentSearch < this.searchCount) {
          return;
        }
        this.response = new Response(
          error ? error : undefined,
          values,
          response ? response.searchResponse : undefined,
          response ? response.values : undefined
        );
        this._emitResponseUpdated(this.response);
      }
    );
    this._emitSearchSent(values);
  }

  /**
   * Clears the error, response, and response values from this object.
   * @param {Object} values Key-value pair parameters.
   */
  clearResponse(values) {
    this.tracking.tracking(values);

    this.response = new Response();
    this._emitResponseUpdated(this.response);
  }

  /**
   * The current response.
   * @return {Response}
   */
  getResponse() {
    return this.response;
  }

  /**
   * The analytics adaptor connected to this pipeline.
   * @return {Analytics}
   */
  getAnalytics() {
    return this.analytics;
  }
}

class Response {
  /**
   * Constructs a Response object.
   * @param {*} error
   * @param {*} queryValues
   * @param {*} response
   * @param {*} values
   */
  constructor(error, queryValues, response, values) {
    /** @private */
    this.error = error;
    /** @private */
    this.queryValues = queryValues;
    /** @private */
    this.response = response;
    /** @private */
    this.values = values;
  }

  /**
   * Is this response empty?
   * @return {boolean}
   */
  isEmpty() {
    return (
      this.error === undefined &&
      this.response === undefined &&
      this.values === undefined &&
      this.queryValues === undefined
    );
  }

  /**
   * Is this response an error?
   * @return {boolean}
   */
  isError() {
    return this.error !== undefined;
  }

  /**
   * The error associated with this response.
   * @return {string|undefined}
   */
  getError() {
    return this.error;
  }

  /**
   * Return the query values used in the search which created this response.
   * @return {Object|undefined}
   */
  getQueryValues() {
    return this.queryValues;
  }

  /**
   * Returns the response, which includes results and aggregates etc.
   * @return {Object|undefined}
   */
  getResponse() {
    return this.response;
  }

  /**
   * Return the pipeline values returned by the search.
   * @return {Object|undefined}
   */
  getValues() {
    return this.values;
  }

  /**
   * Return results from the response.
   * @return {Object|undefined}
   */
  getResults() {
    return this.response !== undefined ? this.response.results : undefined;
  }

  /**
   * Return the total number of results.
   * @return {number|undefined}
   */
  getTotalResults() {
    return this.response !== undefined
      ? parseInt(this.response.totalResults, 10)
      : undefined;
  }

  /**
   * Return time from the response.
   * @return {string|undefined}
   */
  getTime() {
    return this.response !== undefined ? this.response.time : undefined;
  }

  /**
   * Return the aggregates in the response.
   * @return {Object|undefined}
   */
  getAggregates() {
    if (this.response === undefined || this.response.aggregates === undefined) {
      return undefined;
    }
    return this.response.aggregates;
  }
}

export { Pipeline, Response };

{'fraction_non_alphanumeric': 0.13080237379074872, 'fraction_numerical': 0.0006503536297861963, 'mean_word_length': 2.663490172721858, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 12, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '3073358', 'n_tokens_mistral': 3600, 'n_tokens_neox': 3535, 'n_words': 947}