rails_graph 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 49ba4bf9a585c25b02f0b15717a240503ed5edcf61fc751df0f782a5d5fb94c5
-  data.tar.gz: 64610ec973417724321cc32dbfbc8dd7688474625b7c3df354dd9fd760220874
+  metadata.gz: 2aa9e5821416a34e7eca9ba18199bea6007190821c50cabb84147db6232b06e2
+  data.tar.gz: f54cb43940dba30d7ddc5915cb8669601d0e05b9b5eb4582c18a504859101ed0
 SHA512:
-  metadata.gz: 753af0fc868143bdb6c6636fc9a51bb4254486851f65308ad8dea0ee83f60a97a6d699d5050c41b2c9750c654940a81603c9d740b0f624ff919edc851e70b037
-  data.tar.gz: bc15a08314849217fa3e71c8ab9d9640c5b611b56c9ec9f98b22009487e6c6fce2b436b733a304cb7414bcb0081a98b393741c41007a7752bde4d391719a782e
+  metadata.gz: 714e4e3d097f8fda7958665d9f4f8ee0ca78978ecab77be2a59ad84fa1921c3204ae93d111072882068ce33ee3966786a3d7473d787e275077d61785f105815a
+  data.tar.gz: f3889f7437868743a38971f1808e2b3e9d6ac8445a8567f3e96ebb7722d35a24795830f220bdf7eeb1940bfefe472c1cc33f2539a4bd380358af006ad4c62035

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rails_graph (0.1.0)
+    rails_graph (0.3.0)
       activerecord (>= 6.0, < 7.1)
       neo4j-ruby-driver (>= 4.4.0.alpha.8)
       railties (>= 6.0, < 7.1)
@@ -9,83 +9,83 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    actionpack (7.0.4)
-      actionview (= 7.0.4)
-      activesupport (= 7.0.4)
-      rack (~> 2.0, >= 2.2.0)
+    actionpack (7.0.5)
+      actionview (= 7.0.5)
+      activesupport (= 7.0.5)
+      rack (~> 2.0, >= 2.2.4)
       rack-test (>= 0.6.3)
       rails-dom-testing (~> 2.0)
       rails-html-sanitizer (~> 1.0, >= 1.2.0)
-    actionview (7.0.4)
-      activesupport (= 7.0.4)
+    actionview (7.0.5)
+      activesupport (= 7.0.5)
       builder (~> 3.1)
       erubi (~> 1.4)
       rails-dom-testing (~> 2.0)
       rails-html-sanitizer (~> 1.1, >= 1.2.0)
-    activemodel (7.0.4)
-      activesupport (= 7.0.4)
-    activerecord (7.0.4)
-      activemodel (= 7.0.4)
-      activesupport (= 7.0.4)
-    activesupport (7.0.4)
+    activemodel (7.0.5)
+      activesupport (= 7.0.5)
+    activerecord (7.0.5)
+      activemodel (= 7.0.5)
+      activesupport (= 7.0.5)
+    activesupport (7.0.5)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 1.6, < 2)
       minitest (>= 5.1)
       tzinfo (~> 2.0)
     ast (2.4.2)
-    async (2.1.0)
+    async (2.5.1)
       console (~> 1.10)
-      io-event (~> 1.0.0)
+      io-event (~> 1.1)
       timers (~> 4.1)
-    async-io (1.34.0)
+    async-io (1.34.3)
       async
-    async-pool (0.3.12)
-      async (>= 1.25)
     builder (3.2.4)
-    concurrent-ruby (1.1.10)
-    console (1.15.3)
+    concurrent-ruby (1.2.2)
+    connection_pool (2.4.1)
+    console (1.16.2)
       fiber-local
     crass (1.0.6)
     diff-lcs (1.5.0)
-    erubi (1.11.0)
+    erubi (1.12.0)
     fiber-local (1.0.0)
-    i18n (1.12.0)
+    i18n (1.13.0)
       concurrent-ruby (~> 1.0)
-    io-event (1.0.9)
+    io-event (1.2.2)
     json (2.6.2)
-    loofah (2.19.0)
+    loofah (2.21.3)
       crass (~> 1.0.2)
-      nokogiri (>= 1.5.9)
+      nokogiri (>= 1.12.0)
     method_source (1.0.0)
-    mini_portile2 (2.8.0)
-    minitest (5.16.3)
-    neo4j-ruby-driver (4.4.0.alpha.8)
+    mini_portile2 (2.8.2)
+    minitest (5.18.0)
+    neo4j-ruby-driver (4.4.4)
       activesupport
       async-io
-      async-pool
+      connection_pool
       zeitwerk (>= 2.1.10)
-    nokogiri (1.13.8)
-      mini_portile2 (~> 2.8.0)
+    nokogiri (1.15.2)
+      mini_portile2 (~> 2.8.2)
       racc (~> 1.4)
-    nokogiri (1.13.8-arm64-darwin)
+    nokogiri (1.15.2-arm64-darwin)
       racc (~> 1.4)
-    nokogiri (1.13.8-x86_64-linux)
+    nokogiri (1.15.2-x86_64-linux)
       racc (~> 1.4)
     parallel (1.22.1)
     parser (3.1.2.1)
       ast (~> 2.4.1)
-    racc (1.6.0)
-    rack (2.2.4)
-    rack-test (2.0.2)
+    racc (1.6.2)
+    rack (2.2.7)
+    rack-test (2.1.0)
       rack (>= 1.3)
     rails-dom-testing (2.0.3)
       activesupport (>= 4.2.0)
       nokogiri (>= 1.6)
-    rails-html-sanitizer (1.4.3)
-      loofah (~> 2.3)
-    railties (7.0.4)
-      actionpack (= 7.0.4)
-      activesupport (= 7.0.4)
+    rails-html-sanitizer (1.6.0)
+      loofah (~> 2.21)
+      nokogiri (~> 1.14)
+    railties (7.0.5)
+      actionpack (= 7.0.5)
+      activesupport (= 7.0.5)
       method_source
       rake (>= 12.2)
       thor (~> 1.0)
@@ -120,12 +120,12 @@ GEM
     rubocop-ast (1.21.0)
       parser (>= 3.1.1.0)
     ruby-progressbar (1.11.0)
-    thor (1.2.1)
-    timers (4.3.4)
-    tzinfo (2.0.5)
+    thor (1.2.2)
+    timers (4.3.5)
+    tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
     unicode-display_width (2.3.0)
-    zeitwerk (2.6.0)
+    zeitwerk (2.6.8)
 
 PLATFORMS
   arm64-darwin-21

data/README.md

@@ -1,5 +1,7 @@
 # RailsGraph
 
+[![Gem Version](https://badge.fury.io/rb/rails_graph.svg)](https://badge.fury.io/rb/rails_graph)
+
 Graph visualization for Rails applications using [Neo4j Browser](https://neo4j.com/docs/browser-manual/current/). 
 
 Gain visibility on your Rails application data model and identify bottlenecks. It is an ERD, but interactive and queryable using [Cypher](https://neo4j.com/developer/cypher/).
@@ -8,6 +10,7 @@ Gain visibility on your Rails application data model and identify bottlenecks. I
 
 ## Features
 
+- Application Databases and Tables
 - ActiveRecord Models
     - Metadata, e.g size, indexes
     - Columns (disabled by default)
@@ -20,6 +23,7 @@ Gain visibility on your Rails application data model and identify bottlenecks. I
         - Support for Polymorphic Associations
 - ActiveRecord Abstract models
 - Class Hierarchy
+- [Packwerk](https://github.com/Shopify/packwerk) packages (disabled by default)
 
 ## Installation
 
@@ -59,10 +63,19 @@ RailsGraph.configure do |config|
   config.include_classes = [ActsAsTaggableOn::Tag, ActsAsTaggableOn::Tagging]
 
   # Configure Columns parsing
+  # default false
   config.columns = true
 
   # Configure Class Hierarchy parsing
   config.inheritance = true
+
+  # Configure Database entities
+  # default true
+  config.database = false
+
+  # Configure inclusion of Packwerk packages
+  # default false
+  config.include_packwerk = true
 end
 ```
 

data/lib/rails_graph/commands/build_graph.rb

@@ -6,16 +6,26 @@ require_relative "../graph/relationship"
 
 require_relative "../graph/nodes/abstract_model"
 require_relative "../graph/nodes/column"
+require_relative "../graph/nodes/database"
 require_relative "../graph/nodes/model"
+require_relative "../graph/nodes/pack"
+require_relative "../graph/nodes/table"
 require_relative "../graph/nodes/virtual_model"
 
 require_relative "../graph/relationships/association"
 require_relative "../graph/relationships/attribute"
 require_relative "../graph/relationships/inheritance"
+require_relative "../graph/relationships/pack_dependency"
+require_relative "../graph/relationships/pack_model"
 
 require_relative "../helpers/associations"
 require_relative "../helpers/models"
 
+require_relative "./builders/associations"
+require_relative "./builders/models"
+require_relative "./builders/packs"
+require_relative "./builders/databases"
+
 module RailsGraph
   module Commands
     class BuildGraph
@@ -24,89 +34,33 @@ module RailsGraph
       end
 
       def call
-        polymorphic_node = RailsGraph::Graph::Nodes::VirtualModel.new("PolymorphicModel")
-        graph.add_node(polymorphic_node)
+        setup_generic_nodes
 
-        active_record_base_node = RailsGraph::Graph::Nodes::AbstractModel.new(ActiveRecord::Base)
-        graph.add_node(active_record_base_node)
-
-        build_model_nodes
-        build_associations_relationships
-        build_column_nodes if configuration.columns?
-        build_inheritance_relationships if configuration.inheritance?
+        RailsGraph::Commands::Builders::Databases.enrich(graph: graph) if configuration.databases?
+        RailsGraph::Commands::Builders::Models.enrich(inspector: inspector, graph: graph, classes: classes,
+                                                      configuration: configuration)
+        RailsGraph::Commands::Builders::Packs.enrich(graph: graph) if configuration.include_packwerk?
 
         graph
       end
 
       private
 
-      attr_reader :configuration, :classes, :graph
+      attr_reader :configuration, :classes, :graph, :inspector
 
       def initialize(configuration:)
         @configuration = configuration
         @classes = ActiveRecord::Base.descendants + configuration.include_classes
         @graph = RailsGraph::Graph::Graph.new
+        @inspector = RailsGraph::Inspector.new
       end
 
-      def build_model_nodes
-        classes.each do |model|
-          if model.abstract_class
-            node = RailsGraph::Graph::Nodes::AbstractModel.new(model)
-            graph.add_node(node)
-            next
-          end
-
-          node = RailsGraph::Graph::Nodes::Model.new(model)
-          graph.add_node(node)
-        end
-      end
-
-      def build_column_nodes
-        processed = Hash.new(false)
-
-        classes.each do |model|
-          next if model.attribute_names.empty?
-
-          identifier = RailsGraph::Helpers::Models.identifier(model)
-          node = graph.node(identifier)
-
-          next if processed[node.id]
-
-          processed[node.id] = true
-
-          model.columns.each do |column|
-            column_node = RailsGraph::Graph::Nodes::Column.new(column)
-            graph.add_node(column_node)
-
-            relationship = RailsGraph::Graph::Relationships::Attribute.new(node, column_node)
-            graph.add_relationship(relationship)
-          end
-        end
-      end
-
-      def build_inheritance_relationships
-        classes.each do |model|
-          identifier = RailsGraph::Helpers::Models.identifier(model)
-          node = graph.node(identifier)
-
-          superclass_node_identifier = RailsGraph::Helpers::Models.identifier(model.superclass)
-          superclass_node = graph.node(superclass_node_identifier)
-
-          relationship = RailsGraph::Graph::Relationships::Inheritance.new(node, superclass_node)
-          graph.add_relationship(relationship)
-        end
-      end
-
-      def build_associations_relationships
-        classes.each do |model|
-          model.reflect_on_all_associations.each do |association|
-            source_node = RailsGraph::Helpers::Associations.source_node(graph, association)
-            target_node = RailsGraph::Helpers::Associations.target_node(graph, association)
+      def setup_generic_nodes
+        polymorphic_node = RailsGraph::Graph::Nodes::VirtualModel.new("PolymorphicModel")
+        graph.add_node(polymorphic_node)
 
-            relationship = RailsGraph::Graph::Relationships::Association.new(association, source_node, target_node)
-            graph.add_relationship(relationship)
-          end
-        end
+        active_record_base_node = RailsGraph::Graph::Nodes::AbstractModel.new(ActiveRecord::Base)
+        graph.add_node(active_record_base_node)
       end
     end
   end

data/lib/rails_graph/commands/builders/associations.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+module RailsGraph
+  module Commands
+    module Builders
+      class Associations
+        attr_reader :graph, :inspector
+
+        def self.enrich(...)
+          new(...).enrich
+        end
+
+        def enrich
+          build_associations_relationships
+        end
+
+        private
+
+        attr_reader :classes
+
+        def initialize(inspector:, graph:, classes:)
+          @inspector = inspector
+          @graph = graph
+          @classes = classes
+        end
+
+        def build_associations_relationships
+          classes.each do |model|
+            model.reflect_on_all_associations.each do |association|
+              association.check_validity!
+
+              source_node, target_node = fetch_association_nodes(graph, association)
+              next report_invalid_class(model) if source_node.nil? || target_node.nil?
+
+              add_relationship(association, source_node, target_node)
+            rescue ActiveRecord::ActiveRecordError => _e
+              report_invalid_association(association)
+            end
+          end
+        end
+
+        def fetch_association_nodes(graph, association)
+          source_node = RailsGraph::Helpers::Associations.source_node(graph, association)
+          target_node = RailsGraph::Helpers::Associations.target_node(graph, association)
+          [source_node, target_node]
+        end
+
+        def add_relationship(association, source_node, target_node)
+          relationship = RailsGraph::Graph::Relationships::Association.new(association, source_node, target_node)
+          graph.add_relationship(relationship)
+        end
+
+        def report_invalid_association(association)
+          inspector.add_association(association)
+        end
+
+        def report_invalid_class(klass)
+          inspector.add_class(klass)
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/commands/builders/databases.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module RailsGraph
+  module Commands
+    module Builders
+      class Databases
+        def self.enrich(graph:)
+          new(graph: graph).enrich
+
+          graph
+        end
+
+        def enrich
+          build_databases_nodes
+          build_tables_nodes
+        end
+
+        private
+
+        attr_reader :graph
+
+        def initialize(graph:)
+          @graph = graph
+        end
+
+        def build_databases_nodes
+          databases.each do |config|
+            node = RailsGraph::Graph::Nodes::Database.new(config)
+            graph.add_node(node)
+          end
+        end
+
+        def build_tables_nodes
+          ActiveRecord::Base.connection_handler.connection_pools.each do |connection_pool|
+            database_name = connection_pool.db_config.name
+            database_node = graph.node("database_#{database_name}")
+
+            connection_pool.connection.tables.each do |table|
+              table_node = RailsGraph::Graph::Nodes::Table.new(table, database_name)
+              graph.add_node(table_node)
+
+              add_persisted_in_relationship(table_node, database_node)
+            end
+          end
+        end
+
+        def add_persisted_in_relationship(table_node, database_node)
+          relationship = RailsGraph::Graph::Relationship.new(
+            source: table_node,
+            target: database_node,
+            label: "PersistedIn",
+            name: "persisted_in",
+            properties: {}
+          )
+
+          graph.add_relationship(relationship)
+        end
+
+        def databases
+          @databases ||= ActiveRecord::Base.configurations.configs_for(env_name: Rails.env)
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/commands/builders/models.rb

@@ -0,0 +1,118 @@
+# frozen_string_literal: true
+
+module RailsGraph
+  module Commands
+    module Builders
+      class Models
+        attr_reader :graph, :inspector
+
+        def self.enrich(...)
+          new(...).enrich
+        end
+
+        def enrich
+          build_model_nodes
+          build_associations
+          build_column_nodes if configuration.columns?
+          build_model_table_relationships if configuration.databases?
+          build_inheritance_relationships if configuration.inheritance?
+
+          inspector.log
+          graph
+        end
+
+        private
+
+        attr_reader :classes, :configuration
+
+        def initialize(inspector:, graph:, classes:, configuration:)
+          @inspector = inspector
+          @graph = graph
+          @configuration = configuration
+          @classes = classes
+        end
+
+        def build_model_nodes
+          classes.each do |model|
+            if model.abstract_class
+              node = RailsGraph::Graph::Nodes::AbstractModel.new(model)
+              graph.add_node(node)
+              next
+            end
+
+            node = RailsGraph::Graph::Nodes::Model.new(model)
+            graph.add_node(node)
+          end
+        end
+
+        def build_associations
+          Builders::Associations.enrich(inspector: inspector, graph: graph, classes: classes)
+        end
+
+        def build_column_nodes
+          processed = Hash.new(false)
+
+          classes.each do |model|
+            next if model.attribute_names.empty?
+
+            identifier = RailsGraph::Helpers::Models.identifier(model)
+            node = graph.node(identifier)
+
+            next if processed[node.id]
+
+            processed[node.id] = true
+            add_column_nodes(model: model, node: node)
+          end
+        end
+
+        def build_inheritance_relationships
+          classes.each do |model|
+            identifier = RailsGraph::Helpers::Models.identifier(model)
+            node = graph.node(identifier)
+
+            superclass_node_identifier = RailsGraph::Helpers::Models.identifier(model.superclass)
+            superclass_node = graph.node(superclass_node_identifier)
+
+            relationship = RailsGraph::Graph::Relationships::Inheritance.new(node, superclass_node)
+            graph.add_relationship(relationship)
+          end
+        end
+
+        def build_model_table_relationships
+          classes.each do |model|
+            database_name = model.connection_pool.db_config.name
+            table_node = graph.node("table_#{database_name}.#{model.table_name}")
+
+            next unless table_node
+
+            identifier = RailsGraph::Helpers::Models.identifier(model)
+            node = graph.node(identifier)
+
+            relationship = build_represents_table_relationship(node, table_node)
+            graph.add_relationship(relationship)
+          end
+        end
+
+        def build_represents_table_relationship(model, table)
+          RailsGraph::Graph::Relationship.new(
+            source: model,
+            target: table,
+            label: "RepresentsTable",
+            name: "represents_table",
+            properties: {}
+          )
+        end
+
+        def add_column_nodes(model:, node:)
+          model.columns.each do |column|
+            column_node = RailsGraph::Graph::Nodes::Column.new(column)
+            graph.add_node(column_node)
+
+            relationship = RailsGraph::Graph::Relationships::Attribute.new(node, column_node)
+            graph.add_relationship(relationship)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/commands/builders/packs.rb

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+module RailsGraph
+  module Commands
+    module Builders
+      class Packs
+        def self.enrich(graph:)
+          new(graph: graph).enrich
+
+          graph
+        end
+
+        def enrich
+          build_pack_nodes
+          build_pack_relationships
+          build_pack_model_relationships
+        end
+
+        private
+
+        attr_reader :graph
+
+        def initialize(graph:)
+          @graph = graph
+        end
+
+        def build_pack_nodes
+          packs.each do |pack|
+            name, owner = fetch_pack_attributes(pack)
+            node = RailsGraph::Graph::Nodes::Pack.new(name, owner)
+            graph.add_node(node)
+          end
+        end
+
+        def build_pack_relationships
+          packs.each do |pack|
+            source_identifier = pack.name
+            source_node = graph.node(source_identifier)
+
+            pack.dependencies.each do |dep|
+              target_identifier = dep.gsub("packs/", "")
+              target_node = graph.node(target_identifier)
+
+              relationship = RailsGraph::Graph::Relationships::PackDependency.new(source_node, target_node)
+              graph.add_relationship(relationship)
+            end
+          end
+        end
+
+        def build_pack_model_relationships
+          packs.each do |pack|
+            next if pack.name == "goals"
+
+            models = fetch_pack_models(pack: pack)
+            models.each do |model|
+              add_model_relationship(pack: pack, model: model)
+            end
+          end
+        end
+
+        def packs
+          @packs ||=
+            Packwerk::PackageSet.load_all_from("packs").reject do |pack|
+              pack.name == "."
+            end
+        end
+
+        def fetch_pack_attributes(pack)
+          [
+            pack.name,
+            pack.config["metadata"]["owner"] || pack.config["metadata"]["stewards"]
+          ]
+        end
+
+        def fetch_pack_models(pack:)
+          generic_path = "packs/#{pack.name}/app/models"
+          Dir["#{generic_path}/**/*.rb"].map do |path|
+            path
+              .gsub(generic_path, "")
+              .gsub(".rb", "")
+              .prepend(pack.name)
+              .camelize
+              .constantize
+          end
+        end
+
+        def add_model_relationship(pack:, model:)
+          return unless model.respond_to?(:descends_from_active_record?)
+
+          source_identifier = RailsGraph::Helpers::Models.identifier(model)
+          target_identifier = pack.name
+
+          source_node = graph.node(source_identifier)
+          target_node = graph.node(target_identifier)
+
+          relationship = RailsGraph::Graph::Relationships::PackModel.new(source_node, target_node)
+          graph.add_relationship(relationship)
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/configuration.rb

@@ -3,10 +3,12 @@
 module RailsGraph
   class Configuration
     attr_reader :include_classes
-    attr_writer :columns, :inheritance
+    attr_writer :columns, :inheritance, :include_packwerk, :databases
 
     def initialize
       @include_classes = []
+      @include_packwerk = false
+      @databases = true
       @columns = false
       @inheritance = true
     end
@@ -22,5 +24,13 @@ module RailsGraph
     def inheritance?
       @inheritance
     end
+
+    def include_packwerk?
+      @include_packwerk
+    end
+
+    def databases?
+      @databases
+    end
   end
 end

data/lib/rails_graph/exporters/base.rb

@@ -2,6 +2,6 @@
 
 module RailsGraph
   module Exporters
-    class Base; end
+    class Base; end # rubocop:disable Lint/EmptyClass
   end
 end

data/lib/rails_graph/exporters/neo4j.rb

@@ -11,9 +11,7 @@ module RailsGraph
 
         ::Neo4j::Driver::GraphDatabase.driver(host, auth) do |driver|
           driver.session do |session|
-            queries = Cypher.build_queries(graph)
-            queries.shift
-            queries = queries.join("\n")
+            queries = build_query(graph)
 
             session.write_transaction do |tx|
               tx.run(queries, message: "Success!")
@@ -21,6 +19,12 @@ module RailsGraph
           end
         end
       end
+
+      def self.build_query(graph)
+        queries = Cypher.build_queries(graph)
+        queries.shift
+        queries.join("\n")
+      end
     end
   end
 end

data/lib/rails_graph/graph/nodes/database.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require_relative "../node"
+
+module RailsGraph
+  module Graph
+    module Nodes
+      class Database < Node
+        attr_reader :configs
+
+        def initialize(configs)
+          @configs = configs
+
+          super(labels: "Database", name: configs.name, properties: build_properties)
+        end
+
+        def identifier
+          "database_#{name}"
+        end
+
+        private
+
+        def build_properties
+          {
+            pool_size: configs.pool,
+            adapter: configs.adapter,
+            schema_cache_path: configs.schema_cache_path,
+            replica: configs.replica? || false,
+            database_name: configs.database,
+            reaping_frequency: configs.reaping_frequency,
+            min_threads: configs.min_threads,
+            max_threads: configs.max_threads
+          }
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/graph/nodes/pack.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require_relative "../node"
+
+module RailsGraph
+  module Graph
+    module Nodes
+      class Pack < Node
+        attr_reader :pack_owner
+
+        def initialize(name, owner)
+          @pack_owner = owner
+
+          super(labels: "Pack", name: name, properties: build_properties)
+        end
+
+        def identifier
+          name
+        end
+
+        private
+
+        def build_properties
+          { owner: pack_owner }
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/graph/nodes/table.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative "../node"
+
+module RailsGraph
+  module Graph
+    module Nodes
+      class Table < Node
+        attr_reader :name, :database_name
+
+        def initialize(name, database_name)
+          @database_name = database_name
+
+          super(labels: "Table", name: name, properties: build_properties)
+        end
+
+        def identifier
+          "table_#{database_name}.#{name}"
+        end
+
+        private
+
+        def build_properties
+          {
+            database_name: database_name
+          }
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/graph/relationships/pack_dependency.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative "../relationship"
+
+module RailsGraph
+  module Graph
+    module Relationships
+      class PackDependency < Relationship
+        def initialize(source, target)
+          super(
+            source: source,
+            target: target,
+            label: "DependsOnPack",
+            name: target.name
+          )
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/graph/relationships/pack_model.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative "../relationship"
+
+module RailsGraph
+  module Graph
+    module Relationships
+      class PackModel < Relationship
+        def initialize(source, target)
+          super(
+            source: source,
+            target: target,
+            label: "PartOfPack",
+            name: target.name
+          )
+        end
+      end
+    end
+  end
+end

data/lib/rails_graph/helpers/models.rb

@@ -6,6 +6,8 @@ module RailsGraph
       module_function
 
       def identifier(model)
+        return unless model
+
         model.to_s.delete_prefix("::")
       end
     end

data/lib/rails_graph/inspector.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "set"
+
+module RailsGraph
+  class Inspector
+    def initialize
+      @classes = Set.new
+      @associations = Set.new
+    end
+
+    def add_class(klass)
+      @classes.add(klass)
+    end
+
+    def add_association(association)
+      @associations.add(association)
+    end
+
+    def log
+      @classes.each do |klass|
+        puts "[WARN][RailsGraph]: Invalid class configuration: #{klass}"
+      end
+
+      @associations.each do |association|
+        puts "[WARN][RailsGraph]: #{association.name} association defined " \
+             "under #{association.active_record.name} class has invalid configurations!"
+      end
+    end
+  end
+end

data/lib/rails_graph/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsGraph
-  VERSION = "0.1.0"
+  VERSION = "0.3.0"
 end

data/lib/rails_graph.rb

@@ -7,6 +7,7 @@ require "rails_graph/railtie"
 
 require "rails_graph/commands/build_graph"
 require "rails_graph/commands/export_graph"
+require "rails_graph/inspector"
 
 module RailsGraph
   module_function

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_graph
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Ahmad Elassuty
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-10-01 00:00:00.000000000 Z
+date: 2023-05-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -85,6 +85,10 @@ files:
 - graph.svg
 - lib/rails_graph.rb
 - lib/rails_graph/commands/build_graph.rb
+- lib/rails_graph/commands/builders/associations.rb
+- lib/rails_graph/commands/builders/databases.rb
+- lib/rails_graph/commands/builders/models.rb
+- lib/rails_graph/commands/builders/packs.rb
 - lib/rails_graph/commands/export_graph.rb
 - lib/rails_graph/configuration.rb
 - lib/rails_graph/error.rb
@@ -97,15 +101,21 @@ files:
 - lib/rails_graph/graph/node.rb
 - lib/rails_graph/graph/nodes/abstract_model.rb
 - lib/rails_graph/graph/nodes/column.rb
+- lib/rails_graph/graph/nodes/database.rb
 - lib/rails_graph/graph/nodes/model.rb
+- lib/rails_graph/graph/nodes/pack.rb
+- lib/rails_graph/graph/nodes/table.rb
 - lib/rails_graph/graph/nodes/virtual_model.rb
 - lib/rails_graph/graph/relationship.rb
 - lib/rails_graph/graph/relationships/association.rb
 - lib/rails_graph/graph/relationships/attribute.rb
 - lib/rails_graph/graph/relationships/inheritance.rb
+- lib/rails_graph/graph/relationships/pack_dependency.rb
+- lib/rails_graph/graph/relationships/pack_model.rb
 - lib/rails_graph/helpers/associations.rb
 - lib/rails_graph/helpers/models.rb
 - lib/rails_graph/helpers/options_parser.rb
+- lib/rails_graph/inspector.rb
 - lib/rails_graph/railtie.rb
 - lib/rails_graph/tasks/rails_graph.rake
 - lib/rails_graph/version.rb