forked from domingopa/WhatsAPI-Official
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSqliteMessageStore.php
More file actions
106 lines (95 loc) · 3.57 KB
/
SqliteMessageStore.php
File metadata and controls
106 lines (95 loc) · 3.57 KB
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
<?php
interface MessageStoreInterface
{
public function saveMessage($from, $to, $txt, $id, $t);
}
class SqliteMessageStore implements MessageStoreInterface
{
const DATA_FOLDER = 'wadata';
private $db;
public function __construct($number, $customPath = null)
{
$fileName = __DIR__.DIRECTORY_SEPARATOR.self::DATA_FOLDER.DIRECTORY_SEPARATOR.'msgstore-'.$number.'.db';
if ($customPath) {
$fileName = $customPath.'msgstore-'.$number.'.db';
}
$createTable = !file_exists($fileName);
$this->db = new \PDO('sqlite:'.$fileName, null, null, [PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC, PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION]);
if ($createTable) {
$this->db->exec('CREATE TABLE messages (`from` TEXT, `to` TEXT, message TEXT, id TEXT, t TEXT)');
$this->db->exec('CREATE TABLE messages_pending(`id` TEXT PRIMARY KEY,`jid` TEXT, `pending` TINYINT(1) DEFAULT 0)');
} else {
//backward compatibility
$result = $this->db->query("SELECT name FROM sqlite_master WHERE type='table' AND name='messages_pending';")->fetchAll();
if ($result == null || $result == false || count($result) == 0) {
$this->db->exec('CREATE TABLE messages_pending(`id` TEXT PRIMARY KEY,`jid` TEXT, `pending` TINYINT(1) DEFAULT 0)');
}
}
}
public function saveMessage($from, $to, $txt, $id, $t)
{
$sql = 'INSERT INTO messages (`from`, `to`, message, id, t) VALUES (:from, :to, :message, :messageId, :t)';
$query = $this->db->prepare($sql);
$query->execute(
[
':from' => $from,
':to' => $to,
':message' => $txt,
':messageId' => $id,
':t' => $t,
]
);
}
public function setPending($id, $jid)
{
$sql = 'UPDATE messages_pending set `pending` = 1, `jid` = :jid where `id` = :id';
$query = $this->db->prepare($sql);
$query->execute(
[
':id' => $id,
':jid' => $jid,
]
);
$sql = 'INSERT OR IGNORE into messages_pending(`id`,`jid`, `pending`) VALUES(:id,:jid,1)';
$query = $this->db->prepare($sql);
$query->execute(
[
':id' => $id,
':jid' => $jid,
]
);
}
public function getPending($jid)
{
$sql = 'SELECT `id` from messages_pending where `jid` = :jid and `pending` = 1';
$query = $this->db->prepare($sql);
$query->execute(
[
':jid' => $jid,
]
);
$pending_ids = [];
while ($row = $query->fetch(PDO::FETCH_ASSOC)) {
if ($row != null && $row !== false) {
$pending_ids[] = $row['id'];
}
}
if (count($pending_ids) == 0) {
return [];
}
$messages = [];
$qMarks = str_repeat('?,', count($pending_ids) - 1).'?';
$sql = "SELECT * from messages where `id` IN ($qMarks)";
$query = $this->db->prepare($sql);
$query->execute($pending_ids);
while ($row = $query->fetch(PDO::FETCH_ASSOC)) {
if ($row != null && $row !== false) {
$messages[] = $row;
}
}
$sql = 'DELETE FROM messages_pending where `pending` = 1 and jid = :jid';
$query = $this->db->prepare($sql);
$query->execute([':jid' => $jid]);
return $messages;
}
}