baptize 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 11dedf7433eb9c33188a68b80a8d1b79b1a9bf1f
-  data.tar.gz: 5af048e9e82eeafa942d46194b2cc77c72ccf089
+  metadata.gz: 631c1cbd594a828aed8fbc7e6bfdc187460c2be0
+  data.tar.gz: c1e6b3abadd41e0aadcb82c23daa96f9bfd02096
 SHA512:
-  metadata.gz: 2da09b4fb58133cfa2756fbd78aaf06c67bb99f4aa6fa426f38f938dc8ec5c1eb6e8ac0bd133ef41bccc518195aaf80bd3a2f2de43544248f8fec0a62fca5998
-  data.tar.gz: 1c4a577daa8bf401a315587cc610e08957dffad5903ed520596879e7063d7adf32a9d9e1a323bea4985f107085b35e0b6798b1876cea3e098fbcc453b28c99ca
+  metadata.gz: 44943b561ce0f265ae8f7a231b88f56320de66f4ad9714790fdbe126072625264ba708a3f6135aa070a6f501556895e0d798872f6498d8de149b886cd22d434a
+  data.tar.gz: 269aac58005c16f1d619357058458c690c9c5ae0b70c6597717b7e1403cc196afb91848767ae3b55fee9ea20a204d6fd3255171913ab31b54deadf78d35a55fb

data/lib/baptize/plugins/execution.rb

@@ -35,6 +35,32 @@ module Baptize
         call_current_ssh_connection :test, command
       end
 
+      def upload(local, remote, options = {})
+        if fetch(:use_sudo)
+          # perform the transfer in two steps if we're using sudo
+          final = remote
+          remote = "/tmp/baptize_#{File.basename(remote)}"
+        end
+        ssh = fetch(:current_ssh_connection)
+        old_verbosity = nil
+        if fetch(:ssh_verbose)
+          old_verbosity = SSHKit.config.output_verbosity
+          SSHKit.config.output_verbosity = Logger::DEBUG
+        end
+        begin
+          ssh.upload! local, remote, options
+        ensure
+          SSHKit.config.output_verbosity = old_verbosity if old_verbosity
+        end
+        if fetch(:use_sudo)
+          remote_execute "mv #{remote.shellescape} #{final.shellescape}"
+        end
+      end
+
+      def put(buffer, remote, options = {})
+        upload StringIO.new(buffer), remote, options
+      end
+
       private
       def call_current_ssh_connection(action, *args)
         ssh = fetch(:current_ssh_connection)
@@ -45,7 +71,7 @@ module Baptize
         end
         begin
           if fetch(:use_bash)
-            args = ["bash -c " + args.join(" ; ").shellescape]
+            args = ["bash -c " + args.join(" ; ").dump]
           end
           args.unshift(:sudo) if fetch(:use_sudo)
           ssh.send(action, *args)

data/lib/baptize/plugins/helpers.rb

@@ -0,0 +1,31 @@
+module Baptize
+  module Plugins
+
+    module Helpers
+
+      def asset_path(asset)
+        File.expand_path(File.join("assets", asset))
+      end
+
+      def md5_of_file(path, md5)
+        remote_assert "test $(md5sum #{path.shellescape} | cut -f1 -d' ') = #{md5.shellescape}"
+      end
+
+      def escape_sed_arg(s)
+        s.gsub("'", "'\\\\''").gsub("\n", '\n').gsub("/", "\\\\/").gsub('&', '\\\&')
+      end
+
+      def replace_text(pattern, replacement, path)
+        remote_execute "sed -i 's/#{escape_sed_arg(pattern)}/#{escape_sed_arg(replacement)}/g' #{path.shellescape}"
+      end
+
+      def render(path, locals = {})
+        require 'erb'
+        require 'ostruct'
+        ERB.new(File.read(path)).result(locals.kind_of?(Binding) ? locals : OpenStruct.new(locals).instance_eval { binding })
+      end
+
+    end
+
+  end
+end

data/lib/baptize/plugins/verifications.rb

@@ -2,8 +2,36 @@ module Baptize
   module Plugins
 
     module Verifications
-      def fail_verification
-        raise VerificationFailure.new
+      def fail_verification(message = "Verification failed")
+        raise VerificationFailure.new(message)
+      end
+
+      def has_file(path)
+        remote_assert("test -e #{path.shellescape}") or fail_verification("Remote file #{path} does not exist")
+      end
+
+      def has_directory(path)
+        remote_assert("test -d #{path.shellescape}") or fail_verification("Remote directory #{path} does not exist")
+      end
+
+      def file_contains(path, text, options = {})
+        options = {:mode => :text}.merge(options)
+        if options[:mode] == :text
+          command = Array(text.strip.split("\n")).flatten.map {|line| "grep --fixed-strings #{line.shellescape} #{path.shellescape}" }.join(" && ")
+        elsif options[:mode] == :perl
+          command = "grep --perl-regexp #{text.shellescape} #{path.shellescape}"
+        else
+          command = "grep --basic-regexp #{text.shellescape} #{path.shellescape}"
+        end
+        remote_assert command
+      end
+
+      def has_executable(path)
+        remote_assert("which #{path.shellescape}") or fail_verification("No executable #{path} found")
+      end
+
+      def has_user(name)
+        remote_assert("id -u #{name.to_s.shellescape}") or fail_verification("No user #{name}")
       end
     end
 

data/lib/baptize/rake.rb

@@ -1,21 +1,18 @@
-Rake.application.in_namespace('baptize') do
-
-  Rake::Task.define_task(:list) do
-    Baptize::Registry.packages.values.each do |package|
-      puts [package.name, "\t", package.description].join
-    end
+Rake::Task.define_task(:list) do
+  puts "Available packages"
+  Baptize::Registry.packages.values.each do |package|
+    puts [package.name, "\t", package.description].join
   end
+end
 
-  Rake::Task.define_task(:apply) do
-    Baptize::Registry.policies.keys.each do |role|
-      on roles(role), in: :parallel do |host|
-        Baptize::Registry.apply_policy role, host, self
-      end
+Rake::Task.define_task(:apply) do
+  Baptize::Registry.policies.keys.each do |role|
+    on roles(role), in: :parallel do |host|
+      Baptize::Registry.apply_policy role, host, self
     end
   end
-
 end
 
 Rake::Task.define_task(:default) do
-  Rake::Task['baptize:list'].invoke
+  Rake::Task['list'].invoke
 end

data/lib/baptize/version.rb

@@ -1,3 +1,3 @@
 module Baptize
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end # module Baptize

data/lib/baptize.rb

@@ -8,6 +8,7 @@ require 'baptize/registry'
 require 'baptize/package_definition'
 require 'baptize/execution_scope'
 require 'baptize/verification_failure'
+require 'baptize/plugins/helpers'
 require 'baptize/plugins/verifications'
 require 'baptize/plugins/execution'
 require 'baptize/plugins/apt'
@@ -15,6 +16,7 @@ require 'baptize/dsl'
 require 'baptize/application'
 
 Baptize::Registry.plugins << Capistrano::DSL::Env
+Baptize::Registry.plugins << Baptize::Plugins::Helpers
 Baptize::Registry.plugins << Baptize::Plugins::Verifications
 Baptize::Registry.plugins << Baptize::Plugins::Execution
 Baptize::Registry.plugins << Baptize::Plugins::Apt

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: baptize
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Troels Knak-Nielsen
@@ -57,6 +57,7 @@ files:
 - lib/baptize/plugins/apt.rb
 - lib/baptize/plugins/env.rb
 - lib/baptize/plugins/execution.rb
+- lib/baptize/plugins/helpers.rb
 - lib/baptize/plugins/variables.rb
 - lib/baptize/plugins/verifications.rb
 - lib/baptize/rake.rb