from telnetlib import Telnet

class GQRXHandler():
    def __init__(self, hostname: str = "localhost", port: int = 7356):
        self.hostname = hostname
        self.port = port

        self.telnet_connection = None

    def create_telnet_connection(self):
        tel_conn = Telnet(self.hostname, self.port)
        tel_conn.read_until(bytes("Escape character is '^]'."))

        return tel_conn

    def change_freq(self, freq):
        tel_conn = self.create_telnet_connection()
        tel_conn.write(f"F {int(freq)}")
        tel_conn.read_all()
        tel_conn.close()

    def change_squelch(self, squelch):
        tel_conn = self.create_telnet_connection()
        tel_conn.write(f"L SQL {float(squelch)}")
        tel_conn.read_all()
        tel_conn.close()

    def change_mode(self, mode):
        tel_conn = self.create_telnet_connection()
        tel_conn.write(f"M {str(mode)}")
        tel_conn.read_all()
        tel_conn.close()

    def set_all_settings(self, mode, squelch, freq):
        tel_conn = self.create_telnet_connection()
        tel_conn.write(f"F {int(freq)}")
        tel_conn.write(f"M {str(mode)}")
        tel_conn.write(f"L SQL {float(squelch)}")