2 files changed, 22 insertions, 8 deletions
diff --git a/lib/yard/mruby/code_objects/function_object.rb b/lib/yard/mruby/code_objects/function_object.rb
index 35b8e4f..2fdc0b7 100644
--- a/lib/yard/mruby/code_objects/function_object.rb
+++ b/lib/yard/mruby/code_objects/function_object.rb
@@ -2,6 +2,7 @@ module YARD::MRuby::CodeObjects
 
   # A FunctionObject represents a MRuby C API function declaration inside a header inside an include directory
   class FunctionObject < HeaderBaseObject
+    ParameterType = Struct.new(:type,:name)
 
     # Returns the list of parameters parsed out of the method signature
     # with their default values.
@@ -23,13 +24,27 @@ module YARD::MRuby::CodeObjects
     end
 
     def return_type
-      obj.tag(:return).types.first
+      @return_type
     end
 
     def return_type=(type)
-      return if type == 'void'
-      add_tag(YARD::Tags::Tag.new(:return,"", "")) unless has_tag?(:return)
-      tag(:return).types = [type]
+      @return_type = (type == 'void' ? nil : type)
+    end
+
+    def parameter_types
+      @parameter_types || []
+    end
+
+    def parse_parameter_types(parameters)
+      @parameter_types = []
+      return if parameters.match /^\s*void\s*$/
+
+      parameters.split(',').each do |parameter|
+        parameter.scan(/((:?struct\s+)?\w+(:?\s*\*)?)\s*(\w+)?/) do |type,_,_,name|
+          @parameter_types << ParameterType.new(type,name)
+        end
+      end
+
     end
 
 
diff --git a/lib/yard/mruby/handlers/c/header/function_handler.rb b/lib/yard/mruby/handlers/c/header/function_handler.rb
index 079f114..3d401a8 100644
--- a/lib/yard/mruby/handlers/c/header/function_handler.rb
+++ b/lib/yard/mruby/handlers/c/header/function_handler.rb
@@ -4,7 +4,7 @@ module YARD::MRuby::Handlers::C::Header
       MRB_(API|INLINE)\s+
       ((struct\s+)?\w+(\s*\*)?)\s*
       ((\w+\s+)+)?(\w+)\s*
-      \(([\w\s\*,])*\)
+      \(([\w\s\*,]*)\)
     /mx
 
     handles MATCH
@@ -17,15 +17,14 @@ module YARD::MRuby::Handlers::C::Header
     def handle_function(statement)
       header = self.header(statement.file)
 
-["API", "mrb_value", nil, nil, nil, nil, "sample_api_method"]
-      statement.source.scan(MATCH) do |type, retype, _,_,_,_, name, *args|
-        puts args.inspect
+      statement.source.scan(MATCH) do |type, retype, _,_,_,_, name, parameters|
         register FunctionObject.new(header, name) do |obj|
           if statement.comments
             register_docstring(obj, statement.comments.source, statement)
           end
 
           obj.return_type = retype
+          obj.parse_parameter_types(parameters)
 
         end
       end