aboutsummaryrefslogtreecommitdiffstats
path: root/perllib/FixMyStreet/DB
diff options
context:
space:
mode:
Diffstat (limited to 'perllib/FixMyStreet/DB')
-rw-r--r--perllib/FixMyStreet/DB/Result/Problem.pm12
-rw-r--r--perllib/FixMyStreet/DB/Result/User.pm15
-rw-r--r--perllib/FixMyStreet/DB/ResultSet/UserPlannedReport.pm23
3 files changed, 45 insertions, 5 deletions
diff --git a/perllib/FixMyStreet/DB/Result/Problem.pm b/perllib/FixMyStreet/DB/Result/Problem.pm
index 27648ddad..855732f83 100644
--- a/perllib/FixMyStreet/DB/Result/Problem.pm
+++ b/perllib/FixMyStreet/DB/Result/Problem.pm
@@ -1020,4 +1020,16 @@ sub static_map {
};
}
+has shortlisted_user => (
+ is => 'ro',
+ lazy => 1,
+ default => sub {
+ my $self = shift;
+ my $user = $self->result_source->schema->resultset('User')->search(
+ { 'user_planned_reports.report_id' => $self->id },
+ { join => 'user_planned_reports' })->first;
+ return $user;
+ },
+);
+
1;
diff --git a/perllib/FixMyStreet/DB/Result/User.pm b/perllib/FixMyStreet/DB/Result/User.pm
index 8d42d5926..7ec49b074 100644
--- a/perllib/FixMyStreet/DB/Result/User.pm
+++ b/perllib/FixMyStreet/DB/Result/User.pm
@@ -351,11 +351,18 @@ sub adopt {
# Planned reports / shortlist
-# Override the default auto-created function as we only want one live entry per user
+# Override the default auto-created function as we only want one live entry so
+# we need to delete it anywhere else and return an existing one if present.
around add_to_planned_reports => sub {
my ( $orig, $self ) = ( shift, shift );
my ( $report_col ) = @_;
- my $existing = $self->user_planned_reports->search_rs({ report_id => $report_col->{id}, removed => undef })->first;
+
+ $self->result_source->schema->resultset("UserPlannedReport")
+ ->active
+ ->for_report($report_col->id)
+ ->search_rs({ user_id => { '!=', $self->id } })
+ ->remove();
+ my $existing = $self->user_planned_reports->active->for_report($report_col->id)->first;
return $existing if $existing;
return $self->$orig(@_);
};
@@ -363,9 +370,7 @@ around add_to_planned_reports => sub {
# Override the default auto-created function as we don't want to ever delete anything
around remove_from_planned_reports => sub {
my ($orig, $self, $report) = @_;
- $self->user_planned_reports
- ->search_rs({ report_id => $report->id, removed => undef })
- ->update({ removed => \'current_timestamp' });
+ $self->user_planned_reports->active->for_report($report->id)->remove();
};
sub active_planned_reports {
diff --git a/perllib/FixMyStreet/DB/ResultSet/UserPlannedReport.pm b/perllib/FixMyStreet/DB/ResultSet/UserPlannedReport.pm
new file mode 100644
index 000000000..7e16e2dd3
--- /dev/null
+++ b/perllib/FixMyStreet/DB/ResultSet/UserPlannedReport.pm
@@ -0,0 +1,23 @@
+package FixMyStreet::DB::ResultSet::UserPlannedReport;
+use base 'DBIx::Class::ResultSet';
+
+use strict;
+use warnings;
+
+sub active {
+ my $rs = shift;
+ $rs->search({ removed => undef });
+}
+
+sub for_report {
+ my $rs = shift;
+ my $problem_id = shift;
+ $rs->search({ report_id => $problem_id });
+}
+
+sub remove {
+ my $rs = shift;
+ $rs->update({ removed => \'current_timestamp' });
+}
+
+1;