#!/usr/bin/env python3

import websocket
import time
import numpy as np
import datetime

import os

last_date = None
logfile = None

os.makedirs("logs", exist_ok=True)

def write_log(dt, line):
    global last_date, logfile

    date = dt.date()

    # reopen log file if day has changed
    if last_date != date:
        if logfile:
            logfile.close()

        logname = os.path.join("logs", date.strftime("%Y-%m-%d.csv"))
        print(f"(Re-)opening logfile: {logname}")
        logfile = open(logname, "a")

        last_date = date

    if line[-1] != '\n':
        line += '\n'

    logfile.write(line)


last_keepalive = datetime.datetime.now()

def on_open(wsapp):
    wsapp.send("SET auth t=kiwi p=#")
    wsapp.send("SERVER DE CLIENT openwebrx.js W/F")
    wsapp.send("SET send_dB=1")
    wsapp.send("SET zoom=0 start=0")
    wsapp.send("SET maxdb=0 mindb=-100")
    wsapp.send("SET interp=13")
    wsapp.send("SET wf_comp=0")
    wsapp.send("SET wf_speed=1")
    #wsapp.send("SET MARKER min=0.000 max=30000.000 zoom=0 width=1904")
    #wsapp.send("SET aper=1 algo=3 param=0.00")

def on_message(wsapp, message):
    global last_keepalive

    msgtype = message[:3].decode('ascii')
    msgdata = message[4:]
    now = datetime.datetime.now()

    if msgtype == 'MSG':
        msgstr = msgdata.decode('utf-8')
        print(f"Status message: {msgstr}")
    elif msgtype == 'W/F':
        #print(msgdata)
        print(f"Received waterfall data: {len(msgdata)} bytes.")
        #print(f"Header: {msgdata[:12]}")
        specdata = msgdata[12:]
        spectrum = np.frombuffer(specdata, dtype=np.uint8).astype(int)
        spectrum = spectrum - 255 # converts to dBm
        logline = f"{now.timestamp():.3f};" + ";".join([f"{dBm:d}" for dBm in spectrum]) + "\n"

        write_log(now, logline)

        if now - last_keepalive > datetime.timedelta(seconds=10):
            wsapp.send("SET keepalive")
            last_keepalive = now
    else:
        print(f"Unknown type: {msgtype} – data: {msgdata}")

url = f"ws://192.168.75.14:8073/kiwi/{int(time.time())}/W/F"
wsapp = websocket.WebSocketApp(url,
                               on_message=on_message,
                               on_open=on_open)
wsapp.run_forever()