import os
from aws_provider import get_sns_resource, publish_message, get_sqs_resource, \
    send_message_to_sqs


def handler(event, context):
    msg = event.get('message', 'haystack')
    processed_msg = process_message(event)
    if msg == 'queue':
        demo_sending_to_queue(processed_msg)
    demo_sending_to_topic(processed_msg)


def demo_sending_to_queue(processed_msg):
    sqs_resource = get_sqs_resource()
    queue_name = os.getenv('SQS_NAME')
    send_message_to_sqs(sqs_resource, processed_msg, queue_name)


def demo_sending_to_topic(processed_msg):
    sns_resource = get_sns_resource()
    publish_message(sns_resource, processed_msg)


def process_message(event):
    return "I have found the {}".format(
        event.get('message', 'haystack'))