all files / src/ memo.js

94.96% Statements 113/119
94.03% Branches 63/67
90.48% Functions 19/21
92% Lines 69/75
1 statement, 4 branches Ignored     
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246                      1×       1×       1×       1×       1×                     1× 76× 76×   76×   20×   14× 9×   15× 8×     26×   8× 3×   8×   1×               6×                           9×   1×     3×     5×                     14×   14× 4×     10× 10× 10×   1×       9×         9×           15× 5×   10× 2×         26×   26× 2×     24× 24× 9× 6×   3× 15× 7×   8×     10× 2×                 20×                 15×                 14×                 13×                 13×               24×   12×   6×   4×   1×   1×                   7×   1×   3×   1×   1×     1× 1×            
import {default as xdr} from "./generated/digitalbits-xdr_generated";
import isUndefined from "lodash/isUndefined";
import isNull from "lodash/isNull";
import isString from "lodash/isString";
import clone from "lodash/clone";
import {UnsignedHyper} from "js-xdr";
import BigNumber from 'bignumber.js';
 
/**
 * Type of {@link Memo}.
 */
export const MemoNone = "none";
/**
 * Type of {@link Memo}.
 */
export const MemoID = "id";
/**
 * Type of {@link Memo}.
 */
export const MemoText = "text";
/**
 * Type of {@link Memo}.
 */
export const MemoHash = "hash";
/**
 * Type of {@link Memo}.
 */
export const MemoReturn = "return";
 
/**
 * `Memo` represents memos attached to transactions.
 *
 * @param {string} type - `MemoNone`, `MemoID`, `MemoText`, `MemoHash` or `MemoReturn`
 * @param {*} value - `string` for `MemoID`, `MemoText`, buffer of hex string for `MemoHash` or `MemoReturn`
 * @see [Transactions concept](https://developer.digitalbits.io/learn/concepts/transactions.html)
 * @class Memo
 */
export class Memo {
  constructor(type, value = null) {
    this._type = type;
    this._value = value;
 
    switch (this._type) {
      case MemoNone:
        break;
      case MemoID:
        Memo._validateIdValue(value);
        break;
      case MemoText:
        Memo._validateTextValue(value);
        break;
      case MemoHash:
      case MemoReturn:
        Memo._validateHashValue(value);
        // We want MemoHash and MemoReturn to have Buffer as a value
        if (isString(value)) {
          this._value = new Buffer(value, 'hex');
        }
        break;
      default:
        throw new Error("Invalid memo type");
    }
  }
 
  /**
   * Contains memo type: `MemoNone`, `MemoID`, `MemoText`, `MemoHash` or `MemoReturn`
   */
  get type() {
    return clone(this._type);
  }
 
  set type(type) {
    throw new Error("Memo is immutable");
  }
 
  /**
   * Contains memo value:
   * * `null` for `MemoNone`,
   * * `string` for `MemoID`, `MemoText`,
   * * `Buffer` for `MemoHash`, `MemoReturn`
   */
  get value() {
    switch (this._type) {
      case MemoNone:
        return null;
      case MemoID:
      case MemoText:
        return clone(this._value);
      case MemoHash:
      case MemoReturn:
        return new Buffer(this._value);
      default:
        throw new Error("Invalid memo type");
    }
  }
 
  set value(value) {
    throw new Error("Memo is immutable");
  }
 
  static _validateIdValue(value) {
    let error = new Error("Expects a int64 as a string. Got " + value);
 
    if (!isString(value)) {
      throw error;
    }
 
    let number;
    try {
      number = new BigNumber(value);
    } catch (e) {
      throw error;
    }
 
    // Infinity
    Iif (!number.isFinite()) {
      throw error;
    }
 
    // NaN
    Iif (number.isNaN()) {
      throw error;
    }
  }
 
  static _validateTextValue(value) {
    if (!isString(value)) {
      throw new Error("Expects string type got " + typeof(value));
    }
    if (Buffer.byteLength(value, "utf8") > 28) {
      throw new Error("Text should be <= 28 bytes. Got " + Buffer.byteLength(value, "utf8"));
    }
  }
 
  static _validateHashValue(value) {
    let error = new Error("Expects a 32 byte hash value or hex encoded string. Got " + value);
 
    if (value === null || isUndefined(value)) {
      throw error;
    }
 
    let valueBuffer;
    if (isString(value)) {
      if (!/^[0-9A-Fa-f]{64}$/g.test(value)) {
        throw error;
      }
      valueBuffer = new Buffer(value, 'hex');
    } else if (Buffer.isBuffer(value)) {
      valueBuffer = new Buffer(value);
    } else {
      throw error;
    }
 
    if (!valueBuffer.length || valueBuffer.length != 32) {
      throw error;
    }
  }
 
  /**
   * Returns an empty memo (`MemoNone`).
   * @returns {Memo}
   */
  static none() {
    return new Memo(MemoNone);
  }
 
  /**
   * Creates and returns a `MemoText` memo.
   * @param {string} text - memo text
   * @returns {Memo}
   */
  static text(text) {
    return new Memo(MemoText, text);
  }
 
  /**
   * Creates and returns a `MemoID` memo.
   * @param {string} id - 64-bit number represented as a string
   * @returns {Memo}
   */
  static id(id) {
    return new Memo(MemoID, id);
  }
 
  /**
   * Creates and returns a `MemoHash` memo.
   * @param {array|string} hash - 32 byte hash or hex encoded string
   * @returns {Memo}
   */
  static hash(hash) {
    return new Memo(MemoHash, hash);
  }
 
  /**
   * Creates and returns a `MemoReturn` memo.
   * @param {array|string} hash - 32 byte hash or hex encoded string
   * @returns {Memo}
   */
  static return(hash) {
    return new Memo(MemoReturn, hash);
  }
 
  /**
   * Returns XDR memo object.
   * @returns {xdr.Memo}
   */
  toXDRObject() {
    switch (this._type) {
      case MemoNone:
        return xdr.Memo.memoNone();
      case MemoID:
        return xdr.Memo.memoId(UnsignedHyper.fromString(this._value));
      case MemoText:
        return xdr.Memo.memoText(this._value);
      case MemoHash:
        return xdr.Memo.memoHash(this._value);
      case MemoReturn:
        return xdr.Memo.memoReturn(this._value);
    }
  }
 
  /**
   * Returns {@link Memo} from XDR memo object.
   * @param {xdr.Memo}
   * @returns {Memo}
   */
  static fromXDRObject(object) {
    switch (object.arm()) {
      case "id":
        return Memo.id(object.value().toString());
      case "text":
        return Memo.text(object.value());
      case "hash":
        return Memo.hash(object.value());
      case "retHash":
        return Memo.return(object.value());
    }
 
    Eif (typeof object.value() === "undefined") {
      return Memo.none();
    }
 
    throw new Error("Unknown type");
  }
}