Specifying encoding on database creation

[hcoop/domtool2.git] / src / plugins / bind.sml

diff --git a/src/plugins/bind.sml b/src/plugins/bind.sml
index 46bd086..768478f 100644 (file)
--- a/src/plugins/bind.sml
+++ b/src/plugins/bind.sml
@@ -57,7 +57,7 @@ datatype dns_record =
        | AAAA of host * string
        | TXT of host * string
        | AFSDB of string
-       | SRV of string * int * int * int * string
+       | SRV of host * int * int * int * string
 
 fun hostS (Literal s) = s ^ "."
   | hostS Wildcard = "*."
@@ -71,6 +71,14 @@ val host = fn (EApp ((EVar "literal", _), e), _) =>
              SOME Default
            | _ => NONE
 
+val srv_host = fn (EApp ((EVar "srv_literal", _), e), _) =>
+                 Option.map Literal (Env.string e)
+               | (EVar "srv_wildcard", _) =>
+                 SOME Wildcard
+               | (EVar "srv_default", _) =>
+                 SOME Default
+               | _ => NONE
+
 val record = fn (EApp ((EApp ((EVar "dnsA", _), e1), _), e2), _) =>
                (case (host e1, Domain.ip e2) of
                     (SOME v1, SOME v2) => SOME (A (v1, v2))
@@ -90,13 +98,13 @@ val record = fn (EApp ((EApp ((EVar "dnsA", _), e1), _), e2), _) =>
                     (SOME v1, SOME v2) => SOME (AAAA (v1, v2))
                   | _ => NONE)
              | (EApp ((EApp ((EVar "dnsTXT", _), e1), _), e2), _) =>
-               (case (host e1, Env.string e2) of
+               (case (srv_host e1, Env.string e2) of
                     (SOME v1, SOME v2) => SOME (TXT (v1, v2))
                   | _ => NONE)
              | (EApp ((EVar "dnsAFSDB", _), e), _) =>
                Option.map AFSDB (Env.string e)
              | (EApp ((EApp ((EApp ((EApp ((EApp ((EVar "dnsSRV", _), e1), _), e2), _), e3), _), e4), _), e5), _) =>
-               (case (Env.string e1, Env.int e2, Env.int e3, Env.int e4, Env.string e5) of
+               (case (srv_host e1, Env.int e2, Env.int e3, Env.int e4, Env.string e5) of
                     (SOME v1, SOME v2, SOME v3, SOME v4, SOME v5) => SOME (SRV (v1, v2, v3, v4, v5))
                   | _ => NONE)
              | _ => NONE
@@ -162,8 +170,7 @@ fun writeRecord (evs, r) =
                               write "\t";
                               write host;
                               write ".\n")
-             | SRV (from, priority, weight, port, to) => (write from;
-                                                          write ".";
+             | SRV (from, priority, weight, port, to) => (write (hostS from);
                                                           writeDom ();
                                                           write ".\t";
                                                           write (Int.toString ttl);
@@ -351,16 +358,20 @@ val () = Domain.registerDescriber (Domain.considerAll
                                                     heading = "DNS zonefile contents:",
                                                     showEmpty = false}])
 
-fun validHost_ s =
+fun validSrvHost s =
     size s > 0 andalso size s < 20
     andalso CharVector.all (fn ch => Domain.isIdent ch orelse ch = #"-" orelse ch = #"_") s
 
-fun validSRVDomain s =
+fun validSrvDomain s =
     size s > 0 andalso size s < 100
-    andalso List.all validHost_ (String.fields (fn ch => ch = #".") s)
+    andalso List.all validSrvHost (String.fields (fn ch => ch = #".") s)
+
+val _ = Env.type_one "srv_host"
+       Env.string
+       validSrvHost
 
 val _ = Env.type_one "srv_domain"
        Env.string
-       validSRVDomain
+       validSrvDomain
 
 end