イベント通知機能

chibiegg · chibiegg · commit 8e4238b720e2 · 2015-08-13T16:49:25.000+09:00
diff --git a/.node-version b/.node-version
@@ -0,0 +1 @@
+v0.12.7
diff --git a/node/.gitignore b/node/.gitignore
@@ -0,0 +1 @@
+node_modules
diff --git a/node/.python-version b/node/.python-version
@@ -0,0 +1 @@
+2.7.10
diff --git a/node/files/index.html b/node/files/index.html
@@ -0,0 +1,10 @@
+<html>
+  <head>Realtime Example</head>
+  <body>
+    <h1>Realtime Example</h1>
+    <script src="/socket.io/socket.io.js"></script>
+    <script>
+      var socket = io();
+    </script>
+  </body>
+</html>
diff --git a/node/index.js b/node/index.js
@@ -0,0 +1,88 @@
+var app = require('express')();
+var http = require('http').Server(app);
+var io = require('socket.io')(http);
+var request = require('request');
+var bodyParser = require('body-parser');
+
+app.use(bodyParser());
+
+app.get('/realtime/', function(req, res){
+  res.sendFile(__dirname + '/files/index.html');
+});
+
+app.post('/events/', function(req, res){
+  if(req.body){
+    io.emit("event", req.body);
+  }
+  res.status(201);
+  res.send("{}");
+});
+
+io.on("connection", function(socket){
+  // Authentication Check
+  var headers = socket.handshake.headers;
+  request({
+    url:"http://localhost:8000/api/auth/",
+    json:true,
+    headers:{
+      "Cookie":headers.cookie,
+      "Accept":"application/json"
+    }
+  },function (error, response, body){
+    if (!error && response.statusCode == 200 && body.is_staff) {
+      allowed = true;
+    }else{
+      socket.disconnect();
+    }
+  });
+
+
+
+
+});
+
+
+http.listen(8001, function(){
+  console.log('listening on *:8001');
+});
+
+
+
+
+
+// Proxy Server for Development
+var http = require('http');
+var httpProxy = require('http-proxy');
+var url = require('url');
+var proxy = httpProxy.createProxyServer({});
+
+var server = http.createServer(function (req, res) {
+
+  var hostname = req.headers.host.split(":")[0];
+  var pathname = url.parse(req.url).pathname;
+
+  if(pathname.indexOf("/socket.io/") === 0 || pathname.indexOf("/realtime/") === 0 ){
+    proxy.web(req, res, {
+      target: "http://localhost:8001"
+    });
+    return;
+  }
+
+  proxy.web(req, res, {
+    target: "http://localhost:8000"
+  });
+});
+
+server.on('upgrade', function (req, socket, head) {
+  console.log("upgrade");
+  proxy.ws(req, socket, head, {
+    target: {
+      host: 'localhost',
+      port: 8001
+    }
+  });
+});
+
+server.listen(8080, function(){
+  console.log('listening on *:8080');
+});
diff --git a/node/package.json b/node/package.json
@@ -0,0 +1,22 @@
+{
+  "name": "lepus-node",
+  "version": "1.0.0",
+  "description": "LapusはRESTfulなCTFサーバーです．",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git@git.spica.bz:nomuken/lepus.git"
+  },
+  "author": "",
+  "license": "ISC",
+  "devDependencies": {
+    "body-parser": "^1.13.3",
+    "express": "^4.10.2",
+    "http-proxy": "^1.11.1",
+    "request": "^2.60.0",
+    "socket.io": "^1.3.6"
+  }
+}
diff --git a/package.pip b/package.pip
@@ -3,3 +3,4 @@ Markdown
 django-filter
 djangorestframework
 django-debug-toolbar
+requests
diff --git a/src/lepus/middleware.py b/src/lepus/middleware.py
@@ -11,3 +11,5 @@ def process_request(self, request):
 
         if user.is_authenticated() and ip:
             UserConnection.update(user, ip)
+
+from lepus.signals import *
diff --git a/src/lepus/serializers.py b/src/lepus/serializers.py
@@ -63,8 +63,8 @@ def update(self, instance, validated_data):
 class UserSerializer(serializers.ModelSerializer):
     class Meta:
         model = models.User
-        fields = ("id", "username", "password", "team", "points", "last_score_time", "team_name", "team_password")
-        read_only_fields = ("id", "team", "points", "last_score_time")
+        fields = ("id", "username", "password", "team", "points", "last_score_time", "team_name", "team_password", "is_staff")
+        read_only_fields = ("id", "team", "points", "last_score_time", "is_staff")
         extra_kwargs = {'password': {'write_only': True}}
 
     team_name = serializers.CharField(write_only=True, allow_null=False, error_messages={"require":"チーム名は必須です"})
@@ -188,4 +188,3 @@ class NoticeSerializer(serializers.ModelSerializer):
     class Meta:
         model = models.Notice
         fields = ('id', 'title', 'body', 'created_at', 'updated_at')
-        
diff --git a/src/lepus/signals.py b/src/lepus/signals.py
@@ -0,0 +1,26 @@
+# encoding=utf-8
+import requests
+import json
+from django.core.signals import request_finished
+from django.dispatch import Signal, receiver
+from django.db.models.signals import post_save
+from lepus.models import Answer
+
+def send_realtime_event(data):
+    headers = {'Content-type': 'application/json', 'Accept': 'application/json'}
+    response = requests.post("http://localhost:8001/events/", data=json.dumps(data), headers=headers)
+    print(data)
+
+@receiver(post_save, sender=Answer)
+def on_answer_sent(sender, **kwargs):
+    if kwargs["created"]:
+        answer = kwargs["instance"]
+        data = {
+            "type":"answer",
+            "user":answer.user.id,
+            "team":answer.team.id,
+            "answer":answer.answer,
+            "flag":answer.flag.id if answer.flag else None,
+            "is_correct":answer.is_correct
+        }
+        send_realtime_event(data)
