Add class body generation

CHANGELOG.md

@@ -3,8 +3,16 @@
 ## master (unreleased)
 
 
+## 0.4.0 (2019-11-21)
+
+### Added
+
+- Add `body` argument to `Ginny::Class` for code to be generated in the class body.
+
 ## 0.3.0 (2019-11-21)
 
+### Added
+
 - Add the option to wrap generated code in modules.
 
 ## 0.2.1 (2019-11-18)

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ginny (0.3.0)
+    ginny (0.4.0)
 
 GEM
   remote: https://rubygems.org/

README.md

@@ -101,6 +101,7 @@ end
 | --------------- | -------------------- | -------------------------------------------------------------- |
 | name (required) | `String`             | Name of the class.                                             |
 | description     | `String`             | Description of the class. [Markdown][markdown] is supported.   |
+| body            | `String`             | String to write into the body of the class.                    |
 | parent          | `String`             | Name of a class to inherit from. (Ex: `YourNewClass < Parent`) |
 | modules         | `Array<String>`      | List of modules to declare the class inside of                 |
 | attrs           | `Array<Ginny::Attr>` | An array of Attrs.                                             |

lib/ginny/models/class.rb

@@ -17,6 +17,9 @@ class Class
     # An array of {Ginny::Attr}s.
     # @return [Array<Ginny::Attr>]
     attr_accessor :attrs
+    # String to write into the body of the class.
+    # @return [String]
+    attr_accessor :body
 
     # @return [void]
     def initialize()
@@ -35,6 +38,7 @@ def self.create(args = {})
       c.parent      = args[:parent]
       c.modules     = args[:modules] unless args[:modules].nil?
       c.attrs       = Ginny::Attr.from_array(args[:attrs]) if args[:attrs]&.is_a?(Array)
+      c.body        = args[:body] unless args[:body].nil?
       return c
     end
 
@@ -53,6 +57,7 @@ def render()
       parts << (self.description&.length&.positive? ? self.description.comment.strip : nil)
       parts << (self.parent.nil? ? "class #{self.name}" : "class #{self.name} < #{self.parent}")
       parts << self.render_attributes()
+      parts << (self.body&.length&.positive? ? self.body.indent(2) : nil)
       parts << "end"
       if self.modules.length > 0
         body = parts.compact.join("\n").gsub(/\s+$/, "")

lib/ginny/version.rb

@@ -1,3 +1,3 @@
 module Ginny
-  VERSION = "0.3.0".freeze
+  VERSION = "0.4.0".freeze
 end

test/ginny/class_test.rb

@@ -80,4 +80,51 @@ class Human < Mammal
     assert_equal(want.strip, have.strip)
   end
 
+  def test_class_with_body
+    want = <<~RUBY.strip
+      module C4
+        module Elements
+          # - Id: 112
+          # - Name: Pier Name
+          # - Type: AN
+          # - Min/Max: 2/14
+          # - Description: Free-form name of the pier.
+          class PierName < C4::Element::AN
+            # @return [void]
+            def initialize()
+              @id = 112
+              @name = "Pier Name"
+              @type = "AN"
+              self.min = 2
+              self.max = 14
+            end
+          end
+        end
+      end
+    RUBY
+    have = Ginny::Class.create(
+      name: "PierName",
+      parent: "C4::Element::AN",
+      modules: ["C4", "Elements"],
+      description: <<~STRING.strip,
+        - Id: 112
+        - Name: Pier Name
+        - Type: AN
+        - Min/Max: 2/14
+        - Description: Free-form name of the pier.
+      STRING
+      body: <<~RUBY.strip,
+        # @return [void]
+        def initialize()
+          @id = 112
+          @name = "Pier Name"
+          @type = "AN"
+          self.min = 2
+          self.max = 14
+        end
+      RUBY
+    ).render()
+    assert_equal(want.strip, have.strip)
+  end
+
 end