Contents

require 'sequel'
require 'sequel/extensions/schema_dumper'
require 'sequel/extensions/migration'
require 'vendor/okjson'

module Taps
  module Schema
    extend self

    def dump(database_url)
      db = Sequel.connect(database_url)
      db.dump_schema_migration(:indexes => false)
    end

    def dump_table(database_url, table)
      table = table.to_sym
      Sequel.connect(database_url) do |db|
        <<END_MIG
  Class.new(Sequel::Migration) do
    def up
      #{db.extension(:schema_dumper).dump_table_schema(Sequel.identifier(table), :indexes => false)}
    end

    def down
      drop_table("#{table}") if @db.table_exists?("#{table}")
    end
  end
  END_MIG
      end
    end

    def indexes(database_url)
      db = Sequel.connect(database_url)
      db.dump_indexes_migration
    end

    def indexes_individual(database_url)
      idxs = {}
      Sequel.connect(database_url) do |db|
        tables = db.tables
        tables.each do |table|
          idxs[table] = db.extension(:schema_dumper).send(:dump_table_indexes, table, :add_index, {}).split("\n")
        end
      end

      idxs.each do |table, indexes|
        idxs[table] = indexes.map do |idx|
          <<END_MIG
  Class.new(Sequel::Migration) do
    def up
      #{idx}
    end
  end
  END_MIG
        end
      end
      ::OkJson.encode(idxs)
    end

    def load(database_url, schema)
      Sequel.connect(database_url) do |db|
        klass = eval(schema)
        klass.apply(db, :down)
        klass.apply(db, :up)
      end
    end

    def load_indexes(database_url, indexes)
      Sequel.connect(database_url) do |db|
        eval(indexes).apply(db, :up)
      end
    end

    def reset_db_sequences(database_url)
      db = Sequel.connect(database_url)
      return unless db.respond_to?(:reset_primary_key_sequence)
      db.tables.each do |table|
        db.reset_primary_key_sequence(table)
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
taps2-0.5.2	lib/taps/schema.rb
taps2-0.5.1	lib/taps/schema.rb