From 23e7c209b6bc8c5a07208abbbf28465ea6930ef0 Mon Sep 17 00:00:00 2001
From: Devin Matte <devinmatte@gmail.com>
Date: Thu, 15 Aug 2019 11:13:22 -0400
Subject: [PATCH 1/2] Push Notifications Phase 1

---
 config.env.py                      |  5 +++++
 packet/routes/shared.py            | 13 +++++++++++++
 packet/static/js/sw.js             |  2 ++
 packet/static/js/update-sw.js      |  2 ++
 packet/static/manifest.json        |  3 ++-
 packet/templates/include/head.html | 21 +++++++++++++++++++++
 requirements.txt                   |  1 +
 7 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 packet/static/js/sw.js
 create mode 100644 packet/static/js/update-sw.js

diff --git a/config.env.py b/config.env.py
index e403b052..0eb3f4e6 100644
--- a/config.env.py
+++ b/config.env.py
@@ -38,5 +38,10 @@
 MAIL_PASSWORD = environ.get("PACKET_MAIL_PASSWORD", None)
 MAIL_USE_TLS = strtobool(environ.get("PACKET_MAIL_TLS", 'True'))
 
+# OneSignal Config
+ONESIGNAL_USER_AUTH_KEY = environ.get("PACKET_ONESIGNAL_USER_AUTH_KEY", None)
+ONESIGNAL_APP_AUTH_KEY = environ.get("PACKET_ONESIGNAL_APP_AUTH_KEY", None)
+ONESIGNAL_APP_ID = environ.get("PACKET_ONESIGNAL_APP_ID", "6eff123a-0852-4027-804e-723044756f00")
+
 # Slack URL for pushing to #general
 SLACK_WEBHOOK_URL = environ.get("PACKET_SLACK_URL", None)
diff --git a/packet/routes/shared.py b/packet/routes/shared.py
index 3d894baa..c4a04c6e 100644
--- a/packet/routes/shared.py
+++ b/packet/routes/shared.py
@@ -68,3 +68,16 @@ def packets(info=None):
     open_packets.sort(key=packet_sort_key, reverse=True)
 
     return render_template("active_packets.html", info=info, packets=open_packets)
+
+
+@app.route('/sw.js', methods=['GET'])
+@app.route('/OneSignalSDKWorker.js', methods=['GET'])
+def service_worker():
+    return app.send_static_file('js/sw.js')
+
+
+@app.route('/update-sw.js', methods=['GET'])
+@app.route('/OneSignalSDKUpdaterWorker.js', methods=['GET'])
+def update_service_worker():
+    return app.send_static_file('js/update-sw.js')
+
diff --git a/packet/static/js/sw.js b/packet/static/js/sw.js
new file mode 100644
index 00000000..81157226
--- /dev/null
+++ b/packet/static/js/sw.js
@@ -0,0 +1,2 @@
+importScripts('https://cdn.onesignal.com/sdks/OneSignalSDKWorker.js');
+
diff --git a/packet/static/js/update-sw.js b/packet/static/js/update-sw.js
new file mode 100644
index 00000000..81157226
--- /dev/null
+++ b/packet/static/js/update-sw.js
@@ -0,0 +1,2 @@
+importScripts('https://cdn.onesignal.com/sdks/OneSignalSDKWorker.js');
+
diff --git a/packet/static/manifest.json b/packet/static/manifest.json
index 1425b006..d7916ac9 100644
--- a/packet/static/manifest.json
+++ b/packet/static/manifest.json
@@ -20,5 +20,6 @@
       "sizes": "512x512",
       "type": "image/png"
     }
-  ]
+  ],
+  "gcm_sender_id": "482941778795"
 }
diff --git a/packet/templates/include/head.html b/packet/templates/include/head.html
index 677e611f..902889a5 100644
--- a/packet/templates/include/head.html
+++ b/packet/templates/include/head.html
@@ -34,6 +34,27 @@
 
     <link rel="stylesheet" href="{{ url_for('static', filename='css/packet.min.css') }}">
 
+    <!-- Push Notifications -->
+    <script src="https://cdn.onesignal.com/sdks/OneSignalSDK.js" async=""></script>
+    <script>
+        var OneSignal = window.OneSignal || [];
+        OneSignal.push(function() {
+            OneSignal.init({
+		    appId: "{{ config['ONESIGNAL_APP_ID'] }}",
+		autoResubscribe: true,
+                allowLocalhostAsSecureOrigin: true,
+            });
+            OneSignal.showNativePrompt();
+            OneSignal.on("subscriptionChange", function(){
+		OneSignal.getUserId().then(function(result){
+			// TODO: Send to backend
+			console.log(result);
+		});
+	    });
+        });
+    </script>
+
+    <!-- Analytics -->
     <script async src="https://www.googletagmanager.com/gtag/js?id={{ config['ANALYTICS_ID'] }}"></script>
     <script>
         window.dataLayer = window.dataLayer || [];
diff --git a/requirements.txt b/requirements.txt
index 782ac6e1..12adcdc3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,3 +8,4 @@ Flask-Migrate~=2.2.1
 pylint~=2.3.1
 gunicorn~=19.7.1
 csh_ldap~=2.1.0
+onesignal-sdk~=1.0.0

From ee28245eb9dd15a2fdc20489879aaf5d07921f84 Mon Sep 17 00:00:00 2001
From: Devin Matte <devinmatte@gmail.com>
Date: Thu, 15 Aug 2019 11:36:27 -0400
Subject: [PATCH 2/2] Adding model to collect subscriptions

---
 .../versions/53768f0a4850_notifications.py    | 35 +++++++++++++++++++
 packet/models.py                              |  7 ++++
 packet/routes/api.py                          | 16 ++++++++-
 packet/routes/shared.py                       |  1 -
 packet/templates/include/head.html            | 12 +++++--
 5 files changed, 67 insertions(+), 4 deletions(-)
 create mode 100644 migrations/versions/53768f0a4850_notifications.py

diff --git a/migrations/versions/53768f0a4850_notifications.py b/migrations/versions/53768f0a4850_notifications.py
new file mode 100644
index 00000000..3f10330a
--- /dev/null
+++ b/migrations/versions/53768f0a4850_notifications.py
@@ -0,0 +1,35 @@
+"""notifications
+
+Revision ID: 53768f0a4850
+Revises: eecf30892d0e
+Create Date: 2019-08-06 22:15:04.400982
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '53768f0a4850'
+down_revision = 'eecf30892d0e'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('notification_subscriptions',
+    sa.Column('member', sa.String(length=36), nullable=True),
+    sa.Column('freshman_username', sa.String(length=10), nullable=True),
+    sa.Column('token', sa.String(length=256), nullable=False),
+    sa.ForeignKeyConstraint(['freshman_username'], ['freshman.rit_username'], ),
+    sa.PrimaryKeyConstraint('token')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('notification_subscriptions')
+    # ### end Alembic commands ###
+
diff --git a/packet/models.py b/packet/models.py
index c3e6b4dc..b251661c 100644
--- a/packet/models.py
+++ b/packet/models.py
@@ -154,3 +154,10 @@ class MiscSignature(db.Model):
     updated = Column(DateTime, default=datetime.now, onupdate=datetime.now, nullable=False)
 
     packet = relationship("Packet", back_populates="misc_signatures")
+
+
+class NotificationSubscription(db.Model):
+    __tablename__ = "notification_subscriptions"
+    member = Column(String(36), nullable=True)
+    freshman_username = Column(ForeignKey("freshman.rit_username"), nullable=True)
+    token = Column(String(256), primary_key=True, nullable=False)
diff --git a/packet/routes/api.py b/packet/routes/api.py
index abae590c..5e0bc576 100644
--- a/packet/routes/api.py
+++ b/packet/routes/api.py
@@ -7,7 +7,7 @@
 from packet.context_processors import get_rit_name
 from packet.mail import send_report_mail
 from packet.utils import before_request, packet_auth, notify_slack
-from packet.models import Packet, MiscSignature
+from packet.models import Packet, MiscSignature, NotificationSubscription
 
 
 @app.route("/api/v1/sign/<packet_id>/", methods=["POST"])
@@ -40,6 +40,20 @@ def sign(packet_id, info):
     return "Error: Signature not valid.  Reason: Unknown"
 
 
+@app.route("/api/v1/subscribe/", methods=["POST"])
+@packet_auth
+@before_request
+def subscribe(info):
+    data = request.form
+    if app.config["REALM"] == "csh":
+        subscription = NotificationSubscription(token=data['token'], member=info["uid"])
+    else:
+        subscription = NotificationSubscription(token=data['token'], freshman_username=info["uid"])
+    db.session.add(subscription)
+    db.session.commit()
+    return "Token subscribed for " + info["uid"]
+
+
 @app.route("/api/v1/report/", methods=["POST"])
 @packet_auth
 @before_request
diff --git a/packet/routes/shared.py b/packet/routes/shared.py
index c4a04c6e..8f4422ca 100644
--- a/packet/routes/shared.py
+++ b/packet/routes/shared.py
@@ -80,4 +80,3 @@ def service_worker():
 @app.route('/OneSignalSDKUpdaterWorker.js', methods=['GET'])
 def update_service_worker():
     return app.send_static_file('js/update-sw.js')
-
diff --git a/packet/templates/include/head.html b/packet/templates/include/head.html
index 902889a5..2b394dae 100644
--- a/packet/templates/include/head.html
+++ b/packet/templates/include/head.html
@@ -47,8 +47,16 @@
             OneSignal.showNativePrompt();
             OneSignal.on("subscriptionChange", function(){
 		OneSignal.getUserId().then(function(result){
-			// TODO: Send to backend
-			console.log(result);
+			$.ajax({
+            			url: "/api/v1/subscribe/",
+            			method: "POST",
+            			data: {
+                			token: result
+            			},
+            			success: function (data) {
+                			console.log(data);
+            			}
+        		});
 		});
 	    });
         });