better error management

[Net-Plesk.git] / lib / Net / Plesk / Response.pm

diff --git a/lib/Net/Plesk/Response.pm b/lib/Net/Plesk/Response.pm
index 58203f7..53342d5 100644 (file)
--- a/lib/Net/Plesk/Response.pm
+++ b/lib/Net/Plesk/Response.pm
@@ -38,13 +38,28 @@ sub new {
   bless($self, $class);
 
   my $data = shift;
-  $data =~ /^\<\?xml version=\"1.0\"\?\>(.*)$/s; 
-
-  my $xp = XML::XPath->new(xml => $1);
+  if ($data =~ /^\<\?xml version=\"1.0\"\?\>(.*)$/s){
+    $data=$1;
+  }else{
+    $data =~ s/[^\w\s]/ /g;  # yes, we lose stuff
+    $data = '<?xml version="1.0"?>' .
+      '<packet version="' . $self->{'version'} . '">' .
+      "<system><status>error</status><errcode>500</errcode>" .
+      "<errtext>Malformed Plesk response:" . $data .  "</errtext>".
+      "</system></packet>";
+  } 
+
+  my $xp = XML::XPath->new(xml => $data);
   my $nodeset = $xp->find('//result');
   foreach my $node ($nodeset->get_nodelist) {
     push @{$self->{'results'}}, XML::XPath::XMLParser::as_string($node);
   }
+  $nodeset = $xp->find('//system');
+  foreach my $node ($nodeset->get_nodelist) {
+    my $parsed = XML::XPath::XMLParser::as_string($node);
+    $parsed =~ s/\<(\/?)system\>/<$1result>/ig;
+    push @{$self->{'results'}}, $parsed;
+  }
 
   $self;
 }