4poc · December 1, 2016 10:45 · Dec 1, 2016 · Nov 30, 2016
diff --git a/wifibot.py b/wifibot.py
@@ -86,6 +86,3 @@ def on_message(room_object, event):
 t = Thread(None, station_dump_thread)
 t.start()
 t.join()
-while True:
-    pass
-
diff --git a/wifibot.py b/wifibot.py
@@ -0,0 +1,91 @@
+#!/usr/bin/python
+# Wifibot: Read station dumps from a router and shows new/dropped wifi clients in a matrix channel
+# by john & matthias
+
+import re
+import sys
+import time
+import subprocess
+import requests # pip install requests
+from threading import Thread
+from matrix_client.client import MatrixClient # pip install matrix_client
+
+MATRIX_HOST = 'https://[host]'
+MATRIX_USER = '[user]'
+MATRIX_PASS = '[pass]'
+ROOM_ID = '![blub]:[host]'
+STATION_DUMP_URLS = [
+    'http://192.168.0.1/wlan0_station_dump.txt'
+    # configure your router to make that file available:
+    # */1 * * * * iw dev wlan0 station dump > /www/wlan0_station_dump.txt
+]
+STATION_MAP = {
+    '00:11:22:fe:fe:fe': 'user'
+}
+CHECK_INTERVAL = 60 * 1
+
+# connect to matrix:
+client = MatrixClient(MATRIX_HOST)
+client.login_with_password(MATRIX_USER, MATRIX_PASS)
+room = client.join_room(ROOM_ID)
+
+def get_active_stations():
+    output = ''
+    for url in STATION_DUMP_URLS:
+        output += requests.get(url).text
+    stations = re.findall(r"Station ([^ ]+) ", output)
+    return stations
+
+previous = None
+
+def format_stations(stations):
+    s = []
+    for station in stations:
+        if station in STATION_MAP:
+            s.append(STATION_MAP[station])
+        else:
+            s.append(station)
+    return ', '.join(s)
+
+def station_dump_thread():
+    global previous
+    while True:
+        print('gather wifi...')
+        stations = set(get_active_stations())
+        print(stations)
+        if previous:
+            new_stations = stations - previous
+            gone_stations = previous - stations
+            print('new stations:', new_stations)
+            print('gone stations:', gone_stations)
+            if len(new_stations) > 0:
+                room.send_text('new stations:' + format_stations(new_stations))
+            if len(gone_stations) > 0:
+                room.send_text('gone stations:' + format_stations(gone_stations))
+
+        previous = stations
+        time.sleep(CHECK_INTERVAL)
+
+# Called when a message is recieved.
+def on_message(room_object, event):
+    print('event', event)
+    if event['type'] == "m.room.member":
+        if event['membership'] == "join":
+            print("{0} joined".format(event['content']['displayname']))
+    elif event['type'] == "m.room.message":
+        if event['content']['msgtype'] == "m.text":
+            user = event['sender']
+            message = event['content']['body']
+            print("{0}: {1}".format(user, message))
+    else:
+        print(event['type'])
+
+room.add_listener(on_message)
+client.start_listener_thread()
+
+t = Thread(None, station_dump_thread)
+t.start()
+t.join()
+while True:
+    pass
+