From 063da32ae9c459dab8ac172f37c3ad0f50faf775 Mon Sep 17 00:00:00 2001
From: Nico Schottelius <nico@nico-notebook.schottelius.org>
Date: Sun, 26 May 2019 18:26:14 +0200
Subject: [PATCH] Move high score into the class, begin sanitise etcd

---
 game-etcd.py | 123 ++++++++++++++++++++++++++++++++++++---------------
 game.org     |   1 -
 2 files changed, 88 insertions(+), 36 deletions(-)

diff --git a/game-etcd.py b/game-etcd.py
index aae9b13..e9ac214 100644
--- a/game-etcd.py
+++ b/game-etcd.py
@@ -41,44 +41,98 @@ class Ping6(Level):
         """
 
 class Game(object):
-    def __init__(self, name):
-        self.client = etcd.Client(port=2379)
+    def __init__(self, name, etcdclient, etcbase="/ungleichgame/v1"):
+        self.client = etcdclient
         self.app = Flask(name)
         self.app.add_url_rule('/', 'highscore', self.highscore)
+        self.app.add_url_rule('/highscore', 'highscore', self.highscore)
+        self.app.add_url_rule("/register", 'register', self.register, methods=['POST'])
+
+        # etcd paths are below here
+        self.etcbase = etcbase
+        self.userbase = "{}/user".format(self.etcbase)
+
+
+    def read_etcd(self, path, recursive=False):
+        try:
+            data = self.client.read(path, recursive=recursive)
+        except etcd.EtcdKeyNotFound:
+            return None
+        except Exception:
+            abort(Response(status=400, response="Error connecting to etcd"))
+
+        return data
+
+    def get_highscore(self, username=None):
+        """ Returns a dict['username'] = points """
+
+        all_users = {}
+        highscore = {}
+
+        print("getting high")
+
+        if username:
+            path = "{}/{}".format(self.userbase, username)
+            user = self.read_etcd(path)
+            if user:
+                all_users[username] = user
+        else:
+            path = "{}/".format(self.userbase)
+            users = self.read_etcd(path, recursive=True)
+            print("reading from {}".format(path))
+            if users:
+                for child in users.children:
+                    print("adding user {} {} = {}".format(child, child.key, child.value))
+                    all_users[child.key] = child.value
+
+        for k, v in all_users.items():
+            # Ignore all kind of errors - just add the ones that work
+            try:
+                highscore[k] = json.loads(v)['points']
+                print("f?")
+            except Exception as e:
+                print(e)
+
+        return highscore
 
     def highscore(self):
-        return Response("A lof of points")
+        point_list = self.get_highscore()
+        res = []
+        if not point_list:
+            return Response("No winners yet!")
+
+        for k, v in point_list.items():
+            res.append("<p>{} has {} points</p>".format(k, v))
+
+        return Response("\n".join(res))
+
+    def require_args(self, *args):
+        parser = reqparse.RequestParser()
+        for arg in args:
+            parser.add_argument(arg, required=True)
+        return parser.parse_args()
+
+    def register(self):
+        args = self.require_args("network", "user")
+
+        # Needs to be fixed with ungleich-otp
+        username=args['user']
+
+        try:
+            net = ipaddress.IPv6Network(args['network'])
+            network = args['network']
+        except Exception as e:
+            return Response(status=400, response="Cannot register network {}: {}".format(network, e))
+
+        if not net.prefixlen == 64:
+            return Response(status=400, response="{} mask is not /64 - please use a /64 network".format(net))
+
+        self.client.write("/ungleichgame/v1/{}/network".format(username), network)
+        data = self.client.read("/ungleichgame/v1/{}/network".format(username))
+
+        return json.dumps("All good, go to /level/1 to start with level 1! - {}".format(data.value))
 
 
-@app.route("/")
-def high_score():
-    return "High score!"
-
-
-@app.route("/register", methods=['POST'])
-def register():
-    parser = reqparse.RequestParser()
-    parser.add_argument('network', required=True)
-    parser.add_argument('user', required=True)
-    args = parser.parse_args()
-
-    # Needs to be fixed with ungleich-otp
-    username=args['user']
-
-    try:
-        net = ipaddress.IPv6Network(args['network'])
-        network = args['network']
-    except Exception as e:
-        return Response(status=400, response="Cannot register network {}: {}".format(network, e))
-
-    if not net.prefixlen == 64:
-        return Response(status=400, response="{} mask is not /64 - please use a /64 network".format(net))
-
-    client = etcd.Client(port=2379)
-    client.write("/ungleichgame/v1/{}/network".format(username), network)
-    data = client.read("/ungleichgame/v1/{}/network".format(username))
-
-    return json.dumps("All good, go to /level/1 to start with level 1! - {}".format(data.value))
 
 @app.route("/level/1", methods=['GET', 'POST']) # post for username
 def get_ip_address():
@@ -147,8 +201,7 @@ if __name__ == '__main__':
 
     print("{} has {}".format(username, net))
 
-    g = Game(__name__)
+    g = Game(__name__, etcd.Client(port=2379))
     g.app.run(port='5002')
 
-
-    app.run(port='5002')
+    # app.run(port='5002')
diff --git a/game.org b/game.org
index 800b4a4..e24183c 100644
--- a/game.org
+++ b/game.org
@@ -24,4 +24,3 @@
 *** Test: is /64 ping6able?
 * Features
 ** TODO ungleich-otp support
-**