Project

General

Profile

Download (3.26 KB) Statistics
| Branch: | Tag: | Revision:
require "resolv"
# This models a DNS domain and so represents a site.
class Domain < ApplicationRecord
audited
include Authorizable
extend FriendlyId
friendly_id :name
include Taxonomix
include StripLeadingAndTrailingDot
include Parameterizable::ByIdName
include BelongsToProxies
include ParameterAttributes

validates_lengths_from_database
has_many :hostgroups
# order matters! see https://github.com/rails/rails/issues/670
before_destroy EnsureNotUsedBy.new(:interfaces, :hostgroups, :subnets)
has_many :subnet_domains, :dependent => :destroy, :inverse_of => :domain
has_many :subnets, :through => :subnet_domains

belongs_to_proxy :dns,
:feature => 'DNS',
:label => N_('DNS Proxy'),
:description => N_('DNS proxy to use within this domain for managing A records, note that PTR records are managed via Subnet DNS proxy'),
:api_description => N_('DNS proxy ID to use within this domain')

has_many :domain_parameters, :dependent => :destroy, :foreign_key => :reference_id, :inverse_of => :domain
has_many :parameters, :dependent => :destroy, :foreign_key => :reference_id, :class_name => "DomainParameter"
has_many :interfaces, :class_name => 'Nic::Base'
has_many :primary_interfaces, -> { where(:primary => true) }, :class_name => 'Nic::Base'
has_many :hosts, :through => :interfaces
has_many :primary_hosts, :through => :primary_interfaces, :source => :host

accepts_nested_attributes_for :domain_parameters, :allow_destroy => true
include ParameterValidators
validates :name, :presence => true, :uniqueness => true
validates :fullname, :uniqueness => true, :allow_blank => true, :allow_nil => true

scoped_search :on => [:name, :fullname], :complete_value => true
scoped_search :relation => :domain_parameters, :on => :value, :on_key => :name, :complete_value => true, :only_explicit => true, :rename => :params

# with proc support, default_scope can no longer be chained
# include all default scoping here
default_scope lambda {
with_taxonomy_scope do
order("domains.name")
end
}

class Jail < Safemode::Jail
allow :name, :fullname
end

# return the primary name server for our domain based on DNS lookup
# it first searches for SOA record, if it failed it will search for NS records
def nameservers
return [] if Setting[:query_local_nameservers]
dns = Resolv::DNS.new
ns = dns.getresources(name, Resolv::DNS::Resource::IN::SOA).collect {|r| r.mname.to_s}
ns = dns.getresources(name, Resolv::DNS::Resource::IN::NS).collect {|r| r.name.to_s} if ns.empty?
ns.to_a.flatten
end

def resolver
ns = nameservers
Resolv::DNS.new ns.empty? ? nil : {:search => name, :nameserver => ns, :ndots => 1}
end

def proxy
ProxyAPI::DNS.new(:url => dns.url) if dns && dns.url.present?
end

def lookup(query)
Net::DNS.lookup query, proxy, resolver
end

def dot_strip_attrs
['name']
end

# overwrite method in taxonomix, since domain is not direct association of host anymore
def used_taxonomy_ids(type)
return [] if new_record?
Host::Base.joins(:primary_interface).where(:nics => {:domain_id => id}).distinct.pluck(type).compact
end

def hosts_count
Host::Managed.authorized(:view_hosts).joins(:primary_interface).where(:nics => {:domain_id => id}).size
end
end
(13-13/70)