heroku-mongo-backup 0.3.2 → 0.3.6

data/lib/heroku_mongo_backup.rb

@@ -7,6 +7,7 @@ require 'zlib'
 require 'uri'
 require 'yaml'
 require 'rubygems'
+require 'net/ftp'
 
 module HerokuMongoBackup
   require 'heroku_mongo_backup/railtie' if defined?(Rails)
@@ -79,7 +80,25 @@ module HerokuMongoBackup
       @db = connection.db(uri.path.gsub(/^\//, ''))
       @db.authenticate(uri.user, uri.password) if uri.user
     end
-
+    
+    def ftp_connect
+      @ftp = Net::FTP.new(ENV['FTP_HOST'])
+      @ftp.passive = true
+      @ftp.login(ENV['FTP_USERNAME'], ENV['FTP_PASSWORD'])
+    end
+    
+    def ftp_upload
+      @ftp.putbinaryfile(@file_name)
+    end
+    
+    def ftp_download
+      open(@file_name, 'w') do |file|
+        file_content = @ftp.getbinaryfile(@file_name)
+        file.binmode
+        file.write file_content
+      end
+    end
+    
     def s3_connect
       bucket            = ENV['S3_BACKUPS_BUCKET']
       if bucket.nil?
@@ -136,20 +155,37 @@ module HerokuMongoBackup
       puts "Using databased: #{@url}"
   
       self.db_connect
-      self.s3_connect
+
+      if ENV['UPLOAD_TYPE'] == 'ftp'
+        self.ftp_connect
+      else
+        self.s3_connect
+      end
     end
 
     def backup
       self.chdir    
       self.store
-      self.s3_upload
-    end
 
+      if ENV['UPLOAD_TYPE'] == 'ftp'
+        self.ftp_upload
+        @ftp.close
+      else
+        self.s3_upload
+      end
+    end
+    
     def restore file_name
       @file_name = file_name
   
       self.chdir
-      self.s3_download
+      
+      if ENV['UPLOAD_TYPE'] == 'ftp'
+        self.ftp_download
+        @ftp.close
+      else
+        self.s3_download
+      end
       self.load
     end
   end

metadata

@@ -1,19 +1,20 @@
 --- !ruby/object:Gem::Specification
 name: heroku-mongo-backup
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.6
   prerelease: 
 platform: ruby
 authors:
-- Alex Kraves
+- Alex Kravets
+- matyi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-02-27 00:00:00.000000000Z
+date: 2012-05-01 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &70311073645040 !ruby/object:Gem::Requirement
+  requirement: &70361846274660 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +22,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70311073645040
+  version_requirements: *70361846274660
 - !ruby/object:Gem::Dependency
   name: mocha
-  requirement: &70311073644180 !ruby/object:Gem::Requirement
+  requirement: &70361846273820 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +33,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70311073644180
+  version_requirements: *70361846273820
 - !ruby/object:Gem::Dependency
   name: crack
-  requirement: &70311073642720 !ruby/object:Gem::Requirement
+  requirement: &70361846272260 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,9 +44,9 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70311073642720
-description: Rake task for backing up mongo database on heroku and push it to S3.
-  Library can be used as rake task or be easily integrated into daily cron job.
+  version_requirements: *70361846272260
+description: Rake task for backing up mongo database on heroku and push it to S3 or
+  FTP. Library can be used as rake task or be easily integrated into daily cron job.
 email: santyor@gmail.com
 executables: []
 extensions: []
@@ -79,5 +80,5 @@ rubygems_version: 1.8.6
 signing_key: 
 specification_version: 3
 summary: Rake task backups mongo database on Heroku and push gzipped file to Amazon
-  S3.
+  S3 or FTP.
 test_files: []