fork tool

1 files changed, 174 insertions, 0 deletions

diff --git a/mdkapplet-extended-maintenance-helper b/mdkapplet-extended-maintenance-helper
new file mode 100755
index 00000000..e3e19b63
--- /dev/null
+++ b/mdkapplet-extended-maintenance-helper
@@ -0,0 +1,174 @@
+#!/usr/bin/perl
+################################################################################
+# Mandriva Online                                                              # 
+#                                                                              #
+# Copyright (C) 2008 Mandriva                                             #
+#                                                                              #
+# Thierry Vignaud <tvignaud at mandriva dot com>                               #
+#                                                                              #
+# This program is free software; you can redistribute it and/or modify         #
+# it under the terms of the GNU General Public License Version 2 as            #
+# published by the Free Software Foundation.                                   #
+#                                                                              #
+# 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.   #
+################################################################################
+
+use strict;
+use lib qw(/usr/lib/libDrakX /usr/lib/libDrakX/drakfirsttime);
+use standalone; # for explanations
+use common;
+use run_program;
+use feature 'state';
+
+BEGIN { unshift @::textdomains, 'mdkonline' }
+
+use mygtk2 qw(gtknew); #- do not import gtkadd which conflicts with ugtk2 version
+use ugtk2 qw(:all);
+use urpm::cfg;
+use mdkonline;
+use mdkapplet_gui;
+use interactive;
+
+use XML::Simple;
+use URI::Escape;
+use LWP::UserAgent;         
+use HTTP::Request::Common;
+use HTTP::Request;
+use interactive::gtk; 
+
+my $version;
+foreach my $opt (@ARGV) {
+    if ($opt =~ /--(rpm-root|urpmi-root)=(.*)/) {
+        $::rpmdrake_options{$1}[0] = $2;
+    } else {
+        $version = $opt;
+    }
+}
+$version or die "usage: mdkapplet-restricted-helper <distrib_version>\n";
+
+# make it work on 2008.1:
+eval { interactive::gtk::add_padding(Gtk2::Label->new) };
+if ($@) {
+    *interactive::gtk::add_padding = sub { $_[0] };
+}
+
+get_restricted_authentication();
+
+ugtk2::exit(0);
+
+
+sub get_restricted_authentication() {
+    my $w = ugtk2->new(N("Adding an additional package medium"), width => $width + 20);
+    my ($password_w, $email_w, $password);
+
+    $password_w = gtknew('Entry');
+    $password_w->set_visibility(0);
+    $w->{real_window}->signal_connect(destroy => sub { $password = $password_w->get_text });
+
+    my $res =
+      fill_n_run_portable_dialog(
+          $w,
+          [
+              get_banner(N("Adding an additional package medium")),
+              gtknew('Label_Left', text => N("Please fill in your account ID to add an additional package medium"),
+                     @common),
+              gtknew('HButtonBox', layout => 'start', children_tight => [
+                  interactive::gtk::add_padding(
+                      new_link_button('https://my.mandriva.com/info', N("More information on your user account")))
+              ]),
+              gtknew('Table', col_spacings => 5, row_spacings => 5, children => [
+                  [ N("Your email"), $email_w = gtknew('Entry') ],
+                  [ N("Your password"), $password_w ],
+              ]),
+              gtknew('HButtonBox', layout => 'start', children_tight => [
+                  interactive::gtk::add_padding(
+                      new_link_button('https://my.mandriva.com/reset/password/', N("Forgotten password")))
+              ]),
+              create_okcancel($w, N("Next"), N("Cancel")),
+          ]);
+
+    my $email = $email_w->get_text;
+    if ($res) {
+        if ($email && $password) {
+            add_restricted_medium($email, $password);
+        } else {
+            interactive->vnew->ask_warn(N("Error"), N("Password and email cannot be empty."));
+            goto &get_restricted_authentication;
+        }
+    } else {
+        ugtk2::exit(0);
+    }
+}
+
+my $error;
+sub add_restricted_medium {
+    my ($email, $password) = @_;
+
+    my $res =  get_from("https://my.mandriva.com/rest/authenticate",
+                        [ 'username', $email, 'password', $password,
+                          'return', 'userdata' ]);
+    my $ref = xml2perl($res);
+
+    if ($ref->{code} != 0) {
+        my $in = interactive->vnew;
+        $in->ask_warn(N("Error"), N("An error occurred") . "\n" . $ref->{message});
+        goto &get_restricted_authentication;
+    } elsif ($ref->{data}{'can-access-restricted-repositories'} ne 'YES') {
+        no_rights_dialog();
+    } else {
+        $error = 0;
+        my $arch = urpm::cfg::get_arch();
+        actually_add_restricted_medium($ref, $password, $arch)
+          or adding_media_failed();
+        # FIXME: is not enough if we ever support sparc64, ppc64 and the like:
+        if ($arch eq 'x86_64') {
+            actually_add_restricted_medium($ref, $password, 'i586')
+              or adding_media_failed();
+        }
+        if (!$error) {
+            #interactive->vnew->ask_okcancel(N("Error"), N("An error occurred while adding medium"));
+            my $w = ugtk2->new(N("Successfully added media %s.", 'Restricted'), grab => 1);
+            $w->_ask_okcancel(N("Successfully added media %s.", 'Restricted'), N("Ok"), undef);
+            ugtk2::main($w);
+        }
+    }
+}
+
+sub no_rights_dialog() {
+    my $w = ugtk2->new(N("Adding an additional package medium"), width => $width + 20);
+
+    fill_n_run_portable_dialog(
+          $w,
+          [
+              get_banner(N("Adding an additional package medium")),
+              gtknew('Label_Left', text => N("Your Mandriva account does not have Powerpack download subscription enabled."),
+                     @common),
+              gtknew('HButtonBox', layout => 'start', children_tight => [
+                  interactive::gtk::add_padding(
+                      new_link_button('https://my.mandriva.com/powerpack/', N("More Information")))
+              ]),
+              create_okcancel($w, N("Close"), undef),
+          ]);
+}
+
+sub adding_media_failed {
+    $error = 1;
+    interactive->vnew->ask_warn(N("Error"), N("An error occurred while adding medium"));
+    goto &get_restricted_authentication;
+}
+
+sub actually_add_restricted_medium {
+    my ($ref, $password, $arch) = @_;
+    $password = uri_escape($password);
+    my @options = ({ sensitive_arguments => 1 }, 'urpmi.addmedia', '--xml-info', 'always');
+    my $uri = "https://" . uri_escape($ref->{data}{email}) . ":$password\@dl.mandriva.com/rpm/comm/$version/";
+    run_program::raw(@options, "Restricted $arch " . int(rand(100000)), "$uri$arch") or return 0;
+    run_program::raw(@options, '--update', "Restricted Updates $arch " . int(rand(100000)), "${uri}updates/$arch");
+}