more xgrep

lib/xgrep/commander.rb

@@ -1,122 +1,136 @@
 require 'ostruct'
 require 'optparse'
 
-class Commander
+require_relative './custom_env'
+require_relative './node_env'
+require_relative './rails_env'
+require_relative './simple_env'
 
-  attr_accessor :options
+module Xgrep
 
-  def initialize
-    @options = OpenStruct.new
-    options.debug = false
-    options.git_grep = %w{-E}
-    options.pathspec = []
-    options.environment = default_environment
+  class Commander
 
-    @option_parser = OptionParser.new do |op|
-      op.banner = "Usage: xgrep options term(s)"
+    attr_accessor :options
+
+    def initialize (argv)
+      @options = OpenStruct.new
+      options.debug = false
+      options.git_grep = %w{-E}
+      options.pathspec = []
+      options.environment = default_environment
+
+      @option_parser = OptionParser.new do |op|
+        op.banner = "Usage: xgrep options term(s)"
+
+        op.on('-a','--asset') do |argument|
+          options.pathspec << :asset
+        end
+
+        op.on('-b','--db') do |argument|
+          options.pathspec << :db
+        end
+
+        op.on('-c','--controller') do |argument|
+          options.pathspec << :controller
+        end
+
+        op.on('-d','--[no-]debug') do |argument|
+          options.debug = argument
+        end
+
+        op.on('-f','--file') do |argument|
+          options.git_grep << "-L"
+        end
+
+        op.on('-g','--config') do |argument|
+          options.pathspec << :config
+        end
+
+        op.on('-i','--invert') do |argument|
+          options.git_grep << '-v'
+        end
+
+        op.on('-l','--lib') do |argument|
+          options.pathspec << :lib
+        end
+
+        op.on('-m','--model') do |argument|
+          options.pathspec << :model
+        end
+
+        op.on('-o','--core') do |argument|
+          options.pathspec << :core
+        end
+
+        op.on_tail('-h','--help') do |argument|
+          puts op
+          exit
+        end
 
-      op.on('-a','--asset') do |argument|
-        options.pathspec << :asset
-      end
-
-      op.on('-b','--db') do |argument|
-        options.pathspec << :db
-      end
-
-      op.on('-c','--controller') do |argument|
-        options.pathspec << :controller
-      end
-
-      op.on('-d','--[no-]debug') do |argument|
-        options.debug = argument
-      end
-
-      op.on('-f','--file') do |argument|
-        options.git_grep << "-L"
-      end
-
-      op.on('-g','--config') do |argument|
-        options.pathspec << :config
-      end
-
-      op.on('-i','--invert') do |argument|
-        options.git_grep << '-v'
-      end
-
-      op.on('-l','--lib') do |argument|
-        options.pathspec << :lib
-      end
-
-      op.on('-m','--model') do |argument|
-        options.pathspec << :model
-      end
-
-      op.on('-o','--core') do |argument|
-        options.pathspec << :core
-      end
-
-      op.on_tail('-h','--help') do |argument|
-        puts op
-        exit
       end
 
+      @option_parser.parse!(argv)
+      options.terms = argv # must be after parse!
     end
 
-    @option_parser.parse!(ARGV)
-    options.terms = ARGV # must be after parse!
-  end
+    def valid?
+      options.terms.size > 0
+    end
 
-  def valid?
-    options.terms.size > 0
-  end
+    def help
+      puts @option_parser
+      exit
+    end
 
-  def help
-    puts @option_parser
-    exit
-  end
+    def execute
+      env = options.environment
+      env.update_pathspec(options.pathspec)
 
-  def execute
-    env = Object.const_get(options.environment).new
-    env.update_pathspec(options.pathspec)
+      ands = []
+      ors  = []
+      nots = []
+      current_op = ands
 
-    ands = []
-    ors  = []
-    nots = []
-    current_op = ands
+      options.terms.each do |x|
+        case x
+        when 'and'               then current_op = ands
+        when 'or'                then current_op = ors
+        when 'not'               then current_op = nots
+        when /^-+(and|or|not)$/i then current_op << $1
+        else                          current_op << x
+        end
+      end
 
-    options.terms.each do |x|
-      case x
-      when 'and'               then current_op = ands
-      when 'or'                then current_op = ors
-      when 'not'               then current_op = nots
-      when /^-+(and|or|not)$/i then current_op << $1
-      else                          current_op << x
+      ands = ands.map { |x| "-e \"#{x}\"" }
+      ors  = ors.map { |x| "-e \"#{x}\"" }
+      nots = nots.map { |x| "-e \"#{x}\"" }
+
+      ands << "\\( #{ors.join(' --or ')} \\)"        unless ors.empty?
+      ands << "--not \\( #{nots.join(' --or ')} \\)" unless nots.empty?
+
+      command = "git grep #{options.git_grep.join(' ')} #{ands.join(' --and ')} -- #{env.pathspec.sort.join(' ')}"
+      puts
+      if options.debug
+        puts command
+      else
+        system command
+      end
+      puts
+    end
+
+    private
+
+    def default_environment
+      if File.exist?('./.xgrep')
+        Xgrep::CustomEnv.new
+      elsif File.exist?("Gemfile")
+        Xgrep::RailsEnv.new
+      elsif File.exist?("package.json")
+        Xgrep::NodeEnv.new
+      else
+        Xgrep::SimpleEnv.new
       end
     end
-
-    ands = ands.map { |x| "-e \"#{x}\"" }
-    ors  = ors.map { |x| "-e \"#{x}\"" }
-    nots = nots.map { |x| "-e \"#{x}\"" }
-
-    ands << "\\( #{ors.join(' --or ')} \\)"        unless ors.empty?
-    ands << "--not \\( #{nots.join(' --or ')} \\)" unless nots.empty?
-
-    command = "git grep #{options.git_grep.join(' ')} #{ands.join(' --and ')} -- #{env.pathspec.sort.join(' ')}"
   end
 
-  private
-
-  def default_environment
-    if File.exist?('./.xgrep')
-      "CustomEnv"
-    elsif File.exist?("Gemfile") and File.exist?("package.json")
-      "NodeEnv"
-    elsif File.exist?("Gemfile")
-      "RailsEnv"
-    elsif File.exist?("package.json")
-      "NodeEnv"
-    else
-      "RailsEnv"
-    end
-  end
 end

lib/xgrep/custom_env.rb

@@ -0,0 +1,12 @@
+module Xgrep
+  class CustomEnv
+    attr_accessor :pathspec
+
+    def initialize
+      @pathspec = { }
+    end
+
+    def update_pathspec ( pathspec )
+    end
+  end
+end

lib/xgrep/node_env.rb

@@ -0,0 +1,12 @@
+module Xgrep
+  class NodeEnv
+    attr_accessor :pathspec
+
+    def initialize
+      @pathspec = { }
+    end
+
+    def update_pathspec ( pathspec )
+    end
+  end
+end

lib/xgrep/rails_env.rb

@@ -1,6 +1,12 @@
-class RailsEnv
+module Xgrep
+  class RailsEnv
+    attr_accessor :pathspec
 
-  def update_pathspec ( pathspec )
+    def initialize
+      @pathspec = { }
+    end
+
+    def update_pathspec ( pathspec )
+    end
   end
-
 end

lib/xgrep/simple_env.rb

@@ -0,0 +1,12 @@
+module Xgrep
+  class SimpleEnv
+    attr_accessor :pathspec
+
+    def initialize
+      @pathspec = { }
+    end
+
+    def update_pathspec ( pathspec )
+    end
+  end
+end