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
|
<?php
namespace Database;
use PDO;
class User extends Table {
public $UID;
public $Username;
public $Password;
public $Role;
static function create(string $Username, string $Password, string $Role) : int {
return Table::_create(
"Users",
"(Username, Password, Role)",
"(\"$Username\", \"$Password\", \"$Role\")",
);
}
function fromDB(string $username) : User {
return Table::_fromDB(
"SELECT * FROM Users WHERE Username = \"$username\"",
"Database\User"
);
}
static function get_all() : array {
return Table::_get_all("Database\User");
}
}
class Webpage extends Table {
public $WID;
public $Path;
public $URL;
public $Date;
public $Visits;
public $RequesterUID;
static function create(string $Path, string $URL, int $RequesterUID) : int {
return Table::_create(
'Webpages',
'(Path, URL, Date, Visits, RequesterUID)',
"(\"$Path\", \"$URL\", NOW(), 0, \"$RequesterUID\")"
);
}
static function fromDB(string $URL) : Webpage {
return Table::_fromDB(
"SELECT * FROM Webpages WHERE URL = \"$URL\"",
"Database\Webpage"
);
}
}
abstract class Table {
// Cannot be created, because FETCH_CLASS will assign to all attributes
// and then call the constructor
final private function __construct() {}
static protected function _fromDB(string $query_str, string $class) : object {
$conn = Table::connect();
$query = $conn->query($query_str);
// TODO: research if this is enough to close the connection, $query may store a reference
$conn = null;
if ($query->rowCount() == 0) {
throw new Exception("Value for $class doesn't exist!");
}
assert($query->rowCount() == 1, "Vaue for $class must be uniqely specified!");
$query->setFetchMode(PDO::FETCH_CLASS, $class);
return $query->fetch();
}
static protected function _create(string $table, string $columns, string $value) : int {
$conn = Table::connect();
$query = $conn->query("INSERT INTO $table $columns VALUES $value");
// NOTE: If we ever insert more than one values, lastInsertId will returne the first id
$id = $conn->lastInsertId();
$conn = null;
return $id;
}
static protected function _get_all(string $class) : array {
$conn = Table::connect();
$query = $conn->query("SELECT * FROM Users");
$conn = null;
$query->setFetchMode(PDO::FETCH_CLASS, $class);
return $query->fetchAll();
}
static protected function connect() : PDO {
$conn = new PDO(
"mysql:unix_socket=" . getenv('MYSQL_UNIX_SOCKET') . ";dbname=nwfh",
getenv('USER'),
"");
// set the PDO error mode to exception
$conn->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
return $conn;
}
}
|
