package urpm::install; # $Id: install.pm 261966 2009-10-15 11:00:35Z tv $ use strict; use urpm; use urpm::msg; use urpm::util; # size of the installation progress bar my $progress_size = 45; eval { require Term::ReadKey; ($progress_size) = Term::ReadKey::GetTerminalSize(); $progress_size -= 35; $progress_size < 5 and $progress_size = 5; }; sub _hash_intersect_list { my ($h, $l) = @_; my %h; foreach (@$l) { exists $h->{$_} and $h{$_} = $h->{$_}; } \%h; } sub prepare_transaction { my ($_urpm, $set, $blists, $sources) = @_; my @blists_subset = map { +{ %$_, pkgs => _hash_intersect_list($_->{pkgs}, $set->{upgrade}) }; } @$blists; \@blists_subset, _hash_intersect_list($sources, $set->{upgrade}); } sub build_transaction_set_ { my ($urpm, $state, %options) = @_; if ($urpm->{parallel_handler} || !$options{split_length} || keys %{$state->{selected}} < $options{split_level}) { #- build simplest transaction (no split). $urpm->build_transaction_set(undef, $state, split_length => 0); } else { my $db = urpm::db_open_or_die_($urpm); my $sig_handler = sub { undef $db; exit 3 }; local $SIG{INT} = $sig_handler; local $SIG{QUIT} = $sig_handler; #- build transaction set... $urpm->build_transaction_set($db, $state, split_length => $options{split_length}, keep => $options{keep}); } } sub transaction_set_to_string { my ($urpm, $set) = @_; my $format_list = sub { int(@_) . '=' . join(',', @_) }; map { sprintf('remove=%s update=%s', $format_list->(@{$_->{remove} || []}), $format_list->(map { $urpm->{depslist}[$_]->name } @{$_->{upgrade} || []})); } @$set; } # install logger callback sub install_logger { my ($urpm, $type, $id, $subtype, $amount, $total) = @_; my $pkg = defined $id && $urpm->{depslist}[$id]; my $total_pkg = $urpm->{nb_install}; local $| = 1; if ($subtype eq 'start') { $urpm->{logger_progress} = 0; if ($type eq 'trans') { $urpm->{logger_count} ||= 0; my $p = N("Preparing..."); print $p, " " x (33 - length $p); } else { my $pname = $pkg ? $pkg->name : ''; ++$urpm->{logger_count} if $pname; my $cnt = $pname ? $urpm->{logger_count} : '-'; $pname ||= N("[repackaging]"); my $s = sprintf("%9s: %-22s", $cnt . "/" . $total_pkg, $pname); print $s; $s =~ / $/ or printf "\n%9s %-22s", '', ''; } } elsif ($subtype eq 'stop') { if ($urpm->{logger_progress} < $progress_size) { $urpm->{print}('#' x ($progress_size - $urpm->{logger_progress})); $urpm->{logger_progress} = 0; } } elsif ($subtype eq 'progress') { my $new_progress = $total > 0 ? int($progress_size * $amount / $total) : $progress_size; if ($new_progress > $urpm->{logger_progress}) { print '#' x ($new_progress - $urpm->{logger_progress}); $urpm->{logger_progress} = $new_progress; $urpm->{logger_progress} == $progress_size and print "\n"; } } } sub get_README_files { my ($urpm, $trans, $pkg) = @_; foreach my $file ($pkg->doc_files) { my ($kind) = $file =~ m!/README([^/]*)\.urpmi$! or next; my $valid; if ($kind eq '') { $valid = 1; } elsif ($kind eq '.install' && !$pkg->flag_installed) { $valid = 1; } elsif ($kind =~ /(.*)\.(upgrade|update)$/ && $pkg->flag_installed) { if (!$1) { $valid = 1; } else { my $version = $1; foreach my $i (0 .. $trans->NElements - 1) { $trans->Element_name($i) eq $pkg->name or next; my $vr = $trans->Element_version($i) . '-' . $trans->Element_release($i); if (URPM::ranges_overlap("== $vr", "< $version")) { $valid = 1; last; } } } } $valid and $urpm->{readmes}{$file} = $pkg->fullname; } } sub options { my ($urpm) = @_; ( excludepath => $urpm->{options}{excludepath}, excludedocs => $urpm->{options}{excludedocs}, repackage => $urpm->{options}{repackage}, post_clean_cache => $urpm->{options}{'post-clean'}, nosize => $urpm->{options}{ignoresize}, ignorearch => $urpm->{options}{ignorearch}, noscripts => $urpm->{options}{noscripts}, replacefiles => $urpm->{options}{replacefiles}, ); } #- install packages according to each hash (remove, install or upgrade). #- options: #- test, excludepath, nodeps, noorder (unused), delta, #- callback_inst, callback_trans, callback_report_uninst, #- post_clean_cache, verbose #- (more options for trans->run) #- excludedocs, nosize, noscripts, oldpackage, repackage, replacepkgs, justdb, ignorearch sub install { my ($urpm, $remove, $install, $upgrade, %options) = @_; $options{translate_message} = 1; my $db = urpm::db_open_or_die_($urpm, !$options{test}); #- open in read/write mode unless testing installation. my $trans = $db->create_transaction(); if ($trans) { sys_log("transaction on %s (remove=%d, install=%d, upgrade=%d)", $urpm->{root} || '/', scalar(@{$remove || []}), scalar(values %$install), scalar(values %$upgrade)); $urpm->{log}(N("created transaction for installing on %s (remove=%d, install=%d, upgrade=%d)", $urpm->{root} || '/', scalar(@{$remove || []}), scalar(values %$install), scalar(values %$upgrade))); } else { return N("unable to create transaction"); } $trans->set_script_fd($options{script_fd}) if $options{script_fd}; my ($update, @errors) = 0; my @produced_deltas; foreach (@$remove) { if ($trans->remove($_)) { $urpm->{debug} and $urpm->{debug}("trans: scheduling removal of $_"); } else { $urpm->{error}("unable to remove package " . $_); } } my @trans_pkgs; foreach my $mode ($install, $upgrade) { foreach (keys %$mode) { my $pkg = $urpm->{depslist}[$_]; $pkg->update_header($mode->{$_}, keep_all_tags => 1); if ($pkg->payload_format eq 'drpm') { #- handle deltarpms my $true_rpm = urpm::sys::apply_delta_rpm($mode->{$_}, "$urpm->{cachedir}/rpms", $pkg); if ($true_rpm) { push @produced_deltas, ($mode->{$_} = $true_rpm); #- fix path } else { $urpm->{error}(N("unable to extract rpm from delta-rpm package %s", $mode->{$_})); } } if ($trans->add($pkg, update => $update, $options{excludepath} ? (excludepath => [ split /,/, $options{excludepath} ]) : () )) { $urpm->{debug} and $urpm->{debug}( sprintf('trans: scheduling %s of %s (id=%d, file=%s)', $update ? 'update' : 'install', scalar($pkg->fullname), $_, $mode->{$_})); push @trans_pkgs, $pkg; } else { $urpm->{error}(N("unable to install package %s", $mode->{$_})); my $cachefile = "$urpm->{cachedir}/rpms/" . $pkg->filename; if (-e $cachefile) { $urpm->{error}(N("removing bad rpm (%s) from %s", $pkg->name, "$urpm->{cachedir}/rpms")); unlink $cachefile or $urpm->{fatal}(1, N("removing %s failed: %s", $cachefile, $!)); } } } ++$update; } if (!$options{nodeps} && (@errors = $trans->check(%options))) { } elsif (!$options{noorder} && (@errors = $trans->order)) { } else { $urpm->{readmes} = {}; my $index; my $fh; my $is_test = $options{test}; # fix circular reference #- assume default value for some parameter. $options{delta} ||= 1000; #- ensure perl does not create a circular reference below, otherwise all this won't be collected, # and rpmdb won't be closed: my ($callback_open_helper, $callback_close_helper) = ($options{callback_open_helper}, $options{callback_close_helper}); $options{callback_open} = sub { my ($_data, $_type, $id) = @_; $index++; $callback_open_helper and $callback_open_helper->(@_); $fh = urpm::sys::open_safe($urpm, '<', $install->{$id} || $upgrade->{$id}); $fh ? fileno $fh : undef; }; $options{callback_close} = sub { my ($urpm, undef, $pkgid) = @_; return unless defined $pkgid; $callback_close_helper and $callback_close_helper->($db, @_); get_README_files($urpm, $trans, $urpm->{depslist}[$pkgid]) if !$is_test; close $fh if defined $fh; }; #- ensure perl does not create a circular reference below, otherwise all this won't be collected, # and rpmdb won't be closed my ($verbose, $callback_report_uninst) = ($options{verbose}, $options{callback_report_uninst}); $options{callback_uninst} = sub { my ($_urpm, undef, undef, $subtype) = @_; if ($subtype eq 'start') { my ($name, $fullname) = ($trans->Element_name($index), $trans->Element_fullname($index)); my @previous = map { $trans->Element_name($_) } 0 .. ($index - 1); # looking at previous packages in transaction # we should be looking only at installed packages, but it should not give a different result if (member($name, @previous)) { $urpm->{log}("removing upgraded package $fullname"); } else { $callback_report_uninst and $callback_report_uninst->(N("Removing package %s", $fullname)); $urpm->{print}(N("removing package %s", $fullname)) if $verbose >= 0; } $index++; } }; if ($options{verbose} >= 0 && @trans_pkgs) { $options{callback_inst} ||= \&install_logger; $options{callback_trans} ||= \&install_logger; } @errors = $trans->run($urpm, %options); #- don't clear cache if transaction failed. We might want to retry. if (!@errors && !$options{test} && $options{post_clean_cache}) { #- examine the local cache to delete packages which were part of this transaction my $cachedir = "$urpm->{cachedir}/rpms"; my @pkgs = grep { -e "$cachedir/$_" } map { $_->filename } @trans_pkgs; $urpm->{log}(N("removing installed rpms (%s) from %s", join(' ', @pkgs), $cachedir)) if @pkgs; foreach (@pkgs) { unlink "$cachedir/$_" or $urpm->{fatal}(1, N("removing %s failed: %s", $_, $!)); } } if ($options{verbose} >= 0) { foreach (keys %{$urpm->{readmes}}) { $urpm->{print}("-" x 70 . "\n" . N("More information on package %s", $urpm->{readmes}{$_})); $urpm->{print}(scalar cat_utf8(($urpm->{root} || '') . $_)); $urpm->{print}("-" x 70); } } } unlink @produced_deltas; urpm::sys::may_clean_rpmdb_shared_regions($urpm, $options{test}); # explicitely close the RPM DB (needed for drakx -- looks like refcount has hard work): undef $db; undef $trans; @errors; } 1;