thor-tropo 0.0.3 → 1.0.0

data/lib/thor-tropo.rb

@@ -1,19 +1,23 @@
+module ThorTropo
+
+
+$:.unshift  File.expand_path('../thor-tropo', __FILE__)
+
 require 'thor'
 require 'thor/actions'
 require 'thor/scmversion'
 require 'tmpdir'
 require 'archive/tar/minitar'
 require 'zlib'
-#require 'thor/foodcritic'
 require 'berkshelf/thor'
 require 'berkshelf/chef'
+require 'uploader'
+require 'configuration'
 
-
-module ThorTropo
   class Tasks < Thor
     include Thor::Actions
 
-
+    @packaged_cookbook = nil
     class_option :called_path,
       :type => :string,
       :default => Dir.pwd
@@ -23,23 +27,51 @@ module ThorTropo
       :aliases => "-v",
       :default => false
 
-
     namespace "tropo"
 
-    desc "bump", "Bump version"
-    def bump
-      puts "This should bump the version in the metatdata.rb file... eventually"
-    end
-
+    method_option :force,
+      :type    => :boolean,
+      :aliases => "-f",
+      :default => false,
+      :desc    => "Force override of any files on s3 without confirmation",
+      :banner  => "Force overrides of existing files"
+
+    method_option :noop,
+      :type    => :boolean,
+      :aliases => "-n",
+      :default => false,
+      :desc    => "NO-OP mode, dont actually upload anything"
+
+    method_option :ignore,
+      :type     => :boolean,
+      :aliases  => "-i",
+      :default  => false,
+      :desc     => "Ignore any dirty files in directory and package anyways",
+      :banner   => "Ignore dirty repository"
+
+    method_option :clean_mode,
+      :type     => :boolean,
+      :aliases  => "-c",
+      :default  => false,
+      :desc     => "Delete lockfile before running `Berks install`",
+      :banner   => "Delete lockfile before running `Berks install`"
+
+    desc "package", "Package cookbooks and upload to S3"
 
-    desc "package", "Package cookbook"
     def package
-      unless clean?
-        say "There are files that need to be committed first.", :red
-        exit 1
+
+      $config = ThorTropo::Configuration.new(options[:called_path])
+
+      unless options[:ignore]
+        unless clean?
+          say "There are files that need to be committed first.", :red
+          exit 1
+        end
       end
 
       bundle_cookbook
+
+      upload_cookbook @packaged_cookbook, $config.project_name, {:force => options[:force],:noop => options[:noop]}
       #  tag_version {
       #    publish_cookbook(options)
       #  }
@@ -59,25 +91,51 @@ module ThorTropo
 
     no_tasks do
 
+      def upload_cookbook(local_file,path,opts={})
+        uploader = ThorTropo::Uploader.new({
+          :access_key => $config.aws_key,
+          :secret_key => $config.aws_secret,
+          :bucket     => $config.bucket_name
+        })
+
+        uploader.upload :local_file => local_file, :path => path, :force => options[:force], :noop=>options[:noop]
+      end
+
       def clean?
         sh_with_excode("git diff --exit-codeBerkshelf")[1] == 0
       end
 
+      def clean_lockfile
+        berksfile = "#{options[:called_path]}/Berksfile.lock"
+
+        if File.exists? berksfile
+          say "[ TROPO ] - Removing Berksfile.lock before running Berkshelf", :green
+          remove_file berksfile
+        else
+          say "[ TROPO ] - Unable to find berksfile to delete - [ #{berksfile} ]", :yellow
+        end
+      end
+
       def bundle_cookbook
-        tmp = Dir.mktmpdir
+        clean_lockfile if options[:clean_mode]
+
+        say "[ TROPO ] - Packaging cookbooks from Berksfile", :blue
+        @tmp_dir = Dir.mktmpdir
         opts = {
           berksfile: File.join(options[:called_path],"/Berksfile"),
-          path: "#{tmp}/cookbooks"
+          path: "#{@tmp_dir}/cookbooks"
         }
 
         invoke("berkshelf:install", [], opts)
-        output   = File.expand_path(File.join(Dir.home, "#{options[:called_path].split("/")[-1]}-#{current_version}.tar.gz"))
+        output   = File.expand_path(File.join(@tmp_dir, "#{options[:called_path].split("/")[-1]}-#{current_version}.tar.gz"))
 
 
-        Dir.chdir(tmp) do |dir|
+        Dir.chdir(@tmp_dir) do |dir|
           tgz = Zlib::GzipWriter.new(File.open(output, 'wb'))
           Archive::Tar::Minitar.pack('./cookbooks', tgz)
         end
+        @packaged_cookbook = output
+
       end
 
       def current_version

data/lib/thor-tropo/configuration.rb

@@ -0,0 +1,49 @@
+module ThorTropo
+  class Configuration < Thor
+
+  no_commands {
+
+    require 'pathname'
+    require 'yaml'
+
+    attr_reader :bucket_name, :aws_secret, :aws_key, :project_name
+
+    def initialize(path)
+      file_name=".deployer"
+
+      Pathname.new(path).ascend do |dir|
+        config_file = dir + file_name
+        if dir.children.include?(config_file)
+          merge_config(YAML::load_file(config_file))
+        end
+      end
+
+    end
+
+    private
+
+    def merge_config(config)
+
+      if config['bucket_name']
+        @bucket_name = config['bucket_name'] unless @bucket_name
+      end
+
+      if config['aws_secret']
+        @aws_secret = config['aws_secret'] unless @aws_secret
+      end
+
+      if config['aws_key']
+        @aws_key = config['aws_key'] unless @aws_key
+      end
+
+      if config['project_name']
+        @project_name = config['project_name'] unless @project_name
+      end
+
+    end
+  }
+
+  end
+end
+
+

data/lib/thor-tropo/uploader.rb

@@ -0,0 +1,115 @@
+module ThorTropo
+  class Uploader < Thor
+
+    include Thor::Actions
+
+    require 'digest/md5'
+    require 'fog'
+    require 'mime/types'
+
+    require 'pry'
+
+    no_commands {
+
+      def initialize(opts={})
+        @debug_mode   = opts[:debug_mode]
+        @connection   = setup_connection(opts[:access_key],opts[:secret_key])
+        @bucket_name  = get_bucket( opts[:bucket] || 'artifacts.voxeolabs.net' )
+      end
+
+      def upload(opts={})
+
+        local_file  = opts[:local_file]
+        remote_file = opts[:path] + "/" + opts[:local_file].split("/")[-1]
+        force       = opts[:force].nil? ? false : opts[:force]
+        noop        = opts[:noop].nil? ? false : opts[:noop]
+
+        if file_exists? remote_file
+
+          if force
+
+            send_file :remote_file => remote_file, :local_file => local_file, :noop => noop
+
+          else
+
+            response = ask "[ TROPO ] - file already exists on S3, do you want me to overwrite it ? (yes|no) ? ", :yellow
+
+            if response.downcase.match /yes|ok|y/
+              begin
+
+                send_file :remote_file => remote_file, :local_file => local_file, :noop => noop
+
+              rescue Exception => e
+                say "[ TROPO ] Error: #{e}", :red
+                say "[ TROPO ] Backtrace:\n\t#{e.backtrace.join("\n\t")}" if @debug_mode
+              end
+
+            else
+              say "[ TROPO ] - Quiting..", :red
+            end
+          end
+        else
+
+          send_file :remote_file => remote_file, :local_file => local_file, :noop => noop
+
+        end
+      end
+
+      private
+
+      def setup_connection(u,p)
+        Fog::Storage.new({
+            :provider                 => 'AWS',
+            :aws_access_key_id        => u,
+            :aws_secret_access_key    => p
+        })
+      end
+
+      def get_bucket(bucket_name)
+        directory = @connection.directories.create(
+          :key    => bucket_name,
+          :public => true
+        )
+      end
+
+      def file_exists?(key)
+        !@bucket_name.files.head(key).nil?
+      end
+
+      def send_file(opts={})
+        local_file  = opts[:local_file]
+        remote_file = opts[:remote_file]
+
+        say "[ TROPO ] - Uploading file.", :blue
+        unless opts[:noop]
+          file = @bucket_name.files.create(
+            :key    => remote_file,
+            :body   => File.open(local_file),
+            :public => true,
+            :content_type => get_mime_type(local_file),
+            :metadata => {
+              "x-amz-meta-sha256-hash" =>get_sha256_sum(local_file)
+            }
+          )
+          public_url = file.public_url
+
+        else
+          public_url = " *** NOOP ** /#{remote_file}"
+
+        end
+          say "[ TROPO ] - Public URL: #{public_url}", :blue
+
+      end
+
+      def get_sha256_sum(file)
+        Digest::SHA256.file(file).hexdigest
+      end
+
+      def get_mime_type(file)
+        MIME::Types.type_for(file)[0].to_s
+      end
+
+    }
+  end
+
+end

data/thor-tropo.gemspec

@@ -15,12 +15,14 @@ Gem::Specification.new do |gem|
   gem.require_paths = ["lib","bin"]
   gem.version       = ThorTropo::VERSION
 
-  gem.add_dependency 'json', ">= 1.7.0"
+  #gem.add_dependency 'json', ">= 1.7.0"
   gem.add_dependency 'thor'
   gem.add_dependency 'chef', "~> 11.0"
   gem.add_dependency 'berkshelf'
   gem.add_dependency 'thor-scmversion'
   gem.add_dependency 'minitar',           '~> 0.5.4'
+  gem.add_dependency 'aws-sdk'
+  gem.add_dependency 'fog'
 
   gem.add_development_dependency 'foodcritic'
   gem.add_development_dependency 'webmock'
@@ -31,4 +33,5 @@ Gem::Specification.new do |gem|
   gem.add_development_dependency 'rspec'
   gem.add_development_dependency "bundler", "~> 1.3"
   gem.add_development_dependency "rake"
+  gem.add_development_dependency "pry"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: thor-tropo
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 1.0.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,24 +9,8 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-13 00:00:00.000000000 Z
+date: 2013-07-15 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: json
-  requirement: !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: 1.7.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: 1.7.0
 - !ruby/object:Gem::Dependency
   name: thor
   requirement: !ruby/object:Gem::Requirement
@@ -107,6 +91,38 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: 0.5.4
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fog
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: foodcritic
   requirement: !ruby/object:Gem::Requirement
@@ -251,6 +267,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Thor tasks to package a project
 email:
 - johntdyer@gmail.com
@@ -268,6 +300,8 @@ files:
 - VERSION
 - bin/scmversion
 - lib/thor-tropo.rb
+- lib/thor-tropo/configuration.rb
+- lib/thor-tropo/uploader.rb
 - lib/thor-tropo/version.rb
 - lib/thor/tropo.rb
 - thor-tropo.gemspec