Module 2. Build a serverless backend

{
  "Version": "2012-10-17",
  "Statement": [
    {
      "Sid": "",
      "Effect": "Allow",
      "Principal": {
        "Service": [
          "lambda.amazonaws.com",
          “states.amazonaws.com”,
          "apigateway.amazonaws.com"
        ]
      },
      "Action": "sts:AssumeRole"
    }
  ]
}

from __future__ import print_function
import json
import boto3
appstream = boto3.client('appstream')

def lambda_handler(event, context):
    # Log the received event
    print("Received event: " + json.dumps(event, indent=2))
    # Get variables from web form
    strIBName = event['ib_name']
    strIBARN = event['ib_arn']
    strInstType = event['instance_type']
    strSubnetId = event['subnet1id']
    strSecurityGroups = event['securitygroups']
    strUserEmail = event['email']
    strUserName = event['user_name']
    strUserEmail = strUserEmail.strip()
    strUserName= strUserName.strip()  

    try:
        response = appstream.create_image_builder(
            Name=strIBName,
            ImageArn=strIBARN,
            VpcConfig={
                'SubnetIds':[
                    strSubnetId
                    ],
                'SecurityGroupIds':[
                    strSecurityGroups
                    ]
            },
            InstanceType=strInstType)
        print("Started: " + strIBName)
        return 1
    except Exception as e:
        print(e)
        message = 'Error creating image builder'
        print(message)

from __future__ import print_function

import json
import boto3

print('Loading function')
appstream = boto3.client('appstream')

def Status_to_Return(received_state):
    switcher = {
    "PENDING": 0,
    "RUNNING": 1,
    "SNAPSHOTTING": 2,
    "STOPPED": 3,
    "STOPPING": 4,
    "STARTING": 5  
    }
    return switcher.get(received_state, -1)

def lambda_handler(event, context):
    # Log the received event
    print("Received event: " + json.dumps(event, indent=2))
    # Get names to query against
    strIBName = event['ib_name']
    strFleetName = event['fleet_name']
    strType = event['type']
    try:
        #Fleet status check
        if strType == 2:
            print ("Type is Fleet")
            AS2state = appstream.describe_fleets(Names=[strFleetName])
            Status = AS2state['Fleets'][0]['State']
            print ("State: " + Status)
        #Image Builder status check
        elif strType == 1:
            print ("Type is ImageBuilder")
            AS2state = appstream.describe_image_builders(Names=[strIBName])
            Status = AS2state['ImageBuilders'][0]['State']
            print ("State: " + Status)
        return Status_to_Return(Status)
    except Exception as e:
        print(e)
        print("Error getting status")
        return -1

from __future__ import print_function

import json
import boto3

print('Loading function')
appstream = boto3.client('appstream')
ses = boto3.client('ses',region_name=<REGION-CODE>)

def lambda_handler(event, context):
    # Log the received event
    print("Received event: " + json.dumps(event, indent=2))
    # Get resource names
    strUserName = event['user_name']
    strUserName = strUserName.strip()
    strEmail = event['email']
    strEmail = strEmail.strip()
    strIBName = event['ib_name']
    strFleetName = event['fleet_name']
    strStackName = event['stack_name']
    strType = event['type']

    try:
        #Create Fleet Streaming URL
        if strType == 2:
            print ("Type is Fleet")
            mailBody = "Your fleet is now created and ready for verification. Please use the connect link below to connect to the instance."
            url = appstream.create_streaming_url(
                StackName=strStackName,
                FleetName=strFleetName,
                UserId=strUserName,
                Validity=604800
            )            
        #Create Image Builder Streaming URL
        elif strType == 1:
            print ("Type is ImageBuilder")
            mailBody = "If this is your first time installing your application with AppStream2.0, you can check the getting started guide below for assistance. Otherwise, please click the connect link below to get started."
            url = appstream.create_image_builder_streaming_url(
                Name=strIBName,
                Validity=604800
            )
        strStreamingURL = url['StreamingURL']
        print("Streaming URL: " + strStreamingURL)
        #Send Streaming URL in E-mail via SES
        ses.send_email(
            Source='noreply@example.com',
            Destination={
                'ToAddresses': [strEmail],
                        },
            Message={
                'Body': {
                'Html': {
                'Charset': 'UTF-8',
                'Data': '<html><head><title>Example Corp</title></head><body bgcolor="#e0e0e0">' + strUserName + ',<p>Thank you for trying out the Example Corp <strong>Enterprise Application Onboarding</strong> demo. ' + mailBody + '</p><p><a href=" ' + strStreamingURL + ' ">Connect</a></p><a href="https://d1.awsstatic.com/product-marketing/AppStream2.0/Amazon%20AppStream%202.0%20Getting%20Started%20Guide%20April%202018.pdf">Getting Started Guide</a></body></html>',
                },
                },
                'Subject': {
                'Charset': 'UTF-8',
                'Data': 'Example Corp Enterprise Application Onboarding for ' + strUserName                    
                },
            }
        )
        return 1

    except Exception as e:
        print(e)
        print("Error getting streaming URL")
        return -1

from __future__ import print_function

import json
import boto3
import datetime

print('Loading function')
appstream = boto3.client('appstream')

def lambda_handler(event, context):
    # Log the received event
    print("Received event: " + json.dumps(event, indent=2))
    # Get names to query against
    strIBName = event['ib_name']
    try:
        ImagesCMD = appstream.describe_images(Type='PRIVATE')
        Images = ImagesCMD['Images']
        for Image in Images:
            if Image.get('ImageBuilderName','') == strIBName:
                ImageTime = Image['CreatedTime']
                now = datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc)
                howold = now - ImageTime
                HowOldSec = howold.total_seconds()
                if HowOldSec < 3600:
                    imagename = Image['Name']
                    return (imagename)
    except Exception as e:
        print(e)
        print("Error getting Image Name")
        return -1

from __future__ import print_function
import json
import boto3
print('Loading function')
appstream = boto3.client('appstream')

def lambda_handler(event, context):
    # Log the received event
    print("Received event: " + json.dumps(event, indent=2))
    strFleetName = event['fleet_name']
    strStackName = event['stack_name']
    strImageName = event['imagename']
    strInstType = event['instance_type']
    strSubnet1Id = event['subnet1id']
    strSubnet2Id = event['subnet2id']
    strSecurityGroups = event['securitygroups']

    try:
        #Create AppStream fleet
        fleet = appstream.create_fleet(
            Name=strFleetName,
            ImageName=strImageName,
            ComputeCapacity={
                'DesiredInstances': 1
            },            
            VpcConfig={
                'SubnetIds':[
                    strSubnet1Id,
                    strSubnet2Id
                    ],
                'SecurityGroupIds':[
                    strSecurityGroups
                    ]
            },
            InstanceType=strInstType)
        #Create AppStream stack
        stack = appstream.create_stack(
            Name=strStackName,
            StorageConnectors=[
                {
                    'ConnectorType': 'HOMEFOLDERS'
                },
            ],
            UserSettings=[
                {
                    'Action': 'CLIPBOARD_COPY_FROM_LOCAL_DEVICE',
                    'Permission': 'ENABLED'
                },
            ],
            ApplicationSettings={
                'Enabled': False
            }
        )
        #Associate Stack to Fleet
        associate = appstream.associate_fleet(
            FleetName=strFleetName,
            StackName=strStackName
        )
        startfleet = appstream.start_fleet(
            Name=strFleetName
        )
        return 2
    except Exception as e:
        print(e)
        print("Error creating fleet")
        return -1

from __future__ import print_function

import json
import boto3

print('Loading function')
appstream = boto3.client('appstream')

def Status_to_Return(received_state):
    switcher = {
    "PENDING": 0,
    "RUNNING": 1,
    "SNAPSHOTTING": 2,
    "STOPPED": 3,
    "STOPPING": 4
    }
    return switcher.get(received_state, -1)

def lambda_handler(event, context):
    # Log the received event
    print("Received event: " + json.dumps(event, indent=2))
    # Get names to query against
    strIBName = event['ib_name']
    strFleetName = event['fleet_name']
    strType = event['type']
    try:
        #Fleet status check
        if strType == 2:
            print ("Type is Fleet")
            AS2state = appstream.describe_fleets(Names=[strFleetName])
            Status = AS2state['Fleets'][0]['State']
            State = Status_to_Return(Status)
            if State < 3:            
                print (strFleetName + " is " + Status + " attempting to stop")
                AS2stop = appstream.stop_fleet(Name=strFleetName)
            print ("State: " + Status)
        #Image Builder status check
        elif strType == 1:
            print ("Type is ImageBuilder")
            AS2state = appstream.describe_image_builders(Names=[strIBName])
            Status = response['ImageBuilders'][0]['State']
            State = Status_to_Return(Status)
            if State < 3:
                print (strIBName + " is " + Status + " attempting to stop")
                AS2stop = appstream.stop_image_builder(Names=[strIBName])
            print ("State: " + Status)
        return Status_to_Return(Status)
    except Exception as e:
        print(e)
        print("Error stopping services")
        return -1