Changeset 18 for trunk/DNSDB.pm

Timestamp:

10/08/09 17:29:50 (15 years ago)

Author:

Kris Deugau

Message:

/trunk

checkpoint

File:

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

trunk/DNSDB.pm

@@ -23 +23 @@
 @ISA            = qw(Exporter);
 @EXPORT_OK      = qw(
-        &initGlobals &connectDB &finish &addDomain &delDomain &domainName &grpName &getSOA &getRecLine
-        &getDomRecs &addRec &updateRec &delRec &domStatus
+        &initGlobals &connectDB &finish &addDomain &delDomain &domainName &addGroup &grpName &getSOA
+        &getRecLine &getDomRecs &addRec &updateRec &delRec &domStatus
         %typemap %reverse_typemap
         );
@@ -30 +30 @@
 @EXPORT         = (); # Export nothing by default.
 %EXPORT_TAGS    = ( ALL => [qw(
-                &initGlobals &connectDB &finish &addDomain &delDomain &domainName &grpName &getSOA &getRecLine
-                &getDomRecs &addRec &updateRec &delRec &domStatus
+                &initGlobals &connectDB &finish &addDomain &delDomain &domainName &addGroup &grpName &getSOA
+                &getRecLine &getDomRecs &addRec &updateRec &delRec &domStatus
                 %typemap %reverse_typemap
                 )]
@@ -256 +256 @@
 
 
+## DNSDB::addGroup()
+# Add a group
+# Takes a database handle, group name, parent group, and template-vs-cloneme flag
+# Returns a status code and message
+sub addGroup {
+  $errstr = '';
+  my $dbh = shift;
+  my $grpname = shift;
+  my $pargrp = shift;
+
+  # 0 indicates "template", hardcoded.
+  # Any other value clones that group's default records, if it exists.
+  my $torc = shift || 0;        
+
+  # Allow transactions, and raise an exception on errors so we can catch it later.
+  # Use local to make sure these get "reset" properly on exiting this block
+  local $dbh->{AutoCommit} = 0;
+  local $dbh->{RaiseError} = 1;
+
+  # Wrap all the SQL in a transaction
+  eval {
+    my $sth = $dbh->prepare("INSERT INTO groups (parent_group_id,group_name) VALUES (?,?)");
+    $sth->execute($pargrp,$grpname);
+
+    $sth = $dbh->prepare("SELECT group_id FROM groups WHERE group_name=?");
+    $sth->execute($grpname);
+    my ($grpid) = $sth->fetchrow_array();
+
+    $sth = $dbh->prepare("INSERT INTO default_records (group_id,host,type,val,distance,weight,port,ttl) ".
+        "VALUES ($grpid,?,?,?,?,?,?,?)");
+    if ($torc) {
+      my $sth2 = $dbh->prepare("SELECT host,type,val,distance,weight,port,ttl FROM default_records WHERE group_id=?");
+      while (my @clonedata = $sth2->fetchrow_array) {
+        $sth->execute(@clonedata);
+      }
+    } else {
+      # reasonable basic defaults for SOA, MX, NS, and minimal hosting
+      # could load from a config file, but somewhere along the line we need hardcoded bits.
+      $sth->execute('ns1.example.com:hostmaster.example.com', 6, '10800:3600:604800:10800', 0, 0, 0, 86400);
+      $sth->execute('DOMAIN', 1, '192.168.4.2', 0, 0, 0, 7200);
+      $sth->execute('DOMAIN', 15, 'mx.example.com', 10, 0, 0, 7200);
+      $sth->execute('DOMAIN', 2, 'ns1.example.com', 0, 0, 0, 7200);
+      $sth->execute('DOMAIN', 2, 'ns2.example.com', 0, 0, 0, 7200);
+      $sth->execute('www.DOMAIN', 5, 'DOMAIN', 0, 0, 0, 7200);
+    }
+
+die "epic FAIL..  hahahah, just testing!\n";
+    # once we get here, we should have suceeded.
+    $dbh->commit;
+  }; # end eval
+
+  if ($@) {
+    my $msg = $@;
+    eval { $dbh->rollback; };
+    return ('FAIL',$msg);
+  } else {
+    return ('OK','OK');
+  }
+
+} # end addGroup()
+
+
 ## DNSDB::grpName()
 # Return the group name based on a group ID