foxx_generator/foxx_generator/repository_with_graph.js

Generate Foxx APIs using Statecharts and Domain Driven Design

(function () {
  'use strict';
  var Foxx = require('org/arangodb/foxx'),
    _ = require('underscore'),
    RepositoryWithGraph;

  RepositoryWithGraph = Foxx.Repository.extend({
    allWithNeighbors: function (options) {
      var results = this.all(options);
      _.each(results, function (result) {
        this.addLinks(result);
      }, this);
      return results;
    },

    byIdWithNeighbors: function (key) {
      var result = this.byId(key);
      this.addLinks(result);
      return result;
    },

    removeByKey: function (key) {
      this.collection.remove(key);
    },

    addLinks: function (model) {
      var links = {},
        graph = this.graph,
        relations = _.filter(this.relations, function (relation) { return relation.type === 'follow'; });

      _.each(relations, function (relation) {
        var neighbors = graph.neighbors(model.get('_id'), {
          edgeCollectionRestriction: [relation.edgeCollectionName]
        });

        if (relation.cardinality === 'one' && neighbors.length > 0) {
          links[relation.name] = neighbors[0]._key;
        } else if (relation.cardinality === 'many') {
          links[relation.name] = _.pluck(neighbors, '_key');
        }
      });

      model.set('links', links);
    }
  });

  exports.RepositoryWithGraph = RepositoryWithGraph;
}());