#
# Copyright 2010-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#

import logging
import sys

import greengrasssdk

# Setup logging to stdout
logger = logging.getLogger(__name__)
logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)

client = greengrasssdk.client("iot-data")


def message_handler(event, context):
    logger.info("Received message!")
    if "state" in event:
        if event["state"] == "on":
            client.update_thing_shadow(thingName="RobotArm_Thing", payload='{"state":{"desired":{"myState":"on"}}}')
            logger.info("Triggering publish to shadow " "topic to set state to ON")
        elif event["state"] == "off":
            client.update_thing_shadow(thingName="RobotArm_Thing", payload='{"state":{"desired":{"myState":"off"}}}')
            logger.info("Triggering publish to shadow " "topic to set state to OFF")