Rename parameters

Author: ledsun

lib/activerecord_uml/diagram_drawer.rb

@@ -4,8 +4,8 @@
 module ActiverecordUml
   class DiagramDrawer
     CLASS_TEMPLATE = <<EOF
-class <%= class_name %> {
-<% class_name.columns.each do |c| %>
+class <%= klass.name %> {
+<% klass.columns.each do |c| %>
   <%= sprintf("%8s %s", c.type, c.name) %>
 <% end %>
 <% methods.each do |m, parameters| %>
@@ -14,16 +14,16 @@ class <%= class_name %> {
 }
 EOF
 
-    def initialize(class_name)
-      @class_name = class_name
+    def initialize(klass)
+      @klass = klass
     end
 
     def class_diagram
-      class_template.result_with_hash class_name: @class_name, methods: methods
+      class_template.result_with_hash klass: @klass, methods: methods
     end
 
     def relations
-      RelationDrawer.new(@class_name).relations
+      RelationDrawer.new(@klass).relations
     end
 
     private
@@ -33,11 +33,11 @@ def class_template
     end
 
     def methods
-      @class_name.public_instance_methods(false).sort.map do |m|
-        method_parameters = @class_name.new.method(m)
-                                       .parameters
-                                       .filter { |a| a[0] == :req }
-                                       .map { |a| a[1] }
+      @klass.public_instance_methods(false).sort.map do |m|
+        method_parameters = @klass.new.method(m)
+                                  .parameters
+                                  .filter { |a| a[0] == :req }
+                                  .map { |a| a[1] }
 
         if method_parameters.length > 3
           method_parameters = method_parameters.slice(0, 3).append("...")

lib/activerecord_uml/relation_drawer.rb

@@ -3,26 +3,26 @@
 
 module ActiverecordUml
   class RelationDrawer
-    def initialize(class_name)
-      @class_name = class_name
+    def initialize(klass)
+      @klass = klass
     end
 
     def relations
-      @class_name.reflect_on_all_associations(:belongs_to).map do |a|
-        Relation.new(a.class_name, "--*", @class_name, a)
-      end.concat(@class_name.reflect_on_all_associations(:has_many).map do |a|
+      @klass.reflect_on_all_associations(:belongs_to).map do |a|
+        Relation.new(a.class_name, "--*", @klass, a)
+      end.concat(@klass.reflect_on_all_associations(:has_many).map do |a|
         if a.is_a? ActiveRecord::Reflection::ThroughReflection
           if a.source_reflection
             # If a source association is specified, the class name of the source association is displayed.
-            Relation.new(@class_name, "*--*", a.source_reflection.class_name, a)
+            Relation.new(@klass, "*--*", a.source_reflection.class_name, a)
           else
-            Relation.new(@class_name, "*--*", a.class_name, a)
+            Relation.new(@klass, "*--*", a.class_name, a)
           end
         else
-          Relation.new(@class_name, "--*", a.class_name, a)
+          Relation.new(@klass, "--*", a.class_name, a)
         end
-      end).concat(@class_name.reflect_on_all_associations(:has_one).map do |a|
-        Relation.new(@class_name, "--*", a.class_name, a)
+      end).concat(@klass.reflect_on_all_associations(:has_one).map do |a|
+        Relation.new(@klass, "--*", a.class_name, a)
       end)
     end
   end