koedame · unchidev · Oct 10, 2022 · Oct 10, 2022 · Oct 10, 2022 · Oct 10, 2022
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -21,6 +21,7 @@
 	"customizations": {
 		"vscode": {
 			"extensions": [
+				"GitHub.copilot"
 			]
 		}
 	}

diff --git a/README.ja.md b/README.ja.md
@@ -49,6 +49,17 @@ mermaid_erd
 このファイルを共有すれば、Ruby on Rails環境が無くても使用できます。サーバーにアップロードすれば、同じURLを共有することもできます。
 CIと連携して生成から共有までを自動化するのはとても有効な手段です。
 
+## 設定
+
+`./config/mermaid_erd.yml` を設置することで設定を変更できます。
+設定例は [./docs/example.yml](./docs/example.yml) を参照してください。
+
+設定項目は次のとおりです。
+
+| キー | 説明 | 初期値 |
+| --- | --- | --- |
+| `result_path` | 生成されるファイルのパス。 | `mermaid_erd/index.html` |
+
 <!--
 TODO:
 ## Contributing

diff --git a/README.md b/README.md
@@ -51,6 +51,17 @@ If you share this file, it can be used by those who do not have Ruby on Rails en
 
 It would be very smart to generate it automatically using CI.
 
+## Configuration
+
+`./config/mermaid_erd.yml` to customize the configuration.
+See [./docs/example.yml](./docs/example.yml) for an example configuration.
+
+The setting items are as follows.
+
+| key | description | default |
+| --- | --- | --- |
+| `result_path` | Destination of generated files. | `mermaid_erd/index.html` |
+
 <!--
 TODO:
 ## Contributing

diff --git a/docs/example.yml b/docs/example.yml
@@ -0,0 +1,3 @@
+# result path
+# default "mermaid_erd/index.html"
+result_path: doc/erd.html
diff --git a/lib/rails-mermaid_erd.rb b/lib/rails-mermaid_erd.rb
@@ -3,10 +3,17 @@
 require "rake"
 require "rake/dsl_definition"
 require_relative "rails-mermaid_erd/version"
+require_relative "rails-mermaid_erd/configuration"
 
 module RailsMermaidErd
   extend Rake::DSL
 
+  class << self
+    def configuration
+      @configuration ||= RailsMermaidErd::Configuration.new
+    end
+  end
+
   desc "Generate Mermaid ERD."
   task mermaid_erd: :environment do
     result = {
@@ -194,9 +201,10 @@ module RailsMermaidErd
     erb = ERB.new(File.read(File.expand_path("./templates/index.html.erb", __dir__)))
     result_html = erb.result(binding)
 
-    result_dir = Rails.root.join("./mermaid_erd")
-    result_file = result_dir.join("./index.html")
+    result_dir = Rails.root.join(File.dirname(RailsMermaidErd.configuration.result_path))
     FileUtils.mkdir_p(result_dir)
+
+    result_file = Rails.root.join(RailsMermaidErd.configuration.result_path)
     File.write(result_file, result_html)
   end
 end
diff --git a/lib/rails-mermaid_erd/configuration.rb b/lib/rails-mermaid_erd/configuration.rb
@@ -0,0 +1,21 @@
+require "yaml"
+
+module RailsMermaidErd
+  class Configuration
+    attr_accessor :result_path
+
+    def initialize
+      config = {
+        result_path: "mermaid_erd/index.html"
+      }
+
+      config_file = Rails.root.join("config/mermaid_erd.yml")
+      if File.exist?(config_file)
+        custom_config = YAML.load(config_file.read).symbolize_keys
+        config = config.merge(custom_config)
+      end
+
+      @result_path = config[:result_path]
+    end
+  end
+end
diff --git a/spec/fixtures/files/configurations/valid.yml b/spec/fixtures/files/configurations/valid.yml
@@ -0,0 +1,3 @@
+# result path
+# default "mermaid_erd/index.html"
+result_path: doc/erd.html
diff --git a/spec/rails-mermaid_erd/configration_spec.rb b/spec/rails-mermaid_erd/configration_spec.rb
@@ -0,0 +1,23 @@
+require "spec_helper"
+
+describe RailsMermaidErd::Configuration do
+  let(:configuration) { RailsMermaidErd::Configuration.new }
+
+  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")
+    end
+  end
+
+  context "when the config file exist" do
+    before do
+      FileUtils.cp(file_fixture("configurations/valid.yml").to_s, Rails.root.join("config/mermaid_erd.yml").to_s)
+    end
+    after do
+      FileUtils.rm(Rails.root.join("config/mermaid_erd.yml").to_s)
+    end
+    it "return overwrite config value" do
+      expect(configuration.result_path).to eq("doc/erd.html")
+    end
+  end
+end