From 92b174e2b59624f0c9ec54d0d8c486a3d6840952 Mon Sep 17 00:00:00 2001 From: ukmashi Date: Mon, 23 Sep 2024 00:22:15 +0900 Subject: [PATCH] mod uninitialized constant RailsERD::Diagram Error --- lib/rails_erd/cli.rb | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/lib/rails_erd/cli.rb b/lib/rails_erd/cli.rb index a1de4e3..ea0d94f 100644 --- a/lib/rails_erd/cli.rb +++ b/lib/rails_erd/cli.rb @@ -8,6 +8,7 @@ option :generator do long "--generator=Generator" desc "Generator to use (graphviz or mermaid). Defaults to graphviz." + default "graphviz" end option :title do @@ -164,7 +165,8 @@ def start def initialize(path, options) @path, @options = path, options - require "rails_erd/diagram/graphviz" if options.generator == :graphviz + require "rails_erd/diagram/graphviz" if options[:generator] == 'graphviz' + require "rails_erd/diagram/mermaid" if options[:generator] == 'mermaid' end def start @@ -202,7 +204,7 @@ def load_application end def generator - if options.generator == :mermaid + if options[:generator] == 'mermaid' RailsERD::Diagram::Mermaid else RailsERD::Diagram::Graphviz