From 272efbfe30ce0f149756244afd100ec27d281384 Mon Sep 17 00:00:00 2001
From: Vivek Anand <vivekanand1101@gmail.com>
Date: Aug 29 2016 09:08:07 +0000
Subject: Add user's project watch list in index page of the user


---

diff --git a/pagure/lib/__init__.py b/pagure/lib/__init__.py
index 54f67c1..0bd36f6 100644
--- a/pagure/lib/__init__.py
+++ b/pagure/lib/__init__.py
@@ -2946,3 +2946,44 @@ def is_watching(session, user, reponame, repouser=None):
                 return True
 
     return False
+
+
+def user_watch_list(session, user):
+    ''' Returns list of all the projects which the user is watching '''
+
+    user_obj = search_user(session, username=user)
+    if not user_obj:
+        return []
+
+    unwatched = session.query(
+        model.Watcher
+    ).filter(
+        model.Watcher.user_id == user_obj.id
+    ).filter(
+        model.Watcher.watch == False
+    )
+
+    unwatched_list = []
+    if unwatched:
+        unwatched_list = [unwatch.project for unwatch in unwatched.all()]
+
+    watched = session.query(
+        model.Watcher
+    ).filter(
+        model.Watcher.user_id == user_obj.id
+    ).filter(
+        model.Watcher.watch == True
+    )
+
+    watched_list = []
+    if watched:
+        watched_list = [watch.project for watch in watched.all()]
+
+    user_projects = search_projects(session, username=user_obj.user)
+    watch = set(watched_list + user_projects)
+
+    for project in user_projects:
+        if project in unwatched_list:
+            watch.remove(project)
+
+    return sorted(list(watch), key=lambda proj: proj.name)
diff --git a/pagure/templates/index_auth.html b/pagure/templates/index_auth.html
index 7f87988..f8d082f 100644
--- a/pagure/templates/index_auth.html
+++ b/pagure/templates/index_auth.html
@@ -189,6 +189,34 @@
             </div>
             {% endfor %}
         </div>
+        <div class="card" id="watch_list">
+            <div class="card-header">
+                My Watch List <span class="label label-default">{{ watch_list | count }}</span>
+            </div>
+            {% for repo in watch_list %}
+              <div class="list-group">
+                {% if repo.is_fork %}
+                  <a class="list-group-item" href="{{ url_for(
+                    'view_repo', username=repo.user.username, repo=repo.name) }}">
+                    <div class=""><strong><span class="oi" data-glyph="fork">
+                            </span> &nbsp;{{ repo.user.username }}/{{ repo.name }}</strong>
+                    </div>
+                  </a>
+                {% else %}
+                  <a class="list-group-item" href="{{ url_for(
+                    'view_repo', repo=repo.name) }}">
+                    <div class=""><strong><span class="oi" data-glyph="document">
+                          </span> &nbsp;{{ repo.name }}</strong>
+                    </div>
+                  </a>
+                {% endif %}
+              </div>
+            {% else %}
+            <div class="card-block">
+                <p>No project in watch list</p>
+            </div>
+            {% endfor %}
+        </div>
       </div>
     </div>
   </div>
diff --git a/pagure/ui/app.py b/pagure/ui/app.py
index b8568bb..17475fb 100644
--- a/pagure/ui/app.py
+++ b/pagure/ui/app.py
@@ -116,12 +116,17 @@ def index_auth():
         fork=True,
         count=True)
 
+    watch_list = pagure.lib.user_watch_list(
+        SESSION,
+        user=flask.g.fas_user.username)
+
     return flask.render_template(
         'index_auth.html',
         username=flask.g.fas_user.username,
         user=user,
         forks=forks,
         repos=repos,
+        watch_list=watch_list,
         repopage=repopage,
         forkpage=forkpage,
         repos_length=repos_length,
diff --git a/tests/test_pagure_flask_ui_app.py b/tests/test_pagure_flask_ui_app.py
index 1236f14..eb08845 100644
--- a/tests/test_pagure_flask_ui_app.py
+++ b/tests/test_pagure_flask_ui_app.py
@@ -91,7 +91,36 @@ class PagureFlaskApptests(tests.Modeltests):
             self.assertEqual(
                 output.data.count('<p>No group found</p>'), 1)
             self.assertEqual(
-                output.data.count('<div class="card-header">'), 3)
+                output.data.count('<div class="card-header">'), 4)
+
+    def test_watch_list(self):
+        ''' Test for watch list of a user '''
+
+        user = tests.FakeUser(username='pingou')
+        with tests.user_set(pagure.APP, user):
+            output = self.app.get('/')
+            self.assertIn(
+                '<div class="text-xs-center">You have no projects</div>',
+                output.data)
+            self.assertIn(
+                '<p>You have no forks</p>',
+                output.data)
+            self.assertIn(
+                '<p>No project in watch list</p>',
+                output.data)
+
+            tests.create_projects(self.session)
+
+            output = self.app.get('/')
+            self.assertIn(
+                'My Projects <span class="label label-default">2</span>',
+                output.data)
+            self.assertIn(
+                'My Forks <span class="label label-default">0</span>',
+                output.data)
+            self.assertIn(
+                'My Watch List <span class="label label-default">2</span>',
+                output.data)
 
     def test_view_users(self):
         """ Test the view_users endpoint. """
diff --git a/tests/test_pagure_lib.py b/tests/test_pagure_lib.py
index 6952975..85fc630 100644
--- a/tests/test_pagure_lib.py
+++ b/tests/test_pagure_lib.py
@@ -2483,6 +2483,40 @@ class PagureLibtests(tests.Modeltests):
         self.assertFalse(watch)
 
 
+    def test_user_watch_list(self):
+        ''' test user watch list method of pagure.lib '''
+
+        tests.create_projects(self.session)
+
+        # He should be watching
+        user = tests.FakeUser()
+        user.username = 'pingou'
+        watch_list_objs = pagure.lib.user_watch_list(
+            session=self.session,
+            user='pingou',
+        )
+        watch_list = [obj.name for obj in watch_list_objs]
+        self.assertEqual(watch_list, ['test', 'test2'])
+
+        # He isn't in the db, thus not watching anything
+        user.username = 'vivek'
+        watch_list_objs = pagure.lib.user_watch_list(
+            session=self.session,
+            user='vivek',
+        )
+        watch_list = [obj.name for obj in watch_list_objs]
+        self.assertEqual(watch_list, [])
+
+        # He shouldn't be watching anything
+        user.username = 'foo'
+        watch_list_objs = pagure.lib.user_watch_list(
+            session=self.session,
+            user='foo',
+        )
+        watch_list = [obj.name for obj in watch_list_objs]
+        self.assertEqual(watch_list, [])
+
+
 if __name__ == '__main__':
     SUITE = unittest.TestLoader().loadTestsFromTestCase(PagureLibtests)
     unittest.TextTestRunner(verbosity=2).run(SUITE)