owncloud/core

<?php
/**
 * @author Bart Visscher <bartv@thisnet.nl>
 * @author Joas Schilling <coding@schilljs.com>
 * @author Morris Jobke <hey@morrisjobke.de>
 * @author Robin Appelman <icewind@owncloud.com>
 * @author Thomas Müller <thomas.mueller@tmit.eu>
 *
 * @copyright Copyright (c) 2018, ownCloud GmbH
 * @license AGPL-3.0
 *
 * This code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program 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, version 3,
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */

namespace OC\DB;

class AdapterSqlite extends Adapter {
    /**
     * @param string $tableName
     */
    public function lockTable($tableName) {
        $this->conn->executeUpdate('BEGIN EXCLUSIVE TRANSACTION');
    }

    public function unlockTable() {
        $this->conn->executeUpdate('COMMIT TRANSACTION');
    }

    public function fixupStatement($statement) {
        $statement = \preg_replace('( I?LIKE \?)', '$0 ESCAPE \'\\\'', $statement);
        $statement = \preg_replace('/`(\w+)` ILIKE \?/', 'LOWER($1) LIKE LOWER(?)', $statement);
        $statement = \str_replace('`', '"', $statement);
        $statement = \str_ireplace('NOW()', 'datetime(\'now\')', $statement);
        $statement = \str_ireplace('GREATEST(', 'MAX(', $statement);
        $statement = \str_ireplace('UNIX_TIMESTAMP()', 'strftime(\'%s\',\'now\')', $statement);
        return $statement;
    }

    /**
     * Insert a row if the matching row does not exist.
     *
     * @param string $table The table name (will replace *PREFIX* with the actual prefix)
     * @param array $input data that should be inserted into the table  (column name => value)
     * @param array|null $compare List of values that should be checked for "if not exists"
     *                If this is null or an empty array, all keys of $input will be compared
     *                Please note: text fields (clob) must not be used in the compare array
     * @return int number of inserted rows
     * @throws \Doctrine\DBAL\DBALException
     */
    public function insertIfNotExist($table, $input, array $compare = null) {
        if (empty($compare)) {
            $compare = \array_keys($input);
        }
        $fieldList = '`' . \implode('`,`', \array_keys($input)) . '`';
        $query = "INSERT INTO `$table` ($fieldList) SELECT "
            . \str_repeat('?,', \count($input)-1).'? '
            . " WHERE NOT EXISTS (SELECT 1 FROM `$table` WHERE ";

        $inserts = \array_values($input);
        foreach ($compare as $key) {
            $query .= '`' . $key . '`';
            if ($input[$key] === null) {
                $query .= ' IS NULL AND ';
            } else {
                $inserts[] = $input[$key];
                $query .= ' = ? AND ';
            }
        }
        $query = \substr($query, 0, \strlen($query) - 5);
        $query .= ')';

        return $this->conn->executeUpdate($query, $inserts);
    }
}