rubytodo 0.0.2 → 0.0.3

data/bin/rubytodo

@@ -3,6 +3,4 @@
 require 'rubygems'
 require 'rubytodo'
 
-root = ARGV[0] || "."
-
-RubyTodo.list :root => root
+RubyTodo.list ARGV[0] || "."

data/lib/rubytodo.rb

@@ -2,10 +2,15 @@ require 'object'
 require 'ruby_parser'
 
 class RubyTodo
-  def self.list options
-    root = options[:root]
-    puts "Scanning todos in #{root} ..."
-    files = Dir.glob File.join(root, "**", "*")
+  def self.list target
+    files = if File.file? target
+      [target]
+    elsif File.directory? target
+      puts "Scanning todos in #{target} ..."
+      files = Dir.glob File.join(target, "**", "*")
+    else
+      abort "#{target} does not exist."
+    end
     files.reject { |f| f !~ /\.rb\Z/ }.each do |f|
       todos = scan_todos f
       puts f unless todos.nil? || todos.empty?
@@ -14,7 +19,8 @@ class RubyTodo
   end
 
   def self.scan_todos file_path
-    ptree = RubyParser.new.parse File.open(file_path, 'r').read
+    parser = RubyParser.new
+    ptree = parser.parse File.open(file_path, 'r').read
     extract_todos ptree
   end
 

data/lib/rubytodo/version.rb

@@ -1,3 +1,3 @@
 module RubyTodo
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rubytodo
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: