draft sc-16587 (#5690)

**Story card:** [sc-16587](https://app.shortcut.com/simpledotorg/story/16587/add-program-wide-totals-and-division-sub-totals-to-drugstock-dashboard)

Author: Gyan-Gupta-Rtsl

app/controllers/concerns/my_facilities_filtering.rb

@@ -50,7 +50,11 @@ def populate_facility_sizes
     end
 
     def set_selected_facility_group
-      @selected_facility_group = params[:facility_group] ? @facility_groups.find_by(slug: params[:facility_group]) : @facility_groups.first
+      @selected_facility_group = if params[:facility_group]
+        @facility_groups.find_by(slug: params[:facility_group])
+      else
+        (@facility_groups&.first&.protocol_drugs&.where(stock_tracked: true)&.any? ? @facility_groups&.first : @facility_groups&.second)
+      end
     end
 
     def set_selected_zones
@@ -67,7 +71,7 @@ def facilities_by_facility_group(facilities)
     end
 
     def scoped_facility_groups
-      all_district_overview_enabled?(@all_districts_params) ? @facility_groups : @selected_facility_group
+      access_all_districts_overview?(@all_districts_params) ? @facility_groups : @selected_facility_group
     end
 
     def facilities_by_zone(facilities)

app/controllers/my_facilities/drug_stocks_controller.rb

@@ -149,7 +149,8 @@ def prepare_district_reports(report_type)
   end
 
   def set_all_districts_params
-    @all_districts_params = params[:facility_group] == "all-districts"
+    @all_districts_params ||= {}
+    @all_districts_params[:facility_group] = params[:facility_group] || "all-districts"
   end
 
   def drug_stock_enabled_facilities

app/helpers/drug_stock_helper.rb

@@ -107,9 +107,13 @@ def accessible_organization_districts
     end
   end
 
+  def district_handling(all_districts_params)
+    all_districts_params&.dig(:facility_group) == "all-districts" && all_district_overview_feature_enabled? && accessible_organization_facilities
+  end
+
   def facility_group_dropdown_title(facility_group:, overview: false)
     if can_view_all_districts_nav?
-      overview ? "All districts" : (facility_group&.name || "Select Districts")
+      overview && overview&.dig(:facility_group) == "all-districts" ? "All districts" : (facility_group&.name || "Select Districts")
     else
       facility_group&.name
     end

app/helpers/flipper_helper.rb

@@ -8,11 +8,11 @@ def all_district_overview_feature_enabled?
   end
 
   def all_district_overview_enabled?(all_districts_params)
-    all_district_overview_feature_enabled? && all_districts_params
+    all_district_overview_feature_enabled? && all_districts_params == "all-districts"
   end
 
   def access_all_districts_overview?(all_districts_params)
-    all_district_overview_enabled?(all_districts_params) && accessible_organization_facilities.present?
+    all_district_overview_enabled?(all_districts_params&.dig(:facility_group)) && accessible_organization_facilities.present?
   end
 
   def can_view_all_districts_nav?

app/views/shared/_my_facilities_filters.erb

@@ -19,25 +19,27 @@
   <div class="dropdown-menu mh-350px oy-scroll" aria-labelledby="dropdownMenuLink">
     <form id="query-filters" action="<%= request.path %>" method="get">
       <% if can_view_all_districts_nav? %>
-        <button
-          class="dropdown-item <%= "active" if @all_districts_params %>"
-          type="submit"
-          onclick="$('#selected-facility-group').val('all-districts'); $('#selected-size').remove(); $('#selected-zone').remove();"
-        >
-          All districts
-        </button>
-        <div class="dropdown-divider"></div>
-      <% end %>
+      <button
+        class="dropdown-item <%= 'active' if district_handling(@all_districts_params) %>"
+        type="submit"
+        onclick="$('#selected-facility-group').val('all-districts'); $('#selected-size').remove(); $('#selected-zone').remove();"
+      >
+        All districts
+      </button>
+      <div class="dropdown-divider"></div>
+    <% end %>
 
-      <% @facility_groups.each do |facility_group| %>
-        <button
-          class="dropdown-item query-filter-zone <%= "active" if @selected_facility_group&.slug == facility_group.slug %>"
-          type="submit"
-          onclick="$('#selected-facility-group').val('<%= facility_group.slug %>'); $('#selected-size').remove(); $('#selected-zone').remove();"
-        >
-          <%= facility_group.name %>
-        </button>
-      <% end %>
+    <% all_district_selected = district_handling(@all_districts_params) %>
+
+    <% @facility_groups.each do |facility_group| %>
+      <button
+        class="dropdown-item query-filter-zone <%= 'active' if !all_district_selected && @selected_facility_group&.slug == facility_group.slug %>"
+        type="submit"
+        onclick="$('#selected-facility-group').val('<%= facility_group.slug %>'); $('#selected-size').remove(); $('#selected-zone').remove();"
+      >
+        <%= facility_group.name %>
+      </button>
+    <% end %>
     </form>
   </div>
 </div>

spec/helpers/drug_stock_helper_spec.rb

@@ -146,9 +146,20 @@
       before { allow(helper).to receive(:can_view_all_districts_nav?).and_return(true) }
 
       it "returns correct title based on overview and presence of group" do
-        expect(helper.facility_group_dropdown_title(facility_group: group, overview: true)).to eq("All districts")
-        expect(helper.facility_group_dropdown_title(facility_group: group, overview: false)).to eq("Allahbad")
-        expect(helper.facility_group_dropdown_title(facility_group: nil, overview: false)).to eq("Select Districts")
+        expect(helper.facility_group_dropdown_title(
+          facility_group: group,
+          overview: {facility_group: "all-districts"}
+        )).to eq("All districts")
+
+        expect(helper.facility_group_dropdown_title(
+          facility_group: group,
+          overview: {facility_group: "some-other"}
+        )).to eq("Allahbad")
+
+        expect(helper.facility_group_dropdown_title(
+          facility_group: nil,
+          overview: {facility_group: "some-other"}
+        )).to eq("Select Districts")
       end
     end
 

spec/views/my_facilities/drug_stocks/drug_stocks.html.erb_spec.rb

@@ -24,6 +24,7 @@ def action_name
   end
 
   before do
+    assign(:all_districts_params, {facility_group: "all-districts"})
     assign(:current_admin, current_admin)
     assign(:for_end_of_month_display, "Aug-2025")
     assign(:for_end_of_month, Date.new(2025, 8, 31))