snapshots 0.0.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,5 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Simon Menke
+
+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,18 @@
+= snapshots
+
+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) 2009 Simon Menke. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,54 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "snapshots"
+    gem.summary = %Q{Make snapshots of your ActiveRecord database}
+    gem.description = %Q{Dump your rails databases in a database agnostic format (ruby).}
+    gem.email = "simon@mrhenry.be"
+    gem.homepage = "http://github.com/simonmenke/snapshots"
+    gem.authors = ["Simon Menke"]
+    gem.add_development_dependency "thoughtbot-shoulda"
+    gem.add_development_dependency "yard"
+    gem.add_runtime_dependency "thor"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/*_test.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+begin
+  require 'yard'
+  YARD::Rake::YardocTask.new
+rescue LoadError
+  task :yardoc do
+    abort "YARD is not available. In order to run yardoc, you must: sudo gem install yard"
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/bin/snapshots

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+begin
+  require 'rubygems'
+  require 'snapshots'
+rescue LoadError
+  require File.join(File.dirname(__FILE__), *%w[ .. lib snapshots ])
+end
+
+Snapshots::App.start

data/lib/snapshots/app.rb

@@ -0,0 +1,92 @@
+
+module Snapshots
+  class App < Thor
+    
+    class_option :app, :default => '.', :type => :string,
+      :desc   => 'Path to the rails application',
+      :banner => 'rails_root',
+      :group  => 'Global'
+    
+    desc 'list', 'List all snapshots for a rails app.'
+    def list
+      goto_rails(self.options.app) do
+        
+        Snapshots.find.each do |snapshot|
+          snapshot =~ /snapshot_(\d+)\.rb$/
+          timestamp = Time.at($1.to_i)
+          puts "#{$1} @ #{timestamp} : #{snapshot}"
+        end
+        
+      end
+    end
+    
+    desc 'dump', 'Make a new snapshot.'
+    def dump
+      goto_rails(self.options.app) do
+        load_environment!
+        
+        begin
+          Snapshots.dump
+          puts "Snapshot was successfully created."
+        rescue => e
+          puts "An error occured while creating snapshot."
+          puts "#{e.class}: #{e.message}"
+          puts e.backtrace
+        end
+      end
+    end
+    
+    desc 'load VERSION', 'Restore a snapshot.'
+    def load(version)
+      goto_rails(self.options.app) do
+        load_environment!
+        
+        begin
+          Snapshots.load(version)
+          puts "Snapshot #{version} was successfully restored."
+        rescue => e
+          puts "An error occured while restoring snapshot #{version}."
+          puts "#{e.class}: #{e.message}"
+          puts e.backtrace
+        end
+      end
+    end
+    
+    desc 'install', 'Install rake tasks in rails app.'
+    def install
+      goto_rails(self.options.app) do
+        File.open('lib/tasks/snapshots.rake', 'w+') do |f|
+          f.write <<-EOR
+begin
+  require 'rubygems'
+  require 'snapshots/tasks'
+rescue LoadError
+  puts "Please install snapshots: [sudo] gem install snapshots"
+end
+EOR
+        end
+        puts "Rake tasks are successfully installed."
+      end
+    end
+    
+  private
+    
+    def load_environment!
+      $rails_rake_task = true
+      require(File.join(RAILS_ROOT, 'config', 'environment'))
+    end
+    
+    def goto_rails(root_path, &proc)
+      root_path = File.expand_path(root_path)
+      
+      unless File.directory?(root_path) and File.file?(File.join(root_path, 'config', 'environment.rb'))
+        $stderr.puts "This is not a rails application!"
+        exit(1)
+      end
+      
+      Object.const_set('RAILS_ROOT', root_path)
+      Dir.chdir(root_path, &proc)
+    end
+    
+  end
+end

data/lib/snapshots/database_dumper.rb

@@ -0,0 +1,74 @@
+
+module Snapshots
+  class DatabaseDumper < ActiveRecord::SchemaDumper
+    
+    def dump(stream)
+      super(stream)
+      tables_for_data(stream)
+      stream
+    end
+    
+    def header(stream)
+      stream.puts <<-EOC
+# This file was generated by the Snapshots::DatabaseDumper and
+# can be loaded with the Snapshots::DatabaseLoader.
+
+ActiveRecord::Schema.define do
+
+  create_table "schema_migrations", :id => false, :force => true do |t|
+    t.string "version", :null => false
+  end
+
+  add_index "schema_migrations", ["version"], :name => "unique_schema_migrations", :unique => true
+
+EOC
+    end
+    
+  private
+    
+    def tables_for_data(stream)
+      stream.puts "\nSnapshots::DatabaseLoader.load do\n\n"
+      @connection.tables.sort.each do |tbl|
+        next if [ignore_tables].flatten.any? do |ignored|
+          case ignored
+          when String; tbl == ignored
+          when Regexp; tbl =~ ignored
+          else
+            raise StandardError, 'ActiveRecord::SchemaDumper.ignore_tables accepts an array of String and / or Regexp values.'
+          end
+        end 
+        table_for_data(tbl, stream)
+      end
+      stream.puts "end\n"
+    end
+    
+    def table_for_data(table, stream)
+      columns = @connection.columns(table)
+      
+      column_names = columns.collect { |column| column.name }
+      
+      rows = @connection.select_rows("SELECT #{column_names.join(', ')} FROM #{table}")
+      
+      stream.puts "  for_table #{table.inspect} do |t|"
+      rows.each do |row|
+        stream.print "    t.insert(\n"
+        columns.each_with_index do |column, idx|
+          cell = column.type_cast(row[idx])
+          cell = case cell
+          when DateTime then "#{cell.to_formatted_s(:rfc822).inspect}.to_datetime"
+          when Date     then "#{cell.to_formatted_s(:rfc822).inspect}.to_date"
+          when Time     then "#{cell.to_formatted_s(:rfc822).inspect}.to_time"
+          else cell.inspect
+          end
+          stream.print "      :#{column.name} => #{cell}"
+          stream.print ',' unless idx == (columns.size - 1)
+          stream.print "\n"
+        end
+        stream.print "    )\n"
+        stream.flush
+      end
+      stream.puts "  end\n\n"
+    end
+    
+  end
+end

data/lib/snapshots/database_loader.rb

@@ -0,0 +1,61 @@
+
+module Snapshots
+  class DatabaseLoader
+    
+    def self.load(connection=ActiveRecord::Base.connection, &proc)
+      new(connection).load(&proc)
+    end
+    
+    def initialize(connection)
+      @connection = connection
+    end
+    
+    def load(&proc)
+      instance_eval(&proc)
+    end
+    
+    def for_table(name, &proc)
+      Snapshots::TableLoader.load(@connection, name, &proc)
+    end
+    
+  end
+  
+  class TableLoader
+    
+    def self.load(connection, table, &proc)
+      new(connection, table).load(&proc)
+    end
+    
+    def initialize(connection, table)
+      @table  = table
+      @connection = connection
+      @columns = @connection.columns(table)
+    end
+    
+    def load
+      @connection.execute("DELETE FROM #{@table}")
+      yield(self)
+    end
+    
+    def insert(attributes={})
+      quoted_attributes = attributes_with_quotes(attributes)
+      
+      return if quoted_attributes.empty?
+      
+      column_names = @columns.collect { |column| column.name }
+      statement = "INSERT INTO #{@table} (#{column_names.join(', ')}) VALUES(#{quoted_attributes.join(', ')})"
+      
+      @connection.insert(statement, "#{@table} Create")
+    end
+    
+    def attributes_with_quotes(attributes={})
+      quoted = []
+      @columns.each do |column|
+        value = attributes[column.name.to_sym]
+        quoted.push @connection.quote(value, column)
+      end
+      quoted
+    end
+    
+  end
+end

data/lib/snapshots/tasks.rb

@@ -0,0 +1,30 @@
+
+require 'fileutils'
+  
+desc "List all snapshot"
+task :snapshots => :environment do
+  Snapshots.find.each do |snapshot|
+    snapshot =~ /snapshot_(\d+)\.rb$/
+    timestamp = Time.at($1.to_i)
+    puts "#{snapshot} @ #{timestamp}"
+  end
+end
+
+namespace :snapshots do
+  
+  desc "Create a snapshot"
+  task :dump => :environment do
+    Snapshots.dump
+  end
+  
+  desc "Load a snapshot. VERSION specifies the snapshot"
+  task :load => :environment do
+    if ENV['VERSION'].blank?
+      puts "Please use VERSION to specify which snapshot you want to load."
+      
+    else
+      Snapshots.load
+    end
+  end
+  
+end

data/lib/snapshots.rb

@@ -0,0 +1,44 @@
+
+$:.unshift(File.expand_path(File.dirname(__FILE__))) unless $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+require 'rubygems'
+require 'thor'
+
+module Snapshots
+  autoload :DatabaseDumper, 'snapshots/database_dumper'
+  autoload :DatabaseLoader, 'snapshots/database_loader'
+  autoload :App,            'snapshots/app'
+  
+  class << self
+    
+    def find(path="db/snapshots/snapshot_*.rb")
+      Dir.glob(path)
+    end
+    
+    def load(version_or_path)
+      unless String === version_or_path
+        raise ArgumentError, "argument must be a String"
+      end
+      
+      path = if version_or_path =~ /\.rb$/
+        version_or_path
+      else
+        "db/snapshots/snapshot_#{version_or_path}.rb"
+      end
+      
+      unless File.exists?(path)
+        raise LoadError, "missing snapshot: #{path}"
+      end
+      
+      Kernel.load(path)
+    end
+    
+    def dump(snapshot_path="db/snapshots/snapshot_#{Time.now.to_i}.rb", connection=ActiveRecord::Base.connection)
+      FileUtils.mkdir_p(File.dirname(snapshot_path))
+      File.open(snapshot_path, 'w+') do |f|
+        Snapshots::DatabaseDumper.dump(connection, f)
+      end
+    end
+    
+  end
+end

data/snapshots.gemspec

@@ -0,0 +1,66 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{snapshots}
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Simon Menke"]
+  s.date = %q{2009-09-21}
+  s.default_executable = %q{snapshots}
+  s.description = %q{Dump your rails databases in a database agnostic format (ruby).}
+  s.email = %q{simon@mrhenry.be}
+  s.executables = ["snapshots"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "bin/snapshots",
+     "lib/snapshots.rb",
+     "lib/snapshots/app.rb",
+     "lib/snapshots/database_dumper.rb",
+     "lib/snapshots/database_loader.rb",
+     "lib/snapshots/tasks.rb",
+     "snapshots.gemspec",
+     "test/snapshots_test.rb",
+     "test/test_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/simonmenke/snapshots}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Make snapshots of your ActiveRecord database}
+  s.test_files = [
+    "test/snapshots_test.rb",
+     "test/test_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+      s.add_development_dependency(%q<yard>, [">= 0"])
+      s.add_runtime_dependency(%q<thor>, [">= 0"])
+    else
+      s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+      s.add_dependency(%q<yard>, [">= 0"])
+      s.add_dependency(%q<thor>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    s.add_dependency(%q<yard>, [">= 0"])
+    s.add_dependency(%q<thor>, [">= 0"])
+  end
+end

data/test/snapshots_test.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class SnapshotsTest < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'snapshots'
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,102 @@
+--- !ruby/object:Gem::Specification 
+name: snapshots
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Simon Menke
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-09-21 00:00:00 +02:00
+default_executable: snapshots
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: yard
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: thor
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: Dump your rails databases in a database agnostic format (ruby).
+email: simon@mrhenry.be
+engine_dependencies: {}
+
+executables: 
+- snapshots
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/snapshots
+- lib/snapshots.rb
+- lib/snapshots/app.rb
+- lib/snapshots/database_dumper.rb
+- lib/snapshots/database_loader.rb
+- lib/snapshots/tasks.rb
+- snapshots.gemspec
+- test/snapshots_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: http://github.com/simonmenke/snapshots
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Make snapshots of your ActiveRecord database
+test_files: 
+- test/snapshots_test.rb
+- test/test_helper.rb