iain-fill 0.0.1

data/Manifest

@@ -0,0 +1,7 @@
+Manifest
+README.rdoc
+Rakefile
+lib/fill.rb
+lib/fill/configure.rb
+lib/fill/presenter.rb
+lib/fill/procedure.rb

data/README.rdoc

@@ -0,0 +1,83 @@
+= Fill
+
+Rails 2.3.4 introduced db/seeds. Basically that means that the rake task
+"rake db:seed" will run db/seeds.rb, so you can have a centralized place
+to prepare your database by adding seed data.
+
+"Fill" takes this one step further and provides methods to easily define
+data in db/seeds.rb. Here's how:
+
+  Fill.database do  |db|
+
+    db.produce :projects do
+      10.times { Factory(:project) }
+    end
+
+  end
+
+(I use factory_girl here, do whatever you like off course)
+
+Now, when running "rake db:seed", it'll delete all records in the Project
+model and run the contents off the produce block. It'll also measure how
+long it took, resulting in this output:
+
+  +-------+--------+---------+----------+
+  | After | Before | Models  | Time     |
+  +-------+--------+---------+----------+
+  | 10    | 0      | Project | 0.018627 |
+  +-------+--------+---------+----------+
+
+== Why?
+
+At my company, we taught testers, customers and developers to use
+webistrano and run the db:seed" rake task themselves. This way, they can 
+test and experiment as much as they want, and they easily reset the
+database when they fucked it up.
+
+== Usage
+
+=== db.produce
+
+By specifying a block, do whatever you need to fill a model.
+
+Usage: db.produce :model1, :model2 do ... end
+
+Specify the models that you want. The models that you specified will be
+emptied, which is handy if you're building relational models, like users
+and their memberships.
+
+=== db.fill
+
+A simple list of values.
+
+Usage: db.fill :model, :attribute, "value 1", "value 2", "value 3"
+
+For simple models with only one distinct attribute, you can just specify
+the model, attribute and the values.
+
+See also the iain/root_table plugin if you have many of these.
+
+=== db.invoke
+
+Invoke a rake task.
+
+Usage: db.invoke "some:task", :model
+
+=== Global options
+
+* :needs    specify one or many dependencies, tables to be filled before
+            filling the ones you're specifying now.
+* :delete   set to false if you don't want to delete all records before
+            continuing
+
+== Installation
+
+Add this to config/environment.rb:
+
+  config.gem "iain-fill", :lib => "fill", :source => "http://gems.github.com"
+
+And run "rake gems:install" and you're done.
+
+---
+
+Made by Iain, Released under the MIT License

data/Rakefile

@@ -0,0 +1,13 @@
+require 'rake'
+require 'rake/rdoctask'
+
+require 'echoe'
+Echoe.new('fill', '0.0.1') do |p|
+  p.description = "Fill your database"
+  p.url = "http://iain.nl"
+  p.author = "iain"
+  p.email = "iain@iain.nl"
+  p.ignore_pattern = []
+  p.development_dependencies = []
+  p.runtime_dependencies = []
+end

data/fill.gemspec

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{fill}
+  s.version = "0.0.1"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["iain"]
+  s.date = %q{2009-09-21}
+  s.description = %q{Fill your database}
+  s.email = %q{iain@iain.nl}
+  s.extra_rdoc_files = ["README.rdoc", "lib/fill.rb", "lib/fill/configure.rb", "lib/fill/presenter.rb", "lib/fill/procedure.rb"]
+  s.files = ["README.rdoc", "Rakefile", "lib/fill.rb", "lib/fill/configure.rb", "lib/fill/presenter.rb", "lib/fill/procedure.rb", "Manifest", "fill.gemspec"]
+  s.homepage = %q{http://iain.nl}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Fill", "--main", "README.rdoc"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{fill}
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Fill your database}
+
+  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
+    else
+    end
+  else
+  end
+end

data/lib/fill.rb

@@ -0,0 +1,26 @@
+require 'activesupport'
+require 'fill/configure'
+require 'fill/presenter'
+require 'fill/procedure'
+
+module Fill
+
+  class << self
+
+    def database
+      db = Configure.new
+      yield db
+      bm = time { db.perform! }
+      puts Presenter
+      puts "Database filled in %.2f seconds" % bm
+    end
+
+    def time
+      started_at = Time.now
+      yield
+      Time.now - started_at
+    end
+
+  end
+
+end

data/lib/fill/configure.rb

@@ -0,0 +1,69 @@
+module Fill
+
+  class Configure
+
+    def produce(*models, &block)
+      options = models.extract_options!
+      needs = options.delete(:needs) || []
+      register models, Procedure.new(models, options, &block)
+      dependent models, needs
+    end
+
+    def fill(model, field, *values)
+      options = values.extract_options!
+      self.produce model, options do
+        values.each do |value|
+          model.to_s.singularize.camelize.constantize.create!(field => value)
+        end
+      end
+    end
+
+    def invoke(task, *models)
+      self.produce *models do
+        Rake::Task[task].invoke
+      end
+    end
+
+    def environment(env, which, options = {})
+
+    end
+
+    def perform!
+      registered.each_key { |model| perform(model) }
+    end
+
+    private
+
+    def results
+      results = registered.values.uniq.compact.map { |data| data.to_hash }
+    end
+
+    def perform(model)
+      raise "No fill data provided for #{model}" unless registered.has_key? model
+      dependencies[model].each { |dep| perform(dep) } if dependencies.has_key? model
+      registered[model].perform!
+    end
+
+    def register(models, data)
+      models.each do |model|
+        registered.update model => data
+      end
+    end
+
+    def dependent(models, dependent)
+      models.each do |model|
+        dependencies.update model => [dependent].flatten
+      end
+    end
+
+    def registered
+      @registered ||= {}
+    end
+
+    def dependencies
+      @dependencies ||= {}
+    end
+
+  end
+
+end

data/lib/fill/presenter.rb

@@ -0,0 +1,52 @@
+module Fill
+
+  class Presenter
+
+    def self.present(data)
+      presenter.add(data)
+    end
+
+    def self.presenter
+      @presenter ||= new
+    end
+
+    def self.to_s
+      presenter.to_s
+    end
+
+    def add(data)
+      presented.push(data) if data && !presented.include?(data)
+    end
+
+    def hirb?
+      require 'hirb'
+      true
+    rescue LoadError
+      false
+    end
+
+    def presented
+      @presented ||= []
+    end
+
+    def presentable
+      presented.map(&:to_hash)
+    end
+
+    def present_with_hirb
+      Hirb::Helpers::Table.render(presentable, :description => false)
+    end
+
+    def to_s
+      hirb? ? present_with_hirb : present_hash.join("\n")
+    end
+
+    def present_hash
+      hash.map do |row|
+        row.map { |key, value| "#{key}: #{value}" }.join(" - ")
+      end
+    end
+
+  end
+
+end

data/lib/fill/procedure.rb

@@ -0,0 +1,48 @@
+module Fill
+
+  class Procedure
+
+    attr_accessor :block
+
+    def initialize(models, options = {}, &block)
+      @block   = block
+      @options = { :delete => true }.merge(options)
+      @models  = models
+    end
+
+    def perform!
+      @performed ||= perform
+    end
+
+    def to_hash
+      { "Models" => human_models, "Before" => @before.join(', '), "After" => @after.join(', '), "Time" => @time }
+    end
+
+    def human_models
+      @options[:name] ||
+      models.map { |model| model.respond_to?(:human_name) ? model.human_name : model.to_s }.join(', ')
+    end
+
+    def delete_all
+      models.each { |model| model.delete_all } if @options[:delete]
+    end
+
+    def models
+      @models.map { |model| model.to_s.singularize.camelize.constantize }
+    end
+
+    def count
+      models.map { |model| model.count }
+    end
+
+    def perform
+      @before = count
+      @time   = Fill.time { self.delete_all; block.call }
+      @after  = count
+      Presenter.present self
+      true
+    end
+
+  end
+
+end

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification 
+name: iain-fill
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- iain
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-09-21 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Fill your database
+email: iain@iain.nl
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+- lib/fill.rb
+- lib/fill/configure.rb
+- lib/fill/presenter.rb
+- lib/fill/procedure.rb
+files: 
+- README.rdoc
+- Rakefile
+- lib/fill.rb
+- lib/fill/configure.rb
+- lib/fill/presenter.rb
+- lib/fill/procedure.rb
+- Manifest
+- fill.gemspec
+has_rdoc: false
+homepage: http://iain.nl
+licenses: 
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Fill
+- --main
+- README.rdoc
+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: "1.2"
+  version: 
+requirements: []
+
+rubyforge_project: fill
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Fill your database
+test_files: []
+