app_hosts 0.0.3

data/.gitignore

@@ -0,0 +1,5 @@
+.DS_Store
+app_hosts-*.gem
+config
+config/**/*
+pkg

data/Gemfile

@@ -0,0 +1,5 @@
+source 'http://rubygems.org'
+
+group :development, :test do
+  gem 'rspec-rails', '>= 2.0.0.beta.22'
+end

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Andrea Franz
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# AppHosts
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'app_hosts'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install app_hosts
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,12 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+Dir[File.dirname(__FILE__) + '/lib/**/*.rake'].each{ |file| load file }
+
+RSpec::Core::RakeTask.new(:spec) do |t|
+  t.rspec_opts = %w[--color --format documentation]
+  t.pattern = 'spec/*_spec.rb'
+end
+
+task :default => :spec

data/app_hosts.gemspec

@@ -0,0 +1,17 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/app_hosts/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Andrea Franz"]
+  gem.email         = ["andrea@gravityblast.com"]
+  gem.description   = %q{Application based /etc/hosts manager}
+  gem.summary       = %q{}
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "app_hosts"
+  gem.require_paths = ["lib"]
+  gem.version       = AppHosts::VERSION
+end

data/lib/app_hosts.rb

@@ -0,0 +1,2 @@
+require "app_hosts/version"
+require File.dirname(__FILE__) + '/app_hosts/host_manager.rb'

data/lib/app_hosts/host_manager.rb

@@ -0,0 +1,107 @@
+require 'yaml'
+
+module AppHosts
+  class HostManager
+    
+    attr_reader :hosts, :lines
+    
+    def initialize config_file_path
+      @lines = []
+      @parsed = false
+      @config = YAML.load_file config_file_path
+    end
+    
+    def parse file_path=@config['hosts_file_path']
+      @file_path = file_path
+      parse_file file_path
+      @parsed = true
+    end
+    
+    def find_ip_by_name name
+      pair = @config['addresses'].find{|key, value| key == name}
+      pair ? pair.last : nil
+    end
+    
+    def switch_ip_to named_host, file_path=@file_path
+      new_ip = find_ip_by_name named_host
+      raise "Named host not found: #{named_host}" unless new_ip
+      replace_ip @config['host'], new_ip, file_path
+    end
+    
+    def replace_ip host, new_ip, file_path=@file_path
+      raise "You should parse original file before replacing ip address." unless @parsed
+      
+      buffer = ''
+      
+      found = false
+      @lines.each do |line|
+        line = if line[:host] == host
+          found = true
+          build_line new_ip, line[:host], line[:aliases]
+        else
+          line[:text]
+        end
+        buffer << "#{line}\n"
+      end
+        
+      buffer << "#{build_line(new_ip, host)}\n" if !found
+      write_file buffer, file_path      
+    end        
+    
+    def build_line ip, host, aliases=[]
+      "#{ip} #{host} #{aliases.join(' ')}".strip
+    end
+    
+    def replace_ip_to_line line, new_ip
+      new_line = ""
+      line_attributes = parse_line line.strip.chomp
+      if !line_attributes.nil?
+        ip, attributes = line_attributes
+        new_line << build_line(new_ip, attributes[:host], attributes[:aliases])
+      end
+      new_line
+    end
+    
+  private
+    
+    def write_file buffer, file_path
+      buffer.gsub!(/"/, '\\"')
+      buffer.gsub!(/'/){ |match| %|"'"\\'"'"|}
+      cmd = "echo"
+      if @config['use_sudo']
+        system %|sudo sh -c '#{cmd} "#{buffer}\\c" > "#{file_path}"'|
+      else
+        system %|#{cmd} "#{buffer}\\c" > "#{file_path}"|
+      end      
+    end
+    def parse_file file_path
+      each_file_line file_path do |line|        
+        line_attributes = parse_line line
+        if !line_attributes.nil?
+          ip, attributes = line_attributes
+          @lines << { :ip => ip, :host => attributes[:host], :aliases => attributes[:aliases], :text => line }
+        else          
+          @lines << { :text => line }
+        end
+      end
+    end
+    
+    def each_file_line file_path
+      File.open file_path, 'r' do |file|
+        file.each do |line|
+          yield line.chomp.strip
+        end
+      end
+    end
+    
+    def parse_line line
+      return nil if line.size == 0 || line =~ /^\s*#/
+      attributes = line.split.collect &:strip
+      [
+        attributes.shift,
+        :host => attributes.shift,
+        :aliases => attributes
+      ]
+    end        
+  end
+end

data/lib/app_hosts/version.rb

@@ -0,0 +1,3 @@
+module AppHosts
+  VERSION = "0.0.3"
+end

data/lib/tasks/hosts_manager.rake

@@ -0,0 +1,17 @@
+require File.dirname(__FILE__) + '/../app_hosts'
+
+namespace :app_hosts do  
+  desc 'Switch'
+  task :switch, :named_host do |task, args|
+    named_host = args[:named_host].to_s.strip
+    raise "You must specify the named host: rake app_hosts:switch[NAMED_HOST]" if named_host.size == 0
+    in_file = File.dirname(__FILE__) + '/../../spec/fixtures/hosts'
+    out_file = 'test.txt'
+    host = 'www.example.com'
+    ip = '2.2.2.2'
+    
+    manager = AppHosts::HostManager.new 'config/app_hosts.yml'
+    manager.parse
+    manager.switch_ip_to named_host
+  end
+end

data/spec/app_host_spec.rb

@@ -0,0 +1,128 @@
+require 'fileutils'
+require File.dirname(__FILE__) + '/../lib/app_hosts'
+
+describe AppHosts do
+  before(:each) do
+    @manager = AppHosts::HostManager.new File.dirname(__FILE__) + '/fixtures/app_hosts.yml'
+  end
+  
+  it 'should parse line' do
+    line = ' 127.0.0.1 example.com www.example.com beta.example.com '
+    @manager.send(:parse_line, line).should == ['127.0.0.1', {
+      :host => 'example.com',
+      :aliases => ['www.example.com', 'beta.example.com']
+    }]
+  end
+  
+  it 'should not parse comments' do
+    line = ' # 127.0.0.1 example.com www.example.com beta.example.com '
+    @manager.send(:parse_line, line).should be_nil
+  end    
+  
+  it 'should save file lines' do
+    @manager.parse File.dirname(__FILE__) + '/fixtures/hosts'
+    lines = [
+      {:text => "# first line comment"},
+      {:text => ""},
+      {:text => "127.0.0.1 www.example.com example.com", :ip => '127.0.0.1', :host => 'www.example.com', :aliases => ['example.com']},
+      {:text => "127.0.0.2 hello-example.com", :ip => '127.0.0.2', :host => 'hello-example.com', :aliases => []},
+      {:text => "# last line comment"}
+    ]
+    @manager.lines.should =~ lines
+  end
+  
+  it 'should replace ip address to line' do
+    line = '127.0.0.1 example.com www.example.com beta.example.com'
+    new_ip = '127.0.0.2'
+    new_line = "#{new_ip} example.com www.example.com beta.example.com"
+    @manager.replace_ip_to_line(line, new_ip).should == new_line
+  end    
+  
+  it 'should parse config file' do
+    config = {
+      'hosts_file_path' => '/path/to/hosts',
+      'host' => 'www.example.com',
+      'addresses' => {
+        'development' => '127.0.0.1',
+        'staging' => '127.0.0.2',
+        'beta' => '127.0.0.3',
+        'production' => '127.0.0.4'
+      }
+    }
+    @manager.instance_variable_get('@config').should == config
+  end
+  
+  it 'should find host ip by name' do
+    {
+      'development' => '127.0.0.1',
+      'staging' => '127.0.0.2',
+      'beta' => '127.0.0.3',
+      'production' => '127.0.0.4'
+    }.each do |name, ip|
+      @manager.find_ip_by_name(name).should == ip
+    end
+  end
+  
+  it 'should not find undefined address' do
+    @manager.find_ip_by_name('undefined-name').should be_nil
+  end
+  
+  it 'should replace ip for specified host' do
+    tmp_path = File.dirname(__FILE__) + '/../tmp'
+    host = "www.example.com"
+    new_ip = '1.1.1.1'
+    original_file_path = File.dirname(__FILE__) + '/fixtures/hosts'
+    output_file_path = '/tmp/new_hosts.txt'
+    new_content = <<-FILE
+# first line comment
+
+#{new_ip} www.example.com example.com
+127.0.0.2 hello-example.com
+# last line comment
+FILE
+    @manager.should_receive(:write_file).with(new_content, output_file_path)
+    @manager.parse original_file_path
+    @manager.replace_ip host, new_ip, output_file_path    
+  end
+  
+  it 'should replace ip for specified named host' do
+    tmp_path = File.dirname(__FILE__) + '/../tmp'
+    original_file_path = File.dirname(__FILE__) + '/fixtures/hosts'
+    output_file_path = '/tmp/new_hosts.txt'
+    named_host = 'production'
+    new_ip = @manager.instance_variable_get('@config')['addresses']['production']
+    new_content = <<-FILE
+# first line comment
+
+#{new_ip} www.example.com example.com
+127.0.0.2 hello-example.com
+# last line comment
+FILE
+    @manager.should_receive(:write_file).with(new_content, output_file_path)
+    @manager.parse original_file_path
+    @manager.switch_ip_to named_host, output_file_path    
+  end
+  
+  it 'should add ip and hosts if not already specified' do
+    original_file_path = File.dirname(__FILE__) + '/fixtures/hosts'
+    output_file_path = '/tmp/new_hosts.txt'
+    named_host = 'production'
+    
+    config = @manager.instance_variable_get('@config')
+    config['host'] = 'new-example.com'
+    @manager.instance_variable_set('@config', config)
+    
+    new_ip = @manager.instance_variable_get('@config')['addresses']['production']
+    new_content = <<-FILE
+# first line comment
+
+127.0.0.1 www.example.com example.com
+127.0.0.2 hello-example.com
+# last line comment
+#{new_ip} new-example.com
+FILE
+    @manager.should_receive(:write_file).with(new_content, output_file_path)
+    @manager.parse original_file_path
+    @manager.switch_ip_to named_host, output_file_path
+  end
+end

data/spec/fixtures/app_hosts.yml

@@ -0,0 +1,7 @@
+hosts_file_path: /path/to/hosts
+host: www.example.com
+addresses:
+  development: 127.0.0.1
+  staging: 127.0.0.2
+  beta: 127.0.0.3
+  production: 127.0.0.4

data/spec/fixtures/hosts

@@ -0,0 +1,5 @@
+# first line comment
+
+127.0.0.1 www.example.com example.com
+127.0.0.2 hello-example.com
+# last line comment

metadata

@@ -0,0 +1,64 @@
+--- !ruby/object:Gem::Specification
+name: app_hosts
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- Andrea Franz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-11-13 00:00:00.000000000 Z
+dependencies: []
+description: Application based /etc/hosts manager
+email:
+- andrea@gravityblast.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- app_hosts.gemspec
+- lib/app_hosts.rb
+- lib/app_hosts/host_manager.rb
+- lib/app_hosts/version.rb
+- lib/tasks/hosts_manager.rake
+- spec/app_host_spec.rb
+- spec/fixtures/app_hosts.yml
+- spec/fixtures/hosts
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -976389179289596147
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: ''
+test_files:
+- spec/app_host_spec.rb
+- spec/fixtures/app_hosts.yml
+- spec/fixtures/hosts