1 files changed, 176 insertions, 0 deletions
diff --git a/srcs/phpmyadmin/libraries/classes/Plugins/Import/ImportLdi.php b/srcs/phpmyadmin/libraries/classes/Plugins/Import/ImportLdi.php
new file mode 100644
index 0000000..91260c5
--- /dev/null
+++ b/srcs/phpmyadmin/libraries/classes/Plugins/Import/ImportLdi.php
@@ -0,0 +1,176 @@
+<?php
+/* vim: set expandtab sw=4 ts=4 sts=4: */
+/**
+ * CSV import plugin for phpMyAdmin using LOAD DATA
+ *
+ * @package    PhpMyAdmin-Import
+ * @subpackage LDI
+ */
+declare(strict_types=1);
+
+namespace PhpMyAdmin\Plugins\Import;
+
+use PhpMyAdmin\Import;
+use PhpMyAdmin\Message;
+use PhpMyAdmin\Plugins\Import\AbstractImportCsv;
+use PhpMyAdmin\Properties\Options\Items\BoolPropertyItem;
+use PhpMyAdmin\Properties\Options\Items\TextPropertyItem;
+use PhpMyAdmin\Util;
+
+// We need relations enabled and we work only on database
+if (! isset($GLOBALS['plugin_param']) || $GLOBALS['plugin_param'] !== 'table') {
+    $GLOBALS['skip_import'] = true;
+
+    return;
+}
+
+/**
+ * Handles the import for the CSV format using load data
+ *
+ * @package    PhpMyAdmin-Import
+ * @subpackage LDI
+ */
+class ImportLdi extends AbstractImportCsv
+{
+    /**
+     * Constructor
+     */
+    public function __construct()
+    {
+        parent::__construct();
+        $this->setProperties();
+    }
+
+    /**
+     * Sets the import plugin properties.
+     * Called in the constructor.
+     *
+     * @return void
+     */
+    protected function setProperties()
+    {
+        if ($GLOBALS['cfg']['Import']['ldi_local_option'] == 'auto') {
+            $GLOBALS['cfg']['Import']['ldi_local_option'] = false;
+
+            $result = $GLOBALS['dbi']->tryQuery(
+                'SELECT @@local_infile;'
+            );
+            if ($result != false && $GLOBALS['dbi']->numRows($result) > 0) {
+                $tmp = $GLOBALS['dbi']->fetchRow($result);
+                if ($tmp[0] == 'ON') {
+                    $GLOBALS['cfg']['Import']['ldi_local_option'] = true;
+                }
+            }
+            $GLOBALS['dbi']->freeResult($result);
+            unset($result);
+        }
+
+        $generalOptions = parent::setProperties();
+        $this->properties->setText('CSV using LOAD DATA');
+        $this->properties->setExtension('ldi');
+
+        $leaf = new TextPropertyItem(
+            "columns",
+            __('Column names: ')
+        );
+        $generalOptions->addProperty($leaf);
+
+        $leaf = new BoolPropertyItem(
+            "ignore",
+            __('Do not abort on INSERT error')
+        );
+        $generalOptions->addProperty($leaf);
+
+        $leaf = new BoolPropertyItem(
+            "local_option",
+            __('Use LOCAL keyword')
+        );
+        $generalOptions->addProperty($leaf);
+    }
+
+    /**
+     * Handles the whole import logic
+     *
+     * @param array $sql_data 2-element array with sql data
+     *
+     * @return void
+     */
+    public function doImport(array &$sql_data = [])
+    {
+        global $finished, $import_file, $charset_conversion, $table;
+        global $ldi_local_option, $ldi_replace, $ldi_ignore, $ldi_terminated,
+               $ldi_enclosed, $ldi_escaped, $ldi_new_line, $skip_queries, $ldi_columns;
+
+        $compression = $GLOBALS['import_handle']->getCompression();
+
+        if ($import_file == 'none'
+            || $compression != 'none'
+            || $charset_conversion
+        ) {
+            // We handle only some kind of data!
+            $GLOBALS['message'] = Message::error(
+                __('This plugin does not support compressed imports!')
+            );
+            $GLOBALS['error'] = true;
+
+            return;
+        }
+
+        $sql = 'LOAD DATA';
+        if (isset($ldi_local_option)) {
+            $sql .= ' LOCAL';
+        }
+        $sql .= ' INFILE \'' . $GLOBALS['dbi']->escapeString($import_file)
+            . '\'';
+        if (isset($ldi_replace)) {
+            $sql .= ' REPLACE';
+        } elseif (isset($ldi_ignore)) {
+            $sql .= ' IGNORE';
+        }
+        $sql .= ' INTO TABLE ' . Util::backquote($table);
+
+        if (strlen((string) $ldi_terminated) > 0) {
+            $sql .= ' FIELDS TERMINATED BY \'' . $ldi_terminated . '\'';
+        }
+        if (strlen((string) $ldi_enclosed) > 0) {
+            $sql .= ' ENCLOSED BY \''
+                . $GLOBALS['dbi']->escapeString($ldi_enclosed) . '\'';
+        }
+        if (strlen((string) $ldi_escaped) > 0) {
+            $sql .= ' ESCAPED BY \''
+                . $GLOBALS['dbi']->escapeString($ldi_escaped) . '\'';
+        }
+        if (strlen((string) $ldi_new_line) > 0) {
+            if ($ldi_new_line == 'auto') {
+                $ldi_new_line
+                    = (PHP_EOL == "\n")
+                    ? '\n'
+                    : '\r\n';
+            }
+            $sql .= ' LINES TERMINATED BY \'' . $ldi_new_line . '\'';
+        }
+        if ($skip_queries > 0) {
+            $sql .= ' IGNORE ' . $skip_queries . ' LINES';
+            $skip_queries = 0;
+        }
+        if (strlen((string) $ldi_columns) > 0) {
+            $sql .= ' (';
+            $tmp = preg_split('/,( ?)/', $ldi_columns);
+            $cnt_tmp = count($tmp);
+            for ($i = 0; $i < $cnt_tmp; $i++) {
+                if ($i > 0) {
+                    $sql .= ', ';
+                }
+                /* Trim also `, if user already included backquoted fields */
+                $sql .= Util::backquote(
+                    trim($tmp[$i], " \t\r\n\0\x0B`")
+                );
+            } // end for
+            $sql .= ')';
+        }
+
+        $this->import->runQuery($sql, $sql, $sql_data);
+        $this->import->runQuery('', '', $sql_data);
+        $finished = true;
+    }
+}