editwhines.cgi: code style, do not use prepare/execute, use foreign keys

hinted-selects
Vitaliy Filippov 2014-10-09 17:30:16 +04:00
parent f2c048a1aa
commit d43532f58e
1 changed files with 134 additions and 207 deletions

View File

@ -1,6 +1,4 @@
#!/usr/bin/perl -wT
# -*- 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
@ -21,10 +19,6 @@
# Contributor(s): Erik Stambaugh <erik@dasbistro.com>
#
################################################################################
# Script Initialization
################################################################################
use strict;
use lib qw(. lib);
@ -42,10 +36,6 @@ use Bugzilla::Whine::Query;
# require the user to have logged in
my $user = Bugzilla->login(LOGIN_REQUIRED);
###############################################################################
# Main Body Execution
###############################################################################
my $cgi = Bugzilla->cgi;
my $template = Bugzilla->template;
my $vars = {};
@ -53,7 +43,6 @@ my $dbh = Bugzilla->dbh;
my $userid = $user->id;
my $token = $cgi->param('token');
my $sth; # database statement handle
# $events is a hash ref, keyed by event id, that stores the active user's
# events. It starts off with:
@ -77,65 +66,39 @@ my $sth; # database statement handle
my $events = get_events($userid);
# First see if this user may use whines
$user->in_group('bz_canusewhines')
|| ThrowUserError("auth_failure", {group => "bz_canusewhines",
action => "schedule",
object => "reports"});
$user->in_group('bz_canusewhines') || ThrowUserError("auth_failure", {
group => "bz_canusewhines",
action => "schedule",
object => "reports",
});
# May this user send mail to other users?
my $can_mail_others = Bugzilla->user->in_group('bz_canusewhineatothers');
# If the form was submitted, we need to look for what needs to be added or
# removed, then what was altered.
if ($cgi->param('update')) {
if ($cgi->param('update'))
{
check_token_data($token, 'edit_whine');
if ($cgi->param("add_event")) {
if ($cgi->param("add_event"))
{
# we create a new event
$sth = $dbh->prepare("INSERT INTO whine_events " .
"(owner_userid) " .
"VALUES (?)");
$sth->execute($userid);
$dbh->do("INSERT INTO whine_events (owner_userid) VALUES (?)", undef, $userid);
}
else {
for my $eventid (keys %{$events}) {
else
{
for my $eventid (keys %{$events})
{
# delete an entire event
if ($cgi->param("remove_event_$eventid")) {
if ($cgi->param("remove_event_$eventid"))
{
# We need to make sure these belong to the same user,
# otherwise we could simply delete whatever matched that ID.
#
# schedules
my $schedules = Bugzilla::Whine::Schedule->match({ eventid => $eventid });
$sth = $dbh->prepare("DELETE FROM whine_schedules "
. "WHERE id=?");
foreach my $schedule (@$schedules) {
$sth->execute($schedule->id);
}
# queries
$sth = $dbh->prepare("SELECT whine_queries.id " .
"FROM whine_queries " .
"LEFT JOIN whine_events " .
"ON whine_events.id = " .
"whine_queries.eventid " .
"WHERE whine_events.id = ? " .
"AND whine_events.owner_userid = ?");
$sth->execute($eventid, $userid);
my @ids = @{$sth->fetchall_arrayref};
$sth = $dbh->prepare("DELETE FROM whine_queries " .
"WHERE id=?");
for (@ids) {
my $delete_id = $_->[0];
$sth->execute($delete_id);
}
# events
$sth = $dbh->prepare("DELETE FROM whine_events " .
"WHERE id=? AND owner_userid=?");
$sth->execute($eventid, $userid);
$dbh->do("DELETE FROM whine_events WHERE id=? AND owner_userid=?", undef, $eventid, $userid);
}
else {
else
{
# check the subject, body and mailifnobugs for changes
my $subject = ($cgi->param("event_${eventid}_subject") or '');
my $body = ($cgi->param("event_${eventid}_body") or '');
@ -144,31 +107,30 @@ if ($cgi->param('update')) {
trick_taint($subject) if $subject;
trick_taint($body) if $body;
if ( ($subject ne $events->{$eventid}->{'subject'})
|| ($mailifnobugs != $events->{$eventid}->{'mailifnobugs'})
|| ($body ne $events->{$eventid}->{'body'}) ) {
$sth = $dbh->prepare("UPDATE whine_events " .
"SET subject=?, body=?, mailifnobugs=? " .
"WHERE id=?");
$sth->execute($subject, $body, $mailifnobugs, $eventid);
if ($subject ne $events->{$eventid}->{subject} ||
$mailifnobugs != $events->{$eventid}->{mailifnobugs} ||
$body ne $events->{$eventid}->{body})
{
$dbh->do(
"UPDATE whine_events SET subject=?, body=?, mailifnobugs=? WHERE id=?",
undef, $subject, $body, $mailifnobugs, $eventid
);
}
# add a schedule
if ($cgi->param("add_schedule_$eventid")) {
if ($cgi->param("add_schedule_$eventid"))
{
# the schedule table must be locked before altering
$sth = $dbh->prepare("INSERT INTO whine_schedules " .
"(eventid, mailto_type, mailto, " .
"run_day, run_time) " .
"VALUES (?, ?, ?, 'Sun', 2)");
$sth->execute($eventid, MAILTO_USER, $userid);
$dbh->do(
"INSERT INTO whine_schedules (eventid, mailto_type, mailto, " .
"run_day, run_time) VALUES (?, ?, ?, 'Sun', 2)",
undef, $eventid, MAILTO_USER, $userid
);
}
# add a query
elsif ($cgi->param("add_query_$eventid")) {
$sth = $dbh->prepare("INSERT INTO whine_queries "
. "(eventid) "
. "VALUES (?)");
$sth->execute($eventid);
elsif ($cgi->param("add_query_$eventid"))
{
$dbh->do("INSERT INTO whine_queries (eventid) VALUES (?)", undef, $eventid);
}
}
@ -178,44 +140,41 @@ if ($cgi->param('update')) {
# Check schedules for changes
my $schedules = Bugzilla::Whine::Schedule->match({ eventid => $eventid });
my @scheduleids = ();
foreach my $schedule (@$schedules) {
foreach my $schedule (@$schedules)
{
push @scheduleids, $schedule->id;
}
# we need to double-check all of the user IDs in mailto to make
# sure they exist
my $arglist = {}; # args for match_field
for my $sid (@scheduleids) {
if ($cgi->param("mailto_type_$sid") == MAILTO_USER) {
for my $sid (@scheduleids)
{
if ($cgi->param("mailto_type_$sid") == MAILTO_USER)
{
$arglist->{"mailto_$sid"} = {
'type' => 'single',
type => 'single',
};
}
}
if (scalar %{$arglist}) {
if (scalar %{$arglist})
{
Bugzilla::User::match_field($arglist);
}
for my $sid (@scheduleids) {
if ($cgi->param("remove_schedule_$sid")) {
for my $sid (@scheduleids)
{
if ($cgi->param("remove_schedule_$sid"))
{
# having the assignee id in here is a security failsafe
$sth = $dbh->prepare("SELECT whine_schedules.id " .
"FROM whine_schedules " .
"LEFT JOIN whine_events " .
"ON whine_events.id = " .
"whine_schedules.eventid " .
"WHERE whine_events.owner_userid=? " .
"AND whine_schedules.id =?");
$sth->execute($userid, $sid);
my @ids = @{$sth->fetchall_arrayref};
for (@ids) {
$sth = $dbh->prepare("DELETE FROM whine_schedules " .
"WHERE id=?");
$sth->execute($_->[0]);
}
$dbh->do(
"DELETE FROM whine_schedules WHERE id=?".
" AND (SELECT owner_userid FROM whine_events WHERE whine_events.id=eventid)=?",
undef, $sid, $userid
);
}
else {
else
{
my $o_day = $cgi->param("orig_day_$sid") || '';
my $day = $cgi->param("day_$sid") || '';
my $o_time = $cgi->param("orig_time_$sid") || 0;
@ -228,17 +187,21 @@ if ($cgi->param('update')) {
my $mailto_id = $userid;
# get an id for the mailto address
if ($can_mail_others && $mailto) {
if ($mailto_type == MAILTO_USER) {
if ($can_mail_others && $mailto)
{
if ($mailto_type == MAILTO_USER)
{
$mailto_id = login_to_id($mailto);
}
elsif ($mailto_type == MAILTO_GROUP) {
elsif ($mailto_type == MAILTO_GROUP)
{
# The group name is used in a placeholder.
trick_taint($mailto);
$mailto_id = Bugzilla::Group::ValidateGroupName($mailto, ($user))
|| ThrowUserError('invalid_group_name', { name => $mailto });
|| ThrowUserError('invalid_group_name', { name => $mailto });
}
else {
else
{
# bad value, so it will just mail to the whine
# owner. $mailto_id was already set above.
$mailto_type = MAILTO_USER;
@ -247,76 +210,56 @@ if ($cgi->param('update')) {
detaint_natural($mailto_type);
if ( ($o_day ne $day) ||
($o_time ne $time) ||
($o_mailto ne $mailto) ||
($o_mailto_type != $mailto_type) ){
if ($o_day ne $day || $o_time ne $time ||
$o_mailto ne $mailto || $o_mailto_type != $mailto_type)
{
trick_taint($day);
trick_taint($time);
# the schedule table must be locked
$sth = $dbh->prepare("UPDATE whine_schedules " .
"SET run_day=?, run_time=?, " .
"mailto_type=?, mailto=?, " .
"run_next=NULL " .
"WHERE id=?");
$sth->execute($day, $time, $mailto_type,
$mailto_id, $sid);
$dbh->do(
"UPDATE whine_schedules SET run_day=?, run_time=?,".
" mailto_type=?, mailto=?, run_next=NULL WHERE id=?",
undef, $day, $time, $mailto_type, $mailto_id, $sid
);
}
}
}
# Check queries for changes
my $queries = Bugzilla::Whine::Query->match({ eventid => $eventid });
for my $query (@$queries) {
for my $query (@$queries)
{
my $qid = $query->id;
if ($cgi->param("remove_query_$qid")) {
$sth = $dbh->prepare("SELECT whine_queries.id " .
"FROM whine_queries " .
"LEFT JOIN whine_events " .
"ON whine_events.id = " .
"whine_queries.eventid " .
"WHERE whine_events.owner_userid=? " .
"AND whine_queries.id =?");
$sth->execute($userid, $qid);
for (@{$sth->fetchall_arrayref}) {
$sth = $dbh->prepare("DELETE FROM whine_queries " .
"WHERE id=?");
$sth->execute($_->[0]);
}
if ($cgi->param("remove_query_$qid"))
{
$dbh->do(
"DELETE FROM whine_queries WHERE id=?".
" AND (SELECT owner_userid FROM whine_events WHERE whine_events.id=eventid)=?",
undef, $qid, $userid
);
}
else {
else
{
my $o_sort = $cgi->param("orig_query_sort_$qid") || 0;
my $sort = $cgi->param("query_sort_$qid") || 0;
my $o_queryname = $cgi->param("orig_query_name_$qid") || '';
my $queryname = $cgi->param("query_name_$qid") || '';
my $o_title = $cgi->param("orig_query_title_$qid") || '';
my $title = $cgi->param("query_title_$qid") || '';
my $o_onemailperbug =
$cgi->param("orig_query_onemailperbug_$qid") || 0;
my $onemailperbug =
$cgi->param("query_onemailperbug_$qid") ? 1 : 0;
if ( ($o_sort != $sort) ||
($o_queryname ne $queryname) ||
($o_onemailperbug != $onemailperbug) ||
($o_title ne $title) ){
my $o_onemailperbug = $cgi->param("orig_query_onemailperbug_$qid") || 0;
my $onemailperbug = $cgi->param("query_onemailperbug_$qid") ? 1 : 0;
if ($o_sort != $sort || $o_queryname ne $queryname ||
$o_onemailperbug != $onemailperbug || $o_title ne $title)
{
detaint_natural($sort);
trick_taint($queryname);
trick_taint($title);
$sth = $dbh->prepare("UPDATE whine_queries " .
"SET sortkey=?, " .
"query_name=?, " .
"title=?, " .
"onemailperbug=? " .
"WHERE id=?");
$sth->execute($sort, $queryname, $title,
$onemailperbug, $qid);
$dbh->do(
"UPDATE whine_queries SET sortkey=?, query_name=?, title=?, onemailperbug=? WHERE id=?",
undef, $sort, $queryname, $title, $onemailperbug, $qid
);
}
}
}
@ -325,7 +268,7 @@ if ($cgi->param('update')) {
delete_token($token);
}
$vars->{'mail_others'} = $can_mail_others;
$vars->{mail_others} = $can_mail_others;
# Get events again, to cover any updates that were made
$events = get_events($userid);
@ -344,82 +287,66 @@ $events = get_events($userid);
# sort
#
# build the whine list by event id
for my $event_id (keys %{$events}) {
$events->{$event_id}->{'schedule'} = [];
$events->{$event_id}->{'queries'} = [];
for my $event_id (keys %{$events})
{
$events->{$event_id}->{schedule} = [];
$events->{$event_id}->{queries} = [];
# schedules
my $schedules = Bugzilla::Whine::Schedule->match({ eventid => $event_id });
foreach my $schedule (@$schedules) {
my $mailto_type = $schedule->mailto_is_group ? MAILTO_GROUP
: MAILTO_USER;
foreach my $schedule (@$schedules)
{
my $mailto_type = $schedule->mailto_is_group ? MAILTO_GROUP : MAILTO_USER;
my $mailto = '';
if ($mailto_type == MAILTO_USER) {
if ($mailto_type == MAILTO_USER)
{
$mailto = $schedule->mailto->login;
}
elsif ($mailto_type == MAILTO_GROUP) {
elsif ($mailto_type == MAILTO_GROUP)
{
$mailto = $schedule->mailto->name;
}
push @{$events->{$event_id}->{'schedule'}},
{
'day' => $schedule->run_day,
'time' => $schedule->run_time,
'mailto_type' => $mailto_type,
'mailto' => $mailto,
'id' => $schedule->id,
};
push @{$events->{$event_id}->{schedule}}, {
day => $schedule->run_day,
time => $schedule->run_time,
mailto_type => $mailto_type,
mailto => $mailto,
id => $schedule->id,
};
}
# queries
my $queries = Bugzilla::Whine::Query->match({ eventid => $event_id });
for my $query (@$queries) {
push @{$events->{$event_id}->{'queries'}},
{
'name' => $query->name,
'title' => $query->title,
'sort' => $query->sortkey,
'id' => $query->id,
'onemailperbug' => $query->one_email_per_bug,
};
for my $query (@$queries)
{
push @{$events->{$event_id}->{queries}}, {
name => $query->name,
title => $query->title,
sort => $query->sortkey,
id => $query->id,
onemailperbug => $query->one_email_per_bug,
};
}
}
$vars->{'events'} = $events;
$vars->{events} = $events;
# get the available queries
$sth = $dbh->prepare("SELECT name FROM namedqueries WHERE userid=?");
$sth->execute($userid);
$vars->{'available_queries'} = [];
while (my ($query) = $sth->fetchrow_array) {
push @{$vars->{'available_queries'}}, $query;
}
$vars->{'token'} = issue_session_token('edit_whine');
$vars->{'local_timezone'} = Bugzilla->local_timezone->short_name_for_datetime(DateTime->now());
$vars->{available_queries} = $dbh->selectcol_arrayref("SELECT name FROM namedqueries WHERE userid=?", undef, $userid) || [];
$vars->{token} = issue_session_token('edit_whine');
$vars->{local_timezone} = Bugzilla->local_timezone->short_name_for_datetime(DateTime->now());
$template->process("whine/schedule.html.tmpl", $vars)
|| ThrowTemplateError($template->error());
|| ThrowTemplateError($template->error());
exit;
# get_events takes a userid and returns a hash, keyed by event ID, containing
# the subject and body of each event that user owns
sub get_events {
sub get_events
{
my $userid = shift;
my $dbh = Bugzilla->dbh;
my $events = {};
my $sth = $dbh->prepare("SELECT DISTINCT id, subject, body, mailifnobugs " .
"FROM whine_events " .
"WHERE owner_userid=?");
$sth->execute($userid);
while (my ($ev, $sub, $bod, $mno) = $sth->fetchrow_array) {
$events->{$ev} = {
'subject' => $sub || '',
'body' => $bod || '',
'mailifnobugs' => $mno || 0,
};
}
return $events;
return Bugzilla->dbh->selectall_hashref(
"SELECT DISTINCT id, subject, body, mailifnobugs FROM whine_events WHERE owner_userid=?",
'id', undef, $userid
);
}