module Scrivito
module GeneratorHelper
  def self.attribute_definition(arg)
    name, type = arg.to_s.split(":")

    if %w(enum multienum).include?(type)
      "attribute :#{name}, :#{type}, values: []"
    else
      "attribute :#{name}, :#{type}"
    end
  end
end
end