[freeside-commits] freeside/FS/FS/part_export shellcommands.pm, 1.57, 1.58

[Erik Levinson]

Update of /home/cvs/cvsroot/freeside/FS/FS/part_export
In directory wavetail.420.am:/tmp/cvs-serv29101/FS/FS/part_export

Modified Files:
	shellcommands.pm 
Log Message:
shellcommands.pm: move to Net::OpenSSH and add ignored_errors option, RT831

Index: shellcommands.pm
===================================================================
RCS file: /home/cvs/cvsroot/freeside/FS/FS/part_export/shellcommands.pm,v
retrieving revision 1.57
retrieving revision 1.58
diff -u -w -d -r1.57 -r1.58
--- shellcommands.pm	10 Sep 2010 19:31:15 -0000	1.57
+++ shellcommands.pm	24 Jul 2011 16:11:50 -0000	1.58
@@ -80,6 +80,9 @@
                              'Radius group mapping to reason (via template user)',
 			    type  => 'textarea',
 			  },
+  'ignored_errors' => { label   => 'Regexes of errors to ignore, separated by newlines',
+                        type    => 'textarea'
+                      },
 #  'no_queue' => { label => 'Run command immediately',
 #                   type  => 'checkbox',
 #                },
@@ -350,6 +353,7 @@
     host          => $self->machine,
     command       => $command_string,
     stdin_string  => $stdin_string,
+    ignored_errors => $self->option('ignored_errors') || '',
   );
 
   if($self->option($action . '_no_queue')) {
@@ -440,6 +444,7 @@
     host          => $self->machine,
     command       => $command_string,
     stdin_string  => $stdin_string,
+    ignored_errors => $self->option('ignored_errors') || '',
   );
 
   if($self->option('usermod_no_queue')) {
@@ -465,8 +470,22 @@
 }
 
 sub ssh_cmd { #subroutine, not method
-  use Net::SSH '0.08';
-  &Net::SSH::ssh_cmd( { @_ } );
+  use Net::OpenSSH;
+  my $opt = { @_ };
+  my $ssh = Net::OpenSSH->new($opt->{'user'}.'@'.$opt->{'host'});
+  die "Couldn't establish SSH connection: ". $ssh->error if $ssh->error;
+  my ($output, $errput) = $ssh->capture2($opt->{'command'});
+  die "Error running SSH command: ". $ssh->error if $ssh->error;
+  if ($errput && $opt->{'ignored_errors'} && length($opt->{'ignored_errors'})) {
+    my @ignored_errors = split('\n',$opt->{'ignored_errors'});
+    foreach my $ignored_error ( @ignored_errors ) {
+        $errput =~ s/$ignored_error//g;
+    }
+    chomp($errput);
+  }
+  die $errput if $errput;
+  die $output if $output;
+  '';
 }
 
 #sub shellcommands_insert { #subroutine, not method