diff options
Diffstat (limited to 'lib/yard/handlers/c/mruby_method_handler.rb')
| -rw-r--r-- | lib/yard/handlers/c/mruby_method_handler.rb | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/lib/yard/handlers/c/mruby_method_handler.rb b/lib/yard/handlers/c/mruby_method_handler.rb index 01fb345..a8812ae 100644 --- a/lib/yard/handlers/c/mruby_method_handler.rb +++ b/lib/yard/handlers/c/mruby_method_handler.rb @@ -1,7 +1,11 @@ module YARD::Handlers::C class MRubyMethodHandler < MRubyBase - MATCH1 = /mrb_define_method\s* - \( + MATCH1 = /mrb_define_( + method | + singleton_method | + module_function + ) + \s*\( \s*\w+\s*, \s*(\w+)\s*, \s*"(\w+)"\s*, @@ -12,8 +16,8 @@ module YARD::Handlers::C statement_class BodyStatement process do - statement.source.scan(MATCH1) do |var_name, name, func_name| - handle_method(nil, var_name, name, func_name) + statement.source.scan(MATCH1) do |type,var_name, name, func_name| + handle_method(type, var_name, name, func_name) end end |