1 files changed, 34 insertions, 0 deletions
diff --git a/lib/yard/handlers/c/mruby_module_handler.rb b/lib/yard/handlers/c/mruby_module_handler.rb
new file mode 100644
index 0000000..c9243ac
--- /dev/null
+++ b/lib/yard/handlers/c/mruby_module_handler.rb
@@ -0,0 +1,34 @@
+module YARD::MRuby::Handlers::Source
+  class ModuleHandler < Base
+
+    TOP_LEVEL_MODULE = /\*([\w]+)\s*=\s*mrb_define_module\s*
+      \(
+      \s*\w+\s*,
+      \s*"(\w+)"\s*
+      \)
+    /mx
+
+    NAMESPACED_MODULE = /\*([\w]+)\s*=\s*mrb_define_module_under\s*
+      \(
+      \s*\w+\s*,
+      \s*(\w+)\s*,
+      \s*"(\w+)"\s*
+      \)
+    /mx
+
+    handles TOP_LEVEL_MODULE
+    handles NAMESPACED_MODULE
+
+    statement_class BodyStatement
+
+    process do
+      statement.source.scan(TOP_LEVEL_MODULE) do |var_name, module_name|
+        handle_module(var_name, module_name)
+      end
+      statement.source.scan(NAMESPACED_MODULE) do |var_name, in_module, module_name|
+        handle_module(var_name, module_name, in_module)
+      end
+    end
+  end
+end
+