1 files changed, 57 insertions, 0 deletions
diff --git a/Project-Unite/NotificationDaemon.cs b/Project-Unite/NotificationDaemon.cs
new file mode 100644
index 0000000..b8081e2
--- /dev/null
+++ b/Project-Unite/NotificationDaemon.cs
@@ -0,0 +1,57 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Web;
+using Project_Unite.Models;
+
+namespace Project_Unite
+{
+    public static class NotificationDaemon
+    {
+        public static void NotifyFollowers(string uid, string title, string desc, string url)
+        {
+            var db = new ApplicationDbContext();
+            var user = db.Users.FirstOrDefault(x => x.Id == uid);
+            if (user == null)
+                throw new Exception("Cannot find user with ID " + uid + ".");
+            foreach(var follower in user.Followers)
+            {
+                NotifyUser(uid, follower.Follower, title, desc, url);
+            }
+        }
+
+        public static void NotifyEveryone(string uid, string title, string desc, string url)
+        {
+            var db = new ApplicationDbContext();
+            var user = db.Users.FirstOrDefault(x => x.Id == uid);
+            if (user == null)
+                throw new Exception("Cannot find user with ID " + uid + ".");
+            foreach (var usr in db.Users.Where(x=>x.Id!=uid).ToArray())
+            {
+                NotifyUser(uid, usr.Id, title, desc, url);
+            }
+        }
+
+        public static void NotifyUser(string uid, string target, string title, string desc, string url)
+        {
+            var db = new ApplicationDbContext();
+            var user = db.Users.FirstOrDefault(x => x.Id == uid);
+            if (user == null)
+                throw new Exception("Cannot find user with ID " + target + ".");
+            var note = new Notification
+            {
+                Id = Guid.NewGuid().ToString(),
+                UserId = target,
+                Title = title,
+                ActionUrl = url,
+                Description = desc,
+                AvatarUrl = user.AvatarUrl
+            };
+            db.Notifications.Add(note);
+
+            db.SaveChanges();
+        }
+
+
+    }
+}
+\ No newline at end of file