Subversion Repositories camp_sysinfo_client_3

#!/usr/bin/env perl

package sysinfoconf;

#
# Can't use an undefined value as an ARRAY reference at /home/rodolico/scripts/camp_sysinfo_client_3/sysinfoconf.pm line 105, <> line 9.
#

our $VERSION = '1.1.2';
use warnings;
use strict;  

use Data::Dumper;
use File::Basename;

use Exporter;

our @ISA = qw( Exporter );
our @EXPORT = qw( $clientName $serialNumber $hostname @moduleDirs @scriptDirs 
                  $transports $sysinfo3 %sendTypes $binDir $modulesDir
                  $scriptsDir $confDir $binName $confName
                  &showConf &transportsToConfig &getAnswer &yesno  
                  &writeConfig &processParameters $TESTING &checkDirectoryExists
                  &runCommand &setDryRun &enableModules
                );

our $TESTING = 0;
our $dryRun;
our $clientName = '';
our $serialNumber = '';
our $hostname = '';
our @moduleDirs = ( '/opt/camp/sysinfo-client/modules', '/etc/camp/sysinfo-client/modules' );
our @scriptDirs = ( '/opt/camp/sysinfo-client/scripts', '/etc/camp/sysinfo-client/scripts' );
our $transports = {}; # holds transportation mechanisms
# the following are keys which are specific to the different transport channels

our $sysinfo3 = '/etc/camp/sysinfo-client/sysinfo-client.conf';

our $binDir = '/opt/camp/sysinfo-client';
our $modulesDir = $binDir . '/modules';
our $scriptsDir = $binDir . '/scripts';
our $confDir = '/etc/camp/sysinfo-client';

our $binName = 'sysinfo-client';
our $confName = 'sysinfo-client.conf';

sub setDryRun {
   $dryRun = shift;
}

our %sendTypes = ( 
                  'SendEmail' =>    { 'sendScript' => 'sendEmailScript',
                                      'keys' => 
                                      [
                                        'mailTo',
                                        'mailSubject',
                                        'mailCC',
                                        'mailBCC',
                                        'mailServer',
                                        'mailFrom',
                                        'logFile',
                                        'otherCLParams',
                                        'tls',
                                        'smtpUser',
                                        'smtpPass',
                                        'sendEmailScriptLocation'
                                      ],
                                    },
                  'HTTP Upload' =>  { 'sendScript' => 'upload_http',
                                      'keys' => 
                                      [
                                        'URL',
                                        'key for report',
                                        'key for date',
                                        'key for hostname',
                                        'key for client',
                                        'key for serial number'
                                      ]
                                    }
                );


sub enableModules {
   my $moduleDir = shift;
   my %modules;
   if ( opendir( my $dh, "$moduleDir" ) ) {
      %modules = map{ $_ => { 'enabled' => -x "$moduleDir/$_" } } 
                 grep { ! /^\./ } 
                 readdir $dh;
      closedir( $dh );
      foreach my $filename ( keys %modules ) {
         if ( open FILE,"<$moduleDir/$_" ) {
            my @descript = grep{ /^# Description: (.*)/ } <FILE>;
            close FILE;
            chomp @descript;
            $modules{$filename}{'description'} = $descript[0];
         } # if
      } # foreach
   } # if
} # enableModules

sub showConf {
   my $configuration = shift;
#   print Dumper( $configuration ); die;
   my $conf;
   $conf .= "\$clientName = '" .   ( defined( $$configuration{'clientName'}   ) ? $$configuration{'clientName'}   : '' ) . "';\n";
   $conf .= "\$serialNumber = '" . ( defined( $$configuration{'serialNumber'} ) ? $$configuration{'serialNumber'} : '' ) . "';\n";
   $conf .= "\$hostname = '" .     ( defined( $$configuration{'hostname'}     ) ? $$configuration{'hostname'}     : '' ) . "';\n";
   $conf .= "\@moduleDirs = ('" . join( "','", @{ $$configuration{ 'moduleDirs' } } ) . "');\n";
   $conf .= "\@scriptDirs = ('" . join( "','", @{ $$configuration{ 'scriptDirs' } } ) . "');\n";
   $conf .= &transportsToConfig( $$configuration{ 'transports' } );
   return $conf;
}

sub transportsToConfig {
   my $transports = shift;
   my $config;
   foreach my $priority ( sort { $a <=> $b } keys %$transports ) {
      my $name = $$transports{ $priority }{'-name-'};
      $config .= "# Tranport $name at priority $priority\n";
      my $thisOne = $$transports{ $priority };
      foreach my $key ( sort keys %$thisOne ) {
         $config .= "\$\$transports{$priority}{'$key'} = '$$thisOne{$key}';\n";
      } # foreach
   } # foreach
   return $config;
} # transportsToConfig

# prompt the user for a response, then allow them to enter it
# the first response is considered the default and is printed
# in all caps if more than one exists
# first answer is used if they simply press the Enter
# key. The response is returned all lower case if more than one
# exists.
# it is assumed 
sub getAnswer {
   my ( $prompt, @answers ) = @_;
   $answers[0] = '' unless defined( $answers[0] );
   my $default = $answers[0];
   my $onlyOneAnswer = scalar( @answers ) == 1;
   print $prompt . '[ ';
   $answers[0] = uc $answers[0] unless $onlyOneAnswer;
   print join( ' | ', @answers ) . ' ]: ';
   my $thisAnswer = <>;
   chomp $thisAnswer;
   $thisAnswer = $default unless $thisAnswer;
   return $thisAnswer;
}

sub yesno {
   my ( $prompt, $default ) = @_;
   $default = 'yes' unless $default;
   my $answer = &getAnswer( $prompt, $default eq 'yes' ? ('yes','no' ) : ('no', 'yes') );
   return lc( substr( $answer, 0, 1 ) ) eq 'y';
}

# runs a system command. Also, if in testing mode, simply shows what
# would have been done.
sub runCommand {
   while ( my $command = shift ) {
      if ( $dryRun ) {
         print "$command\n";
      } else {
         `$command`;
      }
   }
   return 1;
} # runCommand

# checks if a directory exists and, if not, creates it
my %directories; # holds list of directories already created so no need to do an I/O

sub checkDirectoryExists {
   my ( $filename,$mod,$owner ) = @_;
   $mod = "0700" unless $mod;
   $owner = "root:root" unless $owner;
   print "Checking Directory for $filename with $mod and $owner\n" if $TESTING > 2;
   my ($fn, $dirname) = fileparse( $filename );
   print "\tParsing out $dirname and $filename\n" if $TESTING > 2;
   return '' if exists $directories{$dirname};
   if ( -d $dirname ) {
      $directories{$dirname} = 1;
      return '';
   }
   if ( &runCommand( "mkdir -p $dirname", "chmod $mod $dirname", "chown $owner $dirname" ) ) {
      $directories{$dirname} = 1;
   }
   return '';   
}

sub writeConfig {
   my ( $filename,$content ) = @_;
   my $path;
   ($filename, $path ) = fileparse( $filename );

   my $return;
   `mkdir -p $path` unless -d $path;
   $filename = $path . '/' . $filename;
   if ( -e $filename ) {
      `cp $filename $filename.bak` if ( -e $filename );
      $return .= "Old config copied to $filename.bak\n";
   }
   if ( $dryRun ) {
      $return .= "Not writing to configuration file\n";
   } else {
      open CONF,">$filename" or die "Could not write to $filename: $!\n";
      print CONF $content;
      close CONF;
      `chmod 600 $filename`;
      $return .= "Configuration successfully written to $filename\n";
   }
   return $return;
}

sub processParameters {
   while ( my $parameter = shift ) {
      if ( $parameter eq '-v' ) {
         print "$VERSION\n";
         exit;
      }
   } # while
}



1;