* sipb_xen_database -> invirt.database

[invirt/packages/invirt-dns.git] / code / dnsserver.py

diff --git a/code/dnsserver.py b/code/dnsserver.py
index 8a41a25..7a7cf22 100755 (executable)
--- a/code/dnsserver.py
+++ b/code/dnsserver.py
@@ -6,7 +6,8 @@ from twisted.names import common
 from twisted.internet import defer
 from twisted.python import failure
 
-import sipb_xen_database
+from invirt.config import structs as config
+import invirt.database
 import psycopg2
 import sqlalchemy
 import time
@@ -16,18 +17,24 @@ class DatabaseAuthority(common.ResolverBase):
 
     soa = None
 
-    def __init__(self, domains, database=None):
+    def __init__(self, domains=None, database=None):
         common.ResolverBase.__init__(self)
         if database is not None:
-            sipb_xen_database.connect(database)
-        self.domains = domains
-        self.soa = dns.Record_SOA(mname='ns1.xvm.mit.edu', 
-                                  rname='xvm.mit.edu',
+            invirt.database.connect(database)
+        else:
+            invirt.database.connect()
+        if domains is not None:
+            self.domains = domains
+        else:
+            self.domains = config.dns.domains
+        ns = config.dns.nameservers[0]
+        self.soa = dns.Record_SOA(mname=ns.hostname,
+                                  rname=config.dns.contact.replace('@','.',1),
                                   serial=1, refresh=3600, retry=900,
                                   expire=3600000, minimum=21600, ttl=3600)
-        self.ns = dns.Record_NS(name='ns1.xvm.mit.edu', ttl=3600)
-        record = dns.Record_A(address='18.181.0.62', ttl=3600)
-        self.ns1 = dns.RRHeader('ns1.xvm.mit.edu', dns.A, dns.IN,
+        self.ns = dns.Record_NS(name=ns.hostname, ttl=3600)
+        record = dns.Record_A(address=ns.ip, ttl=3600)
+        self.ns1 = dns.RRHeader(ns.hostname, dns.A, dns.IN,
                                 3600, record, auth=True)
 
     
@@ -45,7 +52,7 @@ class DatabaseAuthority(common.ResolverBase):
                 return value
 
     def _lookup_unsafe(self, name, cls, type, timeout):
-        sipb_xen_database.clear_cache()
+        invirt.database.clear_cache()
         
         ttl = 900
         name = name.lower()
@@ -67,7 +74,7 @@ class DatabaseAuthority(common.ResolverBase):
             host = name[:-len(domain)-1]
             if not host:
                 if type in (dns.A, dns.ALL_RECORDS):
-                    record = dns.Record_A('18.181.0.62', ttl)
+                    record = dns.Record_A(config.dns.nameservers[0].ip, ttl)
                     results.append(dns.RRHeader(name, dns.A, dns.IN, 
                                                 ttl, record, auth=True))
                 elif type == dns.NS:
@@ -79,7 +86,7 @@ class DatabaseAuthority(common.ResolverBase):
                                                 ttl, self.soa, auth=True))
             else:
                 if host:
-                    value = sipb_xen_database.Machine.get_by(name=host)
+                    value = invirt.database.Machine.get_by(name=host)
                     if value is None or not value.nics:
                         return defer.fail(failure.Failure(dns.AuthoritativeDomainError(name)))
                     ip = value.nics[0].ip
@@ -101,9 +108,7 @@ class DatabaseAuthority(common.ResolverBase):
             return defer.fail(failure.Failure(dns.AuthoritativeDomainError(name)))
 
 if '__main__' == __name__:
-    resolver = DatabaseAuthority(['servers.csail.mit.edu',
-                                  'xvm.mit.edu'],
-                                 'postgres://sipb-xen@sipb-xen-dev/sipb_xen')
+    resolver = DatabaseAuthority()
 
     verbosity = 0
     f = server.DNSServerFactory(authorities=[resolver], verbose=verbosity)