Add mysql class

2 files changed, 75 insertions, 0 deletions

diff --git a/app/class/database.class.php b/app/class/database.class.php
index c791088..6c6ecd6 100644
--- a/app/class/database.class.php
+++ b/app/class/database.class.php
@@ -23,6 +23,7 @@
 abstract class Database
 {
     private static $instance = NULL;
+    protected $db;
 
     /*
      * Return the database instance object, creating it if this is the
diff --git a/app/class/mysql.class.php b/app/class/mysql.class.php
new file mode 100644
index 0000000..90a4016
--- /dev/null
+++ b/app/class/mysql.class.php
@@ -0,0 +1,74 @@
+<?php
+
+/*
+ * SCROTT IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+ * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+ * OTHER DEALINGS IN THE SOFTWARE.
+ *
+ * For more information, please refer to UNLICENSE
+ */
+
+require_once "class/database.class.php";
+
+/*
+ * This class implements support for MySQL and MariaDB via PHP's mysqli
+ * driver.
+ */
+class Mysql extends Database
+{
+    /*
+     * Constructor
+     */
+    public function __construct(string $host, string $username, string $password, string $dbName)
+    {
+        $this->db = new mysqli($host, $username, $password, $dbName);
+
+        if ($this->db->connect_error)
+            throw new Exception("Can not connect to MySQL database.  Please check your configuration.");
+    }
+
+    /*
+     * Destructor
+     */
+    public function __destruct()
+    {
+        $this->close();
+    }
+
+    /*
+     * Close connection to DB
+     */
+    public function close()
+    {
+        $this->db->close();
+    }
+
+    /*
+     * Make a query of the database.  Return data as an array of arrays.
+     */
+    public function query(string $query) : array
+    {
+        $arr = array();
+        $res = $this->db->query($query);
+
+        if ($res === true || $res === false)
+            return $arr;
+
+        while (($arr[] = $res->fetch_assoc()));
+        return $arr;
+    }
+
+    /*
+     * Escape a string for use in a query
+     */
+    public function esc(string $str) : string
+    {
+        return $this->db->real_escape_string($str);
+    }
+}
+
+?>