#!/usr/bin/perl

# Drakwizard

# Copyright (C) 2002, 2003 Mandrakesoft
#
# Authors: Arnaud Desmons <adesmons@mandrakesoft.com>
#          Florent Villard <warly@mandrakesoft.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

package MDK::Wizard::Dhcp;
use strict;

use common;
use MDK::Wizard::Varspaceval;
use MDK::Wizard::Wizcommon;

my $wiz = new MDK::Wizard::Wizcommon;
my $wiz_ip_server = $wiz->{net}->itf_get("IPADDR");
my $wiz_tftpserverip = $wiz_ip_server;
my $d = $4 if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
my $s = "$1.$2.$3" if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;

my $o = {
    name => N("DHCP Wizard"),
	var => {
	    ip1 => '',
	    ip2 => '',
		pxe => '1',
	    wiz_authoritative => 0,
	    einterface => $wiz->{net}->default_itf
	},
	needed_rpm => [ 'dhcp-server' ],
	defaultimage => "$ENV{__WIZ_HOME__}dhcp_wizard/images/DHCP.png"
    };

$o->{pages} = {
		  welcome => {
		      name => N("DHCP Wizard") . "\n\n\n" . N("DHCP is a service that automatically assigns networking addresses to your workstations.") . "\n\n\n" . N("This wizard will help you configuring the DHCP services of your server."),
		      post => sub { $wiz->check_dhcp },
		  no_back => 1,
	          next => 'ip_range'
	      },
	      ip_range => {
	          name => N("Range of addresses used by dhcp") . "\n\n\n" . N("Select the range of addresses assigned to the workstations by the DHCP service; unless you have special needs, you can safely accept the proposed values. (ie: 192.168.100.20 192.168.100.40)") . "\n\n" . N("If You want to enable PXE for your dhcp server please check the box."),
                  pre => sub {
			$o->{var}{ip1} ||= compute_range();
			$o->{var}{ip2} ||= compute_range2();
		  },
		  post => \&check,
                  data => [
			   { label => N("Lowest IP Address:"),  val => \$o->{var}{ip1} },
			   { label => N("Highest IP Address:"), val => \$o->{var}{ip2} },
			   { label => N("Enable PXE:"), type => 'bool', val => \$o->{var}{pxe} },
		      ],
	          next => 'summary'
	      },
	      interface => {
	          name => N("Interface the dhcp server must listen to"),
                  pre => sub {
			$o->{var}{interface} ||= $wiz->{net}->default_itf
		  },
                  data => [
		      { list => [ keys %{$wiz->{net}{itf}} ],  val => \$o->{var}{interface} },
		      ],
	          next => 'summary'
	      },
	       dhcp_warning => {
		name => N("Warning") . "\n\n" . N("You are in dhcp, server may not work with your configuration."),
		ignore => 1,
	        next => 'ip_range'
	      },
	      ip_range_error => {
		name => N("Error") . "\n\n" . N("The IP range specified is not correct."),
		ignore => 1,
	        next => 'ip_range'
	      },
	      ip_range_warning => {
		name => N("Warning") . "\n\n" . N("The IP range specified is not in server address range."),
		ignore => 1,
	        next => 'summary'
	      },
	      server_in_range => {
		name => N("Error") . "\n\n" . N("The IP of the server must not be in range."),
		ignore => 1,
	        next => 'ip_range'
	      },
	      summary => {
			  name => N("Configuring the DHCP Server") . "\n\n" . N("The wizard collected the following parameters needed to configure your DHCP service:"),
			  pre => sub {
			    $o->{var}{pxeornot} = $o->{var}{pxe} ? N("enabled") : N("disabled");
			  },
			  data => [
				   { label => N("Lowest IP Address:"), fixed_val => \$o->{var}{ip1} },
				   { label => N("Highest IP Address:"), fixed_val => \$o->{var}{ip2} },
				   { label => N("Interface:"), fixed_val => \$o->{var}{interface} },
				   { label => N("Enable PXE:"), fixed_val => \$o->{var}{pxeornot} },
				  ],
			  post => \&do_it,
			  next => 'end'
			 },
	      end => {
	        name => N("Congratulations") . "\n\n" . N("The wizard successfully configured the DHCP services of your server."),
		end => 1,
	        next => 0
	  },
};

sub compute_range {
    my $n;
    if ($d <= 64) { $n = "65" }
    elsif ($d <= 128) { $n = "129" }
    else { $n = "1" }
    "$s.$n";
}

sub compute_range2 {
    my $n;
    if ($d <= 128) { $n = "254" }
    elsif ($d > 192) { $n = "192" }
    else { $n = "128" }
    "$s.$n";
}

sub check {
    my $check_ip = sub { return $_[0] < 0 || $_[0] > 255 ? 0 : 1 };
    my $r1_trunc = "$1.$2.$3" if $o->{var}{ip1} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    foreach ($1, $2, $3, $4) { $check_ip->($_) or return 'ip_range_error' }
    my $r2_trunc = "$1.$2.$3" if $o->{var}{ip2} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    foreach ($1, $2, $3, $4) { $check_ip->($_) or return 'ip_range_error' }
    my $d1 = $4 if $o->{var}{ip1} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $d2 = $4 if $o->{var}{ip2} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $s_trunc = "$1.$2.$3" if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $ds = $4 if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    if (!$r1_trunc) { standalone::explanations("DHCP wizard : incorrect adress range 1"); return 'ip_range_error' }
    if (!$r2_trunc) { standalone::explanations("DHCP wizard : incorrect adress range 2"); return 'ip_range_error' }
    if ($r1_trunc ne $s_trunc || $r2_trunc ne $s_trunc) {
	standalone::explanations("DHCP wizard : range not in network");
	return 'ip_range_warning';
    }
    if (!$d1 || !$d2 || $d1 > $d2) { standalone::explanations("DHCP wizard : bad range"); return 'ip_range_error' }
    if ($ds >= $d1 && $ds <= $d2) { standalone::explanations("DHCP wizard : server in range"); return 'server_in_range' }
    return 'interface' if keys %{$wiz->{net}{itf}} > 1;
    0
}

sub do_it {
    $::testing and return;
    my $wiz_domain_name = $wiz->{net}->network_get("DOMAINNAME"); 
    my $wiz_host_name = $wiz->{net}->network_get("HOSTNAME"); 
    my $wiz_gateway = $wiz->{net}->network_get("GATEWAY"); 
    my $wiz_dns = $wiz->{net}->network_get("dnsServer");
    if ($wiz_dns eq '127.0.0.1') {
    	$wiz_dns = $wiz_ip_server
    }
    if (!$wiz_gateway) {
        my $t = `LC_ALL=C /sbin/ip route list scope global`;
	($wiz_gateway) = $t =~ /default via (\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}) dev/;
    }
    my $wiz_ip_net = "$1.$2.$3.0" if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $err = check();
    return $err if $err != 0;
    my $wiz_ip_range1 = $o->{var}{ip1};
    my $wiz_ip_range2 = $o->{var}{ip2};
    my $wiz_ip_netmask = $wiz->{net}->itf_get("NETMASK");
    my $wiz_device = $o->{var}{interface};
# patch to rewrite when got new file about dhcp with INTERFACES value
# currently, I put the device to configure as dhcp server
# in /etc/sysconfig/dhcpd

#[ -f /etc/sysconfig/dhcpd ] && cp -f /etc/sysconfig/dhcpd /var/tmp/wiz_bck/orig/dhcpd
    my $file = "/etc/sysconfig/dhcpd";
    -f $file and MDK::Common::cp_af($file, $file.".orig");
    standalone::explanations("now patching etc/sysconfig/dhcpd");
    if (!`grep INTERFACES $file`) {
	my  $tmp = `/bin/mktemp /tmp/Dhcpconf.XXXXXX`	or die "can't make a temp file: $!";
	open(NEW, "> $tmp")				or die "can't open $tmp: $!";
	open(OLD, "< $file")				or die "can't open default: $!";
	while (<OLD>) {
	    if (m|INTERFACE|) {
		print NEW "# $_";
	    }
	    print NEW $_;
	}
	MDK::Common::append_to_file($tmp, "\n# Added by drakwizard\nINTERFACES=$wiz_device");
	close(OLD);
	close(NEW);
	chomp($tmp);
	system("mv $tmp $file");
    }
    $file = "/etc/dhcpd.conf";
    -f $file and MDK::Common::cp_af($file, $file.".orig");
    output($file, map {
	s|__hname__|$wiz_host_name|g;
	s|__dns__|$wiz_dns|g;
	s|__net__|$wiz_ip_net|g;
	s|__ip__|$wiz_ip_server|g;
	s|__mask__|$wiz_ip_netmask|g;
	s|__rng1__|$wiz_ip_range1|g;
	s|__rng2__|$wiz_ip_range2|g;
	s|__dname__|$wiz_domain_name|g;
	s|__gateway__|$wiz_gateway|g;
	s|__tftpserverip__|$wiz_tftpserverip|g;
	$_;
    } cat_("__WIZ_HOME__/dhcp_wizard/scripts/dhcpd.conf.default"));
    standalone::explanations("$file: hname = $wiz_host_name, net = $wiz_ip_net, ip = $wiz_ip_server,
mask = $wiz_ip_netmask, rng1 = $wiz_ip_range1, rng2 = $wiz_ip_range2, dname = $wiz_domain_name");
    MDK::Common::touch("/var/dhcpd/dhcpd.leases");
# modifying webmin config
    $file="/etc/webmin/dhcpd/config";
    if (-f $file) {
	my %mdk = MDK::Wizard::Varspaceval->get($file);
	$mdk{lease_file} = "/var/dhcpd/dhcpd.leases";
	$mdk{interfaces} = $wiz_device;
	standalone::explanations("$file: lease_file = $mdk{lease_file}, interfaces = $mdk{interfaces}");
	MDK::Wizard::Varspaceval->commit($file, \%mdk);
	!$o->{var}{wiz_authoritative} and output($file, map {
	    s|^\s*not\s*authoritative.*|#$&|i;
	    $_
        } cat_("/etc/dhcpd.conf"));
    }

    if ($o->{var}{pxe} == '0') {
      substInFile {
	s/#\s+deny members of "PXE"/deny members of "PXE"/;
      } "/etc/dhcpd.conf";
    }

    if (services::is_service_running('dhcpd')) {
	services::restart('dhcpd')
    } else {
        services::start('dhcpd')
    }
    10;
}

sub new {
    my ($class, $conf) = @_;
    bless {
	    o   => $o,
       }, $class;
}

1;