Change default files. Closes #3

4 files changed, 28 insertions, 0 deletions

diff --git a/lib/yard/mruby.rb b/lib/yard/mruby.rb
index 8d78182..9d24df9 100644
--- a/lib/yard/mruby.rb
+++ b/lib/yard/mruby.rb
@@ -5,3 +5,4 @@ require_relative "mruby/code_objects"
 require_relative "mruby/parser"
 require_relative "mruby/handlers"
 require_relative "mruby/templates"
+require_relative "mruby/cli"
diff --git a/lib/yard/mruby/cli.rb b/lib/yard/mruby/cli.rb
new file mode 100644
index 0000000..ce26311
--- /dev/null
+++ b/lib/yard/mruby/cli.rb
@@ -0,0 +1 @@
+require_relative 'cli/yardoc'
diff --git a/lib/yard/mruby/cli/yardoc.rb b/lib/yard/mruby/cli/yardoc.rb
new file mode 100644
index 0000000..b253428
--- /dev/null
+++ b/lib/yard/mruby/cli/yardoc.rb
@@ -0,0 +1,14 @@
+module YARD::MRuby::CLI
+  module Yardoc
+
+    # Redefines default files for YARD::CLI::Yardoc
+    def parse_arguments(*args)
+      super(*args)
+      self.files = %w{mrblib/**/*.rb src/**/*.c include/**/*.h} if self.files.empty?
+    end
+  end
+
+  # Load Above modifications into YARD::CLI::Yardoc
+  YARD::CLI::Yardoc.include Yardoc
+end
+
diff --git a/spec/yardoc_spec.rb b/spec/yardoc_spec.rb
new file mode 100644
index 0000000..1073178
--- /dev/null
+++ b/spec/yardoc_spec.rb
@@ -0,0 +1,12 @@
+require "spec_helper"
+
+describe YARD::CLI::Yardoc do
+  before do
+    @yardoc = YARD::CLI::Yardoc.new
+    @yardoc.parse_arguments
+  end
+
+  it "should use mrblib/**/*.rb src/**/*.c include/**/*.h as default file glob" do
+    expect(@yardoc.files).to eq %w{mrblib/**/*.rb src/**/*.c include/**/*.h}
+  end
+end