the whole shebang is working now

[loctrkd.git] / gps303 / lookaside.py

diff --git a/gps303/lookaside.py b/gps303/lookaside.py
index d9af989aff2a3b58711938bbb548053355fdd03a..05b7a49a62a7240bfc39477d10a4deab97422021 100644 (file)
--- a/gps303/lookaside.py
+++ b/gps303/lookaside.py
@@ -1,13 +1,59 @@
-"""
-For when responding to the terminal is not trivial
-"""
+""" Estimate coordinates from WIFI_POSITIONING and send back """
 
-from .gps303proto import *
+from datetime import datetime, timezone
+from logging import getLogger
+from struct import pack
+import zmq
+
+from . import common
+from .gps303proto import parse_message, proto_by_name, WIFI_POSITIONING
 from .opencellid import qry_cell
+from .zmsg import Bcast, Resp
+
+log = getLogger("gps303/lookaside")
+
+
+def runserver(conf):
+    zctx = zmq.Context()
+    zsub = zctx.socket(zmq.SUB)
+    zsub.connect(conf.get("collector", "publishurl"))
+    topic = pack("B", proto_by_name("WIFI_POSITIONING"))
+    zsub.setsockopt(zmq.SUBSCRIBE, topic)
+    zpush = zctx.socket(zmq.PUSH)
+    zpush.connect(conf.get("collector", "listenurl"))
+
+    try:
+        while True:
+            zmsg = Bcast(zsub.recv())
+            msg = parse_message(zmsg.packet)
+            log.debug(
+                "IMEI %s from %s at %s: %s",
+                zmsg.imei,
+                zmsg.peeraddr,
+                datetime.fromtimestamp(zmsg.when).astimezone(tz=timezone.utc),
+                msg,
+            )
+            if not isinstance(msg, WIFI_POSITIONING):
+                log.error(
+                    "IMEI %s from %s at %s: %s",
+                    zmsg.imei,
+                    zmsg.peeraddr,
+                    datetime.fromtimestamp(zmsg.when).astimezone(
+                        tz=timezone.utc
+                    ),
+                    msg,
+                )
+                continue
+            lat, lon = qry_cell(
+                conf["opencellid"]["dbfn"], msg.mcc, msg.gsm_cells
+            )
+            resp = Resp(imei=zmsg.imei, packet=msg.response(lat=lat, lon=lon))
+            log.debug("Response for lat=%s, lon=%s: %s", lat, lon, resp)
+            zpush.send(resp.packed)
+
+    except KeyboardInterrupt:
+        pass
+
 
-def prepare_response(conf, msg):
-    if isinstance(msg, WIFI_POSITIONING):
-        lat, lon = qry_cell(conf["opencellid"]["dbfn"],
-                msg.mcc, msg.gsm_cells)
-        return {"lat": lat, "lon": lon}
-    return {}
+if __name__.endswith("__main__"):
+    runserver(common.init(log))