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.

119 lines
3.4 KiB

13 years ago
require 'resolv'
13 years ago
class Domain < ActiveRecord::Base
13 years ago
set_inheritance_column "sti_disabled"
nilify_blanks
# optional IP for create form, results in a type A record
attr_accessor :ip
13 years ago
13 years ago
belongs_to :user, :inverse_of => :domain
has_many :records, :inverse_of => :domain, :dependent => :destroy
13 years ago
cattr_reader :types
@@types = ['NATIVE', 'MASTER', 'SLAVE', 'SUPERSLAVE']
13 years ago
has_one :soa_record,
:class_name => 'SOA',
:conditions => {:type => 'SOA'},
:inverse_of => :domain
13 years ago
for type in Record.types
13 years ago
has_many :"#{type.downcase}_records",
:class_name => type,
:conditions => {:type => type},
:inverse_of => :domain
13 years ago
validates_associated :"#{type.downcase}_records"
end
validates :name, :presence => true, :uniqueness => true, :domainname => {:require_valid_tld => false}
validates :master, :presence => true, :if => :slave?
13 years ago
validates :master, :ip => true, :allow_nil => true, :if => :slave?
13 years ago
validates :type, :inclusion => { :in => @@types, :message => "Unknown domain type" }
validates :soa_record, :presence => {:unless => :slave?}
validates_associated :soa_record #, :allow_nil => true
validates :ns_records, :on => :create, :presence => true, :length => {
:minimum => 2, :maximum => 10, :message => "must have be at least 2, at most 10"}
13 years ago
validates_associated :records
13 years ago
validates :user_id, :presence => true
13 years ago
validate :max_domains_per_user, :on => :create
def max_domains_per_user # domains per user limit for DOS protection
max = Settings.max_domains_per_user.to_i
if user.domains.count >= max
errors.add :base, "as a security measure, you cannot have more than #{max} domains on one account"
end
end
13 years ago
13 years ago
validate :domain_ownership
def domain_ownership
# non-TLD validation
13 years ago
errors[:name] = "cannot be a TLD or a reserved domain" if Tld.include?(name)
# if parent domain is on our system, the user must own it
segments = name.split('.')
if segments.size >= 2
parent = segments[1..-1].join('.')
parent_domain = Domain.find_by_name(parent)
if parent_domain.present? && parent_domain.user_id != user_id
errors[:name] = "issue, the parent domain `#{parent}` is registered to another user"
end
end
13 years ago
end
13 years ago
13 years ago
def slave?; self.type == 'SLAVE' end
13 years ago
before_create do
a_records.build(:content => ip) if ip.present?
end
before_validation(:on => :update) do
if name_changed?
name_was_pattern = /#{Regexp.escape(name_was)}$/
each_update_involved_record do |record|
if record.type == 'SOA'
record.reset_serial
record.name = name
else
record.name = record.name.sub(name_was_pattern, name)
end
record.domain = self
end
end
end
after_update do
if name_changed?
name_was_pattern = /#{Regexp.escape(name_was)}$/
for record in records.all
record.name = record.name.sub(name_was_pattern, name)
record.save!
end
end
end
def each_update_involved_record
yield soa_record
for record in soa_records
yield record
end
for record in records
yield record
end
end
13 years ago
scope :host_domains, where(:name => Settings.host_domains)
13 years ago
def setup(email)
13 years ago
build_soa_record
13 years ago
soa = soa_record
soa.contact ||= email
13 years ago
ns_records.build
13 years ago
ns_records.build
13 years ago
ns1, ns2 = ns_records
13 years ago
ns1.content = Settings.ns.first
ns2.content = (Settings.ns - [ns1.content]).sample
13 years ago
end
13 years ago
end