#!/usr/bin/env python3
import os
import sqlite3
import aprs
import json
import aprslib
import configparser
import time

def read_config():
    config = configparser.ConfigParser()
    config.read('config.ini')
    return config

def get_db_connection():
    conn = sqlite3.connect('database.db')
    conn.row_factory = sqlite3.Row
    return conn

def main():

    # Add the call and location of this station to the packet info
    config = read_config()
    # MYCALL = os.environ.get("MYCALL", "W1CDN")
    # KISS_HOST = os.environ.get("KISS_HOST", "192.168.0.30")
    # KISS_PORT = os.environ.get("KISS_PORT", "8001")


    ki = aprs.TCPKISS(host=config['Settings']['kiss_host'], port=int(config['Settings']['kiss_port']))
    ki.start()


    # Make a simple frame and send it
    frame = aprs.APRSFrame.ui(
        destination="APZ001",
        source=config['Settings']['mycall'],
        path=["WIDE1-1"],
        info=b">Hello World!",
    )
    #ki.write(frame)

    # Watch for new packets to come in
    while True:
        conn = get_db_connection()
        for frame in ki.read(min_frames=1):
            try:
                a = aprslib.parse(str(frame))
                a['station_call'] = config['Settings']['station_call']
                a['station_lat'] = config['Settings']['station_lat']
                a['station_lon'] = config['Settings']['station_lon']
                a['created_unix'] = int(time.time())
                print(a)
                # Make this a string and deal with it later (probably a mistake)
                a['path'] = str(a['path'])
                # Store true/false as 1/0
                if 'alive' in a:
                    if a['alive'] == True:
                        a['alive'] = 1
                    else:
                        a['alive'] = 0
                # Build an INSERT statement based on the fields we have from the frame
                attrib_names = ', '.join('"%s"' % w for w in a.keys())
                attrib_values = ", ".join("?" * len(a.keys()))
                sql = "INSERT INTO frames ("+attrib_names+") VALUES ("+attrib_values+")"
                conn.commit()
                
                try:
                    # Insert data
                    conn.execute(sql, list(a.values()))

                    # TODO update stations table here
                    # Original intent was to include the id from the frames table,
                    # but that would mean making another query.
                    # It's not immediately needed, so I'm skipping it.
                    # Build query
                    # "from" is wrappedin [] because it is a reserved word and using '' doesn't work.
                    # https://www.sqlite.org/lang_keywords.html
                    try:
                        station_update = "'"+a['from'] +"', '"+ str(a['created_unix']) +"', '1'"
                        query3 = "INSERT INTO stations ([from], last_heard_unix, count) \
                        VALUES("+station_update+") \
                        ON CONFLICT([from]) \
                        DO UPDATE SET count = count + 1;"
                        print(query3)
                        # Insert/update data
                        conn.execute(query3)
                        conn.commit()
                    except:
                        print("Stations table couldn't be updated.")

                    # TODO remove packets that are older ('created') than a limit set in config.ini
                    # "5 minutes" also works
                    #conn.execute("DELETE FROM frames WHERE created < DATETIME('now', '"+config['Settings']['keep_time']+"')")
                    #conn.commit()
                except:
                    print("Error with SQLite!")
            except:
                print("Frame could not be parsed.")


        conn.close()



if __name__ == "__main__":
    main()