traver 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 338d95191adfb1ad3addd5e0d12331e772186643
-  data.tar.gz: f22f8a30c993626c0985c4f3bc5e8685caedb6e4
+  metadata.gz: 8aa831df193e85e3df16534f4013983a7a73857d
+  data.tar.gz: 8be0b0fbe6e926eed801c6dc78f87661a98c3886
 SHA512:
-  metadata.gz: c8855182f196e730cd4571c4f32a79d43c95e8d87b9ef79b90d9f827db362c48ae788b67a00083d3e3f3050fe73c0aaf8c29c4704550212bbc51b3e07a6b2524
-  data.tar.gz: 724bcaeb91174f485b5c4d6bbb36818e227c790959b357def7d56ae55edc4d40bd3224c8806f0a5a3e09f2c49f9ce51556e88d0f512b9cec11fa994b141a98e6
+  metadata.gz: 04976719f6a6d1635137bba9e70a890ef40837a7822c95ddfb6c4dfd0983cde6909fd7c7c46044ad9efa36c4c41e12f6be462c5698d5dc2d72727ee0c9f1ea10
+  data.tar.gz: 48cd4fffd40e5dd460997f4ba0bce0d636b56feef4389f1adda876fd79e3ef5d91cbcdb83e580e2ea83c4fb65cf31808ca0c3fd13cc902f56af28be6e326cd89

data/lib/traver.rb

@@ -1,6 +1,8 @@
 require "traver/version"
 require "traver/factory_definer"
 require "traver/object_creator"
+require "traver/graph"
+require "traver/graph_creator"
 
 module Traver
   class Error < Exception; end
@@ -16,4 +18,11 @@ module Traver
     
     object_creator.created_object
   end
+  
+  def self.create_graph(params = {})
+    graph_creator = GraphCreator.new(params)
+    graph_creator.create_graph
+    
+    graph_creator.graph
+  end
 end

data/lib/traver/graph.rb

@@ -0,0 +1,40 @@
+module Traver
+  class Graph
+    def initialize
+      @vertices = {}
+    end
+    
+    def add_vertex(key, object)
+      vertices[key] ||= []
+      vertices[key] << object
+    end
+    
+    def [](key)
+      key = key.to_s
+      
+      if plural?(key)
+        vertices[singularize(key).to_sym]
+      else
+        name, index = key.split(/(\d+)$/)
+        index = (index || 1).to_i
+      
+        vertices[name.to_sym][index - 1]
+      end
+    end
+    
+    def method_missing(method_name, *args)
+      self[method_name]
+    end
+    
+    private
+    attr_reader :vertices
+    
+    def plural?(value)
+      value.end_with?("s")
+    end
+    
+    def singularize(value)
+      value.chomp("s")
+    end
+  end
+end

data/lib/traver/graph_creator.rb

@@ -0,0 +1,20 @@
+module Traver
+  class GraphCreator
+    attr_reader :params
+    attr_reader :graph
+    
+    def initialize(params = {})
+      @params = params
+      @graph = Graph.new
+    end
+    
+    def create_graph
+      object_creator = ObjectCreator.new(params)
+      object_creator.after_create = lambda do |creator|
+        graph.add_vertex(creator.factory_name, creator.created_object)
+      end
+      
+      object_creator.create_object
+    end
+  end
+end

data/lib/traver/object_creator.rb

@@ -1,6 +1,7 @@
 module Traver
   class ObjectCreator
     attr_reader :factory_name, :params
+    attr_accessor :after_create
     attr_reader :created_object
     
     def initialize(options)
@@ -14,6 +15,7 @@ module Traver
       instantiate_object
       set_object_state
       persist_object
+      call_after_create_hook
     end
     
     private
@@ -53,6 +55,7 @@ module Traver
     
     def do_create_object(params)
       object_creator = ObjectCreator.new(params)
+      object_creator.after_create = after_create
       object_creator.create_object
       
       object_creator.created_object
@@ -81,5 +84,9 @@ module Traver
     def persist_object
       created_object.save if defined?(Rails)
     end
+    
+    def call_after_create_hook
+      after_create.call(self) if after_create
+    end
   end
 end

data/lib/traver/version.rb

@@ -1,3 +1,3 @@
 module Traver
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: traver
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Yuri Kaspervich
@@ -97,6 +97,8 @@ files:
 - bin/setup
 - lib/traver.rb
 - lib/traver/factory_definer.rb
+- lib/traver/graph.rb
+- lib/traver/graph_creator.rb
 - lib/traver/object_creator.rb
 - lib/traver/version.rb
 - traver.gemspec