1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
<?php
namespace Codeception\Lib\Driver;
use Codeception\Configuration;
use Codeception\Exception\ModuleException;
class Sqlite extends Db
{
protected $hasSnapshot = false;
protected $filename = '';
protected $con = null;
public function __construct($dsn, $user, $password, $options = null)
{
$filename = substr($dsn, 7);
if ($filename === ':memory:') {
throw new ModuleException(__CLASS__, ':memory: database is not supported');
}
$this->filename = Configuration::projectDir() . $filename;
$this->dsn = 'sqlite:' . $this->filename;
parent::__construct($this->dsn, $user, $password, $options);
}
public function cleanup()
{
$this->dbh = null;
file_put_contents($this->filename, '');
$this->dbh = self::connect($this->dsn, $this->user, $this->password);
}
public function load($sql)
{
if ($this->hasSnapshot) {
$this->dbh = null;
file_put_contents($this->filename, file_get_contents($this->filename . '_snapshot'));
$this->dbh = new \PDO($this->dsn, $this->user, $this->password);
} else {
if (file_exists($this->filename . '_snapshot')) {
unlink($this->filename . '_snapshot');
}
parent::load($sql);
copy($this->filename, $this->filename . '_snapshot');
$this->hasSnapshot = true;
}
}
/**
* @param string $tableName
*
* @return array[string]
*/
public function getPrimaryKey($tableName)
{
if (!isset($this->primaryKeys[$tableName])) {
if ($this->hasRowId($tableName)) {
return $this->primaryKeys[$tableName] = ['_ROWID_'];
}
$primaryKey = [];
$query = 'PRAGMA table_info(' . $this->getQuotedName($tableName) . ')';
$stmt = $this->executeQuery($query, []);
$columns = $stmt->fetchAll(\PDO::FETCH_ASSOC);
foreach ($columns as $column) {
if ($column['pk'] !== '0') {
$primaryKey []= $column['name'];
}
}
$this->primaryKeys[$tableName] = $primaryKey;
}
return $this->primaryKeys[$tableName];
}
/**
* @param $tableName
* @return bool
*/
private function hasRowId($tableName)
{
$params = ['type' => 'table', 'name' => $tableName];
$select = $this->select('sql', 'sqlite_master', $params);
$result = $this->executeQuery($select, $params);
$sql = $result->fetchColumn(0);
return strpos($sql, ') WITHOUT ROWID') === false;
}
}