Merge pull request #27 from jdsn/master

new ParseStatusXML function for fate312712
yast · Nov 19, 2012 · 5fc3146 · 5fc3146
2 parents d5c949e + e9d5523
commit 5fc3146
Showing 1 changed file with 123 additions and 0 deletions.
diff --git a/library/modules/RegistrationStatus.pm b/library/modules/RegistrationStatus.pm
@@ -24,6 +24,8 @@ package RegistrationStatus;
 use strict;
 use Data::Dumper;
 use XML::Simple;
+use XML::XPath;
+use POSIX qw/strftime/;
 
 use YaST::YCP qw(:DATA :LOGGING);
 
@@ -33,6 +35,8 @@ our %TYPEINFO;
 # for more datils about the file format
 my $reg_file = "/var/lib/suseRegister/registration-status.xml";
 
+my $productsd = "/etc/products.d";
+
 # return the default registration status file name
 BEGIN{ $TYPEINFO{RegFile} = ["function", "string"]; }
 sub RegFile {
@@ -65,5 +69,124 @@ sub ReadFile {
     return $data;
 }
 
+# parse registration-status.xml with XML::XPath (more solid and more predictable output than XML-Simple)
+BEGIN{ $TYPEINFO{ParseStatusXML} = ["function", ["map","any","any"], "string"]; }
+sub ParseStatusXML() {
+    my $self = shift;
+    my $file = shift || $reg_file;
+    my $status = {};
+
+    if ( ! -s $file || -d $file ) {
+        y2milestone("Missing or empty registration-status xml file: $file");
+        return { __parser_status => 10, __parser_message => "File missing or empty: $file" };
+    }
+
+    my $xp = XML::XPath->new(filename => $file);
+
+    my ($stSet, $st);
+    eval { $stSet = $xp->find('/status[1]') };
+
+    if ( $@ || ! ((defined $stSet) && ($stSet->size() == 1)) ) {
+        y2error("Could not parse the registration-status xml file: $file");
+        return { __parser_status => 11, __parser_message => "Could not parse status file." };
+    }
+    else {
+        y2milestone("Found status information in the registration-status xml file: $file");
+        $st = $stSet->pop();
+    }
+
+    $status->{generated} = $st->getAttribute('generated') || 0;
+    $status->{_generated_fmt} = strftime("%Y-%m-%d %H:%M:%S", localtime($status->{generated}));
+
+    my ($psSet, $ps);
+    eval { $psSet = $xp->find('/status[1]/productstatus') };
+    if ( $@ || ! ((defined $psSet) && ($psSet->size() > 0)) ) {
+      y2error("The status file ($file) does not contain productstatus information.");
+      return {__parser_status => 12, __parser_message => "No productstatus information found." };
+    }
+
+    $status->{__parser_status} = 0;
+
+    foreach my $n ($psSet->get_nodelist()) {
+        next unless defined $n;
+        my ($_product,$_version,$_arch,$_release,$_messageSet,$_message,$_subscriptionSet,$_subscription);
+        $_product = $n->getAttribute('product') || '';
+        $_version = $n->getAttribute('version') || '';
+        $_arch    = $n->getAttribute('arch')    || '';
+        $_release = $n->getAttribute('release') || '';
+        my $id_string = $_product.'-'.$_version.'-'.$_arch.'-'.$_release;
+        y2milestone("Processing product status for: $id_string");
+
+        eval { $_messageSet = $n->findnodes("message[1]") };
+        if ($@) {
+            y2error("Error while parsing the messasge of a product status.");
+        } else {
+            foreach my $_node ($_messageSet->get_nodelist()) {
+                $_message = $_node->string_value() if (defined $_node);
+            }
+        }
+        $_message ||= '';
+
+        eval { $_subscriptionSet = $n->findnodes("subscription[1]")};
+        if ( $@ || ! ((defined $_subscriptionSet) && ($_subscriptionSet->size() == 1)) ) {
+            y2milestone("The product '$_product' does not have a subscription. It may be fine though.");
+            $_subscription = undef;
+        } else {
+            my $s;
+            $s = $_subscriptionSet->pop();
+            $_subscription = {
+                status     => $s->getAttribute('status')     || '',
+                type       => $s->getAttribute('type')       || '',
+                expiration => $s->getAttribute('expiration') || ''
+            };
+            $_subscription->{_expiration_fmt} = ($_subscription->{expiration} =~ /^\d+$/) ?
+                strftime("%Y-%m-%d %H:%M:%S", localtime($_subscription->{expiration})) : '';
+        }
+
+        my $_prodinfo;
+        my @prodvals = qw(summary shortsummary vendor name version baseversion patchlevel release arch productline);
+        my $prodfile = "$productsd/$_product.prod";
+        if ( -f "$prodfile" && -s "$prodfile" ) {
+            my ($pxml, $pxmlparser, $pxmltree);
+            if ( open(PROD, "<", "$prodfile") ) {
+                $pxml = do { local $/; <PROD> };
+                close PROD;
+                $pxmlparser = XML::XPath::XMLParser->new(xml => $pxml);
+                eval { $pxmltree = $pxmlparser->parse(); };
+                if ($@) {
+                    y2error("Error: Could not parse the products file for the product: $_product");
+                } else {
+                    foreach my $val (@prodvals) {
+                        my ($valn, $valSet);
+                        eval { $valSet = $xp->findnodes("/product/$val", $pxmltree); };
+                        next unless ( (defined $valSet) && ($valSet->size() > 0) );
+                        $valn = $valSet->pop();
+                        $_prodinfo->{$val} = $valn->string_value() if (defined $valn);
+                    }
+                }
+            } else {
+                y2milestone("Product file for product $_product could not be opened.");
+            }
+        } else {
+            y2milestone("No product file for product $_product could be found.");
+        }
+
+        $status->{products}->{$id_string} = {
+            product   => $_product,
+            version   => $_version,
+            arch      => $_arch,
+            release   => $_release,
+            result    => $n->getAttribute('result')    || '',
+            errorcode => $n->getAttribute('errorcode') || '',
+            message   => $_message,
+        };
+        $status->{products}->{$id_string}->{subscription} = $_subscription if defined $_subscription;
+        $status->{products}->{$id_string}->{_productinfo} = $_prodinfo if defined $_prodinfo;
+    }
+
+    return $status;
+}
+
+
 1