heroku_cloud_backup 0.0.2

data/.gitignore

@@ -0,0 +1,6 @@
+.DS_Store
+.rvmrc
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Jack Chu
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= heroku_cloud_backup
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Jack Chu. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/heroku_cloud_backup.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "heroku_cloud_backup/version"
+
+Gem::Specification.new do |s|
+  s.name        = "heroku_cloud_backup"
+  s.version     = HerokuCloudBackup::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Jack Chu"]
+  s.email       = ["jack@jackchu.com"]
+  s.homepage    = ""
+  s.summary     = %q{Backup pg dumps to the cloud}
+  s.description = %q{PG backups into the cloud with fog}
+
+  s.rubyforge_project = "heroku_cloud_backup"
+
+  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"]
+  s.add_runtime_dependency('fog', '>= 0.7.2')
+  s.add_runtime_dependency('heroku', '>= 2.1.4')
+end

data/lib/heroku_cloud_backup/errors.rb

@@ -0,0 +1,10 @@
+module HerokuCloudBackup
+  module Errors
+    class Error < StandardError; end
+    class NotFound < HerokuCloudBackup::Errors::Error; end
+    class InvalidProvider < HerokuCloudBackup::Errors::Error; end
+    class ConnectionError < HerokuCloudBackup::Errors::Error; end
+    class UploadError < HerokuCloudBackup::Errors::Error; end
+    class NoBackups < HerokuCloudBackup::Errors::Error; end
+  end
+end

data/lib/heroku_cloud_backup/railtie.rb

@@ -0,0 +1,12 @@
+require 'heroku_cloud_backup'
+
+module HerokuCloudBackup
+  if defined? Rails::Railtie
+    require 'rails'
+    class Railtie < Rails::Railtie
+      rake_tasks do
+        require "heroku_cloud_backup/tasks"
+      end
+    end
+  end
+end

data/lib/heroku_cloud_backup/tasks.rb

@@ -0,0 +1,8 @@
+require 'heroku_cloud_backup'
+
+namespace :heroku do
+  desc "Example showing PostgreSQL database backups from Heroku to Amazon S3"
+  task :backup => :environment do
+    HerokuCloudBackup.execute
+  end
+end

data/lib/heroku_cloud_backup/version.rb

@@ -0,0 +1,3 @@
+module HerokuCloudBackup
+  VERSION = "0.0.2"
+end

data/lib/heroku_cloud_backup.rb

@@ -0,0 +1,108 @@
+require 'fog'
+require 'open-uri'
+require "heroku"
+require "pgbackups/client"
+require 'heroku_cloud_backup/errors'
+require 'heroku_cloud_backup/railtie'
+require 'heroku_cloud_backup/version'
+
+module HerokuCloudBackup
+  class << self
+
+    def log(message)
+      puts "[#{Time.now}] #{message}"
+    end
+    
+    def backups_url
+      ENV["PGBACKUPS_URL"]
+    end
+    
+    def client
+      @client ||= PGBackups::Client.new(ENV["PGBACKUPS_URL"])
+    end
+    
+    def databases
+      if db = ENV["HEROKU_BACKUP_DATABASES"]
+        db.split(",").map(&:strip)
+      else
+        ["DATABASE_URL"]
+      end
+    end
+    
+    def backup_name(to_url)
+      # translate s3://bucket/email/foo/bar.dump => foo/bar
+      parts = to_url.split('/')
+      parts.slice(4..-1).join('/').gsub(/\.dump$/, '')
+    end
+
+    def execute
+      log "heroku:backup started"
+
+      @bucket_name = ENV['HCB_BUCKET'] || "#{ENV['APP_NAME']}-heroku-backups"
+      @backup_path = ENV['HCB_PREFIX'] || "db"
+      @providers = ENV['HCB_PROVIDERS'] || raise(HerokuCloudBackup::Errors::NotFound.new("Please provide a 'HCB_PROVIDERS' config variable."))
+      b = client.get_latest_backup
+      raise HerokuCloudBackup::Errors::NoBackups.new("You don't have any pgbackups. Please run heroku pgbackups:capture first.") if b.empty?
+
+      @providers.split(',').each do |provider|
+        case provider
+          when 'aws'
+            @hcb_aws_access_key_id = ENV['HCB_AWS_ACCESS_KEY_ID'] || raise(HerokuCloudBackup::Errors::NotFound.new("Please provide a 'HCB_AWS_ACCESS_KEY_ID' config variable."))
+            @hcb_aws_secret_access_key = ENV['HCB_AWS_SECRET_ACCESS_KEY'] || raise(HerokuCloudBackup::Errors::NotFound.new("Please provide a 'HCB_AWS_SECRET_ACCESS_KEY' config variable."))
+            begin
+              @connection = Fog::Storage.new(
+                :provider => 'AWS',
+                :aws_access_key_id => @hcb_aws_access_key_id,
+                :aws_secret_access_key => @hcb_aws_secret_access_key
+              )
+            rescue
+              raise HerokuCloudBackup::Errors::ConnectionError.new("There was an error connecting to your provider.")
+            end
+        else
+          raise HerokuCloudBackup::Errors::InvalidProvider.new("One or more of your providers were invalid. Valid values are 'aws', 'rackspace', or 'google'")
+        end
+
+        directory = @connection.directories.get(@bucket_name)
+
+        if !directory
+          directory = @connection.directories.create(:key => @bucket_name)
+        end
+
+        public_url = b["public_url"]
+        created_at = DateTime.parse b["created_at"]
+        db_name = b["from_name"]
+        name = "#{ENV['APP_NAME']}-#{created_at.strftime('%Y-%m-%d-%H%M%S')}.dump"
+        begin
+          directory.files.create(:key => "#{@backup_path}/#{b["from_name"]}/#{name}", :body => open(public_url))
+        rescue Exception => e
+          raise HerokuCloudBackup::Errors::UploadError.new(e.message)
+        end
+      end
+
+      prune
+
+      log "heroku:backup complete"
+    end
+
+    private
+
+    def prune
+      number_of_files = ENV['HCB_MAX_BACKUPS']
+      if number_of_files && number_of_files.to_i > 0
+        directory = @connection.directories.get(@bucket_name)
+        files = directory.files.all(:prefix => @backup_path)
+        file_count = 0
+        files.reverse.each do |file|
+          if file.key =~ Regexp.new("/#{@backup_path}\/#{ENV['APP_NAME']}-\d{4}-\d{2}-\d{2}-\d{6}\.sql\.gz$/i")
+            file_count += 1
+          else
+            next
+          end
+          if file_count > number_of_files
+            file.destroy
+          end
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,86 @@
+--- !ruby/object:Gem::Specification 
+name: heroku_cloud_backup
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.2
+platform: ruby
+authors: 
+- Jack Chu
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-23 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: fog
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.7.2
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: heroku
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 2.1.4
+  type: :runtime
+  version_requirements: *id002
+description: PG backups into the cloud with fog
+email: 
+- jack@jackchu.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- README.rdoc
+- Rakefile
+- heroku_cloud_backup.gemspec
+- lib/heroku_cloud_backup.rb
+- lib/heroku_cloud_backup/errors.rb
+- lib/heroku_cloud_backup/railtie.rb
+- lib/heroku_cloud_backup/tasks.rb
+- lib/heroku_cloud_backup/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"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: heroku_cloud_backup
+rubygems_version: 1.8.3
+signing_key: 
+specification_version: 3
+summary: Backup pg dumps to the cloud
+test_files: []
+