# Module to respomnd to new nodes we havent seen before with a hello message
# K7MHI Kelly Keeton 2024

import os
import sqlite3
from modules.log import logger
from modules.settings import qrz_db

def initalize_qrz_database():
    try:
        # If the database file doesn't exist, it will be created by sqlite3.connect
        if not os.path.exists(qrz_db):
            logger.info(f"QRZ database file '{qrz_db}' not found. Creating new database.")
        conn = sqlite3.connect(qrz_db)
        c = conn.cursor()
        # Create the table if it doesn't exist
        c.execute('''CREATE TABLE IF NOT EXISTS qrz
                     (qrz_id INTEGER PRIMARY KEY, qrz_call TEXT, qrz_name TEXT, qrz_qth TEXT, qrz_notes TEXT)''')
        conn.commit()
        return True
    except sqlite3.Error as e:
        logger.error(f"Error initializing QRZ database: {e}")
        return False
    finally:
        if 'conn' in locals():
            conn.close()

def never_seen_before(nodeID):
    # check if we have seen this node before and sent a hello message
    conn = sqlite3.connect(qrz_db)
    c = conn.cursor()
    try:
        c.execute("SELECT * FROM qrz WHERE qrz_call = ?", (nodeID,))
        row = c.fetchone()
        conn.close()
        if row is None:
            # we have not seen this node before
            return True
        else:
            # we have seen this node before
            return False
    except sqlite3.OperationalError as e:
        if "no such table" in str(e):
            initalize_qrz_database()
            logger.warning("QRZ database table not found, created new table")
            # we have not seen this node before
            return True
        else:
            raise
    
def hello(nodeID, name):
    # send a hello message
    conn = sqlite3.connect(qrz_db)
    c = conn.cursor()
    try:
        c.execute("INSERT INTO qrz (qrz_call, qrz_name) VALUES (?, ?)", (nodeID, str(name)))
    except sqlite3.OperationalError as e:
        if "no such table" in str(e):
            initalize_qrz_database()
            c.execute("INSERT INTO qrz (qrz_call, qrz_name) VALUES (?, ?)", (nodeID, str(name)))
        else:
            raise
    conn.commit()
    conn.close()
    return True