import stomp
import time
import veneno
import fetcher

class MyListener(stomp.ConnectionListener):

    def on_message(headers, message):

        conn.ack(headers['message-id'])

        print("** new message! **")

        print(message)

        fetcher.fetch(message.body)



    def on_error(headers, message):

        print('received an error "%s"' % message)
        
        conn.ack(headers['message-id'])

conn = stomp.Connection([('osiris.yankee.valorantdigital.com', 61613)])

conn.set_listener('', MyListener())

conn.connect('admin', 'password', wait=True)

conn.subscribe('/queue/sully-gnome-urls', id=1, ack='client', headers={'activemq.prefetchSize': 10})

print("## notification > connected to JMS server")

while True:

    pass

    time.sleep(1)
    
conn.disconnect()