Payment: use Model

2 files changed, 19 insertions, 101 deletions

diff --git a/classes/Offer.php b/classes/Offer.php
index 9487f79..7b23a43 100644
--- a/classes/Offer.php
+++ b/classes/Offer.php
@@ -311,7 +311,7 @@ class Offer {
 		if (is_null($payment)) {
 			return null;
 		} else {
-			return $payment->getDate();
+			return $payment->date;
 		}
 	}
 
diff --git a/classes/Payment.php b/classes/Payment.php
index 11c57a3..e60539f 100644
--- a/classes/Payment.php
+++ b/classes/Payment.php
@@ -24,60 +24,10 @@
 /**
  * An interface to the payment table in the database
  */
-class Payment {
-	/**
-	 * @var pdo $pdo                The PDO class for database communication
-	 * @var int $id                 The id of the payment
-	 * @var int $offerId            The id of the offer this payment is linked to
-	 * @var int $date               A unix timestamp describing the date of the payment
-	 */
-	protected $pdo, $offerId, $id, $date;
-
-	/**
-	 * Create a new instance
-	 *
-	 * @param PDO $pdo              The PDO class, to access the database
-	 * @param int $id               The id of the payment to fetch
-	 *
-	 * @throws PDOException         If something went wrong with the database
-	 * @throws Exception            If the payment could not be found
-	 */
-	public function __construct($pdo, $id) {
-		$this->pdo = $pdo;
-
-		$stmt = $this->pdo->prepare("SELECT * FROM `".Constants::db_prefix."payment` WHERE `id`=?");
-		$stmt->execute(array($id));
-		if ($stmt->rowCount() == 0) {
-			throw new Exception("The payment with id '$id' could not be found.");
-		}
-		$payment = $stmt->fetch(PDO::FETCH_ASSOC);
-
-		$this->id = $payment['id'];
-		$this->offerId = $payment['offerId'];
-		$this->date = strtotime($payment['date']);
-	}
-
-	//------------------------------------------------------------------------------
-	// Getters and setters
-	//------------------------------------------------------------------------------
-
-	/**
-	 * Get the ID of the payment
-	 *
-	 * @return int                  The ID
-	 */
-	public function getId() {
-		return $this->id;
-	}
-
-	/**
-	 * Get the ID of the offer that this payment is linked to
-	 *
-	 * @return int                  The ID
-	 */
-	public function getOfferId() {
-		return $this->offerId;
-	}
+class Payment extends Model {
+	public
+		$table = 'payment',
+		$fillable_columns = ['offerId', 'date'];
 
 	/**
 	 * Get the offer that this payment is linked to
@@ -88,55 +38,23 @@ class Payment {
 		return new Offer($this->pdo, $this->offerId);
 	}
 
-	/**
-	 * Get the date of the payment
-	 *
-	 * @return int                   A unix timestamp describing the date of the payment
-	 */
-	public function getDate() {
-		return $this->date;
-	}
-
-	/**
-	 * Set the date of the payment
-	 *
-	 * @param int $date              The new date for the payment
-	 *
-	 * @throws PDOException          If something went wrong with the database
-	 *
-	 * @return bool                  True on success, false on failure
-	 */
-	public function setDate($date) {
-		$stmt = $this->pdo->prepare("UPDATE `".Constants::db_prefix."payment` SET `date`=? WHERE `id`=?");
-		$stmt->execute([date('Y-m-d H:i:s', $date), $this->id]);
-		if ($stmt->rowCount() == 1) {
-			$this->date = $date;
-			return true;
-		} else {
-			return false;
+	protected function accessor($key, $value) {
+		switch ($key) {
+			case 'date':
+				return strtotime($value);
+				break;
+			default:
+				return parent::accessor($key, $value);
 		}
 	}
 
-	//------------------------------------------------------------------------------
-	// Other functions
-	//------------------------------------------------------------------------------
-
-	/**
-	 * Remove this payment from the database
-	 *
-	 * If this doesn't succeed (i.e. false is returned), that means the payment was removed manually or by another instance of this class
-	 *
-	 * @throws PDOException          If something went wrong with the database
-	 *
-	 * @return bool                  True on success, false on failure
-	 */
-	public function delete() {
-		$stmt = $this->pdo->prepare("DELETE FROM `".Constants::db_prefix."payment` WHERE `id`=?");
-		$stmt->execute(array($this->id));
-		if ($stmt->rowCount() != 1) {
-			return false;
-		} else {
-			return true;
+	protected function mutator($key, $value) {
+		switch ($key) {
+			case 'date':
+				return is_string($value) ? $value : date('Y-m-d H:i:s', $value);
+				break;
+			default:
+				return parent::mutator($key, $value);
 		}
 	}
 }