import boto3
import json
import requests
import time

webhook_url = 'https://open.feishu.cn/open-apis/bot/v2/hook/bde7fdc7-830b-45da-97eb-ee66975f05fc'
def post_to_feishu():
    headers = {
        'Content-Type': 'application/json'
    }

    payload_message = {
        "timestamp": str(int(time.time())),
        "msg_type": "interactive",
        "card": {
            "i18n_elements": {
                "en_us": [
                    {
                        "tag": "div",
                        "text": {
                            "tag": "lark_md",
                            "content": "**AWS Seed Robot**"
                        },
                        "fields": [
                            {
                                "is_short": False,
                                "text": {
                                    "tag": "lark_md",
                                    "content": ""
                                }
                            },
                            {
                                "is_short": True,
                                "text": {
                                    "tag": "lark_md",
                                    "content": "自动关闭 AWS Us-east-1 Seed 异常"
                                }
                            }
                        ]
                    },
                ]
            }
        }
    }

    response = requests.request(
        "POST", webhook_url, headers=headers, data=json.dumps(payload_message))
    print(response.text)

def shut_down_seed():
    asgClient = boto3.client('autoscaling', region_name="us-east-1")

    response = asgClient.describe_auto_scaling_groups()

    asgPrefix = ['staging-us-east-1-f01', "dev-rajeshwer-us-east-1",
                 'dev-us-east-1-zhangxc', 'staging-seed-us-east-1',
                 'dev-seed-us-east-1']
    print(response)

    asgs = response['AutoScalingGroups']

    for asg in asgs:
        asgName = asg['AutoScalingGroupName']

        match = False
        for asg in asgPrefix:
            if asg in asgName:
                match = True
                break

        if not match:
            continue

        print("process asg {}".format(asgName) )
        asgClient.set_desired_capacity(
            AutoScalingGroupName=asgName,
            DesiredCapacity=0,
            HonorCooldown=False
        )

def lambda_handler(event, context):
    try:
        shut_down_seed()
    except:
        post_to_feishu()

    return {
        'statusCode': 200,
        'body': json.dumps('Hello from Lambda!')
    }