summaryrefslogtreecommitdiffhomepage
path: root/lib/yard
diff options
context:
space:
mode:
Diffstat (limited to 'lib/yard')
-rw-r--r--lib/yard/handlers/c/mruby_base.rb11
-rw-r--r--lib/yard/handlers/c/mruby_class_handler.rb4
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