#!/usr/bin/env python3
import oci
import time
import smtplib
import sys
from email.mime.text import MIMEText

import os
from dotenv import load_dotenv
load_dotenv(os.path.join(os.path.dirname(__file__), ".env"))

COMPARTMENT_ID = os.environ["COMPARTMENT_ID"]
AVAILABILITY_DOMAIN = os.environ["AVAILABILITY_DOMAIN"]
SUBNET_ID = os.environ["SUBNET_ID"]
IMAGE_ID = os.environ["IMAGE_ID"]
SSH_KEY = os.environ["SSH_KEY"]
DISPLAY_NAME = os.environ.get("DISPLAY_NAME", "free-arm-instance")
SHAPE = "VM.Standard.A1.Flex"
OCPUS = 4
MEMORY_GB = 24
INTERVAL = 60

EMAIL_TO = os.environ["EMAIL_TO"]
EMAIL_FROM = os.environ["EMAIL_FROM"]
EMAIL_APP_PASSWORD = os.environ["EMAIL_APP_PASSWORD"]

def send_email(subject, body):
    msg = MIMEText(body)
    msg["Subject"] = subject
    msg["From"] = EMAIL_FROM
    msg["To"] = EMAIL_TO
    with smtplib.SMTP_SSL("smtp.gmail.com", 465) as s:
        s.login(EMAIL_FROM, EMAIL_APP_PASSWORD)
        s.send_message(msg)

config = oci.config.from_file()
compute = oci.core.ComputeClient(config)

print("Starting ARM instance monitor...", flush=True)
while True:
    try:
        resp = compute.launch_instance(
            oci.core.models.LaunchInstanceDetails(
                compartment_id=COMPARTMENT_ID,
                availability_domain=AVAILABILITY_DOMAIN,
                shape=SHAPE,
                shape_config=oci.core.models.LaunchInstanceShapeConfigDetails(ocpus=OCPUS, memory_in_gbs=MEMORY_GB),
                image_id=IMAGE_ID,
                subnet_id=SUBNET_ID,
                display_name=DISPLAY_NAME,
                metadata={"ssh_authorized_keys": SSH_KEY},
            )
        )
        instance_id = resp.data.id
        msg = f"ARM instance created!\nID: {instance_id}"
        print(msg, flush=True)
        send_email("OCI ARM Instance Created!", msg)
        sys.exit(0)
    except oci.exceptions.ServiceError as e:
        if "capacity" in str(e.message).lower() or "host" in str(e.message).lower():
            print(f"{time.strftime('%Y-%m-%d %H:%M:%S', time.gmtime(time.time() + 8*3600))} No capacity. Retrying in {INTERVAL}s...", flush=True)
        else:
            print(f"Unexpected error: {e.message}", flush=True)
            send_email("OCI ARM Monitor Error", str(e.message))
            sys.exit(1)
    time.sleep(INTERVAL)