gitlab-org--gitlab-foss/app/models/concerns/strip_attribute.rb

38 lines
698 B
Ruby

# frozen_string_literal: true
# == Strip Attribute module
#
# Contains functionality to remove leading and trailing
# whitespace from the attribute before validation
#
# Usage:
#
# class Milestone < ApplicationRecord
# strip_attributes! :title
# end
#
#
module StripAttribute
extend ActiveSupport::Concern
class_methods do
def strip_attributes!(*attrs)
strip_attrs.concat(attrs)
end
def strip_attrs
@strip_attrs ||= []
end
end
included do
before_validation :strip_attributes!
end
def strip_attributes!
self.class.strip_attrs.each do |attr|
self[attr].strip! if self[attr] && self[attr].respond_to?(:strip!)
end
end
end