1 files changed, 124 insertions, 0 deletions
diff --git a/db.php b/db.php
new file mode 100644
index 0000000..8cd8bc9
--- /dev/null
+++ b/db.php
@@ -0,0 +1,124 @@
+<?php
+
+class Queue {
+    public $name;
+    public $endpoint_url;
+
+    function toPlainText($url_generator) {
+        return "Name: " . $this->name . "\n" .
+            "Endpoint URL: " . (isset($this->endpoint_url) ? $this->endpoint_url : "Not set") . "\n" .
+            "Enqueue: " . $url_generator->enqueue($this->name) . "\n";
+    }
+
+    static function valid_name($str) {
+        return preg_match('/^[a-zA-Z0-9]+$/', $str);
+    }
+}
+
+function item_from_db($id, $queue_name, $created, $headers) {
+    return new Item($id, $queue_name, $created, json_decode($headers));
+}
+
+class Item {
+    public $id;
+    public $queue_name;
+    public $created;
+    public $headers;
+
+    function __construct($id, $queue_name, $created, $headers) {
+        $this->id = $id;
+        $this->queue_name = $queue_name;
+        $this->created = $created;
+        $this->headers = $headers;
+    }
+
+    function toPlainText($url_generator) {
+        $headers = "";
+        foreach($this->headers as $k => $v) {
+            $headers = $headers . " " . $k . ": " . $v . "\n";
+        }
+        return "Headers:\n" . $headers;
+    }
+}
+
+class DB {
+    var $db;
+
+    function __construct() {
+        $this->db = new PDO("sqlite:/tmp/mq/mq.db");
+        $this->db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+        $this->db->exec("PRAGMA foreign_keys = ON;");
+        $this->db->beginTransaction();
+    }
+
+    function commit() {
+        $this->db->commit();
+    }
+
+    function delete_queue_and_items($name) {
+        $stmt = $this->db->prepare("DELETE FROM item WHERE queue_name = :name");
+        $stmt->bindParam(":name", $name);
+        $stmt->execute();
+
+        $stmt = $this->db->prepare("DELETE FROM queue WHERE name = :name");
+        $stmt->bindParam(":name", $name);
+        $stmt->execute();
+    }
+
+    function insert_queue($name, $endpoint_url) {
+        $stmt = $this->db->prepare("INSERT INTO queue(name, endpoint_url) VALUES(:name, :endpoint_url)");
+        $stmt->bindParam(":name", $name);
+        $stmt->bindParam(":endpoint_url", $endpoint_url);
+        $stmt->execute();
+    }
+
+    function update_queue($name, $endpoint_url) {
+        $stmt = $this->db->prepare("UPDATE queue SET endpoint_url = :endpoint_url WHERE name = :name");
+        $stmt->bindParam(":endpoint_url", $endpoint_url);
+        $stmt->bindParam(":name", $name);
+        $stmt->execute();
+    }
+
+    function insert_item($id, array $headers, array $queues) {
+        foreach($queues as $queue) {
+            $stmt = $this->db->prepare("INSERT INTO item(id, queue_name, created, headers) VALUES(:id, :queue_name, datetime('now'), :headers)");
+            $stmt->bindParam(":id", $id);
+            $stmt->bindParam(":queue_name", $queue);
+            $stmt->bindParam(":headers", json_encode($headers));
+            $stmt->execute();
+        }
+    }
+
+    function select_queues() {
+        $stmt = $this->db->prepare("SELECT name, endpoint_url FROM queue ORDER BY name");
+        $stmt->execute();
+        return $stmt->fetchAll(PDO::FETCH_CLASS, "Queue");
+    }
+
+    function select_queue($queue) {
+        $stmt = $this->db->prepare("SELECT name, endpoint_url FROM queue WHERE name = :name");
+        $stmt->setFetchMode(PDO::FETCH_CLASS, "Queue");
+        $stmt->bindParam(":name", $queue);
+        $stmt->execute();
+        return $stmt->fetch(PDO::FETCH_CLASS);
+    }
+
+    function select_item($queue_name, $id) {
+        $stmt = $this->db->prepare("SELECT id, queue_name, created, headers FROM item WHERE queue_name = :queue_name AND id = :id");
+        $stmt->bindParam(":queue_name", $queue_name);
+        $stmt->bindParam(":id", $id);
+        $stmt->execute();
+        return $stmt->fetchAll(PDO::FETCH_FUNC, 'item_from_db');
+    }
+
+    function select_items($queue_name, $limit = 10, $offset = 0) {
+        $stmt = $this->db->prepare("SELECT id, queue_name, created, headers FROM item WHERE queue_name = :queue_name LIMIT :limit OFFSET :offset");
+        $stmt->bindParam(":queue_name", $queue_name);
+        $stmt->bindParam(":limit", $limit);
+        $stmt->bindParam(":offset", $offset);
+        $stmt->execute();
+        return $stmt->fetchAll(PDO::FETCH_FUNC, 'item_from_db');
+    }
+}
+
+?>