Detecting classes and methods from C sources

1 files changed, 18 insertions, 0 deletions

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