snapbot 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aa9a6b0cbc5334ac7078254f385f9c859c0ea22d7ca4821741bade0b4c32cca6
-  data.tar.gz: 8e69a33b99977466a92b9772abaa2e2d889a0d7ff200ed81c5effb01581eb014
+  metadata.gz: bb7768e6e3ac46cacf84d569a0d8026062418160d39c5ae1e214875ac0f4b26d
+  data.tar.gz: 50467bc231c453c7c81129181f90c21759f5acb2f121fbaf07bfd5313dcaea80
 SHA512:
-  metadata.gz: 1de519ae9f8e26a44d081682b83ca89d101daf00f5665c7d04a4ed1bb7c8ee701a72b476b45d9d5760ff2621e498d8bbe0ddfb3bde386fc4f188a16cee3c82db
-  data.tar.gz: 6f549408fbffc23fc5f96f620dc0be49df614b84ce0e5c913fb375422bcec052f36e065a63080f03be3a2992c7f7a519873ab0efcb31bd9610e5b2f2bdf2340e
+  metadata.gz: 0331b6031f3fe4e3ee09aef66a1030bfec968f5df8c7308c2dd09933e0c63cfad55e54dab10e7bf0e9dc568216d34bbd3f8ac32c5bd6e524750ad676fbb49988
+  data.tar.gz: 06757de1f42e455f6b8565661426bba4a94e609d567a5d2a2c6d8cdb8be8e5aed4b16e97334a009d34bda3b400707b2796a9a4b88b507a407b58e1ca8df2a798

data/.rubocop.yml

@@ -2,11 +2,19 @@ AllCops:
   TargetRubyVersion: 2.6
   SuggestExtensions: false
 
+Metrics/AbcSize:
+  Exclude:
+    - spec/support/fixture_database.rb
+
 Metrics/BlockLength:
   Exclude:
     - spec/**/*_spec.rb
     - snapbot.gemspec
 
+Metrics/MethodLength:
+  Exclude:
+    - spec/support/fixture_database.rb
+
 Style/StderrPuts:
   Exclude:
     - lib/snapbot/diagram/renderer.rb

data/lib/snapbot/diagram/dot_generator/relationship.rb

@@ -11,7 +11,7 @@ module Snapbot
         self.destination = destination
       end
 
-      def equals(other)
+      def ==(other)
         source == other.source && destination == other.destination
       end
     end

data/lib/snapbot/diagram/dot_generator.rb

@@ -40,7 +40,7 @@ module Snapbot
       def collect_lets(example)
         @lets_by_value = RSpec::Lets.new(example).collect.each_with_object({}) do |sym, lets_by_value|
           value = example.send(sym) unless @ignore_lets.include?(sym)
-          lets_by_value[value] = sym if value.is_a?(Reflector::BASE_ACTIVERECORD_CLASS)
+          lets_by_value[value] = sym if value.is_a?(reflector.base_activerecord_class)
         end
       end
 

data/lib/snapbot/diagram/renderer.rb

@@ -16,6 +16,8 @@ module Snapbot
       def save
         ensure_graphviz
         FileUtils.rm(OUTPUT_FILENAME, force: true)
+        FileUtils.mkdir_p(File.dirname(OUTPUT_FILENAME))
+
         IO.popen("dot -Tsvg -o #{OUTPUT_FILENAME}", "w") do |pipe|
           pipe.puts(@dot)
         end

data/lib/snapbot/reflector.rb

@@ -5,17 +5,29 @@ require "active_record"
 module Snapbot
   # Reflect models and instances in a way that's useful for generating a diagram
   class Reflector
-    BASE_ACTIVERECORD_CLASS = defined?(ApplicationRecord) ? ApplicationRecord : ActiveRecord::Base
+    def base_activerecord_class
+      defined?(ApplicationRecord) ? ApplicationRecord : ActiveRecord::Base
+    end
 
     def models(only_with_records: true)
       @models ||= begin
         Rails.application.eager_load! if defined?(Rails)
-        BASE_ACTIVERECORD_CLASS.descendants.reject do |c|
-          c.to_s == "Schema" || (only_with_records && c.count.zero?)
+        base_activerecord_class.descendants.reject do |c|
+          activerecord_ignore?(c) || (only_with_records && c.count.zero?)
         end
       end
     end
 
+    ACTIVERECORD_IGNORE = [
+      /^Schema$/,
+      /^HABTM_/,
+      /^ActiveRecord::InternalMetadata$/,
+      /^ActiveRecord::SchemaMigration$/
+    ].freeze
+    def activerecord_ignore?(klass)
+      ACTIVERECORD_IGNORE.any? { |r| klass.name =~ r }
+    end
+
     def instances
       @instances ||= models.each_with_object([]) do |klass, array|
         array << klass.all
@@ -41,7 +53,8 @@ module Snapbot
     def reflect_associations(instance)
       (
         instance.class.reflect_on_all_associations(:has_many).reject { |a| a.name == :schemas } +
-          instance.class.reflect_on_all_associations(:has_one)
+          instance.class.reflect_on_all_associations(:has_one) +
+          instance.class.reflect_on_all_associations(:has_and_belongs_to_many)
       ).flatten
     end
 

data/lib/snapbot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Snapbot
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: snapbot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - Russell Garner
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-08-08 00:00:00.000000000 Z
+date: 2022-08-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: binding_of_caller