#!/usr/bin/perl -w

# Drakwizard

# Copyright (C) 2003 Mandrakesoft
#
# Author: Antoine Ginies <aginies _ateuh _ 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::Pxe;
use lib qw(/usr/lib/libDrakX);
use strict;

use common;
use services;
use MDK::Wizard::Wizcommon;

my $wiz = new MDK::Wizard::Wizcommon;

#my $INSTALLDIR = "/var/install/pxe";
#my $IA64PATH = '/IA64PC';

my $TMPDIR = '/tmp';
my $HOSTNAME = chomp_(`hostname`);
my $TFTPDIR = "/var/lib/tftpboot";
my $CLIENTPATH = '/X86PC/linux';
my $X86 = $TFTPDIR . $CLIENTPATH;
my $IMGPATH = $X86 . '/images';
my $PXEHELP = $X86 . '/help.txt';
my $PXEMESSAGE = $X86 . '/messages';
my $PXEDEFAULT = $X86 . '/pxelinux.cfg/default';

my $CONF = '/etc/pxe.conf';
my $SYSLINUXPATH = '/usr/lib/syslinux/';
my $MEMDISK = $SYSLINUXPATH . '/memdisk';
my $XINETDDIR = "/etc/xinetd.d";

my $o = {
	 name => N("PXE Wizard (devel)"),
	 var => {
		 IMG => '/tmp/t.img',
		 PXENAME => '92',
		 DESCR => 'install 9.2',
		 IMGTOREMOVE => '',
		},
	 needed_rpm => [ 'pxe', 'tftp-server', 'xinetd' ],
	};

my %level = (
             1 => N("PXE - Set PXE server"),
             2 => N("add - Add image in PXE"),
	     3 => N("remove - remove image in PXE"),
#	     4 => N("Modify - Modify image in PXE"),
	    );
# to get menu entry
list_menu();

$o->{pages} = {
	       welcome => {
			   name => N("PXE wizard") . "\n\n" . N("Set a PXE server.") . "\n\n" . N("This wizard will help you configuring the PXE server. This configuration will provide a pxe service, and ability to add/remove/modify boot images."),
			   no_back => 1,
			   pre => sub {
			     $o->{var}{wiz_level} ||= 1;
			   },
			   post => sub {
			     if ($o->{var}{wiz_level} == 2) {
			       return 'addimg' }
			     elsif ($o->{var}{wiz_level} == 3) {
			       return 'removeimg' }
			     elsif ($o->{var}{wiz_level} == 4) {
			       return 'modifyimg' }
			   },
			   data => [
				    { label => N("Which operation:"), val => \$o->{var}{wiz_level}, list => [ keys %level ], format => sub { $level{$_[0]} } },
				   ],
			   no_back => 1,
			   next => 'pxeserver',
			  },
	       addimg => {
			  name => N("Add a boot Image") . "\n\n" . N("Path to image: Full path to image") . "\n\n" . N("PXE Name: Name in PXE menu (one word/number, no space)"),
			  data => [
				   { label => "PXE Name:", val => \$o->{var}{PXENAME} },
				   { label => "PXE Description:", val => \$o->{var}{DESCR} },
				   { label => "Full Path to image:", val => \$o->{var}{IMG} },
				  ],
			  post => \&test_data,
			  next => 'summaryadd',
			 },
	       removeimg => {
			     name => N("Remove a boot Image") . "\n\n" . N("Which one ?"),
			     data => [
				      { label => N("Which one:"), val => \$o->{var}{IMGTOREMOVE} },
				     ],
			     next => 'summaryremove',
			    },
	       modifyimg => {
			     name => N("Add Option to boot image") . "\n\n" . N("on which image ?"),
			     data => [
				      { label => "Wich one ?", val => \$o->{var}{PXENAME} },
				     ],
			     next => 'summarymodify',
			    },
	       pxeserver => {
			     name => N("Set PXE server") . "\n\n" . N("We will use a special dhcpd.conf"),
			     next => 'summaryserver',
			    },
	       error_dir => {
			     name => N('Error Should be a directory'),
			     next => '',
			    },
	       error_img => {
			     name => N('Please provide a Bootable image...'),
			     next => 'addimg',
			    },
	       error_name => {
			      name => N('Please provide a correct name in PXE entry (one word)'),
			      next => 'addimg',
			     },
	       simmilar_entry => {
				  name => N('Similar Name is already used in PXE menu entry') . "\n\n" . N("Please provide another Pxe Menu Name"),
				  next => 'addimg',
				 },
	       summaryserver => {
				 name => N('Now will prepapre all default files to get PXE server'),
				 data => [
					  { label => N("Tftp directory"), fixed_val => \$TFTPDIR },
					  { label => N("Image path"), fixed_val => \$IMGPATH },
					  { label => N("PXE \'default\' file "), fixed_val => \$PXEDEFAULT },
					  { label => N("PXE \'help\' file"), fixed_val => \$PXEHELP },
					 ],
				 post => \&do_it_pxe,
				 next => 'endserver',
				},
	       summarymodify => {
				 name => N('Now will Modify boot options in image'),
				 post => \&do_it_modify,
				 next => 'endmodify',
				},
	       summaryremove => {
				 name => N('Now will remove your PXE boot image'),
				 data => [
					  { label => N("PXE entry to remove:"), fixed_val => \$o->{var}{IMGTOREMOVE} },
					 ],
				 post => \&do_it_remove,
				 next => 'endremove',
				},
	       summaryadd => {
			      name => N('Now will add your PXe boot image'),
				 data => [
					  { label => "PXE Name:", fixed_val => \$o->{var}{PXENAME} },
					  { label => "PXE Description:", fixed_val => \$o->{var}{DESCR} },
					  { label => "Path to image:", fixed_val => \$o->{var}{IMG} },
					 ],
			      post => \&do_it_add,
			      next => 'endadd',
			     },
	       endadd => {
			     name => N('Congratulations'),
			     data => [ { label => N('The wizard successfully add a PXE image.') } ],
                             no_back => 1,
                             end => 1,
                             next => 0
			    },
	       endremove => {
			     name => N('Congratulations'),
			     data => [ { label => N('The wizard successfully remove a PXE image.') } ],
                             no_back => 1,
                             end => 1,
                             next => 0
			    },
	       endmodify => {
			     name => N('Congratulations'),
			     data => [ { label => N('The wizard successfully modify image(s).') } ],
                             no_back => 1,
                             end => 1,
                             next => 0
			    },
	       endserver => {
			     name => N('Congratulations'),
			     data => [ { label => N('The wizard successfully configured Your PXE server.') } ],
                             no_back => 1,
                             end => 1,
                             next => 0
			    },
	      };

sub memlinux_prep {
  if (!-f "$X86/memdisk") {
    cp_af($MEMDISK, $X86);
  }
}

sub test_data {
  $o->{var}{PXENAME} =~ /^\w+$/ or return 'error_name';
  if (!-f $o->{var}{IMG}) {
    return 'error_img'
  }
  if (any { /^$o->{var}{PXENAME}/ } cat_($PXEHELP)) {
    return 'simmilar_entry'
  }
}

sub enable_tftps {
  substInFile { s/disable.*/disable = no/ } "$XINETDDIR/tftp";
}

sub tftp_blksize {
  # $o should be with W or not N
  my ($o) = @_;
  if ($o =~ /W/) {
    substInFile { s/server_args.*/server_args = -r blksize -s $TFTPDIR/ } "$XINETDDIR/tftp";
  } else {
    substInFile { s/server_args.*/server_args = -s $TFTPDIR/ } "$XINETDDIR/tftp";
  }
}


sub crea_wdir {
  if (-e $TMPDIR) { rm_rf($TMPDIR);
	     }
  mkdir_p($TMPDIR);
}

sub add_in_default {
  my ($NAME) = @_;
  if (!any { /${NAME}_BEGIN/ } cat_($PXEDEFAULT)) {
    append_to_file($PXEDEFAULT ,<<EOF);
# TAG: ${NAME}_BEGIN
label $NAME
      KERNEL memdisk
      APPEND initrd=images/$NAME.img
# TAG: ${NAME}_END
EOF
  }
}

sub add_in_help {
  my ($NAME, $DESCR) = @_;
  append_to_file($PXEHELP ,<<EOF);
$NAME : $DESCR
EOF
}

sub add_img {
  my ($SIMG, $PXENAME) = @_;
  cp_af($SIMG, $IMGPATH . "/" . $PXENAME . '.img');
}

sub get_menu_entry {
  my ($NAME) = @_;
  my $conf = cat_($PXEDEFAULT) . "#EndOfFile";
  my ($NAME_bloc) = $conf =~ /(# TAG: ${NAME}_BEGIN.*?# TAG: ${NAME}_END)#EndOfFile/s or die "unable to find Menu entry section";
  print "$NAME_bloc";
}

sub remove_in_menu {
  my ($NAME) = @_;
  my $conf = cat_($PXEDEFAULT) . "#EndOfFile";
  my ($begin, $NAME_bloc, $end) = $conf =~ /(.*?)(# TAG: ${NAME}_BEGIN.*?# TAG: ${NAME}_END)\n(.*)#EndOfFile/s or die "unable to find Menu entry section";
  output($PXEDEFAULT, $begin, $end);
}

sub list_menu {
  foreach (cat_($PXEDEFAULT)) {
    my ($ent) = /\# TAG: (\w+)_BEGIN/;
    ! $ent or print "$ent\n";
  }
}

sub remove_in_help {
  my ($NAME) = @_;
  substInFile {
    s/^$NAME :.*//x;
  } $PXEHELP;
}

sub remove_img {
  my ($NAME) = @_;
  rm_rf($IMGPATH . "/" . $NAME . '.img');
}

sub default_pxe_menu {
  output($PXEHELP, <<EOF);
Available images are:
---------------------
local: local boot
EOF

  output($PXEDEFAULT, <<EOF);
PROMPT 1
DEFAULT local
DISPLAY messages
TIMEOUT 50
F1 help.txt
label local
      LOCALBOOT 0

EOF
}

sub default_pxe_messages {
  output($PXEMESSAGE, <<EOF);

                   Welcome to Mandrake Linux PXE Server
                                                       Pxelinux
              .              .-----------------------------------.
             /|\\            /    Press F1 for available images    \\
            /_|_\\           \\    Hosted by  $HOSTNAME
            \\ | /   _       /'-----------------------------------'
             \\|/   (')     /
              '.    U     /                                (O__
                 .   '.  /                 (o_  (o_  (0_   //\\
                 {o_   (o_  (o_  (o_  (o_  //\\  //\\  //\\  //  )
                 (')_  (`)_ (/)_ (/)_ (/)_ V_/_ V_/_ V_/_ V__/_
            ---------------------------------------------------------

 press F1 for help
EOF

}

sub do_it_add {
  return if $::testing;
  add_in_help($o->{var}{PXENAME}, $o->{var}{DESCR});
  add_in_default($o->{var}{PXENAME});
  add_img($o->{var}{IMG}, $o->{var}{PXENAME});
}

sub do_it_remove {
  return if $::testing;
  remove_in_menu($o->{var}{IMGTOREMOVE});
  remove_in_help($o->{var}{IMGTOREMOVE});
  remove_img($o->{var}{IMGTOREMOVE});
}

sub do_it_modify {
  return if $::testing;
}

sub do_it_pxe {
  return if $::testing;
  my $in = 'interactive'->vnew('su', 'PXE server');
  my $w = $in->wait_message(N("PXE server"), N("Configuring PXE server on your system..."));
  memlinux_prep();
  default_pxe_menu();
  default_pxe_messages();
  enable_tftps();
  tftp_blksize('N');
  map { system("service $_ restart"); $_ } qw(xinetd pxe);
  undef $w;
}


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

1;