Adding report to shortlist removes from others.

1 files changed, 23 insertions, 0 deletions

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;