/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 * 
 * @format
 */
'use strict';

/**
 * An implementation of the `MutableRecordSource` interface (defined in
 * `RelayStoreTypes`) that holds all records in memory.
 */
var RelayInMemoryRecordSource =
/*#__PURE__*/
function () {
  function RelayInMemoryRecordSource(records) {
    this._records = records || {};
  }

  var _proto = RelayInMemoryRecordSource.prototype;

  _proto.clear = function clear() {
    this._records = {};
  };

  _proto["delete"] = function _delete(dataID) {
    this._records[dataID] = null;
  };

  _proto.get = function get(dataID) {
    return this._records[dataID];
  };

  _proto.getRecordIDs = function getRecordIDs() {
    return Object.keys(this._records);
  };

  _proto.getStatus = function getStatus(dataID) {
    if (!this._records.hasOwnProperty(dataID)) {
      return require("./RelayRecordState").UNKNOWN;
    }

    return this._records[dataID] == null ? require("./RelayRecordState").NONEXISTENT : require("./RelayRecordState").EXISTENT;
  };

  _proto.has = function has(dataID) {
    return this._records.hasOwnProperty(dataID);
  };

  _proto.load = function load(dataID, callback) {
    callback(null, this.get(dataID));
  };

  _proto.remove = function remove(dataID) {
    delete this._records[dataID];
  };

  _proto.set = function set(dataID, record) {
    this._records[dataID] = record;
  };

  _proto.size = function size() {
    return Object.keys(this._records).length;
  };

  _proto.toJSON = function toJSON() {
    return this._records;
  };

  return RelayInMemoryRecordSource;
}();

module.exports = RelayInMemoryRecordSource;