graph_mapper.rb in grom-0.1.1

- old
+ new
@@ -1,79 +1,56 @@
 require 'grom'
+require 'rdf/turtle'
 
 module Grom
   module GraphMapper
 
-    def get_graph_data(uri)
-      ttl_data = Net::HTTP.get(URI(uri))
-      create_graph_from_ttl(ttl_data)
+    def get_ttl_data(uri)
+      Net::HTTP.get(URI(uri))
     end
 
-    def convert_to_ttl(data)
-      # p "called convert_to_ttl"
-      result = ""
-      data.each_statement do |statement|
-        result << RDF::NTriples::Writer.serialize(statement)
-      end
-      result
+    def get_id(uri)
+      uri == RDF.type.to_s ? 'type' : uri.to_s.split("/").last
     end
 
-    def create_graph_from_ttl(ttl_data)
-      # p "called create_graph_from_ttl"
-      # p ttl_data
-      ttl_data
-      graph = RDF::Graph.new
-      RDF::NTriples::Reader.new(ttl_data) do |reader|
+    def create_hash_from_ttl(ttl_data)
+      hash = {}
+      RDF::Turtle::Reader.new(ttl_data) do |reader|
         reader.each_statement do |statement|
-          graph << statement
+          statement_mapper(statement, hash)
         end
       end
-      graph
+      hash.values
     end
 
-    def get_id(uri)
-      uri == RDF.type.to_s ? 'type' : uri.to_s.split("/").last
+    def statement_mapper(statement, hash)
+      subject = get_id(statement.subject)
+      hash[subject] ||= { :id => subject }
+      hash[subject][get_id(statement.predicate).to_sym] = statement.object.to_s
     end
 
-    def get_object_and_predicate(statement)
-      predicate = get_id(statement.predicate)
-      { predicate.to_sym => statement.object.to_s }
-    end
-
-    def statements_mapper_by_subject(graph)
-      graph.subjects.map do |subject|
-        individual_graph = RDF::Graph.new
-        pattern = RDF::Query::Pattern.new(subject, :predicate, :object)
-        attributes = graph.query(pattern).map do |statement|
-          individual_graph << statement
-          get_object_and_predicate(statement)
-        end.reduce({}, :merge)
-
-        attributes.merge({id: get_id(subject), graph: individual_graph })
-      end
-    end
-
-    def split_by_subject(graph, associated_class_name)
-      associated_class_type_pattern = RDF::Query::Pattern.new(:subject, RDF.type, RDF::URI.new("#{DATA_URI_PREFIX}/schema/#{associated_class_name}"))
-      associated_graph = RDF::Graph.new
-      through_graph = RDF::Graph.new
-      graph.query(associated_class_type_pattern).subjects.map do |subject|
-        subject_pattern = RDF::Query::Pattern.new(subject, :predicate, :object)
-        graph.query(subject_pattern).each do |statement|
-          associated_graph << statement
+    def through_split_graph(ttl_data)
+      associated_hash, through_hash = {}, {}
+      RDF::Turtle::Reader.new(ttl_data) do |reader|
+        reader.each_statement do |s|
+          if (s.subject.to_s =~ URI::regexp) == 0
+            subject = get_id(s.subject)
+            associated_hash[subject] ||= { :id => subject }
+            associated_hash[subject][get_id(s.predicate).to_sym] = s.object.to_s
+          else
+            through_hash[s.subject.to_s] ||= {}
+            if get_id(s.predicate) == "connect"
+              through_hash[s.subject.to_s][:associated_object_id] = get_id(s.object)
+            elsif get_id(s.predicate) == "objectId"
+              through_hash[s.subject.to_s][:id] = get_id(s.object)
+            else
+              through_hash[s.subject.to_s][get_id(s.predicate).to_sym] = s.object.to_s
+            end
+          end
         end
-        through_graph << graph
-        through_graph.delete(associated_graph)
       end
-      { through_graph: through_graph, associated_class_graph: associated_graph }
-    end
 
-    def get_through_graphs(graph, id)
-      connection_pattern = RDF::Query::Pattern.new(:subject, :predicate, RDF::URI.new("#{DATA_URI_PREFIX}/#{id}"))
-      graph.query(connection_pattern).subjects.map do |subject|
-        subject_pattern = RDF::Query::Pattern.new(subject, :predicate, :object)
-        graph.query(subject_pattern)
-      end
-     end
+      { associated_class_hash: associated_hash, through_class_hash: through_hash }
+    end
 
   end
 end
\ No newline at end of file