1 files changed, 52 insertions, 0 deletions
diff --git a/app/Model/UserNotificationTypeModel.php b/app/Model/UserNotificationTypeModel.php
new file mode 100644
index 00000000..0f377220
--- /dev/null
+++ b/app/Model/UserNotificationTypeModel.php
@@ -0,0 +1,52 @@
+<?php
+
+namespace Kanboard\Model;
+
+/**
+ * User Notification Type
+ *
+ * @package  Kanboard\Model
+ * @author   Frederic Guillot
+ */
+class UserNotificationTypeModel extends NotificationTypeModel
+{
+    /**
+     * SQL table name
+     *
+     * @var string
+     */
+    const TABLE = 'user_has_notification_types';
+
+    /**
+     * Get selected notification types for a given user
+     *
+     * @access public
+     * @param integer  $user_id
+     * @return array
+     */
+    public function getSelectedTypes($user_id)
+    {
+        $types = $this->db->table(self::TABLE)->eq('user_id', $user_id)->asc('notification_type')->findAllByColumn('notification_type');
+        return $this->filterTypes($types);
+    }
+
+    /**
+     * Save notification types for a given user
+     *
+     * @access public
+     * @param  integer  $user_id
+     * @param  string[] $types
+     * @return boolean
+     */
+    public function saveSelectedTypes($user_id, array $types)
+    {
+        $results = array();
+        $this->db->table(self::TABLE)->eq('user_id', $user_id)->remove();
+
+        foreach ($types as $type) {
+            $results[] = $this->db->table(self::TABLE)->insert(array('user_id' => $user_id, 'notification_type' => $type));
+        }
+
+        return ! in_array(false, $results, true);
+    }
+}