web-base/core/Api/NotificationsAPI.class.php

<?php

namespace Api {

  class NotificationsAPI extends Request {

  }

}

namespace Api\Notifications {

  use Api\NotificationsAPI;
  use Api\Parameter\Parameter;
  use Api\Parameter\StringType;
  use Driver\SQL\Condition\Compare;

  class Create extends NotificationsAPI {

    public function __construct($user, $externalCall = false) {
      parent::__construct($user, $externalCall, array(
        'groupId' => new Parameter('groupId', Parameter::TYPE_INT, true),
        'userId' => new Parameter('userId', Parameter::TYPE_INT, true),
        'title' =>  new StringType('title', 32),
        'message' =>  new StringType('message', 256),
      ));
      $this->isPublic = false;
      $this->csrfTokenRequired = true;
    }

    private function checkUser($userId) {
      $sql = $this->user->getSQL();
      $res = $sql->select($sql->count())
        ->from("User")
        ->where(new Compare("uid", $userId))
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();

      if ($this->success) {
        if ($res[0]["count"] == 0) {
          $this->success = false;
          $this->lastError = "User not found";
        }
      }

      return $this->success;
    }

    private function insertUserNotification($userId, $notificationId) {
      $sql = $this->user->getSQL();
      $res = $sql->insert("UserNotification", array("user_id", "notification_id"))
        ->addRow($userId, $notificationId)
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();
      return $this->success;
    }

    private function checkGroup($groupId) {
      $sql = $this->user->getSQL();
      $res = $sql->select($sql->count())
        ->from("Group")
        ->where(new Compare("uid", $groupId))
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();

      if ($this->success) {
        if ($res[0]["count"] == 0) {
          $this->success = false;
          $this->lastError = "Group not found";
        }
      }

      return $this->success;
    }

    private function insertGroupNotification($groupId, $notificationId) {
      $sql = $this->user->getSQL();
      $res = $sql->insert("GroupNotification", array("group_id", "notification_id"))
        ->addRow($groupId, $notificationId)
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();
      return $this->success;
    }

    private function createNotification($title, $message) {
      $sql = $this->user->getSQL();
      $res = $sql->insert("Notification", array("title", "message"))
        ->addRow($title, $message)
        ->returning("uid")
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();

      if ($this->success) {
        return $sql->getLastInsertId();
      }

      return $this->success;
    }

    public function execute($values = array()) {
      if(!parent::execute($values)) {
        return false;
      }

      $userId = $this->getParam("userId");
      $groupId = $this->getParam("groupId");
      $title = $this->getParam("title");
      $message = $this->getParam("message");

      if (is_null($userId) && is_null($groupId)) {
        return $this->createError("Either userId or groupId must be specified.");
      } else if(!is_null($userId) && !is_null($groupId)) {
        return $this->createError("Only one of userId and groupId must be specified.");
      } else if(!is_null($userId)) {
        if ($this->checkUser($userId)) {
          $id = $this->createNotification($title, $message);
          if ($this->success) {
            return $this->insertUserNotification($userId, $id);
          }
        }
      } else if(!is_null($groupId)) {
        if ($this->checkGroup($groupId)) {
          $id = $this->createNotification($title, $message);
          if ($this->success) {
            return $this->insertGroupNotification($groupId, $id);
          }
        }
      }

      return $this->success;
    }
  }

  class Fetch extends NotificationsAPI {

    private array $notifications;

    public function __construct($user, $externalCall = false) {
      parent::__construct($user, $externalCall, array());
      $this->loginRequired = true;
      $this->csrfTokenRequired = true;
    }

    private function fetchUserNotifications() {
      $userId = $this->user->getId();
      $sql = $this->user->getSQL();
      $res = $sql->select($sql->distinct("Notification.uid"), "created_at", "title", "message")
        ->from("Notification")
        ->innerJoin("UserNotification", "UserNotification.notification_id", "Notification.uid")
        ->where(new Compare("UserNotification.user_id", $userId))
        ->where(new Compare("UserNotification.seen", false))
        ->orderBy("created_at")->descending()
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();

      if ($this->success) {
        foreach($res as $row) {
          $id = $row["uid"];
          if (!isset($this->notifications[$id])) {
            $this->notifications[$id] = array(
              "uid" => $id,
              "title" => $row["title"],
              "message" => $row["message"],
              "created_at" => $row["created_at"],
            );
          }
        }
      }

      return $this->success;
    }

    private function fetchGroupNotifications() {
      $userId = $this->user->getId();
      $sql = $this->user->getSQL();
      $res = $sql->select($sql->distinct("Notification.uid"), "created_at", "title", "message")
        ->from("Notification")
        ->innerJoin("GroupNotification", "GroupNotification.notification_id", "Notification.uid")
        ->innerJoin("UserGroup", "GroupNotification.group_id", "UserGroup.group_id")
        ->where(new Compare("UserGroup.user_id", $userId))
        ->where(new Compare("GroupNotification.seen", false))
        ->orderBy("created_at")->descending()
        ->execute();

      $this->success = ($res !== FALSE);
      $this->lastError = $sql->getLastError();

      if ($this->success) {
        foreach($res as $row) {
          $id = $row["uid"];
          if (!isset($this->notifications[$id])) {
            $this->notifications[] = array(
              "uid" => $id,
              "title" => $row["title"],
              "message" => $row["message"],
              "created_at" => $row["created_at"],
            );
          }
        }
      }

      return $this->success;
    }

    public function execute($values = array()) {
      if(!parent::execute($values)) {
        return false;
      }

      $this->notifications = array();
      if ($this->fetchUserNotifications() && $this->fetchGroupNotifications()) {
        $this->result["notifications"] = $this->notifications;
      }

      return $this->success;
    }
  }

}