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
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
|
|
|