s3sizecalc 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 5989b8eeb58eafc76a897bef4b9564d71a0895d4
+  data.tar.gz: d89b75c832ecd11811a5b3c129c15b62e5e85448
+SHA512:
+  metadata.gz: 08b53da0f51e52f2b76fd2ebc39da35e629a738da28d59b7c85ca6f2b86ede4c480234d3f634b6d39508411e5a60e60a8cd7181ee75ca2090db23e66ec1dc0b1
+  data.tar.gz: 3495ac2b34566e37202d32e1371669ab4ae05cb51b7d80d5c4ca40a6c5a36f56b643d00cfdebdbe86f0d6c7504e1fba752eaed8ed6321713a61bcdc6f392b951

data/.gitignore

@@ -0,0 +1,10 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+vendor

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,3 @@
+language: ruby
+rvm:
+  - 2.2.0

data/Gemfile

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

data/README.md

@@ -0,0 +1,54 @@
+# S3SizeCalc
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/s3sizecalc`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 's3sizecalc'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install s3sizecalc
+
+## Usage
+
+```
+# Bucket total
+$ s3sizecalc bucket --bucket my-bucket
+
+# Region total
+$ s3sizecalc region --region ap-northeast-1
+
+# Account total
+$ s3sizecalc all
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `bin/console` for an interactive prompt that will allow you to experiment. Run `bundle exec S3SizeCalc` to use the code located in this directory, ignoring other installed copies of this gem.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release` to create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## TODO
+
+* Concurrent head_object requests
+* Error handling, retries
+* Write spec
+
+## Contributing
+
+1. Fork it ( https://github.com/y13i/s3sizecalc/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,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "s3sizecalc"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "pry"
+Pry.start

data/bin/setup

@@ -0,0 +1,7 @@
+#!/bin/bash
+set -euo pipefail
+IFS=$'\n\t'
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/exe/s3sizecalc

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require "s3sizecalc"
+
+S3SizeCalc::CLI.start

data/lib/s3sizecalc/calculator.rb

@@ -0,0 +1,67 @@
+require "net/http"
+require "timeout"
+require "aws-sdk"
+require "retryable"
+
+module S3SizeCalc
+  class Calculator
+    def initialize(options = {})
+      @options = options
+    end
+
+    def bucket_total(bucket_name)
+      puts "Calculating size of bucket `#{bucket_name}`." if @options.verbose
+
+      s3.bucket(bucket_name).objects.inject 0 do |total, object|
+        size = object.size
+        puts "Size of object `#{object.key}` => #{size} bytes." if @options.very_verbose
+        total += size
+      end
+    end
+
+    def region_total(region = current_region)
+      @current_region = region
+
+      s3(region).buckets.inject 0 do |total, bucket|
+        if s3.client.get_bucket_location(bucket: bucket.name).location_constraint == region
+          total += bucket_total(bucket.name)
+        else
+          total
+        end
+      end
+    end
+
+    def account_total
+      regions = ec2.client.describe_regions[:regions].map {|region| region[:region_name]}
+
+      regions.inject 0 do |total, region|
+        total += region_total(region)
+      end
+    end
+
+    def current_region
+      @current_region || @options.region || own_region
+    end
+
+    private
+
+    def own_region
+      @own_region ||= begin
+        timeout 2 do
+          Net::HTTP.get("169.254.169.254", "latest/meta-data/placement/availability-zone").chop
+        end
+      rescue
+        abort "You must specify a region."
+      end
+    end
+
+    def s3(region = current_region)
+      name = :"@s3_#{region.gsub("-", "_")}"
+      instance_variable_get(name) or instance_variable_set(name, Aws::S3::Resource.new(region: (region)))
+    end
+
+    def ec2
+      @ec2 ||= Aws::EC2::Resource.new(region: (current_region))
+    end
+  end
+end

data/lib/s3sizecalc/cli.rb

@@ -0,0 +1,57 @@
+require "thor"
+require "thor/aws"
+
+module S3SizeCalc
+  class CLI < Thor
+    include Thor::Aws
+
+    class_option :humanize,     type: :boolean, default: true,  aliases: [:h],  desc: "Output size with human friendly unit."
+    class_option :verbose,      type: :boolean, default: true,  aliases: [:v],  desc: "Verbose output."
+    class_option :very_verbose, type: :boolean, default: false,                 desc: "Very verbose output."
+
+    desc :bucket, "Puts bucket total used size."
+    method_option :bucket, required: true, aliases: [:b], desc: "S3 bucket name."
+    def bucket
+      puts "Bucket (#{options.bucket}) total: #{format_size(calculator.bucket_total)}"
+    end
+
+    desc :region, "Puts region total used size."
+    def region
+      puts "Region (#{calculator.current_region}) total: #{format_size(calculator.region_total)}"
+    end
+
+    desc :all, "Puts account total used size."
+    def all
+      puts "Account total: #{format_size(calculator.account_total)}"
+    end
+
+    private
+
+    def calculator
+      @calculator ||= Calculator.new options
+    end
+
+    def format_size(bytes)
+      if options.humanize
+        case
+        when bytes > 2 ** 60
+          "#{sprintf "%.2f", (bytes.to_f / (2 ** 60))} EiB"
+        when bytes > 2 ** 50
+          "#{sprintf "%.2f", (bytes.to_f / (2 ** 50))} PiB"
+        when bytes > 2 ** 40
+          "#{sprintf "%.2f", (bytes.to_f / (2 ** 40))} TiB"
+        when bytes > 2 ** 30
+          "#{sprintf "%.2f", (bytes.to_f / (2 ** 30))} GiB"
+        when bytes > 2 ** 20
+          "#{sprintf "%.2f", (bytes.to_f / (2 ** 20))} MiB"
+        when bytes > 2 ** 10
+          "#{sprintf "%.2f", (bytes.to_f / (2 ** 10))} KiB"
+        else
+          "#{bytes} Bytes"
+        end
+      else
+        "#{bytes} Bytes"
+      end
+    end
+  end
+end

data/lib/s3sizecalc/version.rb

@@ -0,0 +1,3 @@
+module S3SizeCalc
+  VERSION = "0.0.1".freeze
+end

data/lib/s3sizecalc.rb

@@ -0,0 +1,3 @@
+require "s3sizecalc/version"
+require "s3sizecalc/calculator"
+require "s3sizecalc/cli"

data/s3sizecalc.gemspec

@@ -0,0 +1,31 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 's3sizecalc/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "s3sizecalc"
+  spec.version       = S3SizeCalc::VERSION
+  spec.authors       = ["y13i"]
+  spec.email         = ["email@y13i.com"]
+
+  spec.summary       = %(Calculate total size usage of Amazon S3.)
+  spec.description   = %(Calculate bucket/region/account total size usage of Amazon S3.)
+  spec.homepage      = "https://github.com/y13i/s3sizecalc"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.9"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "pry"
+
+  spec.add_dependency "aws-sdk", "~> 2"
+  spec.add_dependency "thor"
+  spec.add_dependency "thor-aws"
+  spec.add_dependency "concurrent-ruby"
+  spec.add_dependency "retryable"
+end

metadata

@@ -0,0 +1,184 @@
+--- !ruby/object:Gem::Specification
+name: s3sizecalc
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- y13i
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-05-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor-aws
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: retryable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Calculate bucket/region/account total size usage of Amazon S3.
+email:
+- email@y13i.com
+executables:
+- s3sizecalc
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/s3sizecalc
+- lib/s3sizecalc.rb
+- lib/s3sizecalc/calculator.rb
+- lib/s3sizecalc/cli.rb
+- lib/s3sizecalc/version.rb
+- s3sizecalc.gemspec
+homepage: https://github.com/y13i/s3sizecalc
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Calculate total size usage of Amazon S3.
+test_files: []