Registering function objects

1 files changed, 21 insertions, 1 deletions

diff --git a/lib/yard/mruby/handlers/header/base.rb b/lib/yard/mruby/handlers/header/base.rb
index d3e5491..58178fe 100644
--- a/lib/yard/mruby/handlers/header/base.rb
+++ b/lib/yard/mruby/handlers/header/base.rb
@@ -1,7 +1,27 @@
 module YARD::MRuby::Handlers
   module Header
-    class Base < YARD::Handlers::Base
+    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(:root, path)
+                            register header
+                            header
+                          end
+
+
+      end
+
+      def headers
+        globals.mruby_headers ||= {}
+      end
+
     end
+
   end
 
   YARD::Handlers::Processor.register_handler_namespace :header, Header