From 26835c0a78fbe94287754bb62388d900a12d5510 Mon Sep 17 00:00:00 2001
From: Thiago Youssef <43591948+thiagoyoussef@users.noreply.github.com>
Date: Fri, 15 Nov 2024 14:10:53 -0300
Subject: [PATCH 1/2] feature: summarizable associations

---
 app/controllers/avo/charts_controller.rb      | 22 ++++++++++-
 app/views/avo/partials/_table_header.html.erb | 10 +----
 spec/features/avo/field_summarizing_spec.rb   | 37 +++++++++++++++++++
 3 files changed, 60 insertions(+), 9 deletions(-)

diff --git a/app/controllers/avo/charts_controller.rb b/app/controllers/avo/charts_controller.rb
index 5e872c8742..e450af5084 100644
--- a/app/controllers/avo/charts_controller.rb
+++ b/app/controllers/avo/charts_controller.rb
@@ -3,10 +3,30 @@
 module Avo
   class ChartsController < BaseController
     def distribution_chart
-      @values_summary = resource.model_class.group(params[:field_id]).reorder("count_all desc").count
+      @values_summary = records.group(params[:field_id]).reorder("count_all desc").count
+
       @field_id = params[:field_id]
 
       render "avo/partials/distribution_chart", layout: "avo/blank"
     end
+
+    private
+
+    def records
+      if is_associated_record?
+        related_records
+      else
+        resource.model_class
+      end
+    end
+
+    def related_records
+      relation_class = BaseResource.get_model_by_name(params[:via_relation_class])
+      parent = relation_class.find(params[:via_record_id])
+
+      association_name = BaseResource.valid_association_name(parent, resource.model_key)
+
+      parent.send(association_name)
+    end
   end
 end
diff --git a/app/views/avo/partials/_table_header.html.erb b/app/views/avo/partials/_table_header.html.erb
index bafb43bc09..a29151f1bb 100644
--- a/app/views/avo/partials/_table_header.html.erb
+++ b/app/views/avo/partials/_table_header.html.erb
@@ -1,9 +1,3 @@
-<%
-  # Currently there isn't a way to summarize records on associations.
-  # We'd love to support this feature so please send in a PR.
-  should_summarize = @parent_record.blank?
-%>
-
 <thead
   data-component-name="avo/partials/table_header"
   class="bg-white border-b border-gray-200 pb-1">
@@ -83,14 +77,14 @@
             <%= field.table_header_label %>
           <% end %>
         <% end %>
-        <% if field.summarizable && should_summarize %>
+        <% if field.summarizable %>
           <div id="summary-header-<%= field.id %>" class="<%= text_classes %> cursor-pointer" data-action="click->toggle#togglePanel">
             <%= svg 'heroicons/solid/chart-bar', class: 'h-3 ml-1' %>
           </div>
           <turbo-frame
             id="summary-frame-<%= field.id %>"
             loading="lazy"
-            src="<%= distribution_chart_path(resource_name: field.resource.model_class, field_id: field.id) %>"
+            src="<%= distribution_chart_path(resource_name: field.resource.model_class, field_id: field.id, via_relation_class: @parent_record.class, via_record_id: @parent_record&.id) %>"
             class="absolute inset-auto bg-white w-full sm:w-[400px] top-0 mt-6 p-4 z-40 shadow-modal rounded overflow-hidden hidden border border-gray-50"
             data-toggle-target="panel"
           >
diff --git a/spec/features/avo/field_summarizing_spec.rb b/spec/features/avo/field_summarizing_spec.rb
index 46537a3971..1c92fd08ac 100644
--- a/spec/features/avo/field_summarizing_spec.rb
+++ b/spec/features/avo/field_summarizing_spec.rb
@@ -42,5 +42,42 @@
       expect(page).not_to have_css 'th[data-table-header-field-id="progress"] div svg'
       expect(page).not_to have_css 'th[data-table-header-field-id="description"] div svg'
     end
+
+    context "when summarizing on association pages" do
+      let(:user) { create :user }
+      let!(:project) { create :project, status: :closed, users: [user] }
+
+      it "only shows values from associated record" do
+        visit avo.resources_user_path(user)
+
+        scroll_to first_tab_group
+        click_tab "Projects", within_target: first_tab_group
+
+        wait_for_turbo_frame_id("has_and_belongs_to_many_field_show_projects")
+
+        expect(page).to have_css "turbo-frame[id='summary-frame-status']", visible: false
+
+        find("#summary-header-status").click
+
+        expect(page).to have_css "turbo-frame[id='summary-frame-status']", visible: true
+
+        # I can't make the lazy loading work, looks like it's not triggered at all
+        wait_for_turbo_frame_id("summary-frame-status")
+
+        expect(page).to have_css "#status-summary", visible: true
+        expect(page).to have_css "#chart-status", visible: true
+
+        within "#status-summary" do
+          expect(page).to_not have_content "REJECTED\n3"
+          expect(page).to have_content "CLOSED\n1"
+          expect(page).to_not have_content "LOADING\n4"
+        end
+
+        find('th[data-table-header-field-id="status"] div svg').click
+
+        expect(page).not_to have_css "#status-summary"
+        expect(page).not_to have_css "#chart-status"
+      end
+    end
   end
 end

From b3f357bb6c64349b9e847e5ec5699be8fbcb434c Mon Sep 17 00:00:00 2001
From: Thiago Youssef <43591948+thiagoyoussef@users.noreply.github.com>
Date: Tue, 19 Nov 2024 09:52:47 -0300
Subject: [PATCH 2/2] summarizable scoped query

---
 app/controllers/avo/charts_controller.rb | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/app/controllers/avo/charts_controller.rb b/app/controllers/avo/charts_controller.rb
index e450af5084..09f96811f2 100644
--- a/app/controllers/avo/charts_controller.rb
+++ b/app/controllers/avo/charts_controller.rb
@@ -3,7 +3,7 @@
 module Avo
   class ChartsController < BaseController
     def distribution_chart
-      @values_summary = records.group(params[:field_id]).reorder("count_all desc").count
+      @values_summary = scoped_query.group(params[:field_id].to_sym).reorder("count_all desc").count
 
       @field_id = params[:field_id]
 
@@ -12,11 +12,13 @@ def distribution_chart
 
     private
 
-    def records
+    def scoped_query
       if is_associated_record?
-        related_records
+        query = resource.authorization&.apply_policy related_records
+
+        Avo::ExecutionContext.new(target: resource.search_query, query: query).handle
       else
-        resource.model_class
+        resource.query_scope
       end
     end