seeder 0.1.0

data/CHANGELOG.txt

@@ -0,0 +1,2 @@
+== 0.1.0
+* Initial release, no tests, MySQL-only

data/LICENSE.txt

@@ -0,0 +1,22 @@
+The MIT LICENSE
+
+Copyright (c) 2011 Wegowise Inc.
+
+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.md

@@ -0,0 +1,49 @@
+Description
+===========
+
+Seeder provides a way for your app to plant seed data in its database.
+It respects unique keys in your database, so that it does the equivalent of
+MySQL's "on duplicate key update" (but this is database agnostic)
+
+
+Install
+=======
+
+    $ gem install seeder
+
+Usage
+=====
+
+Suppose you have a `User` model
+
+And suppose that user model has fields "name", "age", "address", "gender"
+
+Finally, suppose you've set up your database to have a unique key on the fields
+"name" and "address" (so two people can have the same name or live at the
+same address, but not both)
+
+If you wanted to seed your data with a couple users you could do the following:
+
+```ruby
+seed_users = [
+  {
+    :name => "J'onn J'onzz",
+    :age => 94,
+    :address => 'Mars',
+    :gender => 'Male'
+  },
+  {
+    :name => "Barbara Gordon",
+    :age => 35,
+    :address => '14 Gotham Heights',
+    :gender => 'Female'
+  }
+]
+
+SeedData.create {'User' => seed_users}, [:name, :address], User
+```
+
+License
+=======
+
+See LICENSE.txt

data/Rakefile

@@ -0,0 +1,39 @@
+require 'rake'
+require 'fileutils'
+
+def gemspec_name
+  @gemspec_name ||= Dir['*.gemspec'][0]
+end
+
+def gemspec
+  @gemspec ||= eval(File.read(gemspec_name), binding, gemspec_name)
+end
+
+desc "Build the gem"
+task :gem=>:gemspec do
+  sh "gem build #{gemspec_name}"
+  FileUtils.mkdir_p 'pkg'
+  FileUtils.mv "#{gemspec.name}-#{gemspec.version}.gem", 'pkg'
+end
+
+desc "Install the gem locally"
+task :install => :gem do
+  sh %{gem install pkg/#{gemspec.name}-#{gemspec.version}}
+end
+
+desc "Generate the gemspec"
+task :generate do
+  puts gemspec.to_ruby
+end
+
+desc "Validate the gemspec"
+task :gemspec do
+  gemspec.validate
+end
+
+desc 'Run tests'
+task :test do |t|
+  sh 'rspec spec'
+end
+
+task :default => :test

data/lib/seeder.rb

@@ -0,0 +1,62 @@
+class Seeder
+
+  class << self
+
+    def existing_data_keys(ar_model)
+      keys = @primary_keys[ar_model.name]
+      ar_model.all.map{|item| item.attributes.symbolize_keys.values_at(*keys) }
+    end
+
+    def seed_data_keys(ar_model)
+      keys = @primary_keys[ar_model.name]
+      @data[ar_model.name].collect{|attributes| attributes.values_at(*keys) }
+    end
+
+    def items_to_delete(ar_model)
+      existing_data_keys(ar_model) - seed_data_keys(ar_model)
+    end
+
+    def value_to_sql(value)
+      if value.nil? then 'NULL'
+      elsif value.is_a?(String) then "'#{value}'"
+      elsif value == false then 0
+      elsif value == true then 1
+      else value
+      end
+    end
+
+    def add_new_or_changed_data(ar_model)
+      keys = @primary_keys[ar_model.name]
+      find_method = "find_by_#{keys.join('_and_')}"
+      @data[ar_model.name].collect do |attributes|
+        relevant_keys = attributes.keys & ar_model.column_names.map(&:to_sym)
+        existing = ar_model.send(find_method, *attributes.values_at(*keys))
+        next if existing && attributes.values_at(*relevant_keys) == existing.attributes.symbolize_keys.values_at(*relevant_keys)
+        values = attributes.values_at(*relevant_keys)
+        %{
+          INSERT INTO #{ar_model.table_name} (#{relevant_keys.map{|col| "`#{col}`"}.join(', ')}) VALUES (#{values.map{|val| value_to_sql(val)}.join(', ')}) ON DUPLICATE KEY UPDATE #{relevant_keys.map{|col| "`#{col}` = VALUES(`#{col}`)"}.join(', ')}
+        }
+      end.compact
+    end
+
+    def delete_outdated_data(ar_model)
+      keys = @primary_keys[ar_model.name]
+      items_to_delete(ar_model).collect do |field_to_delete|
+        conditions = field_to_delete.to_enum.with_index.collect{|value, index| "#{keys[index]} = '#{value}'"}.join(' AND ')
+        %{ DELETE FROM #{ar_model.table_name} WHERE #{conditions} }
+      end.compact
+    end
+
+    def create(data, keys, models)
+      @data, @primary_keys = data, keys
+      models.each {|model|
+        delete_outdated_data(model).each{|query| sql(query)}
+        add_new_or_changed_data(model).each{|query| sql(query)}
+      }
+    end
+
+    def sql(sql_command)
+      ActiveRecord::Base.connection.execute(sql_command)
+    end
+  end
+end

data/lib/seeder/version.rb

@@ -0,0 +1,3 @@
+class Seeder
+  VERSION = '0.1.0'
+end

data/seeder.gemspec

@@ -0,0 +1,18 @@
+# -*- encoding: utf-8 -*-
+require 'rubygems' unless defined? Gem
+require File.dirname(__FILE__) + "/lib/seeder/version"
+
+Gem::Specification.new do |s|
+  s.name        = "seeder"
+  s.version     = Seeder::VERSION
+  s.authors     = ["Barun Singh", "Gabriel Horner"]
+  s.email       = "bsingh@wegowise.com"
+  s.homepage    = "http://github.com/wegowise/seeder"
+  s.summary = "Seed your data"
+  s.description =  "Keep your app's seed data in one file and update it easily, while respecting key constraints"
+  s.required_rubygems_version = ">= 1.3.6"
+  s.files = Dir.glob(%w[{lib,spec}/**/*.rb [A-Z]*.{txt,rdoc,md} *.gemspec]) + %w{Rakefile}
+  s.extra_rdoc_files = ["README.md", "LICENSE.txt"]
+  s.license = 'MIT'
+  s.add_development_dependency 'rspec', '~> 2.6.0'
+end

metadata

@@ -0,0 +1,67 @@
+--- !ruby/object:Gem::Specification
+name: seeder
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Barun Singh
+- Gabriel Horner
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-09-02 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &2152871700 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+  type: :development
+  prerelease: false
+  version_requirements: *2152871700
+description: Keep your app's seed data in one file and update it easily, while respecting
+  key constraints
+email: bsingh@wegowise.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- README.md
+- LICENSE.txt
+files:
+- lib/seeder/version.rb
+- lib/seeder.rb
+- CHANGELOG.txt
+- LICENSE.txt
+- README.md
+- seeder.gemspec
+- Rakefile
+homepage: http://github.com/wegowise/seeder
+licenses:
+- MIT
+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: 1.3.6
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Seed your data
+test_files: []