aws-ip 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6b8a8763f0c0b8b7d6e6d5226a20b43acfa8bca5
+  data.tar.gz: 7ee5ad3cb96483fed021ae2e8f3e2563dc65ebe6
+SHA512:
+  metadata.gz: fa9e6c7e692f66626c0a87868692bf23b045ca726618a86c0d87db8f50b12c11aab3f5158f6bca8398c59119bd8aa61d5f0533d553ea7c7790834c6f0730c702
+  data.tar.gz: 5b2ca93776681140cd52c648a6d151a64f4a9cc2511e4eafdf8bf88db7fb843f2dffb53f0d5c0d13869017ef33884b2ad8f6a537df834da37512a0348b24e9e8

data/.gitignore

@@ -0,0 +1,14 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log

data/.rspec

@@ -0,0 +1,4 @@
+--color
+--warnings
+--require spec_helper
+--format doc

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in aws-ip.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 rochefort
+
+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,31 @@
+# Aws::Ip
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'aws-ip'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install aws-ip
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it ( https://github.com/[my-github-username]/aws-ip/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request

data/Rakefile

@@ -0,0 +1,7 @@
+require 'bundler/gem_tasks'
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec)
+
+task test: :spec
+task default: :spec

data/aws-ip.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'aws_ip/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'aws-ip'
+  spec.version       = AwsIp::VERSION
+  spec.authors       = ['rochefort']
+  spec.email         = ['terasawan@gmail.com']
+  spec.summary       = %q{find out AWS IP address}
+  spec.description   = spec.summary
+  spec.homepage      = ''
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+
+  spec.add_development_dependency 'bundler', '~> 1.7'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'ipaddress', '~> 0.8.0'
+
+  spec.add_development_dependency 'rspec', '~> 3.1.0'
+  spec.add_development_dependency 'webmock', '~> 1.20.4'
+end

data/bin/aws-ip

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__) + '/../lib'))
+require 'aws_ip'
+
+fail ArgumentError, 'invalid argument' if ARGV.size == 0
+
+aws = AwsIp.new
+puts aws.env
+puts aws.fetch ARGV[0]

data/lib/aws_ip/base.rb

@@ -0,0 +1,49 @@
+require 'ipaddress'
+
+module AwsIp
+  class Base
+    def initialize
+      @range = Range.new
+      @ranges ||= @range.get
+    end
+
+    def update
+      @ranges = @range.get
+    end
+
+    def ranges
+      @ranges['prefixes']
+    end
+    alias_method :all_ranges, :ranges
+
+    def env
+      "syncToken : #{@ranges['syncToken']}\n" \
+      "createDate: #{@ranges['createDate']}\n\n"
+    end
+
+    def fetch(ip_address)
+      ip = parse(ip_address)
+      @ranges['prefixes'].find_all do |prefix|
+        prefix_ip = IPAddress(prefix['ip_prefix'])
+        prefix_ip.include?(ip)
+      end
+    end
+
+    def include?(ip_address)
+      ip = parse(ip_address)
+      result = @ranges['prefixes'].find do |prefix|
+        prefix_ip = IPAddress(prefix['ip_prefix'])
+        prefix_ip.include?(ip)
+      end
+      !!result
+    end
+
+    private
+    def parse(ip_address)
+      IPAddress(ip_address)
+    rescue => e
+      puts e.stacktrace if ENV['DEBUG']
+      abort e.message
+    end
+  end
+end

data/lib/aws_ip/range.rb

@@ -0,0 +1,35 @@
+require 'net/http'
+require 'open-uri'
+require 'json'
+
+module AwsIp
+  class Range
+    BASE_URI = 'https://ip-ranges.amazonaws.com/ip-ranges.json'
+
+    def initialize
+      @uri = URI.parse(BASE_URI)
+      @proxy = @uri.find_proxy
+    end
+
+    def get
+      connection = if @proxy
+        Net::HTTP::Proxy(@proxy.host, @proxy.port, @proxy.user, @proxy.password).new(@uri.host, @uri.port)
+      else
+        Net::HTTP.new(@uri.host, @uri.port)
+      end
+      connection.use_ssl = true
+      response = connection.start { |http| http.get(@uri) }
+      case response
+      when Net::HTTPSuccess
+        json = response.body
+        JSON.parse(json)
+      else
+        # raise excpetion
+        response.value
+      end
+    rescue => e
+      puts e.stacktrace if ENV['DEBUG']
+      abort [e.message, @uri].join(' : ')
+    end
+  end
+end

data/lib/aws_ip/version.rb

@@ -0,0 +1,3 @@
+module AwsIp
+  VERSION = '0.0.1'
+end

data/lib/aws_ip.rb

@@ -0,0 +1,13 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'aws_ip/base'
+require 'aws_ip/range'
+require 'aws_ip/version'
+
+module AwsIp
+  class << self
+    def new
+      AwsIp::Base.new
+    end
+  end
+end

data/spec/aws_ip/base_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+
+RSpec.describe AwsIp::Base do
+  before do
+    stub_base_request
+    @ip = AwsIp.new
+  end
+
+  describe '#fetch' do
+    before do
+      @result = [{"ip_prefix"=>"54.240.200.0/24", "region"=>"ap-northeast-1", "service"=>"AMAZON"}]
+    end
+    context 'in range' do
+      it { expect(@ip.fetch('54.240.200.0')).to eq(@result) }
+      it { expect(@ip.fetch('54.240.200.255')).to eq(@result) }
+    end
+    context 'out of range' do
+      it { expect(@ip.fetch('54.240.201.0')).to eq([]) }
+    end
+  end
+
+  describe '#include?' do
+    context 'in range' do
+      it { expect(@ip.include?('54.240.200.0')).to be_truthy }
+      it { expect(@ip.include?('54.240.200.255')).to be_truthy }
+    end
+    context 'out of range' do
+      it { expect(@ip.include?('54.240.201.0')).to be_falsey }
+    end
+  end
+
+end

data/spec/aws_ip/range_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+RSpec.describe Range do
+  describe 'request behind proxy' do
+    before do
+      allow(ENV).to receive(:[]).with('no_proxy').and_return('')
+      allow(ENV).to receive(:[]).with('https_proxy')
+        .and_return('http://proxy_user:proxy_pass@192.168.1.99:9999')
+      ip = AwsIp::Range.new
+      @proxy = ip.instance_variable_get(:@proxy)
+    end
+    it { expect(@proxy.host).to eq '192.168.1.99' }
+    it { expect(@proxy.user).to eq 'proxy_user' }
+    it { expect(@proxy.password).to eq 'proxy_pass' }
+    it { expect(@proxy.port).to eq 9999 }
+  end
+
+  describe '#get' do
+    context 'when getting json error occurred' do
+      before do
+        stub_request(:get, AwsIp::Range::BASE_URI)
+          .to_return(status: 500, headers: { content_type: 'application/json' }, body: '')
+      end
+      it { expect { AwsIp::Range.new.get }.to raise_error(Exception) }
+    end
+  end
+end