module Scrivito
module GeneratorHelper
  class << self
    def 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
end