canary_bucket = s3.Bucket(
    self,
    "canaryBucket",
    bucket_name = "jeremyritchie.com-canary",
    access_control = s3.BucketAccessControl.PRIVATE,
    encryption = s3.BucketEncryption.S3_MANAGED,
    versioned = False,
    block_public_access = s3.BlockPublicAccess.BLOCK_ALL
)

canary_bucket.add_lifecycle_rule(
    expiration=Duration.days(7),
)

aws-cdk-lib==2.66.0
constructs>=10.0.0,<11.0.0
aws-cdk.aws-synthetics-alpha==2.66.0a0

canary = synthetics.Canary(
    self,
    "HeartbeatCanary",
    canary_name = "website-heartbeat",
    schedule=synthetics.Schedule.rate(Duration.minutes(5)),
    artifacts_bucket_location=synthetics.ArtifactsBucketLocation(
        bucket=canary_bucket,
        prefix="artifacts"
    ),
    test=synthetics.Test.custom(
        code=synthetics.Code.from_asset('./lambda'),
        handler="heartbeat.handler"
    ),
    runtime=synthetics.Runtime.SYNTHETICS_NODEJS_PUPPETEER_3_9
)

const { URL } = require('url');
const synthetics = require('Synthetics');
const log = require('SyntheticsLogger');
const syntheticsConfiguration = synthetics.getConfiguration();
const syntheticsLogHelper = require('SyntheticsLogHelper');

const loadBlueprint = async function () {

    const urls = ['https://jeremyritchie.com','https://jeremyritchie.com/contact','https://jeremyritchie.com/author/jeremy-ritchie','https://jeremyritchie.com/posts/1','https://jeremyritchie.com/blog'];

    // Set screenshot option
    const takeScreenshot = false;

    /* Disabling default step screen shots taken during Synthetics.executeStep() calls
     * Step will be used to publish metrics on time taken to load dom content but
     * Screenshots will be taken outside the executeStep to allow for page to completely load with domcontentloaded
     * You can change it to load, networkidle0, networkidle2 depending on what works best for you.
     */
    syntheticsConfiguration.disableStepScreenshots();
    syntheticsConfiguration.setConfig({
       continueOnStepFailure: true,
       includeRequestHeaders: true, // Enable if headers should be displayed in HAR
       includeResponseHeaders: true, // Enable if headers should be displayed in HAR
       restrictedHeaders: [], // Value of these headers will be redacted from logs and reports
       restrictedUrlParameters: [] // Values of these url parameters will be redacted from logs and reports

    });

    let page = await synthetics.getPage();

    for (const url of urls) {
        await loadUrl(page, url, takeScreenshot);
    }
};

// Reset the page in-between
const resetPage = async function(page) {
    try {
        await page.goto('about:blank',{waitUntil: ['load', 'networkidle0'], timeout: 30000} );
    } catch (e) {
        synthetics.addExecutionError('Unable to open a blank page. ', e);
    }
}

const loadUrl = async function (page, url, takeScreenshot) {
    let stepName = null;
    let domcontentloaded = false;

    try {
        stepName = new URL(url).hostname;
    } catch (e) {
        const errorString = `Error parsing url: ${url}. ${e}`;
        log.error(errorString);
        /* If we fail to parse the URL, don't emit a metric with a stepName based on it.
           It may not be a legal CloudWatch metric dimension name and we may not have an alarms
           setup on the malformed URL stepName.  Instead, fail this step which will
           show up in the logs and will fail the overall canary and alarm on the overall canary
           success rate.
        */
        throw e;
    }

    await synthetics.executeStep(stepName, async function () {
        const sanitizedUrl = syntheticsLogHelper.getSanitizedUrl(url);

        /* You can customize the wait condition here. For instance, using 'networkidle2' or 'networkidle0' to load page completely.
           networkidle0: Navigation is successful when the page has had no network requests for half a second. This might never happen if page is constantly loading multiple resources.
           networkidle2: Navigation is successful when the page has no more then 2 network requests for half a second.
           domcontentloaded: It's fired as soon as the page DOM has been loaded, without waiting for resources to finish loading. Can be used and then add explicit await page.waitFor(timeInMs)
        */
        const response = await page.goto(url, { waitUntil: ['domcontentloaded'], timeout: 30000});
        if (response) {
            domcontentloaded = true;
            const status = response.status();
            const statusText = response.statusText();

            logResponseString = `Response from url: ${sanitizedUrl}  Status: ${status}  Status Text: ${statusText}`;

            //If the response status code is not a 2xx success code
            if (response.status() < 200 || response.status() > 299) {
                throw new Error(`Failed to load url: ${sanitizedUrl} ${response.status()} ${response.statusText()}`);
            }
        } else {
            const logNoResponseString = `No response returned for url: ${sanitizedUrl}`;
            log.error(logNoResponseString);
            throw new Error(logNoResponseString);
        }
    });

    // Wait for 15 seconds to let page load fully before taking screenshot.
    if (domcontentloaded && takeScreenshot) {
        await page.waitFor(15000);
        await synthetics.takeScreenshot(stepName, 'loaded');
        await resetPage(page);
    }
};

const urls = [];

exports.handler = async () => {
    return await loadBlueprint();
};

topic = sns.Topic(
    self,
    "heartbeatTopic",
    topic_name="website-heartbeat"
)

sns.Subscription(
    self,
    "admin",
    topic=topic,
    protocol=sns.SubscriptionProtocol.EMAIL,
    endpoint="jeremyritchie1996@hotmail.com"
)

alarm = cloudwatch.CfnAlarm(
    self,
    'HeartbeatAlarm',
    alarm_name='heartbeat-alarm',
    comparison_operator='LessThanOrEqualToThreshold',
    evaluation_periods=1,
    metric_name='SuccessPercent',
    namespace='CloudWatchSynthetics',
    period=3600,
    statistic='Average',
    threshold=99,
    alarm_actions=[topic.topic_arn],
    dimensions=[cloudwatch.CfnAlarm.DimensionProperty(
        name='CanaryName',
        value=canary.canary_name,
    )],
)