1 files changed, 35 insertions, 0 deletions
diff --git a/lib/yard/mruby/handlers/source/class_handler.rb b/lib/yard/mruby/handlers/source/class_handler.rb
new file mode 100644
index 0000000..185aaa6
--- /dev/null
+++ b/lib/yard/mruby/handlers/source/class_handler.rb
@@ -0,0 +1,35 @@
+module YARD::MRuby::Handlers::Source
+  class ClassHandler < Base
+
+    TOP_LEVEL_CLASS = /\*([\w]+)\s*=\s*mrb_define_class\s*
+      \(
+      \s*\w+\s*,
+      \s*"(\w+)"\s*,
+      \s*([\w\->]+)\s*
+      \)
+    /mx
+
+    NAMESPACED_CLASS = /\*([\w]+)\s*=\s*mrb_define_class_under\s*
+      \(
+      \s*\w+\s*,
+      \s*(\w+)\s*,
+      \s*"(\w+)"\s*,
+      \s*([\w\->]+)\s*
+      \)
+    /mx
+
+    handles TOP_LEVEL_CLASS
+    handles NAMESPACED_CLASS
+
+    statement_class BodyStatement
+
+    process do
+      statement.source.scan(TOP_LEVEL_CLASS) do |var_name, class_name, parent|
+        handle_class(var_name, class_name, parent)
+      end
+      statement.source.scan(NAMESPACED_CLASS) do |var_name, in_module, class_name, parent|
+        handle_class(var_name, class_name, parent, in_module)
+      end
+    end
+  end
+end