module Scrivito
module GeneratorHelper
  def self.attribute_definition(arg)
    if %w(enum multienum).include?(arg.type.to_s)
      "attribute :#{arg.name}, :#{arg.type}, values: []"
    else
      "attribute :#{arg.name}, :#{arg.type}"
    end
  end
end
end