blanket 0.0.3 → 0.1.0

data/bin/blanket-cfg

@@ -36,6 +36,7 @@ $stderr <<  BAD_ARGS unless OPTIONS[:task] && OPTIONS[:source] && OPTIONS[:sink]
 
 task = OPTIONS[:task]
 source_class = OPTIONS[:source]
+sink_class = OPTIONS[:sink]
 @task_dir = FileUtils.pwd + "/" +task
 puts "Creating #{@task_dir}"
 FileUtils.mkdir_p @task_dir
@@ -51,6 +52,14 @@ end
 write_config_for(:source)
 write_config_for(:sink)
 
+requirements = ["'capistrano/recipes/blanket'"]
+[:source, :sink].each do |phase|
+  phase_class = Object.const_get(OPTIONS[phase])
+  (requirements << phase_class.additional_requirements) if phase_class.respond_to? :additional_requirements
+end
+
+capfile_requires = requirements.inject("") {|statement, r| statement += "require #{r}\r\n" }
+
 
 def unindent(string)
   indentation = string[/\A\s*/]
@@ -58,9 +67,7 @@ def unindent(string)
 end
 
 files = {
-  "Capfile" => unindent(<<-FILE),
-    require 'capistrano/recipes/blanket'
-  FILE
+  "Capfile" => unindent(capfile_requires)
 }
 
 files.each do |file, content|

data/lib/blanket/init.rb

@@ -22,3 +22,5 @@ require File.dirname(__FILE__) + "/plugins/sources/single_file.rb"
 require File.dirname(__FILE__) + "/plugins/sources/remote_directory.rb"
 require File.dirname(__FILE__) + "/plugins/sources/postgresql.rb"
 require File.dirname(__FILE__) + "/plugins/sinks/s3.rb"
+require File.dirname(__FILE__) + "/plugins/sinks/scp.rb"
+require File.dirname(__FILE__) + "/plugins/sinks/sftp.rb"

data/lib/blanket/plugins/sinks/s3.rb

@@ -9,6 +9,10 @@ class S3 < Sink
     @reader = reader
   end
   
+  def self.additional_requirements
+    "'capistrano/recipes/s3'"
+  end
+  
   def self.attribute_symbols
     [:access_key_id, :secret_access_key, :sink_type, :bucket]
   end

data/lib/blanket/plugins/sinks/scp.rb

@@ -0,0 +1,39 @@
+class Scp < Sink
+  
+  def initialize(reader)
+    @reader = reader
+  end
+  
+  def self.attribute_symbols
+    [:sink_type, :host, :user, :password, :remote_path, :local_path ]
+  end
+  
+  def self.default_sink_type
+    "Scp"
+  end
+  
+  def self.default_host
+    "yourhost.com"
+  end
+  
+  def self.default_user
+    "username"
+  end
+  
+  def self.default_password
+    "password"
+  end
+  
+  def self.default_remote_path
+    "/path/to/remote/backup/directory"
+  end
+  
+  def self.default_local_path
+    "/path/to/local/blanket"
+  end
+  
+  def cleanup_command
+    noop
+  end
+  
+end

data/lib/blanket/plugins/sinks/sftp.rb

@@ -0,0 +1,43 @@
+class Sftp < Sink
+  
+  def initialize(reader)
+    @reader = reader
+  end
+  
+  def self.additional_requirements
+    "'capistrano/recipes/sftp'"
+  end
+  
+  def self.attribute_symbols
+    [:sink_type, :host, :user, :password, :remote_path, :local_path ]
+  end
+  
+  def self.default_sink_type
+    "Sftp"
+  end
+  
+  def self.default_host
+    "yourhost.com"
+  end
+  
+  def self.default_user
+    "username"
+  end
+  
+  def self.default_password
+    "password"
+  end
+  
+  def self.default_remote_path
+    "/path/to/remote/backup/directory"
+  end
+  
+  def self.default_local_path
+    "/path/to/local/blanket"
+  end
+  
+  def cleanup_command
+    noop
+  end
+  
+end

data/lib/capistrano/recipes/blanket.rb

@@ -42,14 +42,14 @@ Capistrano::Configuration.instance(:must_exist).load do
       end
       set :sink, Object.const_get(sink_type).new(reader)
     end
-  
-    desc "Send file to sink"
-    task :export do
-      upload_filename = Pathname.new(variables[:source].local_backup_path).basename.to_s
-      variables[:sink].drain(bucket, upload_filename)
+    
+    desc "Export file using put"
+    task :export, :hosts => lambda{ host } do
+      put variables[:source].remote_backup_path, variables[:sink].remote_path
     end
   
   end
   before "sink:export", "source:download"
   before "sink:export", "sink:load_config"
+
 end

data/lib/capistrano/recipes/s3.rb

@@ -0,0 +1,14 @@
+gem 'blanket'
+require 'blanket/init'
+Capistrano::Configuration.instance(:must_exist).load do
+
+  namespace :sink do
+      
+    desc "Upload file to S3"
+    task :export do
+      upload_filename = Pathname.new(variables[:source].local_backup_path).basename.to_s
+      variables[:sink].drain(bucket, upload_filename)
+    end
+  
+  end
+end

data/lib/capistrano/recipes/sftp.rb

@@ -0,0 +1,18 @@
+gem 'blanket'
+require 'blanket/init'
+Capistrano::Configuration.instance(:must_exist).load do
+
+  namespace :sink do
+      
+    desc "Upload file to S3"
+    task :export do
+      puts "Uploading #{variables[:source].local_backup_path} to #{variables[:sink].remote_path}"
+      Net::SFTP.start(host, user, password) do |sftp|
+        sftp.put_file variables[:source].local_backup_path, variables[:sink].remote_path
+      end
+    end
+  
+  end
+end
+
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: blanket
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors: 
 - Jim Van Fleet
@@ -57,6 +57,8 @@ files:
 - lib/blanket/utils.rb
 - lib/capistrano/recipes
 - lib/capistrano/recipes/blanket.rb
+- lib/capistrano/recipes/s3.rb
+- lib/capistrano/recipes/sftp.rb
 - lib/blanket/plugins/sources
 - lib/blanket/plugins/sources/confluence.rb
 - lib/blanket/plugins/sources/mysql.rb
@@ -65,6 +67,8 @@ files:
 - lib/blanket/plugins/sources/remote_directory.rb
 - lib/blanket/plugins/sources/postgresql.rb
 - lib/blanket/plugins/sinks/s3.rb
+- lib/blanket/plugins/sinks/scp.rb
+- lib/blanket/plugins/sinks/sftp.rb
 - spec/fixtures
 - spec/confluence_spec.rb
 - spec/dependency_spec.rb