heirloom 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/.rvmrc

@@ -0,0 +1 @@
+rvm use ruby-1.9.3-p125@heirloom --create

data/Gemfile

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

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/heirloom

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require 'heirloom'
+require 'heirloom/cli'
+
+Heirloom::CLI.start

data/heirloom.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "heirloom/version"
+
+Gem::Specification.new do |s|
+  s.name        = "heirloom"
+  s.version     = Heirloom::VERSION
+  s.authors     = ["Brett Weaver"]
+  s.email       = ["brett@weav.net"]
+  s.homepage    = ""
+  s.summary     = %q{I help with artifacts}
+  s.description = %q{I help build and manage artifacts}
+
+  s.rubyforge_project = "heirloom"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  s.add_development_dependency "rspec"
+
+  s.add_runtime_dependency 'fog'
+  s.add_runtime_dependency 'grit'
+  s.add_runtime_dependency 'logger'
+  s.add_runtime_dependency 'minitar'
+  s.add_runtime_dependency 'trollop'
+end

data/lib/heirloom.rb

@@ -0,0 +1,189 @@
+require "heirloom/aws"
+require "heirloom/version"
+
+require 'socket'
+require 'fog'
+require 'grit'
+require 'logger'
+require 'time'
+require 'zlib'
+require 'archive/tar/minitar'
+include Archive::Tar
+include Grit
+
+module Heirloom
+
+  class Heirloom
+    attr_accessor :heirloom_type, :commit, :source_dir, :prefix
+
+    def self.list
+      sdb = self.connect_to_sdb
+
+      r = {}
+
+      sdb.domains.each do |domain|
+        r[domain] = sdb.select("select * from #{domain}")
+      end
+
+      r
+    end
+
+    def self.info(args)
+      sdb = self.connect_to_sdb
+
+      sdb.select("select * from #{args[:class]} where itemName() = '#{args[:sha]}'")
+    end
+
+    def self.delete(args)
+      sdb = self.connect_to_sdb
+
+      sdb.delete args[:class], args[:sha]
+    end
+
+    def self.connect_to_sdb
+      @access_key = ENV['AWS_ACCESS_KEY_ID']
+      @secret_key = ENV['AWS_SECRET_ACCESS_KEY']
+
+      AWS::SimpleDb.new(@access_key, @secret_key)
+    end
+
+    def initialize(args)
+      @sdb = self.class.connect_to_sdb
+
+      self.heirloom_type = args[:heirloom_type]
+      self.source_dir = args[:source_dir]
+      self.prefix = args[:prefix]
+    end
+
+    def build_and_upload_to_s3(args)
+      get_commit args[:sha]
+      raise 'Artifact already uploaded to S3' if already_in_sdb?
+      build_local_artifact
+      create_sdb_record
+      upload_to_s3
+      delete_local_artifact
+    end
+
+    protected
+
+    def get_commit(sha = nil)
+      r = Repo.new source_dir
+      if sha
+        self.commit = r.commits(sha).first
+      else
+        self.commit = r.commits.first
+      end
+    end
+
+    def build_local_artifact
+      tgz = Zlib::GzipWriter.new File.open(artifact_path, 'wb')
+      Minitar.pack(source_dir, tgz)
+    end
+
+    def delete_local_artifact
+      File.delete(artifact_path)
+    end
+
+    def create_sdb_domain
+      @sdb.create_domain(domain)
+    end
+
+    def create_sdb_record
+      create_sdb_domain 
+      @sdb.put_attributes domain, sha, { 'built_by' => "#{user}@#{hostname}",
+                                         'built_at' => Time.now.utc.iso8601,
+                                         'tests' => 'pending',
+                                         'sha' => sha,
+                                         'abbreviated_sha' => abbreviated_sha,
+                                         'message' => message,
+                                         'author' => author }
+    end
+
+    def already_in_sdb?
+      @sdb.select("select * from #{domain} where itemName() = '#{sha}'").count > 0
+    end
+
+    def upload_to_s3
+      target_buckets = buckets
+
+      # Upload the artifact to each bucket
+      target_buckets.each_pair do |bucket, options|
+
+        # Connect to s3 in region of bucket
+        region = options['region']
+        connection = Fog::Storage.new :provider                 => 'AWS',
+                                      :aws_access_key_id        => ENV['AWS_ACCESS_KEY_ID'],
+                                      :aws_secret_access_key    => ENV['AWS_SECRET_ACCESS_KEY'],
+                                      :region                   => region
+
+        # Get bucket
+        b = connection.directories.get bucket
+
+        # Upload the artifact
+        b.files.create :key    => "#{folder}/#{artifact}",
+                       :body   => File.open(artifact_path),
+                       :public => false
+                       
+        # Add the artifact location
+        @sdb.put_attributes domain, sha, { "#{region}-s3-url" => "s3://#{bucket}/#{folder}/#{artifact}" }
+
+        # Add the http url
+        @sdb.put_attributes domain, sha, { "#{region}-http-url" => "http://#{bucket}/#{folder}/#{artifact}" }
+
+        # Add the https url
+        @sdb.put_attributes domain, sha, { "#{region}-https-url" => "https://#{bucket}/#{folder}/#{artifact}" }
+      end
+    end
+
+    private
+
+    def buckets
+      {
+        "#{prefix}-us-west-1" => { 
+          'region' => 'us-west-1'
+        }
+      }
+    end
+
+    def artifact
+      "#{sha}.tar.gz"
+    end
+
+    def artifact_path
+      "/tmp/#{artifact}"
+    end
+
+    def domain
+      heirloom_type
+    end
+
+    def folder
+      heirloom_type
+    end
+
+    def author
+      commit.author.name
+    end
+
+    def message
+      commit.message
+    end
+    
+    def sha
+      commit.id
+    end
+
+    def abbreviated_sha
+      commit.id_abbrev
+    end
+
+    def user
+      ENV['USER']
+    end
+
+    def hostname
+      Socket.gethostname
+    end
+
+  end
+end

data/lib/heirloom/aws.rb

@@ -0,0 +1,35 @@
+module Heirloom
+
+  class AWS
+    class SimpleDb
+
+      def initialize(access_key, secret_key)
+        @sdb = Fog::AWS::SimpleDB.new :aws_access_key_id => access_key,
+                                      :aws_secret_access_key => secret_key,
+                                      :region => 'us-west-1'
+      end
+
+      def domains
+        @sdb.list_domains.body['Domains']
+      end
+
+      def create_domain(domain)
+        @sdb.create_domain(domain) unless domains.include? domain
+      end
+
+      def put_attributes(domain, key, args)
+        @sdb.put_attributes domain, key, args
+      end
+
+      def select(query)
+        @sdb.select(query).body['Items']
+      end
+
+      def delete(domain, key)
+        @sdb.delete_attributes domain, key
+      end
+
+    end
+  end
+
+end

data/lib/heirloom/cli.rb

@@ -0,0 +1,48 @@
+require 'trollop'
+
+module Heirloom
+  module CLI
+    def self.start
+      @opts = Trollop::options do
+        banner <<-EOS
+build and manage artifacts
+
+Usage:
+
+heirloom [options] build
+heirloom list
+EOS
+        opt :help, "Display Help"
+        opt :class, "Class of artifact.  This should match the SCM repo", :type => :string
+        opt :dir, "Directory which contains git repo", :type => :string
+        opt :prefix, "Bucket prefix", :type => :string
+        opt :sha, "Git Sha", :type => :string
+      end
+
+      cmd = ARGV.shift
+
+      case cmd
+      when 'build'
+        raise 'Missing required args' unless @opts[:sha] && @opts[:class]
+        h = Heirloom.new :heirloom_type => @opts[:class],
+                         :source_dir => @opts[:dir] ||= ".",
+                         :prefix => @opts[:prefix]
+
+        h.build_and_upload_to_s3(:sha => @opts[:sha])
+      when 'list'
+        Heirloom.list.each do |a|
+          puts a.to_yaml
+        end
+      when 'info'
+        raise 'Missing required args' unless @opts[:sha] && @opts[:class]
+        puts Heirloom.info(:class => @opts[:class], :sha => @opts[:sha]).to_yaml
+      when 'delete'
+        raise 'Missing required args' unless @opts[:sha] && @opts[:class]
+        puts Heirloom.delete(:class => @opts[:class], :sha => @opts[:sha]).to_yaml
+        puts "!!Make sure to manualy delete any artifacts from S3!!!"
+      else
+        puts "Unkown command: '#{cmd}'."
+      end
+    end
+  end
+end

data/lib/heirloom/version.rb

@@ -0,0 +1,3 @@
+module Heirloom
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: heirloom
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Brett Weaver
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70129331233340 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70129331233340
+- !ruby/object:Gem::Dependency
+  name: fog
+  requirement: &70129331232880 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70129331232880
+- !ruby/object:Gem::Dependency
+  name: grit
+  requirement: &70129331228060 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70129331228060
+- !ruby/object:Gem::Dependency
+  name: logger
+  requirement: &70129331227600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70129331227600
+- !ruby/object:Gem::Dependency
+  name: minitar
+  requirement: &70129331227180 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70129331227180
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: &70129331226740 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70129331226740
+description: I help build and manage artifacts
+email:
+- brett@weav.net
+executables:
+- heirloom
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rvmrc
+- Gemfile
+- Rakefile
+- bin/heirloom
+- heirloom.gemspec
+- lib/heirloom.rb
+- lib/heirloom/aws.rb
+- lib/heirloom/cli.rb
+- lib/heirloom/version.rb
+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: 2848188182829183969
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 2848188182829183969
+requirements: []
+rubyforge_project: heirloom
+rubygems_version: 1.8.16
+signing_key: 
+specification_version: 3
+summary: I help with artifacts
+test_files: []