diff options
Diffstat (limited to 'lib/yard')
| -rw-r--r-- | lib/yard/handlers/c/mruby_base.rb | 11 | ||||
| -rw-r--r-- | lib/yard/handlers/c/mruby_class_handler.rb | 4 |
2 files changed, 13 insertions, 2 deletions
diff --git a/lib/yard/handlers/c/mruby_base.rb b/lib/yard/handlers/c/mruby_base.rb index 45f0dd6..ccf85b4 100644 --- a/lib/yard/handlers/c/mruby_base.rb +++ b/lib/yard/handlers/c/mruby_base.rb @@ -24,5 +24,16 @@ module YARD::Handlers::C super end + + def handle_class(var_name, class_name, parent, stmt, in_module = nil) + object = super(var_name, class_name, parent, in_module) + + if stmt.comments + register_docstring(object, stmt.comments.source, stmt) + end + + object + end + end end diff --git a/lib/yard/handlers/c/mruby_class_handler.rb b/lib/yard/handlers/c/mruby_class_handler.rb index 9e3d1f5..72d516c 100644 --- a/lib/yard/handlers/c/mruby_class_handler.rb +++ b/lib/yard/handlers/c/mruby_class_handler.rb @@ -25,10 +25,10 @@ module YARD::Handlers::C process do statement.source.scan(TOP_LEVEL_CLASS) do |var_name, class_name, parent| - handle_class(var_name, class_name, parent) + handle_class(var_name, class_name, parent, statement) end statement.source.scan(NAMESPACED_CLASS) do |var_name, in_module, class_name, parent| - handle_class(var_name, class_name, parent, in_module) + handle_class(var_name, class_name, parent, statement, in_module) end end end |
