"""Use PushToken to send messages from snowflake.
    Ref: https://aiosfstream.readthedocs.io/en/latest/
"""
import asyncio

from aiosfstream import SalesforceStreamingClient


async def stream_events():
    # connect to Streaming API
    async with SalesforceStreamingClient(
            consumer_key="REPLACE-ME",
            consumer_secret="REPLACE-ME",
            username="kafkaintegration@sonymusic.com.fullcopysb",
            password="REPLACE-ME",
            sandbox=True
    ) as client:
        print("\n Authentication successful")
        # subscribe to topics
        await client.subscribe("/data/Fan__ChangeEvent")
        # await client.subscribe("/topic/FansPushTopic") # did not work
        print("\n subscribe successful")

        # listen for incoming messages
        async for message in client:
            print("\n message==", message)
            topic = message["channel"]
            data = message["data"]
            print(f"\n {topic}: {data}")


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.run_until_complete(stream_events())