Add support for ensurable in types_extras_handler

Puppet's `file` type uses the `ensurable` method in an extra file, so add
support for that.

Author: joshcooper

lib/puppet-strings/yard/handlers/ruby/type_extras_handler.rb

@@ -5,14 +5,15 @@
 require 'puppet-strings/yard/code_objects'
 require 'puppet-strings/yard/util'
 
-# Implements the handler for Puppet resource type newparam/newproperty calls written in Ruby.
+# Implements the handler for Puppet resource type newparam/newproperty/ensurable calls written in Ruby.
 class PuppetStrings::Yard::Handlers::Ruby::TypeExtrasHandler < PuppetStrings::Yard::Handlers::Ruby::TypeBase
   # The default docstring when ensurable is used without given a docstring.
   DEFAULT_ENSURABLE_DOCSTRING = 'The basic property that the resource should be in.'
 
   namespace_only
   handles method_call(:newparam)
   handles method_call(:newproperty)
+  handles method_call(:ensurable)
 
   process do
 
@@ -38,9 +39,15 @@ class PuppetStrings::Yard::Handlers::Ruby::TypeExtrasHandler < PuppetStrings::Ya
     method1_name = statement[0].children.drop(1).find{ |c| c.type == :ident }.source
     return unless ['Type', 'Puppet::Type'].include?(module_name) && method1_name == 'type'
 
+    # ensurable is syntatic sugar for newproperty
     typename = get_name(statement[0], 'Puppet::Type.type')
-    method2_name = caller_method
-    propertyname = get_name(statement, "Puppet::Type.type().#{method2_name}")
+    if caller_method == 'ensurable'
+      method2_name = 'newproperty'
+      propertyname = 'ensure'
+    else
+      method2_name = caller_method
+      propertyname = get_name(statement, "Puppet::Type.type().#{method2_name}")
+    end
 
     typeobject = get_type_yard_object(typename)
 

spec/unit/puppet-strings/yard/handlers/ruby/type_extras_handler_spec.rb

@@ -49,4 +49,24 @@
       expect(object.parameters[0].isnamevar).to eq(true)
     end
   end
+
+  describe 'parsing source with ensurable' do
+    let(:source) { <<~SOURCE
+      Puppet::Type.newtype(:database) do
+        desc 'database'
+      end
+      Puppet::Type.type(:database).ensurable do
+        desc 'What state the database should be in.'
+      end
+    SOURCE
+    }
+
+    it 'generates a doc string for an ensurable' do
+      expect(subject.size).to eq(1)
+      object = subject.first
+      expect(object.properties.size).to eq(1)
+      expect(object.properties[0].name).to eq('ensure')
+      expect(object.properties[0].docstring).to eq('What state the database should be in.')
+    end
+  end
 end