aasm/lib/generators/aasm/orm_helpers.rb

42 lines
667 B
Ruby

module AASM
module Generators
module OrmHelpers
def model_contents
if column_name == 'aasm_state'
<<RUBY
include AASM
aasm do
end
RUBY
else
<<RUBY
include AASM
aasm :column => '#{column_name}' do
end
RUBY
end
end
private
def column_exists?
table_name.singularize.humanize.constantize.column_names.include?(column_name.to_s)
rescue NameError
false
end
def model_exists?
File.exist?(File.join(destination_root, model_path))
end
def model_path
@model_path ||= File.join("app", "models", "#{file_path}.rb")
end
end
end
end