import json
import boto3
import os
import datetime
from botocore.exceptions import ClientError


# Pinpoint Project Id
application_id = "4fcac97e77254e918fa8f9d919afde1e"

region = os.environ['AWS_REGION']
client = boto3.client('pinpoint',region_name=region)

def get_segment_id(segment_name):
    try:
        response = client.get_segments(
            ApplicationId=application_id
        )
        segment_list = response['SegmentsResponse']['Item']
        for one in segment_list:
            if one['Name'] == segment_name:
                segment_id = one['Id']
    except ClientError as e:
        print(e.response['Error']['Message'])
    else:
        print(segment_id)
        # print(json.dumps(response))
    return segment_id

def create_campaign(title, message, segment_id, icon_url, image_url):
    print(segment_id)
    try:
        response = client.create_campaign(
            ApplicationId=application_id,
            WriteCampaignRequest={
                'MessageConfiguration': {
                    'DefaultMessage': {
                        'Action': 'OPEN_APP',
                        'Body': message,
                        'Title': title,
                        'ImageIconUrl': icon_url,
                        'ImageUrl': image_url
                        # 'ImageUrl': 'http://www.earlyadopter.co.kr/wp-content/uploads/2019/11/apple-airpods-pro-early-adopter-review-1.jpg'
                        # 'MediaUrl': 'https://m.media-amazon.com/images/G/01/kindle/merch/2019/ONTHEGO/19951312/PUGE0013_Amazon_Puget_US_REV_2019_45_HD-forDP.mp4?_=1'
                    },
                },
                'Name': title,
                'Description': "캠페인 메시지",
                'SegmentId': segment_id,
                'Schedule': {
                    'StartTime': "IMMEDIATE"
                    # 'Frequency': 'ONCE',
                    # 'StartTime': datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc).isoformat()
                }
            }
        )
    except ClientError as e:
        print(e.response['Error']['Message'])
    else:
        print (json.dumps(response))

def lambda_handler(event, context):
    if "body" in event:
        event = json.loads(event['body'])

    # Query parmeters
    title = event["title"]
    message = event["message"]
    segment = event["segment"]
    icon_url = event["icon"]
    image_url = event["image"]

    segment_id = get_segment_id(segment)
    response = create_campaign(title, message, segment_id, icon_url, image_url)

    return {
        'statusCode': 200,
        'body': json.dumps({'result': response}),
        'headers': {
            'Content-Type': 'application/json',
            'Access-Control-Allow-Origin': '*'
        }
    }

{
  "segment": "cart",
  "title": "안녕하세요 {{User.UserId}}님",
  "message": "신제품 Echo Bud를 소개 합니다.",
  "icon": "https://apprecs.org/ios/images/app-icons/256/6d/580990573.jpg",
  "image": "https://pplware.sapo.pt/wp-content/uploads/2019/09/Amazon_Echo_Buds_02.jpg"
}

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Sid": "VisualEditor0",
            "Effect": "Allow",
            "Action": [
                "logs:CreateLogStream",
                "logs:PutLogEvents"
            ],
            "Resource": "arn:aws:logs:us-east-1:{account-id}:log-group:/aws/lambda/PinpointPushTest:*"
        },
        {
            "Sid": "VisualEditor1",
            "Effect": "Allow",
            "Action": "mobiletargeting:*",
            "Resource": "*"
        },
        {
            "Sid": "VisualEditor2",
            "Effect": "Allow",
            "Action": "logs:CreateLogGroup",
            "Resource": "arn:aws:logs:us-east-1:{account-id}:*"
        }
    ]
}

'use strict';
console.log('Loading function');

exports.handler = async (event, context) => {
    /* Process the list of records and transform them */
    const output = event.records.map((record) => {
        const entry = (new Buffer.from(record.data, 'base64')).toString('utf8');
        var json_data = JSON.parse(entry);
        json_data.event_timestamp   = new Date(json_data.event_timestamp).toISOString();
        json_data.arrival_timestamp = new Date(json_data.arrival_timestamp).toISOString();
        console.log(json_data);
        const result = JSON.stringify(json_data) + '\n';
        const payload = (new Buffer.from(result, 'utf8')).toString('base64');

        /* Transformed event */
        return {
            recordId: record.recordId,
            result: 'Ok',
            data: payload,
        };
    });
    console.log(`Processing completed.  Successful records ${output.length}.`);
    console.log(output);
    return { records: output };
};