Broadcast location, gps and approximated

[loctrkd.git] / gps303 / lookaside.py

diff --git a/gps303/lookaside.py b/gps303/lookaside.py
index d9af989aff2a3b58711938bbb548053355fdd03a..085cbe9a31b65c65057d6fa150158560658ef4f4 100644 (file)
--- a/gps303/lookaside.py
+++ b/gps303/lookaside.py
@@ -1,13 +1,72 @@
-"""
-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, LocEvt, Resp
+
+log = getLogger("gps303/lookaside")
+
+
+def runserver(conf):
+    zctx = zmq.Context()
+    zpub = zctx.socket(zmq.PUB)
+    zpub.bind(conf.get("lookaside", "publishurl"))
+    zsub = zctx.socket(zmq.SUB)
+    zsub.connect(conf.get("collector", "publishurl"))
+    for protoname in (
+        "GPS_POSITIONING",
+        "GPS_OFFLINE_POSITIONING",
+        "WIFI_POSITIONING",
+    ):
+        topic = pack("B", proto_by_name(protoname))
+        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 isinstance(msg, WIFI_POSITIONING):
+                is_gps = False
+                lat, lon = qry_cell(
+                    conf["opencellid"]["dbfn"], msg.mcc, msg.gsm_cells
+                )
+                resp = Resp(
+                    imei=zmsg.imei, packet=msg.Out(lat=lat, lon=lon).packed
+                )
+                log.debug("Response for lat=%s, lon=%s: %s", lat, lon, resp)
+                zpush.send(resp.packed)
+            else:
+                is_gps = True
+                lat = msg.latitude
+                lon = msg.longitude
+            zpub.send(
+                LocEvt(
+                    imei=zmsg.imei,
+                    devtime=msg.devtime,
+                    is_gps=is_gps,
+                    lat=lat,
+                    lon=lon,
+                ).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))