#!/usr/bin/env perl
# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
#-------------------------------------------------------
package xCAT_plugin::DNSsn;
use xCAT::Table;

use xCAT::Utils;

use xCAT::MsgUtils;
use Getopt::Long;

#-------------------------------------------------------

=head1 
  xCAT plugin package to setup of DNS 


#-------------------------------------------------------

=head3  handled_commands 

Check to see if on a Service Node
Check database to see if this node is a DNS server
Call  setup_DNS

=cut

#-------------------------------------------------------

sub handled_commands
{
    my $rc = 0;
    if (xCAT::Utils->isServiceNode())
    {
        my @nodeinfo   = xCAT::Utils->determinehostname;
        my $nodename   = pop @nodeinfo; # get hostname
        my @nodeipaddr = @nodeinfo;  # get ip addresses
        my $service    = "nameservers";

        $rc = xCAT::Utils->isServiceReq($nodename, $service, \@nodeipaddr);
        if ($rc == 1)
        {

            # service needed on this Service Node
            $rc = &setup_DNS($nodename);    # setup DNS
            if ($rc == 0)
            {
                xCAT::Utils->update_xCATSN($service);
            }
        }
    }
    return $rc;
}

#-------------------------------------------------------

=head3  process_request 

  Process the command

=cut

#-------------------------------------------------------
sub process_request
{
    return;
}

#-----------------------------------------------------------------------------

=head3 setup_DNS 

    Sets up Domain Name service  
	http://www.adminschoice.com/docs/domain_name_service.htm#Introduction

=cut

#-----------------------------------------------------------------------------
sub setup_DNS
{
    my ($nodename) = @_;

    # backup the original
    if ((-e "/etc/named.conf") && (!(-e "/etc/named.conf.ORIG")))
    {
        `cp /etc/named.conf /etc/named.conf.ORIG`;
    }

    # read DB for nodeinfo
    my $master;
    my $os;
    my $arch;
    my $retdata = xCAT::Utils->readSNInfo($nodename);
    if ($retdata->{'arch'})
    {    # no error
        $master = $retdata->{'master'};
        $os     = $retdata->{'os'};
        $arch   = $retdata->{'arch'};

        # build the named.conf file
        my $namedconfig = "/etc/named.conf";
        my @nameconfigtemplate;
        $nameconfigtemplate[0] = "options {directory \"/var/named\";\n";
        $nameconfigtemplate[1] =
          "     dump-file \"/var/named/data/cache_dump.db\";\n";
        $nameconfigtemplate[2] =
          "      statistics-file \"/var/named/data/named_stats.txt\";\n";
        $nameconfigtemplate[3] = "     forward only;\n";
        $nameconfigtemplate[4] = "     forwarders{$master;\n      };\n";
        $nameconfigtemplate[5] = "};\n";
        $nameconfigtemplate[6] = "\n";
        $nameconfigtemplate[7] = "controls {\n";
        $nameconfigtemplate[8] =
          "    inet 127.0.0.1 allow { localhost; } keys { rndckey; };\n";
        $nameconfigtemplate[9]  = "};\n";
        $nameconfigtemplate[10] = "\n";
        $nameconfigtemplate[11] = "zone \".\" IN {\n";
        $nameconfigtemplate[12] = "   type hint;\n";
        $nameconfigtemplate[13] = "   file \"named.ca\";\n";
        $nameconfigtemplate[14] = "};\n\n";
        $nameconfigtemplate[15] = "zone \"localdomain\" IN {\n";
        $nameconfigtemplate[16] = "    type master;\n";
        $nameconfigtemplate[17] = "    file \"localdomain.zone\";\n";
        $nameconfigtemplate[18] = "    allow-update { none; };\n";
        $nameconfigtemplate[19] = "};\n";
        $nameconfigtemplate[20] = "\n";
        $nameconfigtemplate[21] = "zone \"localhost\" IN {\n";
        $nameconfigtemplate[22] = "    type master;\n";
        $nameconfigtemplate[23] = "    file \"localhost.zone\";\n";
        $nameconfigtemplate[24] = "    allow-update { none; };\n";
        $nameconfigtemplate[25] = "};\n";
        $nameconfigtemplate[26] = "\n";
        $nameconfigtemplate[27] = "zone \"0.0.127.in-addr.arpa\" IN {\n";
        $nameconfigtemplate[28] = "    type master;\n";
        $nameconfigtemplate[29] = "    file \"named.local\";\n";
        $nameconfigtemplate[30] = "    allow-update { none; };\n";
        $nameconfigtemplate[31] = "};\n";
        $nameconfigtemplate[32] = "\n";
        $nameconfigtemplate[33] =
          "zone \"0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa\" IN { \n";
        $nameconfigtemplate[34] = "    type master; \n";
        $nameconfigtemplate[35] = "    file \"named.ip6.local\";\n";
        $nameconfigtemplate[36] = "    allow-update { none; };\n";
        $nameconfigtemplate[37] = "};\n";
        $nameconfigtemplate[38] = "\n";
        $nameconfigtemplate[39] = "zone \"255.in-addr.arpa\" IN {\n ";
        $nameconfigtemplate[40] = "    type master; \n";
        $nameconfigtemplate[41] = "    file \"named.broadcast\";\n";
        $nameconfigtemplate[42] = "    allow-update { none; };\n";
        $nameconfigtemplate[43] = "};\n";
        $nameconfigtemplate[44] = "\n";
        $nameconfigtemplate[45] = "zone \"0.in-addr.arpa\" IN {\n ";
        $nameconfigtemplate[46] = "    type master;\n ";
        $nameconfigtemplate[47] = "     file \"named.zero\";\n";
        $nameconfigtemplate[48] = "    allow-update { none; };\n";
        $nameconfigtemplate[49] = "};\n";
        $nameconfigtemplate[50] = "\n";
        $nameconfigtemplate[51] = "include \"/etc/rndc.key\";\n ";

        open(DNSCFG, ">$namedconfig")
          or xCAT::MsgUtils->message('S',
                                    "Cannot open $named.conf for DNS setup \n");
        print DNSCFG @nameconfigtemplate;
        close DNSCFG;

        # turn DNS on

        $cmd = "cp /etc/named.conf /var/named/chroot/etc";
        xCAT::Utils->runcmd($cmd, -1);
        if ($::RUNCMD_RC != 0)
        {
            xCAT::MsgUtils->message("S", "Error from $cmd");
            return 1;
        }
        $cmd = "chkconfig named on";
        xCAT::Utils->runcmd($cmd, -1);
        if ($::RUNCMD_RC != 0)
        {
            xCAT::MsgUtils->message("S", "Error from $cmd");
            return 1;
        }
        $cmd = "/etc/rc.d/init.d/named restart";
        xCAT::Utils->runcmd($cmd, -1);
        if ($::RUNCMD_RC != 0)
        {
            xCAT::MsgUtils->message("S", "Error from $cmd");
            return 1;
        }

    }
    else
    {    # error reading DB
        return 1;
    }
    return 0;
}
1;