schema_doc 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: de7315ab4e254938328c0acfdce54a58f9169c329d5c473ffaad3b81aca91675
-  data.tar.gz: 5a6b26ed9c8b57a03ae054f0e68b66885951d5cda75fe187b7cfeb441b95881c
+  metadata.gz: eadb829eb9f3581debcd1a1916db1f39ef23782a3f10257394453ba9e28d82d9
+  data.tar.gz: 0ac48d9c7a82bac4d46435065bd1064fd1bea25b5c25f7d3a351131d54b2c6ed
 SHA512:
-  metadata.gz: d8bd9c1154c5be36875336f29a90f0ffc85e1b1c43a1d3664f05b0b2d18b0498022d1e6e85e0f101c483829d1983cfcd9cb9a0b99a5970ab2ae04abb5def625e
-  data.tar.gz: b331c3d7ec6e178460dab62eb9f1a2c80383cbf7949b46178be2e280984517b878dbac406485a149a3ec6d10e2dc62820eb07d76fc61a43c5defdadc08f6db61
+  metadata.gz: cf247b70e5250c9504f19e7c48bcd7ac15846e099fbcd3e10d4241b5c235c57e1619f381302d6ac1d165f6a91e64fda1a0f222f97b7d3491ccdb08f24ba02e8a
+  data.tar.gz: ff8843e7a210fece539ec0aab79dc363f47c1a0e3e2bf5a0f81f457e6108d9beaeccbcba79e244fba9fe65bbf9d2a1d1ae75393cc377761e025be373388fcaa9

data/README.md

@@ -38,6 +38,16 @@ end
 
 SchemaDoc will output [THIS](https://github.com/mizoR/schema_doc/tree/master/example/sample_output.md).
 
+## Optional Configuration
+
+If you want to customize the models that are listed, you can create a `config/initializers/schema_doc.rb` as shown in the example below:
+
+```ruby
+SchemaDoc.configure do |config|
+  config.model_classes = Proc.new { ApplicationRecord.descendants.select{ |klass| klass.superclass.name == ApplicationRecord.name } }
+end
+```
+
 ## Contributing
 
 1. Fork it

data/lib/schema_doc.rb

@@ -1,8 +1,25 @@
+require "schema_doc/configuration"
 require "schema_doc/version"
 require "schema_doc/document"
 require "schema_doc/railtie" if defined?(Rails)
 
 module SchemaDoc
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  def self.reset
+    @configuration = Configuration.new
+  end
+
+  def self.configure
+    yield(configuration)
+  end
+
   def self.root
     File.expand_path('../..', __FILE__)
   end

data/lib/schema_doc/configuration.rb

@@ -0,0 +1,9 @@
+module SchemaDoc
+  class Configuration
+    attr_accessor :model_classes
+    
+    def initialize
+      @model_classes = Proc.new { ActiveRecord::Base.descendants }
+    end
+  end
+end

data/lib/schema_doc/document.rb

@@ -29,7 +29,7 @@ module SchemaDoc
       end
 
       def model_classes
-        ActiveRecord::Base.descendants
+        SchemaDoc.configuration.model_classes.try(:call)
       end
 
       def model_classes_group_by_table_name

data/lib/schema_doc/version.rb

@@ -1,3 +1,3 @@
 module SchemaDoc
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: schema_doc
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - mizokami
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-07-07 00:00:00.000000000 Z
+date: 2020-07-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -96,6 +96,7 @@ files:
 - Rakefile
 - example/sample_output.md
 - lib/schema_doc.rb
+- lib/schema_doc/configuration.rb
 - lib/schema_doc/document.rb
 - lib/schema_doc/railtie.rb
 - lib/schema_doc/tasks.rake