Fix row evolution for flattened social networks report (#14266)

1 files changed, 26 insertions, 16 deletions

diff --git a/plugins/Referrers/API.php b/plugins/Referrers/API.php
index 3a7e19910e..b59ac8a36b 100644
--- a/plugins/Referrers/API.php
+++ b/plugins/Referrers/API.php
@@ -622,24 +622,34 @@ class API extends \Piwik\Plugin\API
         }));
         $urlsTable = $urlsTable->mergeSubtables();
 
-        foreach ($dataTable->getRows() as $row) {
-            $row->removeSubtable();
-
-            $social = $row->getColumn('label');
-            $newTable = $urlsTable->getEmptyClone();
-
-            $rows = $urlsTable->getRows();
-            foreach ($rows as $id => $urlsTableRow) {
-                $url = $urlsTableRow->getColumn('label');
-                if (Social::getInstance()->isSocialUrl($url, $social)) {
-                    $newTable->addRow($urlsTableRow);
-                    $urlsTable->deleteRow($id);
+        if ($dataTable instanceof DataTable\Map) {
+            $dataTables = $dataTable->getDataTables();
+            $urlsTables = $urlsTable->getDataTables();
+        } else {
+            $dataTables = [$dataTable];
+            $urlsTables = [$urlsTable];
+        }
+
+        foreach ($dataTables as $label => $dataTable) {
+            foreach ($dataTable->getRows() as $row) {
+                $row->removeSubtable();
+
+                $social = $row->getColumn('label');
+                $newTable = $urlsTables[$label]->getEmptyClone();
+
+                $rows = $urlsTables[$label]->getRows();
+                foreach ($rows as $id => $urlsTableRow) {
+                    $url = $urlsTableRow->getColumn('label');
+                    if (Social::getInstance()->isSocialUrl($url, $social)) {
+                        $newTable->addRow($urlsTableRow);
+                        $urlsTables[$label]->deleteRow($id);
+                    }
                 }
-            }
 
-            if ($newTable->getRowsCount()) {
-                $newTable->filter('Piwik\Plugins\Referrers\DataTable\Filter\UrlsForSocial', array($expanded));
-                $row->setSubtable($newTable);
+                if ($newTable->getRowsCount()) {
+                    $newTable->filter('Piwik\Plugins\Referrers\DataTable\Filter\UrlsForSocial', array($expanded));
+                    $row->setSubtable($newTable);
+                }
             }
         }