module Neo4jr
  class Service < Sinatra::Base
            
    get '/info' do
      Neo4jr::DB.stats.to_json
    end
    
    get '/nodes' do
      nodes = Neo4jr::DB.execute do |neo|
        nodes = neo.all_nodes.map{|m| m.to_hash }        
      end
      nodes.to_json
    end
  
    post '/nodes' do
      node = Neo4jr::DB.execute do |neo|
        node = neo.create_node(params)
      end
      node.to_hash.to_json
    end
    
    get '/nodes/:node_id' do
      Neo4jr::DB.getNodeById(params.delete('node_id')).to_json
    end
      
    put '/nodes/:node_id' do
      node = Neo4jr::DB.execute do |neo|
        node = neo.getNodeById(params.delete('node_id'))
        node.update_properties(params)
      end
      node.to_hash.to_json
    end
    
    delete '/nodes/:node_id' do
      Neo4jr::DB.execute do |neo|
        node = neo.getNodeById(params['node_id'])
        node.delete
      end
    end
    
    get '/nodes/:node_id/relationships' do
      relationships = Neo4jr::DB.execute do |neo|
        node              = neo.getNodeById(params.delete('node_id'))
        if relationship_type = params.delete('type')
          relationship_type = RelationshipType.instance(relationship_type)
          node.getRelationships(relationship_type.to_a).hashify_objects
        else
          node.getRelationships.hashify_objects
        end
      end
      relationships.to_json
    end
    
    post '/nodes/:node_id/relationships' do
      relationships = Neo4jr::DB.execute do |neo|
        node              = neo.getNodeById(params.delete('node_id'))
        to_node           = neo.getNodeById(params.delete('to'))
        relationship_type = RelationshipType.instance(params.delete('type'))
        relationship      = node.create_relationship_to to_node, relationship_type
        relationship.update_properties(params)
        node.getRelationships(relationship_type.to_a).hashify_objects
      end
      relationships.to_json
    end

    #optional direction & depth
    get '/nodes/:node_id/path' do
      paths = Neo4jr::DB.execute do |neo|
        relationship  = Neo4jr::RelationshipType.instance(params.delete('type'))
        start_node    = neo.getNodeById(params.delete('node_id'))
        end_node      = neo.getNodeById(params.delete('to'))
        depth         = params.delete('depth') || 2
        direction     = Neo4jr::Direction.from_string(params.delete('direction') || 'both')
        shortest_path = AllSimplePaths.new(start_node, end_node, depth.to_i, direction, relationship.to_a)
        paths = shortest_path.getPaths
        paths.map{|p| p.map{|n| n.to_hash }}
      end
      paths.to_json
    end
    
    #optional
    get '/nodes/:node_id/recommendations' do
      suggestions = Neo4jr::DB.execute do |neo|
        relationship = Neo4jr::RelationshipType.incoming(params.delete('type'))
        start_node   = neo.getNodeById(params.delete('node_id'))
        level        = (params.delete('level') || 1).to_i
        order        = Order::BREADTH_FIRST
        return_when  = Return.when do |current_position|
          current_position.depth > level
        end
        traverser = start_node.traverse(order, Stop.at(level + 1), return_when, relationship)
        traverser.map{|node| node.to_hash }
      end
      suggestions.to_json
    end
  end
end