@skypilot/scraper/lib/scraper/ScriptBuilder.js

Node-based scriptable web scraper

"use strict";

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

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

var _normalizeCommands = require("./normalizeCommands");

var _normalizeQuery = require("./normalizeQuery");

var _normalizeQueryDict = require("./normalizeQueryDict");

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; }

class ScriptBuilder {
  constructor() {
    _defineProperty(this, "_commands", []);
  }

  get commands() {
    return (0, _normalizeCommands.normalizeCommands)(this._commands);
  }

  chain(builder) {
    this._commands.push(...builder.commands);

    return this;
  }

  click(queryOrSelector, options) {
    this._commands.push(_objectSpread({
      action: 'click',
      query: (0, _normalizeQuery.normalizeQuery)(queryOrSelector)
    }, (0, _sugarbowl.mergeIf)(options, {
      options
    })));

    return this;
  }

  follow(queryOrSelector, options) {
    this._commands.push(_objectSpread({
      action: 'follow',
      query: (0, _normalizeQuery.normalizeQuery)(queryOrSelector)
    }, (0, _sugarbowl.mergeIf)(options, {
      options
    })));

    return this;
  }

  goTo(url) {
    this._commands.push({
      action: 'goTo',
      url
    });

    return this;
  }

  query(queryDict, options) {
    this._commands.push(_objectSpread({
      action: 'query',
      queryDict: (0, _normalizeQueryDict.normalizeQueryDict)(queryDict)
    }, (0, _sugarbowl.mergeIf)(options, {
      options
    })));

    return this;
  }

  runOnAll(queryOrSelector, commandsOrBuilder, options) {
    this._commands.push(_objectSpread({
      action: 'runOnAll',
      query: (0, _normalizeQuery.normalizeQuery)(queryOrSelector),
      commands: (0, _normalizeCommands.normalizeCommands)(commandsOrBuilder)
    }, (0, _sugarbowl.mergeIf)(options, {
      options
    })));

    return this;
  }

  set(dict) {
    this._commands.push({
      action: 'set',
      state: dict
    });

    return this;
  }

  write(collectionName) {
    this._commands.push(_objectSpread({
      action: 'write'
    }, (0, _sugarbowl.mergeIf)(collectionName, {
      collectionName
    })));

    return this;
  }

}

exports.ScriptBuilder = ScriptBuilder;