#434

backend/app/services/counts_table_export.rb

@@ -10,6 +10,10 @@ class CountsTableExport
     raise ArgumentError if (params.keys - CountsTableQuery::KEYS - EXTRA).any?
     raise ArgumentError unless params.values.all? { _1.is_a?(String) }
 
+    unless params.key?('children')
+      params = params.merge('children' => CountsExportProjection.default_children(
+          params.fetch('resource', 'posts')).join(','))
+    end
     @params = params
     @labels = JSON.parse(params.fetch('labels', '{}'))
     raise ArgumentError unless @labels.is_a?(Hash)
@@ -27,8 +31,11 @@ class CountsTableExport
     @query = CountsTableQuery.new(params.slice(*CountsTableQuery::KEYS)
         .merge('page' => '1', 'limit' => '100'))
     @summary = @query.mode == 'summary'
-    @columns = params.fetch('columns', '').split(',')
-    @headers = JSON.parse(params.fetch('headers', '[]'))
+    default_columns = @summary ? @query.axes.all.map { _1.fetch('field') } + ['count'] :
+        CountsExportProjection.default_columns(resource)
+    @columns = params.fetch('columns', default_columns.join(',')).split(',')
+    default_headers = @columns.map { @labels.fetch(_1, _1) }
+    @headers = JSON.parse(params.fetch('headers', default_headers.to_json))
     raise ArgumentError unless @headers.is_a?(Array) && @headers.all? { _1.is_a?(String) }
     if @summary && @headers.any? && @headers.length != @query.axes.all.length + 1
       raise ArgumentError
@@ -45,13 +52,16 @@ class CountsTableExport
     end
   end
 
-  def body
+  def body(progress: nil)
     Enumerator.new do |output|
       ApplicationRecord.transaction(isolation: :repeatable_read) do
         @language_names = Language.pluck(:code, :name).to_h
+        processed = 0
         started = false
         first = true
         @query.each_page do |page|
+          total = @summary ? nil : page[:count]
+          progress&.call(processed, total)
           unless started
             metadata = page.except(:versions, :cells, :axis_labels, :column_values).merge(
                 resource:, filters: @params.slice(*CountsTableQuery::KEYS),
@@ -84,8 +94,11 @@ class CountsTableExport
               output << CSV.generate_line(values.map { safe(_1) }, row_sep: "\r\n")
             end
             first = false
+            processed += 1
+            progress&.call(processed, total)
           end
         end
+        progress&.call(processed, processed)
         output << ']}' if format == 'json'
       end
     end