Changeset 59 for devel/dploy-pxe/bin/dploy-add2pxe

Timestamp:

06/02/08 19:14:30 (4 years ago)

Author:

bruno

Message:

• pod content for dploy-add2pxe
• type is changed to lowercase before usage
• Fix various typos ($url inexistent, ...)
• use a new $destdir var to designate the dir in which place tftpboot files

File:

• devel/dploy-pxe/bin/dploy-add2pxe (modified) (5 diffs)

devel/dploy-pxe/bin/dploy-add2pxe

@@ -18 +18 @@
 use Dploy::Base;
 
-my %opts;
-my $url = undef;
-my $ksdevice = "eth0";
-my $prefix = undef;
+=pod
+
+=head1 NAME
+
+dploy-add2pxe - Update PXE configuration file for dploy.org
+
+=head1 DESCRIPTION
+
+dploy-add2pxe helps you keeping your PXE configuration up to date 
+
+=head1 SYNOPSIS
+
+dploy-add2pxe -t mondorescue|linuxcoe -m MAC-address -u ISO-image-URL
+
+dploy-add2pxe --type mondorescue|linuxcoe --mac MAC-address --url ISO-image-URL
+
+=head1 OPTIONS
+
+=over 4
+
+=item B<-h|--help>
+
+Print a brief help message and exits.
+
+=cut
+
+
+my %opts;                   # GetOpt Hash
+my $url = undef;            # url of the boot ISO image
+my $ksdevice = "eth0";      # NIC device to boot on
+my $prefix = undef;         # Prefix of the MondoRescue image
+my $cliparams;              # Command Line parames of the MondoRescue image
+my $destdir = undef;        # Destination driectory for tftp files
 
 pb_syntax_init("dploy-add2pxe Version PBVER-PBREV\n");
@@ -35 +64 @@
 }
 
-pb_conf_add("PBCONF/dploy.conf");
+# type is always lowercase
+$opts{'type'} = lc($opts{'type'});
+
+my $dploydir = "PBCONF";
+pb_conf_add("$dploydir/dploy.conf");
 
 # Root of the PXE setup
-my $pxe=pb_conf_get("pathpxe");
+my ($pxe)=pb_conf_get("pathpxe");
 my $pathpxe = $pxe->{"default"};
 
@@ -52 +85 @@
 
 # Get the ID given by LinuxCOE/MondoRescue to prepare kernel/initrd/ks
-my $iso=basename($url);
+my $iso=basename($opts{'url'});
 my $id=$iso;
 $id =~ s/img([0-9]+)\.iso/$1/;
 my $k="k-$id";
 my $i="i-$id";
-my $ks=$url;
+my $ks=$opts{'url'};
 $ks =~ s/img([0-9]+)\.iso/ks$1.cfg/;
 
 # mount the ISO to get kernel and initrd
 mkdir "$mnt",0755;
-if ($opts{'type'} == "LinuxCOE") {
+if ($opts{'type'} =~ /linuxcoe/) {
     my $bn = basename($mrpath);
     unlink("$ENV{'PBTMP'}/$bn");
-    pb_system("cd \$PBTMP ; wget $url","Downloading $url");
+    pb_system("cd \$PBTMP ; wget $opts{'url'}","Downloading $opts{'url'}");
     pb_system("sudo /bin/mount -o loop $ENV{'PBTMP'}/$bn $mnt","Mounting $bn");
-} elsif ($opts{'type'} == "MondoRescue") {
+} elsif ($opts{'type'} =~ /mondorescue/) {
     pb_system("sudo /bin/mount -t nfs $mrserver:$mrpath $mnt2","Mountig NFS $mrserver:$mrpath");
     pb_system("sudo /bin/mount -o loop $mnt2/$prefix-1.iso $mnt","Mounting LOOP $mnt2/$prefix-1.iso");
@@ -76 +109 @@
 
 # Install script will have to check (and create if not existing) /tftpboot/tmp
-copy("$mnt/isolinux/vmlinuz","$pathpxe/../tmp/$k") or die "Copy failed: $!";
-copy("$mnt/isolinux/initrd.img","$pathpxe/../tmp/$i") or die "Copy failed: $!";
+# $pathpxe is normaly setup correctly by package install
+$destdir = "$pathpxe/../$opts{'type'}";
+copy("$mnt/isolinux/vmlinuz","$destdir/$k") or die "Copy of $mnt/isolinux/vmlinuz to $destdir/$k failed: $!";
+copy("$mnt/isolinux/initrd.img","$destdir/$i") or die "Copy of $mnt/isolinux/initrd.img to $destdir/$i failed: $!";
 
 # Get the Command Line parameters for MondoRescue restoration
-my $cliparams;
-open(ISOLINUX, "$mnt/isolinux.cfg") || die "Unable to open $mnt/isolinux.cfg for writing";
-while ( <ISOLINUX> ) {
-    if (/label nuke/) {
-        $cliparams = <ISOLINUX>;        # kernel line
-        $cliparams = <ISOLINUX>;        # append line
+if ($opts{'type'} =~ /MondoRescue/i) {
+    open(ISOLINUX, "$mnt/isolinux.cfg") || die "Unable to open $mnt/isolinux.cfg for writing";
+    while ( <ISOLINUX> ) {
+        if (/label nuke/) {
+            $cliparams = <ISOLINUX>;        # kernel line
+            $cliparams = <ISOLINUX>;        # append line
+        }
     }
+    $cliparams =~ s/initrd=[A-z.] /initrd=mondorescue\/$i /;
+    close ISOLINUX;
 }
-$cliparams =~ s/initrd=[a-z.] /initrd=initrd=mondorescue\/$i /;
 
-close ISOLINUX;
 # All done with ISO image... umounting
 pb_system("sudo /bin/umount $mnt","Umount ISO image");
@@ -110 +146 @@
 
 EOF
-close(PXE);
 
 # Customize pxe file depending on Job type submited
-open(PXE, ">> $pathpxe/01-$mac") || die "Unable to open $pathpxe/01-$mac for writing";
-if ($opts{'type'} == "LinuxCOE") {
-    print PXE "label linuxcoe\n";
-    print PXE "  kernel linuxcoe/$k\n";
-    print PXE "  append ksdevice=$ksdevice ks=$ks initrd=linuxcoe/$i text\n";
+if ($opts{'type'} =~ /linuxcoe/) {
+    print PXE << "EOF";
+label linuxcoe
+  kernel linuxcoe/$k
+  append ksdevice=$ksdevice ks=$ks initrd=linuxcoe/$i text
+EOF
 } else {
-    print PXE "label mondorescue\n";
-    print PXE "  kernel mondorescue/$k\n";
-    print PXE "  append $cliparams pxe prefix=$prefix ipconf=$ksdevice:dhcp nfsmount=$mrserver:$mrpath\n";
+    print PXE << "EOF";
+label mondorescue
+  kernel mondorescue/$k
+  append $cliparams pxe prefix=$prefix ipconf=$ksdevice:dhcp nfsmount=$mrserver:$mrpath
+EOF
 }
 close(PXE);