graphable 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rvmrc

@@ -0,0 +1,2 @@
+RVM_STRING="rvm use ruby-1.9.3-p125-perf@graphable --create --verbose"
+[ -e './.rvmrc_local' ] && source './.rvmrc_local' || $RVM_STRING

data/Gemfile

@@ -0,0 +1,4 @@
+source 'http://rubygems.org'
+
+# Specify your gem's dependencies in graphable.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,7 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+
+task :default do
+  require 'graphable'
+  puts "Loaded!"
+end

data/graphable.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/graphable/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Joe Fredette"]
+  gem.email         = ["jfredett@gmail.com"]
+  gem.description   = %q{A library for extracting static graph representations of data from rails-y databases}
+  gem.summary       = %q{A library for extracting static graph representations of data from rails-y databases}
+  gem.homepage      = "http://www.github.com/jfredett/graphable"
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  gem.name          = "graphable"
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency "active_support"
+  gem.add_dependency "neography"
+
+  gem.version       = Graphable::VERSION
+end

data/lib/graphable.rb

@@ -0,0 +1,85 @@
+require 'active_support'
+Dir["./lib/graphable/*.rb"].each do |f| require f end
+
+
+module Graphable
+  extend ActiveSupport::Concern
+
+  included do
+    Graphable.register NodeCreator.new(self)
+    indexes :id
+  end
+
+  def self.register(registrant)
+    registry << registrant
+  end
+
+  def self.neo
+    Neography::Rest.new(ENV["NEO4J_URL"] || "http://localhost:7474")  
+  end
+
+  def self.registry
+    @registry ||= []
+  end
+
+  def self.build!
+    puts "Building Graph"
+    registry.select { |f| f.is_a? NodeCreator  }.map(&:call)
+    registry.select { |f| f.is_a? IndexCreator }.map(&:call)
+    registry.select { |f| f.is_a? EdgeCreator  }.map(&:call)
+    nil
+  end
+
+  def self.index_cache
+    @index_cache ||= {}
+  end
+
+  def self.completed_indicies
+    @completed_indicies ||= {}
+  end
+
+  def self.completed_index(klass, method)
+    completed_indicies[[klass, method]] = true
+  end
+
+  def self.has_indexed?(klass, method)
+    completed_indicies[[klass,method]]
+  end
+
+
+
+  module InstanceMethods
+    def to_node
+      attributes.to_hash.tap do |hash|
+        hash.each { |k, _| hash.delete(k) if k.to_s =~ /_id$/ } #remove FKs
+        hash[:type] = self.class.name
+      end
+    end
+  end
+
+  module ClassMethods
+    def index_name
+      "#{name.downcase.pluralize}_index"
+    end
+
+    def indexes(*methods)
+       methods.each do |method|
+         Graphable.register IndexCreator.new(self, method) 
+       end
+    end
+
+    def has_edge(hash = {}, &block)
+      source_klass = self
+      target_klass = hash[:to]
+      hash[:block] = block
+      if hash.has_key?(:through)
+        Graphable.register EdgeCreator.through(source_klass, target_klass, hash.delete(:through), hash)
+      elsif hash.has_key?(:via)
+        Graphable.register EdgeCreator.via(source_klass, target_klass, hash.delete(:via), hash)
+      else
+        raise "Invalid Edge type, must be :through or :via"
+      end
+    end
+  end
+
+end

data/lib/graphable/edge_creator.rb

@@ -0,0 +1,102 @@
+require 'forwardable'
+
+module Graphable
+  class EdgeCreator
+    extend Forwardable 
+
+    def self.through(*args)
+      # for has_many :through relations
+      ThroughEdgeCreator.new(*args) 
+    end
+
+    def self.via(*args)
+      #for habtm/has_many relations
+      ViaEdgeCreator.new(*args)
+    end
+
+    def initialize(source, target, method, opts)
+      @source = source
+      @target = target
+      @method = method
+      @name = opts[:name] || method
+      @metadata_proc = opts[:block]
+      @target_name = opts[:target_method]
+    end
+
+    protected 
+
+    def load_node(n)
+      binding.pry unless n
+      Graphable.index_cache[n]
+    end
+
+    def build_relationship(source_node, target_node, metadata = {})
+      # dodges an issue with neography, nil-values can't be sent up. that's
+      # okay, since trying to access a field which doesn't exist should be nil
+      # anyway.
+      metadata.reject! { |_,v| v.nil? } 
+
+      Neography::Relationship.create(@name, source_node, target_node, metadata) 
+    end
+
+    delegate :all => :@source
+    alias_method :sources, :all
+
+    def target_name
+      @target_name ||= @target.name.pluralize.downcase.to_sym
+    end
+  end
+
+  class ThroughEdgeCreator < EdgeCreator
+    def call 
+      puts "Building edges for #{@source.name} -> #{@target.name}"
+      sources.each_slice(100) do |slice|
+        slice.each do |source|
+          source_node = load_node(source)
+
+          intermediates_for(source).each do |intermediate_target|
+            metadata = @metadata_proc.call(intermediate_target) if @metadata_proc
+            metadata = intermediate_target.send(:edge_metadata) if intermediate_target.respond_to?(:edge_metadata)
+
+            target_node = load_node(intermediate_target.send(target_name)) rescue binding.pry
+
+            build_relationship(source_node, target_node, metadata || {})
+          end
+        end
+      end
+    end
+
+    def targets_for(source)
+      res = intermediates_for(source).map(&target_name)
+      res = res.to_a if res.respond_to?(:to_a)
+      if res.respond_to?(:each) then res else [res] end
+    end
+
+    def intermediates_for(source)
+      source.send(@method)
+    end
+  end
+
+  class ViaEdgeCreator < EdgeCreator
+    def call 
+      puts "Building edges for #{@source.name} -> #{@target.name}"
+      sources.each do |source|
+        source_node = load_node(source)
+
+        targets_for(source).each do |target|
+          metadata = @metadata_proc.call(target) if @metadata_proc
+
+          target_node = load_node(target) rescue binding.pry
+          build_relationship(source_node, target_node, metadata || {})
+        end
+      end
+    end
+
+    def targets_for(source)
+      res = source.send(@method)
+      res = res.to_a if res.respond_to?(:to_a)
+      if res.respond_to?(:each) then res else [res] end
+    end
+  end
+
+end

data/lib/graphable/index_creator.rb

@@ -0,0 +1,26 @@
+module Graphable
+  class IndexCreator
+    extend Forwardable
+
+    attr_reader :klass
+    delegate [:name, :all, :index_name] => :@klass
+
+    def initialize(klass, method)
+      @klass = klass
+      @method = method
+    end
+
+    def call
+      return if Graphable.has_indexed?(@klass, @method)
+
+      puts "Building index for #{name}"
+      Graphable.neo.create_node_index(index_name, 'fulltext')
+
+      all.to_a.each do |object|
+        Graphable.neo.add_node_to_index(index_name, @method, object.send(@method), Graphable.index_cache[object])
+      end
+
+      Graphable.completed_index(@klass, @method)
+    end
+  end
+end

data/lib/graphable/node_creator.rb

@@ -0,0 +1,23 @@
+require 'forwardable'
+
+module Graphable
+  class NodeCreator
+    extend Forwardable
+
+    attr_reader :klass
+    delegate [:name, :all] => :@klass
+
+    def initialize(klass)
+      @klass = klass
+    end
+
+    def call 
+      puts "Building nodes for #{name}"
+      all.each_slice(100) do |slice|
+        slice.each do |object|
+          Graphable.index_cache[object] = Neography::Node.create(object.to_node) 
+        end
+      end
+    end
+  end
+end

data/lib/graphable/version.rb

@@ -0,0 +1,3 @@
+module Graphable
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: graphable
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Joe Fredette
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-03-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: active_support
+  requirement: &2152759120 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152759120
+- !ruby/object:Gem::Dependency
+  name: neography
+  requirement: &2152758480 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2152758480
+description: A library for extracting static graph representations of data from rails-y
+  databases
+email:
+- jfredett@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rvmrc
+- Gemfile
+- Rakefile
+- graphable.gemspec
+- lib/graphable.rb
+- lib/graphable/edge_creator.rb
+- lib/graphable/index_creator.rb
+- lib/graphable/node_creator.rb
+- lib/graphable/version.rb
+homepage: http://www.github.com/jfredett/graphable
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: A library for extracting static graph representations of data from rails-y
+  databases
+test_files: []