3 files changed, 56 insertions, 0 deletions
diff --git a/lib/yard/mruby/handlers/c/header/base.rb b/lib/yard/mruby/handlers/c/header/base.rb
new file mode 100644
index 0000000..f0771ca
--- /dev/null
+++ b/lib/yard/mruby/handlers/c/header/base.rb
@@ -0,0 +1,30 @@
+module YARD::MRuby::Handlers
+  module C
+    module Header
+      class Base < YARD::Handlers::C::Base
+        include YARD::MRuby::CodeObjects
+
+        def header(path)
+          # Remove include prefix
+          path = path.gsub(/^.*include\//,'')
+
+          headers[path] ||= begin
+                              header = HeaderObject.new(HEADERS_ROOT, path)
+                              register header
+                              header
+                            end
+
+
+        end
+
+        def headers
+          globals.mruby_headers ||= {}
+        end
+
+      end
+
+    end
+
+    YARD::Handlers::Processor.register_handler_namespace :header, Header
+  end
+end
diff --git a/lib/yard/mruby/handlers/c/header/define_handler.rb b/lib/yard/mruby/handlers/c/header/define_handler.rb
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/lib/yard/mruby/handlers/c/header/define_handler.rb
diff --git a/lib/yard/mruby/handlers/c/header/function_handler.rb b/lib/yard/mruby/handlers/c/header/function_handler.rb
new file mode 100644
index 0000000..6bdb8a6
--- /dev/null
+++ b/lib/yard/mruby/handlers/c/header/function_handler.rb
@@ -0,0 +1,26 @@
+module YARD::MRuby::Handlers::C::Header
+  class FunctionHandler < Base
+    MATCH = /
+      MRB_(API|INLINE)\s+((\w+\s+)+)(\w+)\s*\(
+    /mx
+
+    handles MATCH
+    statement_class ToplevelStatement
+
+    process do
+      handle_function(statement)
+    end
+
+    def handle_function(statement)
+      header = self.header(statement.file)
+
+      statement.source.scan(MATCH) do |type, prefix, _, name, *args|
+        register FunctionObject.new(header, name) do |obj|
+          if statement.comments
+            register_docstring(obj, statement.comments.source, statement)
+          end
+        end
+      end
+    end
+  end
+end