"""Lambda watchdog function module."""

import sentry_sdk
from sentry_sdk.integrations.aws_lambda import AwsLambdaIntegration
from lambdacommon.common_config import logger

import config
from connectors import neo4j
from logic import verify_neo4j


sentry_sdk.init(
    config.SENTRY_DSN,
    integrations=[AwsLambdaIntegration()]
)


driver = neo4j.get_driver()


def handler(event, context):
    """Lambda entry point."""
    try:
        neo4j_errors = verify_neo4j.verify_neo4j_cluster(driver)
        if neo4j_errors:
            raise verify_neo4j.VerifyNeo4JException(neo4j_errors)
        logger.info('Neo4J cluster is good.')
        return {'status': 'OK'}
    except Exception as e:
        logger.exception(str(e))
        raise e