2 files changed, 16 insertions, 0 deletions
diff --git a/Gemfile b/Gemfile
index 24c4b19283..05f1cd44f5 100644
--- a/Gemfile
+++ b/Gemfile
@@ -6,6 +6,7 @@ gem "activesupport", :group => :compat_testing, :platform => "ruby"
 
 group(:docgen) do
   gem "ronn"
+  gem "yard"
 end
 
 group(:development, :test) do
diff --git a/Rakefile b/Rakefile
index c223a60618..085887fab5 100644
--- a/Rakefile
+++ b/Rakefile
@@ -118,5 +118,20 @@ end
 
 task :docs => "docs:all"
 
+begin
+  require 'yard'
+  DOC_FILES = [ "README.rdoc", "LICENSE", "spec/tiny_server.rb", "lib/**/*.rb" ]
+  namespace :yard do
+    desc "Create YARD documentation"
+
+    YARD::Rake::YardocTask.new(:html) do |t|
+      t.files = DOC_FILES
+      t.options = ['--format', 'html']
+    end
+  end
+
+rescue LoadError
+  puts "yard is not available. (sudo) gem install yard to generate yard documentation."
+end