From 3ba270bd0798694a6b93771086088b6bc11ea163 Mon Sep 17 00:00:00 2001
From: cuom1999 <lephuocdinh99@gmail.com>
Date: Mon, 10 Jan 2022 23:02:12 +0700
Subject: [PATCH] General rank for friends

---
 judge/views/contests.py        |  2 +-
 templates/contest/ranking.html | 24 +++++++++++++++++++++++-
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/judge/views/contests.py b/judge/views/contests.py
index 7bdccac..8dcd43b 100644
--- a/judge/views/contests.py
+++ b/judge/views/contests.py
@@ -661,7 +661,7 @@ def get_contest_ranking_list(request, contest, participation=None, ranking_list=
     problems = list(contest.contest_problems.select_related('problem').defer('problem__description').order_by('order'))
 
     users = ranker(ranking_list(contest, problems), key=attrgetter('points', 'cumtime', 'tiebreaker'))
-    
+
     if show_current_virtual:
         if participation is None and request.user.is_authenticated:
             participation = request.profile.current_contest
diff --git a/templates/contest/ranking.html b/templates/contest/ranking.html
index 41d47ea..a4227a8 100644
--- a/templates/contest/ranking.html
+++ b/templates/contest/ranking.html
@@ -319,6 +319,22 @@
             }
         }
 
+        function get_initial_rank() {
+            var ranks = $('.rank-td').map(function() {return this.innerHTML}).get();
+            var usernames = $('.user-name .user a').map(function() {return this.text}).get();
+            window.user_rank = new Map();
+            for (var i = 0; i < ranks.length; i++) {
+                window.user_rank[usernames[i]] = ranks[i];
+            }
+        }
+
+        function add_initial_friend_rank() {
+            var usernames = $('.user-name .user a').map(function() {return this.text}).get();
+            $('.rank-td').each(function(i) {
+                this.innerHTML += ' (' + window.user_rank[usernames[i]] + ')';
+            });
+        }
+
         function update_ranking() {
             var friend = $('#show-friends-checkbox').is(':checked');
             var virtual = $('#show-virtual-checkbox').is(':checked');
@@ -328,8 +344,13 @@
                 success: function(HTML) {
                     $('#users-table').html(HTML);
                     highlightFirstSolve();
-                    renew_filter();
                     $('#loading-gif').hide();
+                    if (!virtual && !friend) {
+                        get_initial_rank();
+                    }
+                    if (friend) {
+                        add_initial_friend_rank();
+                    }
                 },
                 fail: function() {
                     console.log('Fail to update ranking');
@@ -394,6 +415,7 @@
 
             highlightFirstSolve();
             renew_filter();
+            get_initial_rank();
             {% if participation_tab %}
                 $('#show-virtual-checkbox').hide();
                 $('#show-virtual-label').hide();