You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

75 lines
2.4 KiB

class Record < ActiveRecord::Base
belongs_to :domain, :inverse_of => :records
belongs_to :user, :inverse_of => :records
cattr_reader :types
@@types = %w(SOA NS A MX TXT CNAME)
validates :domain, :name, :presence => true
validates :type, :inclusion => {:in => @@types, :message => "Unknown record type"}
validates :content, :uniqueness => {:scope => [:domain_id, :type, :name]}
# RFC 2181, 8
validates :ttl, :numericality => {
# :greater_than_or_equal_to => 0,
:greater_than_or_equal_to => Settings.min_ttl.to_i,
:less_than => 2**31,
:only_integer => true
}, :allow_blank => true
validates :authentication_token, :presence => true, :uniqueness => true
validate :max_records_per_domain, :on => :create
def max_records_per_domain # domains per user limit for DOS protection
max = Settings.max_records_per_domain.to_i
if domain.records.count >= max
errors.add :base, "as a security measure, you cannot have more than #{max} records on one domain"
end
end
before_validation :generate_token, :on => :create
before_validation :prepare_name!
before_save :update_change_date
after_save :update_soa_serial
after_initialize do
self.ttl ||= Settings.default_ttl
end
# By default records don't support priorities.
# Those who do can overwrite this in their own classes.
def supports_priority?; false end
def shortname; name.gsub(/\.?#{self.domain.name}$/, '') end
def shortname=(value); self.name = value end
def to_label; "#{type} #{content}" end
# Generate a token by looping and ensuring does not already exist.
# stolen from Devise
def generate_token
self.authentication_token = loop do
token = Devise.friendly_token
break token unless self.class.exists?(:authentication_token => token)
end
end
protected
def prepare_name!
return if domain.nil? or domain.name.blank?
self.name = domain.name if name.blank? or name == '@'
self.name << ".#{domain.name}" unless name =~ /#{Regexp.escape(domain.name)}$/
end
# Update the change date for automatic serial number generation
def update_change_date; self.change_date = Time.now.to_i end
def update_soa_serial #:nodoc:
unless type == 'SOA' || @serial_updated || domain.slave?
domain.soa_record.update_serial!
@serial_updated = true
end
end
def name_equals_domain_name
errors.add :name, "must be equal to domain's name" unless name == domain.name
end
end