1 files changed, 38 insertions, 0 deletions
diff --git a/perllib/FixMyStreet/DB/ResultSet/Alert.pm b/perllib/FixMyStreet/DB/ResultSet/Alert.pm
new file mode 100644
index 000000000..62e6120e4
--- /dev/null
+++ b/perllib/FixMyStreet/DB/ResultSet/Alert.pm
@@ -0,0 +1,38 @@
+package FixMyStreet::DB::ResultSet::Alert;
+use base 'DBIx::Class::ResultSet';
+
+use strict;
+use warnings;
+
+sub timeline_created {
+    my ( $rs, $restriction ) = @_;
+
+    my $prefetch = 
+        FixMyStreet::App->model('DB')->schema->storage->sql_maker->quote_char ?
+        [ qw/alert_type user/ ] :
+        [ qw/alert_type/ ];
+
+    return $rs->search(
+        {
+            whensubscribed => { '>=', \"ms_current_timestamp()-'7 days'::interval" },
+            confirmed => 1,
+            %{ $restriction },
+        },
+        {
+            prefetch => $prefetch,
+        }
+    );
+}
+
+sub timeline_disabled {
+    my ( $rs, $restriction ) = @_;
+
+    return $rs->search(
+        {
+            whendisabled => { '>=', \"ms_current_timestamp()-'7 days'::interval" },
+            %{ $restriction },
+        },
+    );
+}
+
+1;