import logging

import stomp
import time
import yaml
import msgspec
from pathlib import Path
from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
from platformdirs import PlatformDirs


parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
parser.add_argument("--config", "-c", default="kontor-docker")
parser.add_argument('--verbose', '-v', action='count', default=0)
parser.add_argument("--server", "-s", default="127.0.0.1")
args = parser.parse_args()
logger = logging.getLogger(__name__)

def get_logger(level: int, config: str):
    """
    create Logger with configuration from config file
    """
    dirs = PlatformDirs(config)
    logging_config = Path(dirs.user_config_dir, "logging-config.yaml")
    with open(logging_config, "rt", encoding="UTF-8") as f:
        config_dict = yaml.safe_load(f.read())
    logging.config.dictConfig(config_dict)
    log = logging.getLogger("development")
    if level is not None:
        match level:
            case 0:
                log.setLevel(logging.INFO)
            case 1:
                log.setLevel(logging.DEBUG)
            case _:
                log.setLevel(logging.CRITICAL)
    return log


class Link(msgspec.Struct):
    url: str

class MyListener(stomp.ConnectionListener):
    def __init__(self, log):
        self.log = log
        pass

    def on_error(self, frame):
        self.log.info("received an error %s", frame.body)

    def on_message(self, frame):
        self.log.info("received a message %s", frame.body)
        link = msgspec.json.decode(frame.body, type=Link)
        self.log.info("found link: %s", link.url)


if __name__ == '__main__':
    logger = get_logger(args.verbose, args.config)
    logger.info("kontor.read_queue started")
    host = [(args.server, 61616)]
    conn = stomp.Connection(host_and_ports=host)
    conn.set_listener('', MyListener(logger))
    conn.connect(username='artemis', passcode='artemis', wait=True)
    conn.subscribe(destination='add_link', id=1, ack='auto', headers={})
    time.sleep(5)
    conn.disconnect()
    logger.info("kontor.read_queue finished")