bundle exec add_index_to_type_column.rb db/schema.rb

add_index_to_type_column.rb

#!/usr/bin/env ruby
require 'rubygems'
require 'active_support/core_ext/string'

ARGF.each_line("create_table") do |line|
  line =~ / "([^"]+)"/
  table = $1
  if line =~ /t.string\s+"type"/
    next if line =~ /t.index \["type"\]/
    # check if nothing inherits from that class
    unless system("ag '<\\s*:?:?#{table.classify}$' app/ 2>&1 > /dev/null")
      filename= "app/models/#{table.singularize}.rb"
      # check if NewWithTypeSti mixin is included
      if File.exists?(filename)
        if File.open(filename).grep(/NewWithTypeSti/).empty?
          puts "# no STI for #{table} ?"
          print '#'
        end
      else
        puts "# #{filename} does not exist"
        print '#'
      end
    end
    puts "    add_index :#{table}, :type"
  end
end