Handlinc class comments

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