Add formatData and formatResponse for post data formatting

Author: patrick-yi-82

src/DataTableAction.php

@@ -50,6 +50,22 @@ class DataTableAction extends Action
      */
     public $applyFilter;
 
+    /**
+     * Format data
+     * Signature is following:
+     * function ($query, $columns)
+     * @var callable
+     */
+    public $formatData;
+
+    /**
+     * Format response
+     * Signature is following:
+     * function ($response)
+     * @var callable
+     */
+    public $formatResponse;
+
     public function init()
     {
         if ($this->query === null) {
@@ -89,12 +105,13 @@ public function run()
                 'draw' => (int)$draw,
                 'recordsTotal' => (int)$originalQuery->count(),
                 'recordsFiltered' => (int)$dataProvider->getTotalCount(),
-                'data' => $filterQuery->all(),
+                'data' => $this->formatData($filterQuery, $columns),
             ];
         } catch (\Exception $e) {
             return ['error' => $e->getMessage()];
         }
-        return $response;
+
+        return $this->formatResponse($response);
     }
 
     /**
@@ -143,4 +160,31 @@ public function applyOrder(ActiveQuery $query, $columns, $order)
         }
         return $query;
     }
+
+    /**
+     * @param ActiveQuery $query
+     * @param array $columns
+     * @return ActiveQuery
+     */
+    public function formatData(ActiveQuery $query, $columns)
+    {
+        if ($this->formatData !== null) {
+            return call_user_func($this->formatData, $query, $columns);
+        }
+
+        return $query->all();
+    }
+
+    /**
+     * @param array $response
+     * @return ActiveQuery
+     */
+    public function formatResponse($response)
+    {
+        if ($this->formatResponse !== null) {
+            return call_user_func($this->formatResponse, $response);
+        }
+
+        return $response;
+    }
 }