Contents

class Ridgepole::Dumper
  def initialize(options = {})
    @options = options
  end

  def dump
    stream = StringIO.new
    conn = ActiveRecord::Base.connection
    target_tables = @options[:tables]

    if target_tables
      conn.tables.each do |tbl|
        next if target_tables.include?(tbl)
        ActiveRecord::SchemaDumper.ignore_tables << tbl
      end
    end

    ActiveRecord::SchemaDumper.dump(conn, stream)

    ActiveRecord::SchemaDumper.ignore_tables.clear if target_tables

    dsl = stream.string.lines.select {|line|
      line !~ /\A#/ &&
      line !~ /\AActiveRecord::Schema\.define/ &&
      line !~ /\Aend/
    }.join.undent.strip

    definitions = []

    each_table(dsl) do |name, definition|
      if target?(name)
        definitions << definition
        yield(name, definition) if block_given?
      end
    end

    definitions.join("\n\n")
  end

  private

  def each_table(dsl)
    name = nil
    definition = []

    pass = proc do
      if name
        yield(name, definition.join.strip)
        name = nil
        definition = []
      end
    end

    dsl.lines.each do |line|
      if line =~ /\Acreate_table/
        pass.call
        name = line.split(/[\s,'"]+/)[1]
        definition << line
      elsif name
        definition << line
      end
    end

    pass.call
  end

  def target?(table_name)
    not @options[:tables] or @options[:tables].include?(table_name)
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
ridgepole-0.1.7	lib/ridgepole/dumper.rb
ridgepole-0.1.6	lib/ridgepole/dumper.rb
ridgepole-0.1.5	lib/ridgepole/dumper.rb
ridgepole-0.1.4	lib/ridgepole/dumper.rb
ridgepole-0.1.3	lib/ridgepole/dumper.rb