From aed91b0fb2fe3c2cd3a74e2e5bc23c8bd95ef1db Mon Sep 17 00:00:00 2001
From: Seba Gamboa <me@sagmor.com>
Date: Wed, 23 Sep 2015 14:24:03 -0300
Subject: Making specs works

---
 lib/yard/handlers/c/mruby_module_handler.rb | 34 +++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 lib/yard/handlers/c/mruby_module_handler.rb

(limited to 'lib/yard/handlers/c/mruby_module_handler.rb')

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
+
-- 
cgit v1.2.3