Merge pull request #22487 from mattwire/repeattransactiontemplatecontribution

[civicrm-core.git] / CRM / Import / DataSource / CSV.php

diff --git a/CRM/Import/DataSource/CSV.php b/CRM/Import/DataSource/CSV.php
index 2f40999c8023b7ee9b3783e5fce205c70a3465c6..79e68a0d7ff21af016d3f4153316d18906047766 100644 (file)
--- a/CRM/Import/DataSource/CSV.php
+++ b/CRM/Import/DataSource/CSV.php
@@ -19,21 +19,20 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
     NUM_ROWS_TO_INSERT = 100;
 
   /**
-   * Provides information about the data source.
+   * Form fields declared for this datasource.
    *
-   * @return array
-   *   collection of info about this data source
+   * @var string[]
    */
-  public function getInfo() {
-    return ['title' => ts('Comma-Separated Values (CSV)')];
-  }
+  protected $submittableFields = ['skipColumnHeader', 'uploadField'];
 
   /**
-   * Set variables up before form is built.
+   * Provides information about the data source.
    *
-   * @param CRM_Core_Form $form
+   * @return array
+   *   collection of info about this data source
    */
-  public function preProcess(&$form) {
+  public function getInfo(): array {
+    return ['title' => ts('Comma-Separated Values (CSV)')];
   }
 
   /**
@@ -49,9 +48,7 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
   public function buildQuickForm(&$form) {
     $form->add('hidden', 'hidden_dataSource', 'CRM_Import_DataSource_CSV');
 
-    $config = CRM_Core_Config::singleton();
-
-    $uploadFileSize = CRM_Utils_Number::formatUnitSize($config->maxFileSize . 'm', TRUE);
+    $uploadFileSize = CRM_Utils_Number::formatUnitSize(Civi::settings()->get('maxFileSize') . 'm', TRUE);
     //Fetch uploadFileSize from php_ini when $config->maxFileSize is set to "no limit".
     if (empty($uploadFileSize)) {
       $uploadFileSize = CRM_Utils_Number::formatUnitSize(ini_get('upload_max_filesize'), TRUE);
@@ -71,38 +68,33 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
   }
 
   /**
-   * Process the form submission.
-   *
-   * @param array $params
-   * @param string $db
-   * @param \CRM_Core_Form $form
+   * Initialize the datasource, based on the submitted values stored in the user job.
    *
+   * @throws \API_Exception
    * @throws \CRM_Core_Exception
    */
-  public function postProcess(&$params, &$db, &$form) {
-    $file = $params['uploadFile']['name'];
-    $result = self::_CsvToTable($db,
-      $file,
-      CRM_Utils_Array::value('skipColumnHeader', $params, FALSE),
-      CRM_Utils_Array::value('import_table_name', $params),
-      CRM_Utils_Array::value('fieldSeparator', $params, ',')
+  public function initialize(): void {
+    $result = self::_CsvToTable(
+      $this->getSubmittedValue('uploadFile')['name'],
+      $this->getSubmittedValue('skipColumnHeader'),
+      $this->getSubmittedValue('fieldSeparator') ?? ','
     );
+    $this->addTrackingFieldsToTable($result['import_table_name']);
 
-    $form->set('originalColHeader', CRM_Utils_Array::value('original_col_header', $result));
-    $form->set('importTableName', $result['import_table_name']);
+    $this->updateUserJobMetadata('DataSource', [
+      'table_name' => $result['import_table_name'],
+      'column_headers' => $this->getSubmittedValue('skipColumnHeader') ? $result['column_headers'] : [],
+      'number_of_columns' => $result['number_of_columns'],
+    ]);
   }
 
   /**
    * Create a table that matches the CSV file and populate it with the file's contents
    *
-   * @param object $db
-   *   Handle to the database connection.
    * @param string $file
    *   File name to load.
    * @param bool $headers
    *   Whether the first row contains headers.
-   * @param string $tableName
-   *   Name of table from which data imported.
    * @param string $fieldSeparator
    *   Character that separates the various columns in the file.
    *
@@ -111,10 +103,8 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
    * @throws \CRM_Core_Exception
    */
   private static function _CsvToTable(
-    &$db,
     $file,
     $headers = FALSE,
-    $tableName = NULL,
     $fieldSeparator = ','
   ) {
     $result = [];
@@ -138,7 +128,7 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
     // create the column names from the CSV header or as col_0, col_1, etc.
     if ($headers) {
       //need to get original headers.
-      $result['original_col_header'] = $firstrow;
+      $result['column_headers'] = $firstrow;
 
       $strtolower = function_exists('mb_strtolower') ? 'mb_strtolower' : 'strtolower';
       $columns = array_map($strtolower, $firstrow);
@@ -183,9 +173,6 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
       }
     }
 
-    if ($tableName) {
-      CRM_Core_DAO::executeQuery("DROP TABLE IF EXISTS $tableName");
-    }
     $table = CRM_Utils_SQL_TempTable::build()->setDurable();
     $tableName = $table->getName();
     CRM_Core_DAO::executeQuery("DROP TABLE IF EXISTS $tableName");
@@ -245,7 +232,7 @@ class CRM_Import_DataSource_CSV extends CRM_Import_DataSource {
 
     //get the import tmp table name.
     $result['import_table_name'] = $tableName;
-
+    $result['number_of_columns'] = $numColumns;
     return $result;
   }