From cc49ade5e7029e4d8d8d62b7c09590042b43d65d Mon Sep 17 00:00:00 2001
From: Seba Gamboa <me@sagmor.com>
Date: Thu, 24 Sep 2015 15:27:31 -0300
Subject: Registering function objects

---
 lib/yard/mruby/handlers/header/function_handler.rb | 23 ++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'lib/yard/mruby/handlers/header/function_handler.rb')

diff --git a/lib/yard/mruby/handlers/header/function_handler.rb b/lib/yard/mruby/handlers/header/function_handler.rb
index e69de29..9b98258 100644
--- a/lib/yard/mruby/handlers/header/function_handler.rb
+++ b/lib/yard/mruby/handlers/header/function_handler.rb
@@ -0,0 +1,23 @@
+module YARD::MRuby::Handlers::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|
+        end
+      end
+    end
+  end
+end
-- 
cgit v1.2.3