Changeset 275 for trunk/DNSDB.pm

Timestamp:

03/15/12 15:34:38 (12 years ago)

Author:

Kris Deugau

Message:

/trunk

Convert domStatus() and caller to zoneStatus() for reverse DNS
Add status change to revzones page handling
See #26

File:

• trunk/DNSDB.pm (modified) (3 diffs)

trunk/DNSDB.pm

@@ -48 +48 @@
         &parentID
         &isParent
-        &domStatus &importAXFR
+        &zoneStatus &importAXFR
         &export
         &mailNotify
@@ -70 +70 @@
                 &parentID
                 &isParent
-                &domStatus &importAXFR
+                &zoneStatus &importAXFR
                 &export
                 &mailNotify
@@ -2722 +2722 @@
 
 
-## DNSDB::domStatus()
-# Sets and/or returns a domain's status
-# Takes a database handle, domain ID and optionally a status argument
-# Returns undef on errors.
-sub domStatus {
+## DNSDB::zoneStatus()
+# Returns and optionally sets a zone's status
+# Takes a database handle, domain/revzone ID, forward/reverse flag, and optionally a status argument
+# Returns status, or undef on errors.
+sub zoneStatus {
   my $dbh = shift;
   my $id = shift;
-  my $newstatus = shift;
+  my $revrec = shift;
+  my $newstatus = shift || 'mu';
 
   return undef if $id !~ /^\d+$/;
 
-  my $sth;
-
 # ooo, fun!  let's see what we were passed for status
-  if ($newstatus) {
-    $sth = $dbh->prepare("update domains set status=? where domain_id=?");
-    # ass-u-me caller knows what's going on in full
-    if ($newstatus =~ /^[01]$/) {       # only two valid for now.
-      $sth->execute($newstatus,$id);
-    } elsif ($newstatus =~ /^domo(?:n|ff)$/) {
-      $sth->execute(($newstatus eq 'domon' ? 1 : 0),$id);
-    }
-  }
-
-  $sth = $dbh->prepare("select status from domains where domain_id=?");
-  $sth->execute($id);
-  my ($status) = $sth->fetchrow_array;
+  if ($newstatus ne 'mu') {
+    $newstatus = 0 if $newstatus eq 'domoff';
+    $newstatus = 1 if $newstatus eq 'domon';
+    $dbh->do("UPDATE ".($revrec eq 'n' ? 'domains' : 'revzones')." SET status=? WHERE ".
+        ($revrec eq 'n' ? 'domain_id' : 'rdns_id')."=?", undef, ($newstatus,$id) );
+  }
+
+  my ($status) = $dbh->selectrow_array("SELECT status FROM ".
+        ($revrec eq 'n' ? "domains WHERE domain_id=?" : "revzones WHERE rdns_id=?"),
+        undef, ($id) );
   return $status;
-} # end domStatus()
+} # end zoneStatus()