Handle #define directives

3 files changed, 17 insertions, 11 deletions

diff --git a/lib/yard/mruby/code_objects/define_object.rb b/lib/yard/mruby/code_objects/define_object.rb
index 2c05d91..4220acd 100644
--- a/lib/yard/mruby/code_objects/define_object.rb
+++ b/lib/yard/mruby/code_objects/define_object.rb
@@ -1,7 +1,7 @@
 module YARD::MRuby::CodeObjects
 
   # A DefineObject represents a MRuby C API define macro declaration inside a header inside an include directory
-  class DefineObject < YARD::CodeObjects::Base
+  class DefineObject < HeaderBaseObject
   end
 end
 
diff --git a/lib/yard/mruby/code_objects/function_object.rb b/lib/yard/mruby/code_objects/function_object.rb
index 7d85439..ecf7dd5 100644
--- a/lib/yard/mruby/code_objects/function_object.rb
+++ b/lib/yard/mruby/code_objects/function_object.rb
@@ -1,7 +1,7 @@
 module YARD::MRuby::CodeObjects
 
   # A FunctionObject represents a MRuby C API function declaration inside a header inside an include directory
-  class FunctionObject < YARD::CodeObjects::Base
+  class FunctionObject < HeaderBaseObject
 
     # Returns the list of parameters parsed out of the method signature
     # with their default values.
@@ -14,15 +14,6 @@ module YARD::MRuby::CodeObjects
       super
     end
 
-    def header
-      self.namespace
-    end
-
-    # Function's shouln't be namespaced
-    def path
-      self.name.to_s
-    end
-
     def attr_info
       nil
     end
diff --git a/lib/yard/mruby/code_objects/header_base_object.rb b/lib/yard/mruby/code_objects/header_base_object.rb
new file mode 100644
index 0000000..403db3d
--- /dev/null
+++ b/lib/yard/mruby/code_objects/header_base_object.rb
@@ -0,0 +1,15 @@
+module YARD::MRuby::CodeObjects
+
+  # A FunctionObject represents a MRuby C API function declaration inside a header inside an include directory
+  class HeaderBaseObject < YARD::CodeObjects::Base
+
+    def header
+      self.namespace
+    end
+
+    # Header objects's shouln't be namespaced
+    def path
+      self.name.to_s
+    end
+  end
+end