Updating to let admins set the notification title

Author: ColdHeat

__init__.py

@@ -26,12 +26,13 @@ def notify_send():
             return render_template('notify.html')
         elif request.method == 'POST':
             msg = request.form.get('msg')
+            title = request.form.get('title')
             client_secret = get_config('thunderpush_client_secret')
             server_secret = get_config('thunderpush_server_secret')
             thunderpush_url = get_config('thunderpush_url')
             thunderpush_port = get_config('thunderpush_port')
             t = Thunder(apikey=client_secret, apisecret=server_secret, host=thunderpush_url, port=thunderpush_port)
-            print t.send_message_to_channel(channel='all_teams', message={'title': 'CTFd Notification', 'msg': msg})
+            t.send_message_to_channel(channel='all_teams', message={'title': title, 'msg': msg})
             return '', 200
 
     @notify.route('/notify/static/ctfd-notify.js', methods=['GET'])
@@ -43,4 +44,12 @@ def notify_static_generator():
         return Response(js, mimetype='application/javascript')
 
     app.register_blueprint(notify)
-    app.register_blueprint(notify_static, url_prefix='/notify')
+    app.register_blueprint(notify_static, url_prefix='/notify')
+    scripts = [
+        "/notify/static/sockjs-0.3.4.min.js",
+        "/notify/static/thunderpush.js",
+        "/notify/static/push.min.js",
+        "/notify/static/ctfd-notify.js"
+    ]
+    for s in scripts:
+        utils.register_plugin_script(s)

templates/ctfd-notify.js

@@ -41,6 +41,5 @@ if ( push_status === Push.Permission.DEFAULT ) {
         console.log(message);
         var message = html.format(message.title, message.msg);
         $('#notification_wrapper').append(message);
-
     });
 }

templates/notify.html

@@ -9,6 +9,10 @@ <h1 class="text-center">Send Notification</h1>
 </div>
 <div class="row">
     <form id="thunderpush_notify" method="POST" action="/admin/plugins/notify/send">
+        <div class="form-group">
+            <label>Notification Title</label>
+            <input class="form-control" type="text" name="title" placeholder="Notification Title">
+        </div>
         <div class="form-group">
             <label>Notification Message</label>
             <input class="form-control" type="text" name="msg" placeholder="Notification Message">
@@ -23,13 +27,16 @@ <h1 class="text-center">Send Notification</h1>
 {% endblock %}
 
 {% block scripts %}
+<script src="{{ request.script_root }}/notify/static/sockjs-0.3.4.min.js"></script>
+<script src="{{ request.script_root }}/notify/static/thunderpush.js"></script>
+<script src="{{ request.script_root }}/notify/static/push.min.js"></script>
+<script src="{{ request.script_root }}/notify/static/ctfd-notify.js"></script>
+
 <script type="text/javascript">
     $('#thunderpush_notify').submit(function(e){
         e.preventDefault();
         $.post(window.location.pathname, $(this).serialize(), function(data){
-            if (data == 1){
-                window.location.reload();
-            }
+            console.log(data);
         });
     });
 </script>