albertosaurus/power_enum

# Copyright (c) 2011 Arthur Shagall
# Released under the MIT license.  See LICENSE for details.

module PowerEnum::Migration # :nodoc:

  # Extensions for CommandRecorder
  module CommandRecorder
    # Records create_enum
    def create_enum(*args)
      record(:create_enum, args)
    end

    # Records remove_enum
    def remove_enum(*args)
      record(:remove_enum, args)
    end

    # The inversion of create_enum is remove_enum
    # @param [Array] args Arguments to create_enum
    # @return [Array] [:remove_enum, [enum_name]]
    def invert_create_enum(args)
      enum_name = args[0]
      [:remove_enum, [enum_name]]
    end
  end

end