X-Git-Url: http://www.average.org/gitweb/?a=blobdiff_plain;f=gps303%2Flookaside.py;h=2e759d1ef109ec2a227d5f772ee981a6c0b2bf52;hb=bf48ccad4b4b91e7d7e09d1087f5953bc2db97d7;hp=d9af989aff2a3b58711938bbb548053355fdd03a;hpb=b613110bb16c95d9b641882c2ad6869e3ced1a0c;p=loctrkd.git

diff --git a/gps303/lookaside.py b/gps303/lookaside.py
index d9af989..2e759d1 100644
--- a/gps303/lookaside.py
+++ b/gps303/lookaside.py
@@ -1,13 +1,62 @@
-"""
-For when responding to the terminal is not trivial
-"""
-
-from .gps303proto import *
-from .opencellid import qry_cell
-
-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 {}
+""" Estimate coordinates from WIFI_POSITIONING and send back """
+
+from configparser import ConfigParser
+from datetime import datetime, timezone
+from importlib import import_module
+from logging import getLogger
+from os import umask
+from struct import pack
+import zmq
+
+from . import common
+from .zx303proto import parse_message, proto_name, WIFI_POSITIONING
+from .zmsg import Bcast, Resp, topic
+
+log = getLogger("gps303/lookaside")
+
+
+def runserver(conf: ConfigParser) -> None:
+    qry = import_module("." + conf.get("lookaside", "backend"), __package__)
+    qry.init(conf)
+    # Is this https://github.com/zeromq/pyzmq/issues/1627 still not fixed?!
+    zctx = zmq.Context()  # type: ignore
+    zsub = zctx.socket(zmq.SUB)  # type: ignore
+    zsub.connect(conf.get("collector", "publishurl"))
+    zsub.setsockopt(zmq.SUBSCRIBE, topic(proto_name(WIFI_POSITIONING)))
+    zpush = zctx.socket(zmq.PUSH)  # type: ignore
+    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,
+            )
+            try:
+                lat, lon = qry.lookup(
+                    msg.mcc, msg.mnc, msg.gsm_cells, msg.wifi_aps
+                )
+                resp = Resp(
+                    imei=zmsg.imei,
+                    when=zmsg.when,  # not the current time, but the original!
+                    packet=msg.Out(latitude=lat, longitude=lon).packed,
+                )
+                log.debug("Response for lat=%s, lon=%s: %s", lat, lon, resp)
+                zpush.send(resp.packed)
+            except Exception as e:
+                log.warning("Lookup for %s resulted in %s", msg, e)
+
+    except KeyboardInterrupt:
+        zsub.close()
+        zpush.close()
+        zctx.destroy()  # type: ignore
+        qry.shut()
+
+
+if __name__.endswith("__main__"):
+    runserver(common.init(log))