rails-schema 0.1.4 → 0.1.6

data/lib/rails/schema/assets/style.css

@@ -99,7 +99,7 @@ body {
 #toolbar {
   position: fixed;
   top: 0;
-  left: var(--sidebar-width);
+  left: 0;
   right: 0;
   height: var(--toolbar-height);
   background: var(--bg-secondary);
@@ -108,7 +108,7 @@ body {
   align-items: center;
   padding: 0 16px;
   gap: 8px;
-  z-index: 100;
+  z-index: 200;
 }
 
 #toolbar button {
@@ -135,6 +135,13 @@ body {
   text-align: center;
 }
 
+.toolbar-divider {
+  width: 1px;
+  height: 24px;
+  background: var(--border-color);
+  margin: 0 4px;
+}
+
 #toolbar .shortcuts-hint {
   font-size: 11px;
   color: var(--text-muted);
@@ -148,7 +155,8 @@ body {
   border-right: 1px solid var(--border-color);
   display: flex;
   flex-direction: column;
-  height: 100vh;
+  margin-top: var(--toolbar-height);
+  height: calc(100vh - var(--toolbar-height));
   z-index: 101;
 }
 
@@ -163,15 +171,39 @@ body {
   margin-bottom: 12px;
 }
 
+#search-wrapper {
+  position: relative;
+}
+
 #search-input {
   width: 100%;
-  padding: 8px 12px;
+  padding: 8px 28px 8px 12px;
   border: 1px solid var(--border-color);
   border-radius: 6px;
   background: var(--bg-primary);
   color: var(--text-primary);
   font-size: 13px;
   outline: none;
+  box-sizing: border-box;
+}
+
+#search-clear {
+  position: absolute;
+  right: 4px;
+  top: 50%;
+  transform: translateY(-50%);
+  background: none;
+  border: none;
+  color: var(--text-secondary);
+  font-size: 16px;
+  cursor: pointer;
+  padding: 2px 6px;
+  line-height: 1;
+  display: none;
+}
+
+#search-clear:hover {
+  color: var(--text-primary);
 }
 
 #search-input:focus {
@@ -200,6 +232,10 @@ body {
   background: var(--bg-tertiary);
 }
 
+#collapse-groups-btn {
+  margin-left: auto;
+}
+
 #model-list {
   flex: 1;
   overflow-y: auto;
@@ -346,6 +382,14 @@ body {
   pointer-events: none;
 }
 
+.edge-label.belongs_to { fill: var(--edge-belongs-to); }
+.edge-label.has_many { fill: var(--edge-has-many); }
+.edge-label.has_one { fill: var(--edge-has-one); }
+.edge-label.has_and_belongs_to_many { fill: var(--edge-habtm); }
+.edge-label.embeds_many { fill: var(--edge-embeds-many); }
+.edge-label.embeds_one { fill: var(--edge-embeds-one); }
+.edge-label.embedded_in { fill: var(--edge-embedded-in); }
+
 .edge-label-bg {
   fill: var(--bg-primary);
   opacity: 0.85;
@@ -430,6 +474,10 @@ body {
 .column-list .col-name {
   font-family: "SF Mono", "Fira Code", monospace;
   font-weight: 500;
+  overflow: hidden;
+  text-overflow: ellipsis;
+  white-space: nowrap;
+  min-width: 0;
 }
 
 .column-list .col-name.pk {
@@ -440,6 +488,8 @@ body {
   color: var(--text-muted);
   font-family: "SF Mono", "Fira Code", monospace;
   font-size: 11px;
+  flex-shrink: 0;
+  margin-left: 8px;
 }
 
 .assoc-list {
@@ -499,6 +549,15 @@ body {
   gap: 4px;
 }
 
+.legend-toggle {
+  cursor: pointer;
+}
+
+.legend-toggle input[type="checkbox"] {
+  margin: 0;
+  cursor: pointer;
+}
+
 .legend-line {
   width: 20px;
   height: 2px;
@@ -545,3 +604,59 @@ body {
   );
   background-color: transparent;
 }
+
+/* Sidebar group headers */
+.group-header {
+  display: flex;
+  align-items: center;
+  padding: 6px 16px;
+  font-size: 12px;
+  font-weight: 700;
+  border-bottom: 1px solid var(--border-color);
+  color: var(--text-secondary);
+  text-transform: uppercase;
+  letter-spacing: 0.5px;
+  cursor: pointer;
+  user-select: none;
+  gap: 6px;
+}
+
+.group-header:hover {
+  background: var(--bg-tertiary);
+}
+
+.group-header.active {
+  background: var(--accent-light);
+}
+
+.group-count {
+  font-size: 11px;
+  color: var(--text-secondary);
+  font-weight: normal;
+  margin-left: auto;
+  flex-shrink: 0;
+}
+
+.group-toggle {
+  font-size: 10px;
+  width: 12px;
+  flex-shrink: 0;
+}
+
+.group-color-dot {
+  width: 10px;
+  height: 10px;
+  border-radius: 50%;
+  display: inline-block;
+  flex-shrink: 0;
+}
+
+.group-header-label {
+  overflow: hidden;
+  text-overflow: ellipsis;
+  white-space: nowrap;
+}
+
+.group-models {
+  border-bottom: 1px solid var(--border-color);
+}

data/lib/rails/schema/assets/template.html.erb

@@ -11,11 +11,15 @@
     <div id="sidebar">
       <div id="sidebar-header">
         <h1><%= title %></h1>
-        <input type="text" id="search-input" placeholder="Search models... ( / )" autocomplete="off">
+        <div id="search-wrapper">
+          <input type="text" id="search-input" placeholder="Search models... ( / )" autocomplete="off">
+          <button id="search-clear" type="button" aria-label="Clear search">&times;</button>
+        </div>
       </div>
       <div id="sidebar-actions">
         <button id="select-all-btn">Select All</button>
         <button id="deselect-all-btn">Deselect All</button>
+        <button id="collapse-groups-btn" style="display:none" title="Collapse all groups">&#9660;</button>
       </div>
       <div id="model-list"></div>
     </div>
@@ -28,7 +32,7 @@
         <span class="legend-item"><span class="legend-line habtm"></span> habtm (M:M)</span>
         <span class="legend-item" data-mode="mongoid"><span class="legend-line embeds_many"></span> embeds_many</span>
         <span class="legend-item" data-mode="mongoid"><span class="legend-line embeds_one"></span> embeds_one</span>
-        <span class="legend-item" data-mode="active_record"><span class="legend-line through"></span> :through</span>
+        <label class="legend-item legend-toggle" data-mode="active_record"><input type="checkbox" id="toggle-through" checked><span class="legend-line through"></span> :through</label>
         <span class="legend-item"><span class="legend-line polymorphic"></span> polymorphic</span>
       </div>
       <div class="spacer"></div>
@@ -37,6 +41,8 @@
       <button id="zoom-out-btn" title="Zoom out (-)">−</button>
       <button id="fit-btn" title="Fit to screen (F)">Fit</button>
       <button id="theme-btn" title="Toggle theme">Theme</button>
+      <span class="toolbar-divider"></span>
+      <button id="export-mermaid-btn" title="Export as Mermaid (Markdown)">Export Mermaid</button>
       <span class="shortcuts-hint">/ search · Esc deselect · F fit</span>
     </div>
 

data/lib/rails/schema/configuration.rb

@@ -3,15 +3,31 @@
 module Rails
   module Schema
     class Configuration
-      attr_accessor :output_path, :exclude_models, :title, :theme, :expand_columns, :schema_format
+      attr_accessor :output_path, :exclude_models, :exclude_model_if, :title, :theme, :expand_columns, :schema_format,
+                    :model_schema_group, :collapse_groups, :show_through_edges
 
       def initialize
         @output_path = "docs/schema.html"
         @exclude_models = []
+        @exclude_model_if = nil
         @title = "Database Schema"
         @theme = :auto
         @expand_columns = false
         @schema_format = :auto
+        @model_schema_group = nil
+        @collapse_groups = true
+        @show_through_edges = true
+      end
+
+      def resolved_group_proc
+        case @model_schema_group
+        when nil then nil
+        when :namespaces
+          ->(model) { model.name.split("::")[0..-2] }
+        when Proc then @model_schema_group
+        else
+          raise ArgumentError, "model_schema_group must be nil, :namespaces, or a Proc"
+        end
       end
     end
   end

data/lib/rails/schema/extractor/model_scanner.rb

@@ -36,13 +36,9 @@ module Rails
 
         def eager_load_via_zeitwerk!
           loader = ::Rails.autoloaders.main
-          models_path = ::Rails.root&.join("app", "models")&.to_s
-
-          if models_path && File.directory?(models_path) && loader.respond_to?(:eager_load_dir)
-            loader.eager_load_dir(models_path)
-          else
-            loader.eager_load
-          end
+          dirs = model_directories
+          dirs.each { |dir| loader.eager_load_dir(dir) } if dirs.any? && loader.respond_to?(:eager_load_dir)
+          loader.eager_load if no_concrete_descendants?
         rescue StandardError, LoadError => e
           warn "[rails-schema] Zeitwerk eager_load failed (#{e.class}: #{e.message}), " \
                "trying Rails.application.eager_load!"
@@ -60,22 +56,29 @@ module Rails
         def eager_load_model_files!
           return unless defined?(::Rails.root) && ::Rails.root
 
-          models_path = ::Rails.root.join("app", "models")
-          return unless models_path.exist?
+          dirs = model_directories
+          return if dirs.empty?
 
-          Dir.glob(models_path.join("**/*.rb")).sort.each do |file|
+          dirs.flat_map { |d| Dir.glob(File.join(d, "**/*.rb")) }.sort.each do |file|
             require file
           rescue StandardError, LoadError => e
             warn "[rails-schema] Could not load #{file}: #{e.class}: #{e.message}"
           end
         end
 
+        def model_directories
+          dirs = PackwerkDiscovery.new.model_paths
+          default_path = ::Rails.root&.join("app", "models")&.to_s
+          dirs << default_path if default_path && !dirs.include?(default_path)
+          dirs.select { |d| d && File.directory?(d) }.uniq
+        end
+
+        def no_concrete_descendants?
+          ActiveRecord::Base.descendants.none? { |m| !m.abstract_class? && m.name }
+        end
+
         def table_known?(model)
-          if @schema_data
-            @schema_data.key?(model.table_name)
-          else
-            safe_table_exists?(model)
-          end
+          @schema_data ? @schema_data.key?(model.table_name) : safe_table_exists?(model)
         end
 
         def safe_table_exists?(model)
@@ -86,12 +89,10 @@ module Rails
         end
 
         def excluded?(model)
+          return true if @configuration.exclude_model_if&.call(model)
+
           @configuration.exclude_models.any? do |pattern|
-            if pattern.end_with?("*")
-              model.name.start_with?(pattern.delete_suffix("*"))
-            else
-              model.name == pattern
-            end
+            pattern.end_with?("*") ? model.name.start_with?(pattern.delete_suffix("*")) : model.name == pattern
           end
         end
 

data/lib/rails/schema/extractor/mongoid/model_scanner.rb

@@ -120,6 +120,8 @@ module Rails
           end
 
           def excluded?(model)
+            return true if @configuration.exclude_model_if&.call(model)
+
             @configuration.exclude_models.any? do |pattern|
               if pattern.end_with?("*")
                 model.name.start_with?(pattern.delete_suffix("*"))

data/lib/rails/schema/extractor/packwerk_discovery.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Rails
+  module Schema
+    module Extractor
+      class PackwerkDiscovery
+        def model_paths
+          return [] unless defined?(::Rails.root) && ::Rails.root
+
+          find_package_directories.flat_map do |dir|
+            %w[app/models app/public].map { |sub| File.join(dir, sub) }
+          end
+        end
+
+        private
+
+        def find_package_directories
+          patterns = package_paths
+          return [] if patterns.empty?
+
+          patterns.flat_map do |pattern|
+            Dir.glob(::Rails.root.join(pattern).to_s).select do |path|
+              File.directory?(path) && File.exist?(File.join(path, "package.yml"))
+            end
+          end
+        end
+
+        def package_paths
+          packwerk_yml = ::Rails.root.join("packwerk.yml")
+          return [] unless File.exist?(packwerk_yml)
+
+          require "yaml"
+          config = YAML.safe_load(File.read(packwerk_yml)) || {}
+          config["package_paths"] || ["**/"]
+        rescue StandardError
+          []
+        end
+      end
+    end
+  end
+end

data/lib/rails/schema/renderer/html_generator.rb

@@ -61,7 +61,10 @@ module Rails
         def config_json
           config = {
             expand_columns: @configuration.expand_columns,
-            theme: @configuration.theme.to_s
+            theme: @configuration.theme.to_s,
+            grouping_enabled: !@configuration.model_schema_group.nil?,
+            collapse_groups: @configuration.collapse_groups,
+            show_through_edges: @configuration.show_through_edges
           }
           JSON.generate(config).gsub("</", '<\/')
         end

data/lib/rails/schema/transformer/edge.rb

@@ -5,6 +5,7 @@ module Rails
     module Transformer
       class Edge
         attr_reader :from, :to, :association_type, :label, :foreign_key, :through, :polymorphic
+        attr_accessor :reverse_label, :reverse_association_type
 
         def initialize(from:, to:, association_type:, label:, foreign_key: nil, through: nil, polymorphic: false)
           @from = from
@@ -14,6 +15,8 @@ module Rails
           @foreign_key = foreign_key
           @through = through
           @polymorphic = polymorphic
+          @reverse_label = nil
+          @reverse_association_type = nil
         end
 
         def to_h
@@ -24,7 +27,9 @@ module Rails
             label: @label,
             foreign_key: @foreign_key,
             through: @through,
-            polymorphic: @polymorphic
+            polymorphic: @polymorphic,
+            reverse_label: @reverse_label,
+            reverse_association_type: @reverse_association_type
           }
         end
       end

data/lib/rails/schema/transformer/graph_builder.rb

@@ -6,18 +6,19 @@ module Rails
   module Schema
     module Transformer
       class GraphBuilder
-        def initialize(column_reader: Extractor::ColumnReader.new, association_reader: Extractor::AssociationReader.new)
+        def initialize(column_reader: Extractor::ColumnReader.new, association_reader: Extractor::AssociationReader.new,
+                       configuration: ::Rails::Schema.configuration)
           @column_reader = column_reader
           @association_reader = association_reader
+          @group_proc = configuration.resolved_group_proc
         end
 
         def build(models)
           model_ids = assign_unique_ids(models)
-          name_to_id = {}
-          model_ids.each { |m, uid| name_to_id[m.name] ||= uid }
+          name_to_id = build_name_to_id(model_ids)
 
           nodes = model_ids.map { |m, uid| build_node(m, uid) }
-          edges = model_ids.flat_map { |m, uid| build_edges(m, uid, name_to_id) }
+          edges = deduplicate_edges(model_ids.flat_map { |m, uid| build_edges(m, uid, name_to_id) })
 
           {
             nodes: nodes.map(&:to_h),
@@ -28,6 +29,10 @@ module Rails
 
         private
 
+        def build_name_to_id(model_ids)
+          model_ids.each_with_object({}) { |(m, uid), map| map[m.name] ||= uid }
+        end
+
         def assign_unique_ids(models)
           counts = models.group_by(&:name).transform_values(&:size)
           models.map do |m|
@@ -37,10 +42,12 @@ module Rails
         end
 
         def build_node(model, unique_id)
+          group = @group_proc ? Array(@group_proc.call(model)) : []
           Node.new(
             id: unique_id,
             table_name: model.table_name,
-            columns: @column_reader.read(model)
+            columns: @column_reader.read(model),
+            group: group
           )
         end
 
@@ -60,6 +67,51 @@ module Rails
           end
         end
 
+        def deduplicate_edges(edges)
+          edges = deduplicate_habtm(edges)
+          deduplicate_has_many_belongs_to(edges)
+        end
+
+        def deduplicate_habtm(edges)
+          habtm_first = {}
+
+          edges.each_with_object([]) do |edge, result|
+            next result << edge unless edge.association_type == "has_and_belongs_to_many"
+
+            pair = [edge.from, edge.to].sort
+            if habtm_first.key?(pair)
+              habtm_first[pair].reverse_label = edge.label
+            else
+              habtm_first[pair] = edge
+              result << edge
+            end
+          end
+        end
+
+        def deduplicate_has_many_belongs_to(edges)
+          hm_lookup = build_has_many_lookup(edges)
+
+          edges.each_with_object([]) do |edge, result|
+            if edge.association_type == "belongs_to"
+              hm_edge = hm_lookup[[edge.from, edge.to, edge.foreign_key]]
+              if hm_edge
+                hm_edge.reverse_label = edge.label
+                hm_edge.reverse_association_type = edge.association_type
+                next
+              end
+            end
+            result << edge
+          end
+        end
+
+        def build_has_many_lookup(edges)
+          edges.each_with_object({}) do |edge, lookup|
+            next unless %w[has_many has_one].include?(edge.association_type)
+
+            lookup[[edge.to, edge.from, edge.foreign_key]] = edge
+          end
+        end
+
         def build_metadata(models)
           {
             generated_at: Time.now.utc.iso8601,

data/lib/rails/schema/transformer/node.rb

@@ -4,20 +4,19 @@ module Rails
   module Schema
     module Transformer
       class Node
-        attr_reader :id, :table_name, :columns
+        attr_reader :id, :table_name, :columns, :group
 
-        def initialize(id:, table_name:, columns: [])
+        def initialize(id:, table_name:, columns: [], group: [])
           @id = id
           @table_name = table_name
           @columns = columns
+          @group = group
         end
 
         def to_h
-          {
-            id: @id,
-            table_name: @table_name,
-            columns: @columns
-          }
+          h = { id: @id, table_name: @table_name, columns: @columns }
+          h[:group] = @group unless @group.empty?
+          h
         end
       end
     end

data/lib/rails/schema/version.rb

@@ -2,6 +2,6 @@
 
 module Rails
   module Schema
-    VERSION = "0.1.4"
+    VERSION = "0.1.6"
   end
 end

data/lib/rails/schema.rb

@@ -6,6 +6,7 @@ require_relative "schema/transformer/node"
 require_relative "schema/transformer/edge"
 require_relative "schema/extractor/schema_file_parser"
 require_relative "schema/extractor/structure_sql_parser"
+require_relative "schema/extractor/packwerk_discovery"
 require_relative "schema/extractor/model_scanner"
 require_relative "schema/extractor/column_reader"
 require_relative "schema/extractor/association_reader"

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: rails-schema
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.6
 platform: ruby
 authors:
 - Andrei Kislichenko
-autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-03-08 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -80,6 +79,7 @@ files:
 - lib/rails/schema/extractor/mongoid/column_reader.rb
 - lib/rails/schema/extractor/mongoid/model_adapter.rb
 - lib/rails/schema/extractor/mongoid/model_scanner.rb
+- lib/rails/schema/extractor/packwerk_discovery.rb
 - lib/rails/schema/extractor/schema_file_parser.rb
 - lib/rails/schema/extractor/structure_sql_parser.rb
 - lib/rails/schema/railtie.rb
@@ -97,7 +97,6 @@ metadata:
   source_code_uri: https://github.com/andrew2net/rails-schema
   changelog_uri: https://github.com/andrew2net/rails-schema/blob/main/CHANGELOG.md
   rubygems_mfa_required: 'true'
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -112,8 +111,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.6
-signing_key:
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Interactive HTML visualization of your Rails database schema
 test_files: []