Initial Upload
This commit is contained in:
176
lib/Helper/DataSetUploadHandler.php
Normal file
176
lib/Helper/DataSetUploadHandler.php
Normal file
@@ -0,0 +1,176 @@
|
||||
<?php
|
||||
/*
|
||||
* Copyright (C) 2024 Xibo Signage Ltd
|
||||
*
|
||||
* Xibo - Digital Signage - https://xibosignage.com
|
||||
*
|
||||
* This file is part of Xibo.
|
||||
*
|
||||
* Xibo is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* any later version.
|
||||
*
|
||||
* Xibo is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with Xibo. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
namespace Xibo\Helper;
|
||||
|
||||
use Carbon\Carbon;
|
||||
use Exception;
|
||||
use Xibo\Support\Exception\AccessDeniedException;
|
||||
use Xibo\Support\Exception\InvalidArgumentException;
|
||||
use Xibo\Support\Sanitizer\SanitizerInterface;
|
||||
|
||||
/**
|
||||
* Class DataSetUploadHandler
|
||||
* @package Xibo\Helper
|
||||
*/
|
||||
class DataSetUploadHandler extends BlueImpUploadHandler
|
||||
{
|
||||
/**
|
||||
* @param $file
|
||||
* @param $index
|
||||
*/
|
||||
protected function handleFormData($file, $index)
|
||||
{
|
||||
/* @var \Xibo\Controller\DataSet $controller */
|
||||
$controller = $this->options['controller'];
|
||||
|
||||
/* @var SanitizerInterface $sanitizer */
|
||||
$sanitizer = $this->options['sanitizer'];
|
||||
|
||||
// Handle form data, e.g. $_REQUEST['description'][$index]
|
||||
$fileName = $file->name;
|
||||
|
||||
$controller->getLog()->debug('Upload complete for ' . $fileName . '.');
|
||||
|
||||
// Upload and Save
|
||||
try {
|
||||
|
||||
// Authenticate
|
||||
$controller = $this->options['controller'];
|
||||
$dataSet = $controller->getDataSetFactory()->getById($this->options['dataSetId']);
|
||||
|
||||
if (!$controller->getUser()->checkEditable($dataSet)) {
|
||||
throw new AccessDeniedException();
|
||||
}
|
||||
|
||||
// Get all columns
|
||||
$columns = $dataSet->getColumn();
|
||||
|
||||
// Filter columns where dataSetColumnType is "Value"
|
||||
$filteredColumns = array_filter($columns, function ($column) {
|
||||
return $column->dataSetColumnTypeId == '1';
|
||||
});
|
||||
|
||||
// Check if there are any value columns defined in the dataset
|
||||
if (count($filteredColumns) === 0) {
|
||||
$controller->getLog()->error('Import failed: No value columns defined in the dataset.');
|
||||
throw new InvalidArgumentException(__('Import failed: No value columns defined in the dataset.'));
|
||||
}
|
||||
|
||||
// We are allowed to edit - pull all required parameters from the request object
|
||||
$overwrite = $sanitizer->getCheckbox('overwrite');
|
||||
$ignoreFirstRow = $sanitizer->getCheckbox('ignorefirstrow');
|
||||
|
||||
$controller->getLog()->debug('Options provided - overwrite = %d, ignore first row = %d', $overwrite, $ignoreFirstRow);
|
||||
|
||||
// Enumerate over the columns in the DataSet and set a row value for each
|
||||
$spreadSheetMapping = [];
|
||||
|
||||
foreach ($dataSet->getColumn() as $column) {
|
||||
/* @var \Xibo\Entity\DataSetColumn $column */
|
||||
if ($column->dataSetColumnTypeId == 1) {
|
||||
// Has this column been provided in the mappings?
|
||||
|
||||
$spreadSheetColumn = 0;
|
||||
if (isset($_REQUEST['csvImport_' . $column->dataSetColumnId]))
|
||||
$spreadSheetColumn = (($index === null) ? $_REQUEST['csvImport_' . $column->dataSetColumnId] : $_REQUEST['csvImport_' . $column->dataSetColumnId][$index]);
|
||||
|
||||
// If it has been left blank, then skip
|
||||
if ($spreadSheetColumn != 0)
|
||||
$spreadSheetMapping[($spreadSheetColumn - 1)] = $column->heading;
|
||||
}
|
||||
}
|
||||
|
||||
// Delete the data?
|
||||
if ($overwrite == 1)
|
||||
$dataSet->deleteData();
|
||||
|
||||
$firstRow = true;
|
||||
$i = 0;
|
||||
$handle = fopen($controller->getConfig()->getSetting('LIBRARY_LOCATION') . 'temp/' . $fileName, 'r');
|
||||
while (($data = fgetcsv($handle)) !== FALSE ) {
|
||||
$i++;
|
||||
|
||||
// remove any elements that doesn't contain any value from the array
|
||||
$filteredData = array_filter($data, function($value) {
|
||||
return !empty($value);
|
||||
});
|
||||
|
||||
// Skip empty lines without any delimiters or data
|
||||
if (empty($filteredData)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
$row = [];
|
||||
|
||||
// The CSV file might have headings, so ignore the first row.
|
||||
if ($firstRow) {
|
||||
$firstRow = false;
|
||||
|
||||
if ($ignoreFirstRow == 1)
|
||||
continue;
|
||||
}
|
||||
|
||||
for ($cell = 0; $cell < count($data); $cell++) {
|
||||
|
||||
// Insert the data into the correct column
|
||||
if (isset($spreadSheetMapping[$cell])) {
|
||||
// Sanitize the data a bit
|
||||
$item = $data[$cell];
|
||||
|
||||
if ($item == '')
|
||||
$item = null;
|
||||
|
||||
$row[$spreadSheetMapping[$cell]] = $item;
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
$dataSet->addRow($row);
|
||||
} catch (\PDOException $PDOException) {
|
||||
$controller->getLog()->error('Error importing row ' . $i . '. E = ' . $PDOException->getMessage());
|
||||
$controller->getLog()->debug($PDOException->getTraceAsString());
|
||||
|
||||
throw new InvalidArgumentException(__('Unable to import row %d', $i), 'row');
|
||||
}
|
||||
}
|
||||
|
||||
// Close the file
|
||||
fclose($handle);
|
||||
|
||||
// TODO: update list content definitions
|
||||
|
||||
// Save the dataSet
|
||||
$dataSet->lastDataEdit = Carbon::now()->format('U');
|
||||
$dataSet->save(['validate' => false, 'saveColumns' => false]);
|
||||
|
||||
// Tidy up the temporary file
|
||||
@unlink($controller->getConfig()->getSetting('LIBRARY_LOCATION') . 'temp/' . $fileName);
|
||||
|
||||
} catch (Exception $e) {
|
||||
$file->error = $e->getMessage();
|
||||
|
||||
// Don't commit
|
||||
$controller->getState()->setCommitState(false);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user