Added FieldHandler

lib/yard-mongoid.rb

@@ -1,5 +1 @@
-module Yard
-  module Mongoid
-    # Your code goes here...
-  end
-end
+require 'yard/mongoid'

lib/yard/mongoid.rb

@@ -0,0 +1,8 @@
+module YARD
+  module Mongoid
+    autoload :VERSION, 'lib/mongoid/version'
+    autoload :Extensions, 'lib/mongoid/extensions'
+  end
+end
+
+require 'lib/mongoid/field_handler'

lib/yard/mongoid/extensions.rb

@@ -0,0 +1,21 @@
+module YARD
+  module Mongoid
+    module Extensions
+    protected
+      def effected_namespace
+        if statement.type == :command_call
+          context = statement.jump(:var_ref)
+
+          unless context.source == 'self'
+            return ensure_loaded!(
+              Registry.resolve(namespace, context.source)
+            )
+          end
+        end
+
+        return namespace
+      end
+    end
+
+  end
+end

lib/yard/mongoid/field_handler.rb

@@ -0,0 +1,44 @@
+module YARD
+  module Mongoid
+    class FieldHandler < YARD::Handlers::Ruby::Base
+      include Extensions
+
+      handles method_call(:field)
+
+      def process
+        nobj = effected_namespace
+        mscope = scope
+        name = statement.parameters[0].last
+
+        if name.type == :symbol
+          name = name.source[1..-1]
+
+          register MethodObject.new(nobj, name, mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def #{name}"
+          end
+
+          register MethodObject.new(nobj, "#{name}=", mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def #{name}=(value)"
+            o.parameters = [['value', nil]]
+          end
+
+          register MethodObject.new(nobj, "#{name}?", mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def #{name}?"
+          end
+
+          register MethodObject.new(nobj, "#{name}?", mscope) do |o|
+            o.visibility = :public
+            o.source = statement.source
+            o.signature = "def #{name}?"
+          end
+        end
+      end
+    end
+  end
+end