@skypilot/scraper/lib/scraper/State.js

Node-based scriptable web scraper

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.State = void 0;

var _sugarbowl = require("@skypilot/sugarbowl");

var _deepmerge = _interopRequireDefault(require("deepmerge"));

var _rfdc = _interopRequireDefault(require("rfdc"));

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); keys.push.apply(keys, symbols); } return keys; }

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

const deepClone = (0, _rfdc.default)();

class State {
  constructor(initialData) {
    _defineProperty(this, "data", void 0);

    this.data = initialData ? deepClone(initialData) : {};
  }

  clear(options = {}) {
    const {
      statePath
    } = options;

    if (statePath) {
      this.set({}, {
        statePath
      });
    } else {
      this.data = {};
    }

    return this.data;
  }

  clone() {
    return new State(this.get());
  }

  get() {
    return deepClone(this.data);
  }

  set(data, options = {}) {
    const {
      statePath
    } = options;
    this.data = statePath ? _objectSpread(_objectSpread({}, this.data), {}, {
      [statePath]: deepClone(data)
    }) : deepClone(data);
    return this.get();
  }

  update(dataToMerge, options = {}) {
    const {
      statePath
    } = options;
    this.data = (0, _sugarbowl.omitUndefined)((0, _deepmerge.default)(this.data, statePath ? {
      [statePath]: dataToMerge
    } : dataToMerge));
    return this.get();
  }

}

exports.State = State;