diff options
Diffstat (limited to 'extensions/REMO/Extension.pm')
| -rw-r--r-- | extensions/REMO/Extension.pm | 547 |
1 files changed, 277 insertions, 270 deletions
diff --git a/extensions/REMO/Extension.pm b/extensions/REMO/Extension.pm index df4e70c80..7ca74e081 100644 --- a/extensions/REMO/Extension.pm +++ b/extensions/REMO/Extension.pm @@ -37,304 +37,311 @@ 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'}; + my ($self, $args) = @_; + my $page = $args->{'page_id'}; + my $vars = $args->{'vars'}; - if ($page eq 'remo-form-payment.html') { - _remo_form_payment($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; + 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}); + } } - else { - $vars->{'token'} = issue_session_token('remo_form_payment:'); + + # 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' }, + "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; - } + 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"}; + my $s = shift; + $s =~ s/"/""/g; + return qq{"$s"}; } sub _csv_line { - return join(",", map { _csv_quote($_) } @_); + return join(",", map { _csv_quote($_) } @_); } sub _csv_encode { - return join("\r\n", map { _csv_line(@$_) } @_) . "\r\n"; + 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); + 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 "$@"; } - 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!" - ); + if (@attachments) { - $bug->update($bug->creation_ts); - Bugzilla::BugMail::Send($bug->id, { changer => Bugzilla->user }); + # 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; |