add Utils.pm for common functions

2 files changed, 118 insertions, 0 deletions

diff --git a/MANIFEST b/MANIFEST
index 34efb36..ef127c6 100644
--- a/MANIFEST
+++ b/MANIFEST
@@ -3,6 +3,7 @@ lib/MDV/Distribconf/Build.pm
 lib/MDV/Distribconf.pm
 lib/MDV/Distribconf/MediaCFG.pm
 lib/MDV/Distribconf/Checks.pm
+lib/MDV/Distribconf/Utils.pm
 bin/checkdistrib
 Makefile.PL
 MANIFEST
diff --git a/lib/MDV/Distribconf/Utils.pm b/lib/MDV/Distribconf/Utils.pm
new file mode 100644
index 0000000..e1120b6
--- /dev/null
+++ b/lib/MDV/Distribconf/Utils.pm
@@ -0,0 +1,117 @@
+package MDV::Distribconf::Utils;
+
+use strict;
+use warnings;
+use MDV::Packdrakeng;
+use Digest::MD5;
+
+our ($VERSION) = (qq$Revision$ =~ /(\d+)/)[0];
+
+=head1 NAME
+
+MDV::Distribconf::Utils
+
+=head1 DESCRIPTION
+
+Contains basic functions use by Distribconf
+
+=head1 FUNCTIONS
+
+=head2 hdlist_vs_dir($hdlistfile, @dirs)
+
+Return two array ref about rpm include only in hdlist or in directories
+
+=cut
+
+sub hdlist_vs_dir {
+    my ($hdlist, @dir) = @_;
+    my (@only_pack, @only_dir);
+    my @rpms;
+    foreach my $dir (@dir) {
+        push(@rpms, map { m:.*/+(.*): ; $1 } glob("$dir/*.rpm"));
+    }
+    @rpms = sort { $b cmp $a } @rpms;
+    if (my $pack = MDV::Packdrakeng->open(archive => $hdlist)) {
+        my (undef, $files, undef) = $pack->getcontent();
+        my @hdrs = sort { $b cmp $a } map { "$_.rpm" } @{$files || []};
+        my ($r, $h) = ("", "");
+        do {
+            my $comp = $r cmp $h;
+            # print "< $r - $h > $comp\n";
+            if ($comp < 0) { push(@only_pack, $h); }
+            elsif ($comp > 0) { push(@only_dir, $r); }
+
+            if ($comp <= 0) {
+                $h = shift(@hdrs) || "";
+            } 
+            if ($comp >= 0) {
+                $r = shift(@rpms) || "";
+            }
+        } while (scalar(@rpms) || scalar(@hdrs));
+    } else {
+        return;
+    }
+    return (\@only_pack, \@only_dir);
+}
+
+=head2 checkmd5($md5file, @files)
+
+Return an array ref to unsync file found and a hashref containing
+files and their found md5.
+
+=cut
+
+sub checkmd5 {
+    my ($md5file, @files) = @_;
+    my %foundmd5;
+    foreach my $file (@files) {
+        my ($basename) = $file =~ m:.*/+([^/]*)$:; #: vi syntax coloring
+        if (open(my $hfile, "<", $file)) {
+            my $ctx = Digest::MD5->new;
+            $ctx->addfile($hfile);
+            close($hfile);
+            $foundmd5{$basename} = $ctx->hexdigest;
+        } else {
+            $foundmd5{$basename} = '';
+        }
+    }
+    open(my $hmd5, "< $md5file") or return([ keys %foundmd5 ], \%foundmd5);
+    my %md5;
+    while (<$hmd5>) {
+        chomp;
+        s/#.*//;
+        /^(.{32})  (.*)/ or next;
+        $md5{$2} = $1;
+    }
+    close($hmd5);
+    my @badfiles = grep { $foundmd5{$_} ne ($md5{$_} || '') } keys %foundmd5;
+    return (\@badfiles, \%foundmd5);
+}
+
+1;
+
+__END__
+
+=head1 AUTHOR
+
+Olivier Thauvin <nanardon@mandriva.org>
+
+=head1 LICENSE AND COPYRIGHT
+
+(c) 2005 Olivier Thauvin ; (c) 2005, 2006 Mandriva
+
+This program is free software; you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation; either version 2, or (at your option)
+any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+=cut