heroku_pg_loader 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a6cf49270202e1edc2a5ca3f0858eef8cf6ee294
+  data.tar.gz: ab6a2255562156f612d9e2795c0e853b512fc178
+SHA512:
+  metadata.gz: 56f6d0cef29e38ef5f9850bf25a32edcfef3295f834b38c2a437e069a2e571980c9a5c9d08314c8dca877053b276b10a625e42c3ef21ed194a5171892f2900d7
+  data.tar.gz: 494c4f30fcd70433f76d332350ab43bf1ec6c7625eb0b8a3a744926950778f0b677f5cd8e8c1361468a0bc4bcbf0d9009f6fd2064ba8e46ac17bc279c311fc88

data/bin/heroku_pg_load

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'heroku_pg_loader'
+
+HerokuPgLoader.new.start

data/lib/heroku_pg_loader.rb

@@ -0,0 +1,118 @@
+require 'yaml'
+require 'uri'
+
+class HerokuPgLoader
+
+	def start
+		begin
+			get_app_name
+			get_development_database
+
+			load_database
+
+
+		rescue Exception => e
+			puts "Error: #{e}"
+		ensure
+			dump_file.close
+			dump_file.delete
+		end
+	end
+
+	private
+
+	def get_url_contents(url_string)
+		require 'open-uri'
+		open(url_string).read
+	end
+
+	def load_database
+		run_command "heroku pgbackups:capture --app #{@heroku_app_name} --expire"
+
+		heroku_backup_url = run_command "heroku pgbackups:url --app #{@heroku_app_name}"
+
+		puts "Dumping to: #{dump_file.path}"
+
+		dump_file.write get_url_contents(heroku_backup_url)
+
+		run_command "echo \"DROP DATABASE #{@db_name}; CREATE DATABASE #{@db_name};\" | psql -U #{@db_username} -d postgres"
+
+		run_command "pg_restore --verbose --clean --no-acl --no-owner -h 127.0.0.1 -U #{@db_username} -d #{@db_name} #{dump_file.path}"
+	end
+
+	def dump_file
+		if @dump_file.nil?
+			path = "#{working_dir}/latest.dump"
+			puts "Creating dump file #{path}"
+			@dump_file = File.open(path, 'w')
+		end
+		@dump_file
+	end
+
+	def run_command(command)
+		puts "[Running] #{command}"
+		output = `#{command}`; result = $?.success?
+		raise "Error" unless result
+
+		output
+	end
+
+	def get_development_database
+
+		# Read database.yml
+		db_config = YAML.load_file("#{working_dir}/config/database.yml")
+
+		dev_config = db_config["development"]
+		if dev_config["adapter"] == "postgresql"
+			@db_name = dev_config["database"]
+			@db_username = dev_config["username"]
+			@db_password = dev_config["password"]
+		end
+
+		@db_name = prompt "Local Database Name:\n (WARNING: THIS DATABASE WILL BE ERASED)",
+		@db_name,
+		"Database name not entered"
+
+		@db_username = prompt "Database Username:",
+		@db_username,
+		"Database username not entered"
+
+		@db_password = prompt "Database Password:",
+		@db_password,
+		"Database Password not entered"
+	end
+
+	def working_dir
+		File.expand_path File.dirname(__FILE__)
+	end
+
+	def prompt(message, current_value, error_message)
+		puts "#{message} [#{current_value}]"
+		print ">"
+		input = gets.chomp
+		current_value = input if input != ""
+		raise error_message if current_value.nil?
+
+		current_value
+	end
+
+	def get_app_name
+		detect_app_name
+
+		@heroku_app_name = prompt "Heroku App Name:",
+				@heroku_app_name,
+				"No Heroku App Name entered"
+	end
+
+	def detect_app_name
+		git_remote = `git remote -v | grep heroku`
+
+		/git@heroku.com:(\w*).git/.match(git_remote)
+		@heroku_app_name = $1
+
+		if @heroku_app_name
+			puts "Heroku App Detected: #{@heroku_app_name}"
+		end
+	end
+
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: heroku_pg_loader
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Gerry Eng
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-26 00:00:00.000000000 Z
+dependencies: []
+description: A simple gem toload your Heroku Postgres production database into a local
+  database
+email: m@gerryeng.com
+executables:
+- heroku_pg_load
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/heroku_pg_loader.rb
+- bin/heroku_pg_load
+homepage: http://gerryeng.com
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: A simple gem to load your Heroku Postgres production database into a local
+  database
+test_files: []