From aa8a920288e64fed3e7e095a4e4d465ba5ee1fd7 Mon Sep 17 00:00:00 2001
From: Hugo Peixoto <hugo.peixoto@gmail.com>
Date: Sun, 30 Jul 2023 17:45:26 +0100
Subject: [PATCH] Add bytes and count per contact

---
 index.html.erb | 7 +++++++
 main.rb        | 3 +++
 2 files changed, 10 insertions(+)

diff --git a/index.html.erb b/index.html.erb
index cbeb379..bd3c705 100644
--- a/index.html.erb
+++ b/index.html.erb
@@ -88,6 +88,13 @@ $ PROCS=10 /caminho/para/o/saca-sapos download 1000 eu@exemplo.pt
   <p>Tamanho total estimado: <%= h bytes_downloaded / videos_downloaded * videos_known %></p>
   <p>Número de pessoas a descarregar vídeos: <%= users %></p>
 
+  <p>Tamanho descarregado por cada pessoa:</p>
+  <ol>
+    <% downloads_per_user.each_with_index do |downloads, index| %>
+      <li><%= h downloads["size"] %> (<%= downloads["total"] %> vídeos)</li>
+    <% end %>
+  </ol>
+
   <footer>
     <hr />
 
diff --git a/main.rb b/main.rb
index 96c515b..b5137ec 100644
--- a/main.rb
+++ b/main.rb
@@ -31,6 +31,7 @@ get '/' do
 
   videos = ActiveRecord::Base.connection.execute("select count(1) AS total from videos;")[0]
   users = ActiveRecord::Base.connection.execute("select count(distinct email) AS total from downloads;")[0]
+  downloads_per_user = ActiveRecord::Base.connection.execute("select SUM(size) AS size, COUNT(1) AS total FROM (select email, video_id, MAX(size) as size from downloads group by email, video_id) x GROUP BY email ORDER BY 1 DESC;")
 
   stats = {
     videos_known: videos["total"].to_i,
@@ -38,6 +39,8 @@ get '/' do
 
     bytes_downloaded: downloads["bytes"].to_i,
 
+    downloads_per_user: downloads_per_user,
+
     users: users["total"],
   }