pik 0.1.0

data/lib/pik/commands/add_command.rb

@@ -0,0 +1,65 @@
+module Pik
+
+  class  Add < Command
+  	    
+    it 'Adds another ruby location to pik.'
+    include ConfigFileEditor
+
+    attr_reader :interactive 
+
+    def execute(path=nil)
+      return add_interactive if interactive
+      path = @args.first || ::Config::CONFIG['bindir']
+      add(path)
+    end
+
+    def add(path)
+      path = Pathname.new(path)
+      path = path.dirname if path.file?
+      if ruby_exists_at?(path) 
+        if config[get_version(path)]
+          puts "This version has already been added."
+        else
+          version = get_version(path)
+          path    = path.expand_path.to_ruby
+          puts "Adding:  #{version}'\n Located at:  #{path}\n"
+          @config[version][:path] = path
+        end
+      else
+        puts "Couldn't find a Ruby version at #{path}"
+      end
+    end
+
+    def command_options
+      super
+      options.banner += "[path_to_ruby]"
+      options.separator ""      
+      options.on("--interactive", "-i", "Add interactively") do |value|
+        @interactive = value
+      end 
+    end    
+
+    def add_interactive
+      @hl.choose do |menu|  
+        menu.prompt = ""
+        menu.choice('e]nter a path'){
+          dir = @hl.ask("Enter a path to a ruby/bin dir (enter to quit)")
+          execute(dir) unless dir.empty? || !@hl.agree("Add '#{dir}'? [Yn] ")
+          add_interactive
+        }
+        menu.choice('s]earch'){
+          search_dir = @hl.ask("Enter a search path")
+          files = ruby_glob(search_dir + '**')
+          files.each{|file| 
+            dir = File.dirname(file)
+            add(dir) if @hl.agree("Add '#{dir}'? [Yn] ")
+          }
+          add_interactive
+        }
+        menu.choice('q]uit'){raise QuitError}
+      end        
+    
+    end
+  end
+
+end

data/lib/pik/commands/batch_file_editor.rb

@@ -0,0 +1,62 @@
+module Pik
+
+  module BatchFileEditor
+    
+    attr_reader :batch
+        
+    def initialize(args=ARGV,config=nil)
+      super
+      batch_file = File.join(PIK_HOME, "#{File.basename($0)}_#{$$}.bat")
+      @batch = BatchFile.new( batch_file )
+      editors << @batch
+    end
+    
+    def close
+      update_gem_batch
+      super
+    end
+
+    def set(items)
+      items.each do |k, v|
+        @batch.set(k => v)
+        WindowsEnv.user.set(k => v) if global
+      end
+    end  
+      
+    def switch_path_to(new_ver)
+      dir = current_ruby_bin_path
+      new_path = SearchPath.new(ENV['PATH']).replace_or_add(dir, new_ver[:path])
+      if new_ver[:gem_home]
+        new_path.replace_or_add(current_gem_bin_path, Pathname.new(new_ver[:gem_home]) + 'bin')
+      else
+        new_path.remove(current_gem_bin_path)
+      end
+      @batch.set('PATH' => new_path.join )
+    end
+
+    def switch_gem_home_to(gem_home_dir)
+      gem_home_dir ||= ''
+      gem_path = Pathname.new(gem_home_dir).to_windows 
+      @batch.set('GEM_PATH' => gem_path )
+      @batch.set('GEM_HOME' => gem_path )
+    end
+  
+    def echo_ruby_version(verb='Using')
+      @batch.file_data << "for /f \"delims=\" %%a in ('ruby -v') do @echo #{verb} %%a "
+    end
+  
+    def echo_running_with_ruby_version
+      echo_ruby_version('Running with')
+    end 
+    
+    def update_gem_batch
+      BatchFile.open("#{$0}.bat") do |gem_bat|
+        # call new .pik/pik batch
+        gem_bat.call(%Q("#{@batch.path.to_windows}")) if @batch
+        gem_bat.write        
+      end
+    end
+  
+  end
+  
+end

data/lib/pik/commands/checkup_command.rb

@@ -0,0 +1,67 @@
+module Pik
+
+  class  Checkup < Command  
+    
+    aka :cu
+    it "Checks your environment for current Ruby best practices."
+    
+    def execute
+      home
+      rubyopt
+      path
+      pathext
+    end
+
+    def to_s
+      ERB.new(@output.join("\n\n")).result
+    end
+
+    def rubyopt
+      unless WindowsEnv.user['rubyopt'].empty? && WindowsEnv.system['rubyopt'].empty?
+        fail('rubyopt')        
+      else
+        pass('rubyopt')
+      end
+    end
+
+    def home
+      if WindowsEnv.user['home'].empty?
+        fail('home')        
+      else
+        pass('home')
+      end
+    end
+
+    def path
+      dirs = (WindowsEnv.user['path'] + WindowsEnv.system['path']).split(';')
+      dirs = dirs.select{|dir| File.exist?( File.join(dir,'ruby.exe') ) }
+      unless dirs.size == 1
+        fail('path')        
+      else
+        pass('path')
+      end
+    end
+
+    def pathext
+      p_ext = WindowsEnv.system['pathext'].downcase
+      unless p_ext.include?('.rb') && p_ext.include?('.rbw')
+        fail('pathext')        
+      else
+        pass('pathext')
+      end
+    end
+
+    def pass(test)
+      print '.'
+      $stdout.flush
+      # @output << @text[test][:pass]
+    end
+
+    def fail(test)
+      print 'F'
+      # @output << @text[test][:fail]
+    end
+
+  end
+  
+end

data/lib/pik/commands/command.rb

@@ -0,0 +1,166 @@
+require 'optparse'
+
+module Pik
+
+  class QuitError < StandardError
+  end
+
+  class  Command
+    
+    attr_reader :config
+    
+    attr_reader :options
+    
+    attr_accessor :output
+    
+    def self.cmd_name
+      name.split('::').last.downcase.to_sym
+    end
+    
+    def self.description
+      new.options.to_s
+    end
+    
+    def self.inherited(subclass)
+      Commands.add(subclass)
+    end
+    
+    def self.it(str)
+      @summary = str
+    end
+    
+    def self.summary
+      @summary 
+    end
+    
+    def self.aka(*aliases)
+      @names = names + aliases
+    end
+    
+    def self.names
+      @names ||= [cmd_name]
+    end
+  
+    def self.clean_gem_batch
+      BatchFile.open("#{$0}.bat") do |gem_bat|
+        # remove old calls to .pik/pik batches
+        gem_bat.remove_line( /call.+pik.+bat/i )
+        gem_bat.write  
+      end
+    end
+
+    def self.choose_from(patterns, config)
+      if patterns.empty?
+        possibles = config.keys  
+      else
+        possibles = patterns.map{|p| config.keys.grep(Regexp.new(Regexp.escape(p.to_s))) }
+        possibles = possibles.inject{|m,v| m & v }.flatten.uniq
+      end
+      case possibles.size
+      when 0
+        return nil
+      when 1
+        return possibles.first
+      else
+        hl.say('Select which Ruby you want:')
+        ver = hl.choose(*possibles)
+        return ver
+      end
+    end
+    
+    def self.hl
+      @hl ||= HighLine.new
+    end
+  
+    def initialize(args=ARGV, config_=nil)
+      @args    = args
+      @options = OptionParser.new
+      @config  = config_ || ConfigFile.new
+      @hl      = HighLine.new
+      add_sigint_handler
+      options.program_name = "#{File.basename($0)} #{self.class.names.join('|')}"
+      command_options
+      parse_options
+      create(PIK_HOME) unless PIK_HOME.exist?
+      # Add.new(nil, config).execute if config.empty?
+      delete_old_pik_batches
+    end
+
+    def close
+      editors.each{|e| e.write }
+    end
+    
+    def editors
+      @editors ||= []
+    end
+    
+    def command_options
+      options.separator ""
+      options.separator self.class.summary
+      options.separator ""
+    end
+    
+    def parse_options
+      options.parse! @args
+    end
+   
+    def current_version?(string)
+      string == get_version
+    end
+        
+    def get_version(path=current_ruby_bin_path)
+      cmd = ruby_exe(path).basename
+      cmd = Pathname.new(path) + cmd
+      ruby_ver = `#{cmd} -v`
+      ruby_ver =~ /ruby (\d\.\d\.\d)/
+      major    = $1.gsub('.','')
+      "#{major}: #{ruby_ver.strip}"
+    end
+        
+    def current_ruby_bin_path
+      Pathname.new(::RbConfig::CONFIG['bindir'] )
+    end
+
+    def ruby_exists_at?(path)
+      !!ruby_exe(path)
+    end
+    
+    def ruby_exe(path)
+      ruby_glob(path).first
+    end
+    
+    def ruby_glob(path)
+      glob = "#{Pathname.new(path).to_ruby}/{ruby.exe}"
+      Pathname.glob(glob)
+    end
+    
+    def current_gem_bin_path
+      default_gem_home + 'bin'
+    end
+
+    def default_gem_home
+      Pathname.new(Gem.default_path.first).to_windows
+    end
+    
+    def create(home)
+      puts "creating #{home}"
+      home.mkpath
+    end
+   
+    def delete_old_pik_batches( cutoff=(Time.now - (2 * 60 * 60)) )
+      Dir[(PIK_HOME + "*.bat").to_windows.to_s].each do |f| 
+        File.delete(f) if File.ctime(f) < cutoff 
+      end
+    end
+
+    # Installs a sigint handler.
+  
+    def add_sigint_handler
+      trap 'INT' do
+        raise QuitError
+      end
+    end
+    
+  end  
+    
+end

data/lib/pik/commands/config_command.rb

@@ -0,0 +1,28 @@
+module Pik
+
+  class  Config < Command
+    
+    it "Adds/modifies configuration options."
+    include BatchFileEditor
+    include ConfigFileEditor
+    
+    attr_accessor :global
+
+    def execute
+      item, value = @args.shift.downcase.split('=')
+      case item
+      when 'rubyopt'
+        case value
+        when 'on'  then set('RUBYOPT' => '-rubygems')
+        when 'off' then set('RUBYOPT' => nil)
+        end
+      when 'gem_home'
+        config[get_version][:gem_home] = @args.include?('default') ? default_gem_home : value
+      else
+        puts "Unknown configuration option"
+      end
+    end
+
+  end 
+  
+end

data/lib/pik/commands/config_file_editor.rb

@@ -0,0 +1,12 @@
+module Pik
+
+  module ConfigFileEditor
+
+    def initialize(args=ARGV,config=nil)
+      super
+      editors << @config
+    end
+  
+  end
+  
+end

data/lib/pik/commands/default_command.rb

@@ -0,0 +1,22 @@
+
+module Pik
+  
+  class Default < Command
+  
+    it "switches back to the default settings"
+  
+    include BatchFileEditor
+  
+    def execute
+      sys = WindowsEnv.system
+      usr = WindowsEnv.user
+      
+      @batch.set('PATH' => sys['PATH'] + ';' + usr['PATH'] )
+      @batch.set('GEM_PATH' => usr['GEM_PATH'] || sys['GEM_PATH'] )
+      @batch.set('GEM_HOME' => usr['GEM_HOME'] || sys['GEM_HOME'] )
+      echo_ruby_version
+    end
+  
+  end
+
+end

data/lib/pik/commands/gemsync_command.rb

@@ -0,0 +1,69 @@
+module Pik
+
+  class  GemSync < Command
+  
+    it "Duplicates gems from the current version to the one specified."
+    include BatchFileEditor
+    
+    def execute
+      target  = self.class.choose_from(@args, config)
+      current = get_version
+      install_gems(current, target) if target
+    end
+    
+    def install_gems(current, target)
+      switch_path_to(config[target])
+      switch_gem_home_to(config[target][:gem_home]) 
+      target_cache = gem_cache(target)
+      
+      gem_cache(current).find do |file|
+        if file.file?
+          if (target_cache + file.basename).exist? 
+            @batch.echo "Gem #{file.basename} already installed"
+          else
+            gem_install(file)
+          end
+        end
+      end
+      switch_path_to(config[current])
+      switch_gem_home_to(config[target][:gem_home]) 
+    end
+    
+    def gem_cache(version)
+      conf = config[version]
+      path = if conf[:gem_home] 
+        Pathname.new( conf[:gem_home] )
+      else
+        cmd = ruby_exe(conf[:path]).to_s + " -rubygems -e \"puts Gem.default_path.last\""
+        Pathname.new( `#{cmd}`.chomp ) 
+      end
+      puts path + "cache"
+      path + "cache"
+    end
+    
+    def gem_install(file)
+      @batch.echo "Installing #{file.basename}"
+      @batch.call "gem install -q --no-rdoc --no-ri #{file}"
+    end
+    
+    def command_options
+      super
+      sep =<<SEP
+  Examples:
+
+    C:\\>ruby -v
+    ruby 1.8.6 (2009-03-31 patchlevel 368) [i386-mingw32]
+    
+    C:\\>pik gemsync 191 p2
+    Gem ZenTest-4.1.4.gem already installed
+    Installing xml-simple-1.0.12.gem
+    Successfully installed xml-simple-1.0.12
+    1 gem installed
+    ...
+SEP
+      options.separator sep
+    end
+  
+  end
+
+end