import os
import streamlit as st
import googleapiclient.discovery
from google.oauth2 import service_account

def auth():
    with open("/tmp/token.json", "w") as token_f:
        google_key = os.getenv("GOOGLE_KEY")
        token_f.write(google_key)
        
    credentials = service_account.Credentials.from_service_account_file('/tmp/token.json')
    
    return credentials

def get_status(credentials, instance_name):
    status = False
    client = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
    response = client.instances().list(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c').execute()
    for item in response['items']:
        if item["name"] == instance_name:
            try:
                ip_address = item['networkInterfaces'][0]['accessConfigs'][0]['natIP']
            except Exception as e:
                ip_address = None
            status = True if item["status"] == "RUNNING" else False
    return status, ip_address

def get_server(credentials, instance_name):
    service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
    request = service.instances().get(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name)
    response = request.execute()
    return response

def schedule_server(credentials, instance_name):
    service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
    request = service.instances().setScheduling(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name, body={"automaticRestart": False,"maxRunDuration": {"seconds": 7200}})    
    response = request.execute()
    return response

def activate_server(credentials, instance_name):
    service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
    request = service.instances().start(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c', instance=instance_name)
    response = request.execute()
    return response

def deactivate_server(credentials, instance_name):
    service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
    request = service.instances().stop(project='nus-cisco-corp-lab-wp1',zone='asia-southeast1-c',instance=instance_name, discardLocalSsd=False)
    response = request.execute()
    return response

st.title("Lucky Reactor Panel")

instance_name = "lucky-reactor"
credentials = auth()
status, ip_address = get_status(credentials, instance_name)

st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running 🚀" if status else "sleeping 😴") + ".")

if not status:            
    with st.form("activate_form"):
        token = st.text_input("Token (FYI: The reactor costs USD $10 per hour to operate)")
        submitted = st.form_submit_button("Ignite 🚀")
        
        if submitted and token == os.getenv("EASY_TOKEN"):
            st.write("Lucky Reactor has been ignited. Please wait a few minutes (3~5) to preceed...")     
            response = schedule_server(credentials, instance_name)
            response = activate_server(credentials, instance_name)
else:
    st.write(f"You can access Lucky Reactor via http://{ip_address}:7860")
    info = get_server(credentials, instance_name)
    st.write(info)
    with st.form("deactivate_form"):
        token = st.text_input("Token")
        submitted = st.form_submit_button("Terminate 🕊️")
        
        if submitted and token == os.getenv("EASY_TOKEN"):
            st.write("Lucky Reactor has been terminated.")            
            response = deactivate_server(credentials, instance_name)