30247a3800
This makes it easier for developers new to the codebase to easily get important information on the model's table in the same file as the model code. To ensure the model annotations stay accurate, run them on test_up/test_down. In CI, regenerate the annotations, and check for no changes, similar to how the linters work.
74 lines
2.4 KiB
Ruby
74 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../../model"
|
|
|
|
class DnsZone < Sequel::Model
|
|
one_to_one :strand, key: :id
|
|
many_to_one :project
|
|
many_to_many :dns_servers
|
|
one_to_many :records, class: :DnsRecord
|
|
one_to_one :active_billing_record, class: :BillingRecord, key: :resource_id do |ds| ds.active end
|
|
|
|
include ResourceMethods
|
|
include SemaphoreMethods
|
|
include Authorization::HyperTagMethods
|
|
include Authorization::TaggableMethods
|
|
|
|
semaphore :refresh_dns_servers
|
|
|
|
def hyper_tag_name(project)
|
|
"project/#{project.ubid}/dns-zone/#{ubid}"
|
|
end
|
|
|
|
def insert_record(record_name:, type:, ttl:, data:)
|
|
record_name = add_dot_if_missing(record_name)
|
|
DnsRecord.create_with_id(dns_zone_id: id, name: record_name, type: type, ttl: ttl, data: data)
|
|
|
|
incr_refresh_dns_servers
|
|
end
|
|
|
|
def delete_record(record_name:, type: nil, data: nil)
|
|
fail "Type needs to be specified if data is specified!" if data && type.nil?
|
|
|
|
record_name = add_dot_if_missing(record_name)
|
|
records = records_dataset.where(name: record_name)
|
|
records = records.where(type: type) if type
|
|
records = records.where(data: data) if data
|
|
|
|
DB[:dns_record].multi_insert(
|
|
records.map {
|
|
{
|
|
id: DnsRecord.generate_uuid,
|
|
dns_zone_id: id,
|
|
name: _1.name,
|
|
type: _1.type,
|
|
ttl: _1.ttl,
|
|
data: _1.data,
|
|
tombstoned: true
|
|
}
|
|
}
|
|
)
|
|
|
|
incr_refresh_dns_servers
|
|
end
|
|
|
|
def add_dot_if_missing(record_name)
|
|
(record_name[-1] == ".") ? record_name : record_name + "."
|
|
end
|
|
end
|
|
|
|
# Table: dns_zone
|
|
# Columns:
|
|
# id | uuid | PRIMARY KEY
|
|
# created_at | timestamp with time zone | NOT NULL DEFAULT now()
|
|
# project_id | uuid | NOT NULL
|
|
# name | text | NOT NULL
|
|
# last_purged_at | timestamp with time zone | NOT NULL DEFAULT now()
|
|
# Indexes:
|
|
# dns_zone_pkey | PRIMARY KEY btree (id)
|
|
# Referenced By:
|
|
# cert | cert_dns_zone_id_fkey | (dns_zone_id) REFERENCES dns_zone(id)
|
|
# dns_record | dns_record_dns_zone_id_fkey | (dns_zone_id) REFERENCES dns_zone(id)
|
|
# dns_servers_dns_zones | dns_servers_dns_zones_dns_zone_id_fkey | (dns_zone_id) REFERENCES dns_zone(id)
|
|
# load_balancer | load_balancer_custom_hostname_dns_zone_id_fkey | (custom_hostname_dns_zone_id) REFERENCES dns_zone(id)
|