From b8705a7bf2d3040b6dbf5f70e28f85b4de13a2e3 Mon Sep 17 00:00:00 2001
From: alxd <github@alxd.org>
Date: Fri, 30 Nov 2018 13:00:29 +0100
Subject: [PATCH] change getNotifications to use filter

---
 .../notifications/components/filter_bar.js       | 16 ++++++++--------
 .../mastodon/features/notifications/index.js     |  9 +++++++--
 2 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/app/javascript/mastodon/features/notifications/components/filter_bar.js b/app/javascript/mastodon/features/notifications/components/filter_bar.js
index 6b46064757b5a..4887fcbefaf5f 100644
--- a/app/javascript/mastodon/features/notifications/components/filter_bar.js
+++ b/app/javascript/mastodon/features/notifications/components/filter_bar.js
@@ -23,8 +23,8 @@ export default class FilterBar extends React.PureComponent {
           />
         </button>
         <button
-          className={selectedFilter === 'mentions' ? 'active' : ''}
-          onClick={() => selectFilter('mentions')}
+          className={selectedFilter === 'mention' ? 'active' : ''}
+          onClick={() => selectFilter('mention')}
         >
           <FormattedMessage
             id='notifications.filter.mentions'
@@ -32,8 +32,8 @@ export default class FilterBar extends React.PureComponent {
           />
         </button>
         <button
-          className={selectedFilter === 'favourites' ? 'active' : ''}
-          onClick={() => selectFilter('favourites')}
+          className={selectedFilter === 'favourite' ? 'active' : ''}
+          onClick={() => selectFilter('favourite')}
         >
           <FormattedMessage
             id='notifications.filter.favourites'
@@ -41,8 +41,8 @@ export default class FilterBar extends React.PureComponent {
           />
         </button>
         <button
-          className={selectedFilter === 'boosts' ? 'active' : ''}
-          onClick={() => selectFilter('boosts')}
+          className={selectedFilter === 'reblog' ? 'active' : ''}
+          onClick={() => selectFilter('reblog')}
         >
           <FormattedMessage
             id='notifications.filter.boosts'
@@ -50,8 +50,8 @@ export default class FilterBar extends React.PureComponent {
           />
         </button>
         <button
-          className={selectedFilter === 'follows' ? 'active' : ''}
-          onClick={() => selectFilter('follows')}
+          className={selectedFilter === 'follow' ? 'active' : ''}
+          onClick={() => selectFilter('follow')}
         >
           <FormattedMessage
             id='notifications.filter.follows'
diff --git a/app/javascript/mastodon/features/notifications/index.js b/app/javascript/mastodon/features/notifications/index.js
index 1b55cdabc88c5..03e48af4946e2 100644
--- a/app/javascript/mastodon/features/notifications/index.js
+++ b/app/javascript/mastodon/features/notifications/index.js
@@ -20,11 +20,16 @@ const messages = defineMessages({
   title: { id: 'column.notifications', defaultMessage: 'Notifications' },
 });
 
-// TODO: Modify this to filter notifications
 const getNotifications = createSelector([
+  state => state.getIn(['settings', 'notifications', 'filter']),
   state => ImmutableList(state.getIn(['settings', 'notifications', 'shows']).filter(item => !item).keys()),
   state => state.getIn(['notifications', 'items']),
-], (excludedTypes, notifications) => notifications.filterNot(item => item !== null && excludedTypes.includes(item.get('type'))));
+], (allowedType, excludedTypes, notifications) => {
+  if (allowedType === 'all') {
+    return notifications.filterNot(item => item !== null && excludedTypes.includes(item.get('type')));
+  }
+  return notifications.filter(item => item !== null && allowedType === item.get('type'))
+});
 
 const mapStateToProps = state => ({
   notifications: getNotifications(state),