@ua/react-native-airship/src/AttributeEditor.ts

Airship plugin for React Native apps.

/* Copyright Airship and Contributors */

'use strict';

/**
 * Attribute operation
 * @hidden
 */
export interface AttributeOperation {
  /**
   * The operation name.
   */
  action: string;
  /**
   * The attribute key.
   */
  key: string;
  /**
   * The attribute value, if available.
   */
  value?: string | number | Date;
  /**
   * The attribute type, if available.
   */
  type?: 'string' | 'number' | 'date';
}

/**
 * Editor for attributes.
 */
export class AttributeEditor {
  onApply: (operations: AttributeOperation[]) => Promise<void>;
  operations: AttributeOperation[];

  /**
   * AttributeEditor constructor
   *
   * @hidden
   * @param onApply The apply function
   */
  constructor(onApply: (operations: AttributeOperation[]) => Promise<void>) {
    this.onApply = onApply;
    this.operations = [];
  }

  /**
   * Adds an attribute.
   *
   * @param value The attribute value.
   * @param name The attribute name.
   * @return The attribute editor instance.
   */
  setAttribute(
    name: string,
    value: string | number | boolean | Date
  ): AttributeEditor {
    var attributeValue: string | number | Date;
    var attributeType: 'string' | 'number' | 'date';

    if (typeof value === 'boolean') {
      // No boolean attribute type. Convert value to string.
      attributeValue = value.toString();
      attributeType = 'string';
    } else {
      attributeValue = value;
      if (typeof value === 'string') {
        attributeType = 'string';
      } else if (typeof attributeValue === 'number') {
        attributeType = 'number';
      } else if (value instanceof Date) {
        // JavaScript's date type doesn't pass through the JS to native bridge.
        // Dates are instead serialized as milliseconds since epoch.
        attributeType = 'date';
        attributeValue = value.getTime();
      } else {
        throw 'Unsupported attribute type: ' + typeof attributeValue;
      }
    }

    const operation = {
      action: 'set',
      value: attributeValue,
      key: name,
      type: attributeType,
    };
    this.operations.push(operation);

    return this;
  }

  /**
   * Removes an attribute.
   * @param name The name of the attribute to remove.
   * @return The attribute editor instance.
   */
  removeAttribute(name: string): AttributeEditor {
    const operation = { action: 'remove', key: name };
    this.operations.push(operation);
    return this;
  }

  /**
   * Applies the attribute operations.
   */
  apply(): Promise<void> {
    return this.onApply(this.operations);
  }
}