# -*- Mode: perl; indent-tabs-mode: nil -*-
#
# The contents of this file are subject to the Mozilla Public
# License Version 1.1 (the "License"); you may not use this file
# except in compliance with the License. You may obtain a copy of
# the License at http://www.mozilla.org/MPL/
#
# Software distributed under the License is distributed on an "AS
# IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
# implied. See the License for the specific language governing
# rights and limitations under the License.
#
# The Original Code is the REMO Bugzilla Extension.
#
# The Initial Developer of the Original Code is Mozilla Foundation
# Portions created by the Initial Developer are Copyright (C) 2011 the
# Initial Developer. All Rights Reserved.
#
# Contributor(s):
#   Byron Jones <glob@mozilla.com>
#   David Lawrence <dkl@mozilla.com>

package Bugzilla::Extension::REMO;

use 5.10.1;
use strict;
use warnings;

use base qw(Bugzilla::Extension);

use Bugzilla::Constants;
use Bugzilla::Util qw(trick_taint trim detaint_natural);
use Bugzilla::Token;
use Bugzilla::Error;
use List::Util qw(first);

our $VERSION = '0.01';

sub page_before_template {
    my ($self, $args) = @_;
    my $page = $args->{'page_id'};
    my $vars = $args->{'vars'};

    if ($page eq 'remo-form-payment.html') {
        _remo_form_payment($vars);
    }
}

sub _remo_form_payment {
    my ($vars) = @_;
    my $input = Bugzilla->input_params;

    my $user = Bugzilla->login(LOGIN_REQUIRED);

    if ($input->{'action'} eq 'commit') {
        my $template = Bugzilla->template;
        my $cgi      = Bugzilla->cgi;
        my $dbh      = Bugzilla->dbh;

        my $bug_id = $input->{'bug_id'};
        detaint_natural($bug_id);
        my $bug = Bugzilla::Bug->check($bug_id);

        # Detect if the user already used the same form to submit again
        my $token = trim($input->{'token'});
        if ($token) {
            my ($creator_id, $date, $old_attach_id) = Bugzilla::Token::GetTokenData($token);
            if (!$creator_id
                || $creator_id != $user->id
                || $old_attach_id !~ "^remo_form_payment:")
            {
                # The token is invalid.
                ThrowUserError('token_does_not_exist');
            }

            $old_attach_id =~ s/^remo_form_payment://;
            if ($old_attach_id) {
                ThrowUserError('remo_payment_cancel_dupe',
                               { bugid => $bug_id, attachid => $old_attach_id });
            }
        }

        # Make sure the user can attach to this bug
        if (!$bug->user->{'canedit'}) {
            ThrowUserError("remo_payment_bug_edit_denied",
                           { bug_id => $bug->id });
        }

        # Make sure the bug is under the correct product/component
        if ($bug->product ne 'Mozilla Reps'
            || $bug->component ne 'Budget Requests')
        {
            ThrowUserError('remo_payment_invalid_product');
        }

        my ($timestamp) = $dbh->selectrow_array("SELECT NOW()");

        $dbh->bz_start_transaction;

        # Create the comment to be added based on the form fields from rep-payment-form
        my $comment;
        $template->process("pages/comment-remo-form-payment.txt.tmpl", $vars, \$comment)
            || ThrowTemplateError($template->error());
        $bug->add_comment($comment, { isprivate => 0 });

        # Attach expense report
        # FIXME: Would be nice to be able to have the above prefilled comment and
        # the following attachments all show up under a single comment. But the longdescs
        # table can only handle one attach_id per comment currently. At least only one
        # email is sent the way it is done below.
        my $attachment;
        if (defined $cgi->upload('expenseform')) {
            # Determine content-type
            my $content_type = $cgi->uploadInfo($cgi->param('expenseform'))->{'Content-Type'};

            $attachment = Bugzilla::Attachment->create(
                { bug           => $bug,
                  creation_ts   => $timestamp,
                  data          => $cgi->upload('expenseform'),
                  description   => 'Expense Form',
                  filename      => scalar $cgi->upload('expenseform'),
                  ispatch       => 0,
                  isprivate     => 0,
                  mimetype      => $content_type,
            });

            # Insert comment for attachment
            $bug->add_comment('', { isprivate  => 0,
                                    type       => CMT_ATTACHMENT_CREATED,
                                    extra_data => $attachment->id });
        }

        # Attach receipts file
        if (defined $cgi->upload("receipts")) {
            # Determine content-type
            my $content_type = $cgi->uploadInfo($cgi->param("receipts"))->{'Content-Type'};

            $attachment = Bugzilla::Attachment->create(
                { bug           => $bug,
                  creation_ts   => $timestamp,
                  data          => $cgi->upload('receipts'),
                  description   => "Receipts",
                  filename      => scalar $cgi->upload("receipts"),
                  ispatch       => 0,
                  isprivate     => 0,
                  mimetype      => $content_type,
            });

            # Insert comment for attachment
            $bug->add_comment('', { isprivate  => 0,
                                    type       => CMT_ATTACHMENT_CREATED,
                                    extra_data => $attachment->id });
        }

        $bug->update($timestamp);

        if ($token) {
            trick_taint($token);
            $dbh->do('UPDATE tokens SET eventdata = ? WHERE token = ?', undef,
                     ("remo_form_payment:" . $attachment->id, $token));
        }

        $dbh->bz_commit_transaction;

        # Define the variables and functions that will be passed to the UI template.
        $vars->{'attachment'} = $attachment;
        $vars->{'bugs'} = [ new Bugzilla::Bug($bug_id) ];
        $vars->{'header_done'} = 1;
        $vars->{'contenttypemethod'} = 'autodetect';

        my $recipients = { 'changer' => $user };
        $vars->{'sent_bugmail'} = Bugzilla::BugMail::Send($bug_id, $recipients);

        print $cgi->header();
        # Generate and return the UI (HTML page) from the appropriate template.
        $template->process("attachment/created.html.tmpl", $vars)
            || ThrowTemplateError($template->error());
        exit;
    }
    else {
        $vars->{'token'} = issue_session_token('remo_form_payment:');
    }
}

my %CSV_COLUMNS = (
    "Date Required"   => { pos =>  1, value => '%cf_due_date' },
    "Requester"       => { pos =>  2, value => 'Rizki Kelimutu' },
    "Email 1"         => { pos =>  3, value => 'rkelimutu@mozilla.com' },
    "Mozilla Space"   => { pos =>  4, value => 'Remote' },
    "Team"            => { pos =>  5, value => 'Participation' },
    "Department Code" => { pos =>  6, value => '1002' },
    "Purpose"         => { pos =>  7, value => 'Rep event: %eventpage' },
    "Item 1"          => { pos =>  8  },
    "Item 2"          => { pos =>  9  },
    "Item 3"          => { pos =>  10 },
    "Item 4"          => { pos =>  11 },
    "Item 5"          => { pos =>  12 },
    "Item 6"          => { pos =>  13 },
    "Item 7"          => { pos =>  14 },
    "Item 8"          => { pos =>  15 },
    "Item 9"          => { pos =>  16 },
    "Item 10"         => { pos =>  17 },
    "Item 11"         => { pos =>  18 },
    "Item 12"         => { pos =>  19 },
    "Item 13"         => { pos =>  20 },
    "Item 14"         => { pos =>  21 },
    "Recipient Name"  => { pos =>  22, value => '%shiptofirstname %shiptolastname' },
    "Email 2"         => { pos =>  23, value => sub { Bugzilla->user->email } },
    "Address 1"       => { pos =>  24, value => '%shiptoaddress1' },
    "Address 2"       => { pos =>  25, value => '%shiptoaddress2' },
    "City"            => { pos =>  26, value => '%shiptocity' },
    "State"           => { pos =>  27, value => '%shiptostate' },
    "Zip"             => { pos =>  28, value => '%shiptopcode' },
    "Country"         => { pos =>  29, value => '%shiptocountry' },
    "Phone number"    => { pos =>  30, value => '%shiptophone' },
    "Notes"           => { pos =>  31, value => '%shipadditional' },
);

sub _expand_value {
    my $value = shift;
    if (ref $value && ref $value eq 'CODE') {
        return $value->();
    }
    else {
        my $cgi = Bugzilla->cgi;
        $value =~ s/%(\w+)/$cgi->param($1)/ge;
        return $value;
    }
}

sub _csv_quote {
    my $s = shift;
    $s =~ s/"/""/g;
    return qq{"$s"};
}

sub _csv_line {
    return join(",", map { _csv_quote($_) } @_);
}

sub _csv_encode {
    return join("\r\n", map { _csv_line(@$_) } @_) . "\r\n";
}

sub post_bug_after_creation {
    my ($self, $args) = @_;
    my $vars = $args->{vars};
    my $bug = $vars->{bug};
    my $template = Bugzilla->template;

    my $format = Bugzilla->input_params->{format};

    return unless defined $format;

    if ($format eq 'remo-swag') {
        # If the attachment cannot be successfully added to the bug,
        # we notify the user, but we don't interrupt the bug creation process.
        my $error_mode_cache = Bugzilla->error_mode;
        Bugzilla->error_mode(ERROR_MODE_DIE);

        my @attachments;
        eval {
            my $xml;
            $template->process("bug/create/create-remo-swag.xml.tmpl", {}, \$xml)
                || ThrowTemplateError($template->error());

            push @attachments, Bugzilla::Attachment->create(
                { bug         => $bug,
                  creation_ts => $bug->creation_ts,
                  data        => $xml,
                  description => 'Remo Swag Request (XML)',
                  filename    => 'remo-swag.xml',
                  ispatch     => 0,
                  isprivate   => 0,
                  mimetype    => 'text/xml',
            });

            my @columns_raw = sort { $CSV_COLUMNS{$a}{pos} <=> $CSV_COLUMNS{$b}{pos} } keys %CSV_COLUMNS;
            my @data        = map { _expand_value( $CSV_COLUMNS{$_}{value} ) } @columns_raw;
            my @columns     = map { s/^(Item|Email) \d+$/$1/g; $_ } @columns_raw;
            my $csv         = _csv_encode(\@columns, \@data);

            push @attachments, Bugzilla::Attachment->create({
                bug         => $bug,
                creation_ts => $bug->creation_ts,
                data        => $csv,
                description => 'Remo Swag Request (CSV)',
                filename    => 'remo-swag.csv',
                ispatch     => 0,
                isprivate   => 0,
                mimetype    => 'text/csv',
            });
        };
        if ($@) {
            warn "$@";
        }

        if (@attachments) {
            # Insert comment for attachment
            foreach my $attachment (@attachments) {
                $bug->add_comment('', { isprivate  => 0,
                                        type       => CMT_ATTACHMENT_CREATED,
                                        extra_data => $attachment->id });
            }
            $bug->update($bug->creation_ts);
            delete $bug->{attachments};
        }
        else {
            $vars->{'message'} = 'attachment_creation_failed';
        }

        Bugzilla->error_mode($error_mode_cache);
    }

    elsif ($format eq 'mozreps') {
        my $needinfo_type = first { $_->name eq 'needinfo' } @{$bug->flag_types};
        return unless $needinfo_type;
        my %original_cc = map { $_ => 1 } Bugzilla->cgi->param('cc');
        my @cc_users    = grep { $_->is_enabled && $original_cc{$_->login}} @{$bug->cc_users};
        my @new_flags   = map {
            { type_id   => $needinfo_type->id,
              status    => '?',
              requestee => $_->login }
        } @cc_users;
        ThrowUserError('remo_missing_voucher') unless @cc_users;

        $bug->set_flags(\@new_flags, []) if @new_flags;
        $bug->add_comment(
            join(", ", map { $_->name || $_->login } @cc_users) .
            ": You have been added as a voucher to this Reps application.\n" .
            "Please provide a comment describing why you endorse this application.\n" .
            "Thanks!"
        );

        $bug->update($bug->creation_ts);
        Bugzilla::BugMail::Send($bug->id, { changer => Bugzilla->user });
    }
}

__PACKAGE__->NAME;