Detecting classes and methods from C sources

10 files changed, 126 insertions, 0 deletions

diff --git a/lib/yard/mruby/handlers/header.rb b/lib/yard/mruby/handlers/header.rb
new file mode 100644
index 0000000..bb39563
--- /dev/null
+++ b/lib/yard/mruby/handlers/header.rb
@@ -0,0 +1 @@
+ require_relative 'header/base' 
diff --git a/lib/yard/mruby/handlers/header/base.rb b/lib/yard/mruby/handlers/header/base.rb
new file mode 100644
index 0000000..35859f7
--- /dev/null
+++ b/lib/yard/mruby/handlers/header/base.rb
@@ -0,0 +1,11 @@
+module YARD
+  module MRuby
+    module Handlers
+      module Header
+        class Base < YARD::Handlers::Base
+          include YARD::Parser::C
+        end
+      end
+    end
+  end
+end
diff --git a/lib/yard/mruby/handlers/header/define_handler.rb b/lib/yard/mruby/handlers/header/define_handler.rb
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/lib/yard/mruby/handlers/header/define_handler.rb
diff --git a/lib/yard/mruby/handlers/header/function_handler.rb b/lib/yard/mruby/handlers/header/function_handler.rb
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/lib/yard/mruby/handlers/header/function_handler.rb
diff --git a/lib/yard/mruby/handlers/header/header_handler.rb b/lib/yard/mruby/handlers/header/header_handler.rb
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/lib/yard/mruby/handlers/header/header_handler.rb
diff --git a/lib/yard/mruby/handlers/source.rb b/lib/yard/mruby/handlers/source.rb
new file mode 100644
index 0000000..af8b0db
--- /dev/null
+++ b/lib/yard/mruby/handlers/source.rb
@@ -0,0 +1,5 @@
+ require_relative 'source/base'
+ require_relative 'source/class_handler'
+ require_relative 'source/module_handler'
+ require_relative 'source/method_handler'
+
diff --git a/lib/yard/mruby/handlers/source/base.rb b/lib/yard/mruby/handlers/source/base.rb
new file mode 100644
index 0000000..4578079
--- /dev/null
+++ b/lib/yard/mruby/handlers/source/base.rb
@@ -0,0 +1,18 @@
+module YARD
+  module MRuby
+    module Handlers
+      module Source
+        class Base < YARD::Handlers::C::Base
+          DEFAULT_NAMESPACES = {
+            'mrb->object_class' => 'Object'
+          }
+
+          def namespace_for_variable(var)
+            return DEFAULT_NAMESPACES[var] if DEFAULT_NAMESPACES[var]
+            super
+          end
+        end
+      end
+    end
+  end
+end
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
diff --git a/lib/yard/mruby/handlers/source/method_handler.rb b/lib/yard/mruby/handlers/source/method_handler.rb
new file mode 100644
index 0000000..f3eab8c
--- /dev/null
+++ b/lib/yard/mruby/handlers/source/method_handler.rb
@@ -0,0 +1,22 @@
+module YARD::MRuby::Handlers::Source
+  class MethodHandler < Base
+    MATCH1 = /mrb_define_method\s*
+      \(
+      \s*\w+\s*,
+      \s*(\w+)\s*,
+      \s*"(\w+)"\s*,
+      \s*(\w+)\s*,
+    /mx
+
+    handles MATCH1
+    statement_class BodyStatement
+
+    process do
+      puts statement.inspect
+      statement.source.scan(MATCH1) do |var_name, name, func_name|
+        handle_method(nil, var_name, name, func_name)
+      end
+    end
+
+  end
+end
diff --git a/lib/yard/mruby/handlers/source/module_handler.rb b/lib/yard/mruby/handlers/source/module_handler.rb
new file mode 100644
index 0000000..c9243ac
--- /dev/null
+++ b/lib/yard/mruby/handlers/source/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
+