gremlin-script/src/functions/steps/collectionstep.js

Generate Gremlin scripts with ease

'use strict';
var inherits = require('util').inherits;

var _ = require('lodash');

var GremlinStep = require('./step');


function CollectionStep() {
  GremlinStep.apply(this, arguments);
}

inherits(CollectionStep, GremlinStep);

CollectionStep.prototype.toGroovy = function() {
  var str = '.'+ this.name;
  var argumentList = [];
  var args = this.arguments;
  var firstArg = args[0];

  if (_.isArray(firstArg)) {
    // Handle .method([g.v(1), g.v(2)]) signature
    var pipelines = _.map(firstArg, function(a) { return a.toGroovy(); }).join(',');
    str += "([" + pipelines + "])";
  } else {
    str += "('"+ this.arguments[0] + "')";
  }

  return str;
};

module.exports = CollectionStep;