rails-mermaid_erd 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 70e02f24014b859b87f976e91a1ac6c7d9a2abb55b645993fcdba7280b027d03
-  data.tar.gz: ba506228775db665a9637d11d0fdf1b8af14a3f7db80da610e54e36ec4ac77c1
+  metadata.gz: 1a0b049ccbaa1ec74fd4f0bc0c55e9b1ca066ac4f9e7b624591e1626edbbcb47
+  data.tar.gz: 0c2a7bf4cb12c29f7623e2ceaef53de385f9a2dd9f6d3596786849623e9979c6
 SHA512:
-  metadata.gz: ccb7f93e4c2b7063cfe9d3ac394a5cb05268c956986616930356bba4a703f06e1b18da281ecc29f78f102444c5b26645c0cf6c109803b39279e02da3d11c6ac5
-  data.tar.gz: e35178ebb61084b313e1e3586732e45834b14a73d7b16af17dadfc6de8bed921f8c24ce3f58a22a6d412783efce02788c5c1b25ce8b4de0452b856e3273f6719
+  metadata.gz: 7894f5a5b2717404fb8593295138466678158de760e4704235dbd64ba968dfb180334df4610f615edf670df564d7aca1bfa869a6ba5f54e033883dba80add955
+  data.tar.gz: 44011954159232651066fb8a12b58ea27877055a12a7c677641dd37981bfb98ecf3d3caed1c0797ee0fb38f3d61bc8e9050585782d31203d262c5dede90fee46

data/lib/rails-mermaid_erd/version.rb

@@ -1,3 +1,3 @@
 module RailsMermaidErd
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/lib/rails-mermaid_erd.rb

@@ -24,6 +24,9 @@ module RailsMermaidErd
     ::Rails.application.eager_load!
     ::ActiveRecord::Base.descendants.each do |defined_model|
       next unless defined_model.table_exists?
+
+      next if defined_model.name.include?("HABTM_")
+
       model = {
         TableName: defined_model.table_name,
         ModelName: defined_model.name,
@@ -91,6 +94,25 @@ module RailsMermaidErd
         end
       end
 
+      defined_model.reflect_on_all_associations(:has_and_belongs_to_many).each do |h|
+        reverse_relation = if h.options[:class_name]
+          result[:Relations].find { |r| r[:RightModelName] == model[:ModelName] && r[:LeftModelName] == h.options[:class_name] }
+        else
+          result[:Relations].find { |r| r[:RightModelName] == model[:ModelName] && r[:LeftModelName] == h.name.to_s.classify }
+        end
+        if reverse_relation
+          reverse_relation[:Comment] = "HABTM"
+        else
+          result[:Relations] << {
+            LeftModelName: model[:ModelName],
+            LeftValue: "}o",
+            RightModelName: h.name.to_s.classify,
+            RightValue: "o{",
+            Comment: "HABTM"
+          }
+        end
+      end
+
       defined_model.reflect_on_all_associations(:belongs_to).each do |h|
         if h.options[:class_name]
           reverse_relation = result[:Relations].find { |r| r[:RightModelName] == model[:ModelName] && r[:LeftModelName] == h.options[:class_name] }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails-mermaid_erd
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - 肥溜め
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-10-10 00:00:00.000000000 Z
+date: 2022-10-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails