koedame · shifakhan · Oct 9, 2024 · Oct 9, 2024
diff --git a/docs/example.yml b/docs/example.yml
@@ -1,3 +1,4 @@
 # result path
 # default "mermaid_erd/index.html"
 result_path: doc/erd.html
+all_models_selected: false
diff --git a/lib/rails-mermaid_erd.rb b/lib/rails-mermaid_erd.rb
@@ -22,6 +22,7 @@ def configuration
     version = VERSION
     app_name = ::Rails.application.class.try(:parent_name) || ::Rails.application.class.try(:module_parent_name)
     logo = File.read(File.expand_path("./assets/logo.svg", __dir__))
+    configuration = RailsMermaidErd.configuration
     erb = ERB.new(File.read(File.expand_path("./templates/index.html.erb", __dir__)))
     result_html = erb.result(binding)
 

diff --git a/lib/rails-mermaid_erd/configuration.rb b/lib/rails-mermaid_erd/configuration.rb
@@ -1,11 +1,12 @@
 require "yaml"
 
 class RailsMermaidErd::Configuration
-  attr_accessor :result_path
+  attr_accessor :result_path, :all_models_selected
 
   def initialize
     config = {
-      result_path: "mermaid_erd/index.html"
+      result_path: "mermaid_erd/index.html",
+      all_models_selected: true,
     }
 
     config_file = Rails.root.join("config/mermaid_erd.yml")
@@ -14,6 +15,14 @@ def initialize
       config = config.merge(custom_config)
     end
 
+    @all_models_selected = config[:all_models_selected]
     @result_path = config[:result_path]
   end
+
+  def to_json
+    {
+      allModelsSelected: @all_models_selected,
+      resultPath: @result_path
+    }.to_json
+  end
 end
diff --git a/lib/templates/index.html.erb b/lib/templates/index.html.erb
@@ -267,6 +267,7 @@
   <script src="https://unpkg.com/vue@3.2.40/dist/vue.global.js"></script>
 
   <script>window.SCHEMA_DATA=<%= result.to_json %></script>
+  <script>window.CONFIG_DATA=<%= configuration.to_json %></script>
   <script>
     window.i18n = {
       en: {
@@ -351,6 +352,7 @@
           ...window.SCHEMA_DATA,
           Models: window.SCHEMA_DATA.Models.sort((a, b) => a.ModelName < b.ModelName ? -1 : 1)
         }
+        const configData = { ...window.CONFIG_DATA }
         const selectModels = Vue.ref([])
         const isPreviewRelations = Vue.ref(false)
         const isShowRelationComment = Vue.ref(false)
@@ -406,9 +408,11 @@
 
         const reset = () => {
           selectModels.value = []
-          schemaData.Models.forEach((model) => {
-            selectModels.value.push(model.ModelName)
-          })
+          if (configData.allModelsSelected) {
+            schemaData.Models.forEach((model) => {
+              selectModels.value.push(model.ModelName)
+            })
+          }
           isPreviewRelations.value = false
           isShowRelationComment.value = false
           isShowKey.value = false

diff --git a/spec/fixtures/files/configurations/valid.yml b/spec/fixtures/files/configurations/valid.yml
@@ -1,3 +1,4 @@
 # result path
 # default "mermaid_erd/index.html"
 result_path: doc/erd.html
+all_models_selected: false
diff --git a/spec/rails-mermaid_erd/configration_spec.rb b/spec/rails-mermaid_erd/configration_spec.rb
@@ -6,6 +6,7 @@
   context "when the config file does not exist" do
     it "return default config value" do
       expect(configuration.result_path).to eq("mermaid_erd/index.html")
+      expect(configuration.all_models_selected).to eq(true)
     end
   end
 
@@ -18,6 +19,7 @@
     end
     it "return overwrite config value" do
       expect(configuration.result_path).to eq("doc/erd.html")
+      expect(configuration.all_models_selected).to eq(false)
     end
   end
 end