#!/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);
    # WARNING: WON'T WORK IF EVER EMBEDDED
    $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 (unlikely) 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");
}