import os

import requests
from dotenv import load_dotenv
from flask import Flask, request, jsonify

load_dotenv()

# Getting our settings from env
ARBITRIUM_TOKEN = os.getenv("ARBITRIUM_TOKEN")
APPLICATION = os.getenv("APPLICATION")
VERSION = os.getenv("VERSION")

app = Flask(__name__)


@app.route('/servers', methods=['POST'])
def start_server():
    # How you access the user's IP is unimportant for us, but you will need it to create a deployment.
    # Here the user sent its IP in the request to get the TODO application, along with its username. 
    # We will inject it into the environment variables of our deployment.
    data = request.get_json()
    ip = data.get("ip")
    username = data.get("username")

    headers = {
        "Content-Type": "application/json",
        "Authorization": ARBITRIUM_TOKEN
    }

    # Building request body from the API specifications (/v1/deploy)
    deployment_request = {
        # These are mandatory to create a deployment
        "app_name": APPLICATION,
        "version_name": VERSION,
        "ip_list": [
            ip
        ],
        # These are not mandatory, but recommended
        "env_vars": [
            {
                "key": "message",
                "value": f"Welcome {username} to your personal TODO application!"
            }
        ],
        # Arbitrium will make a POST request to this URL once the deployment container is accessible.
        "webhook_url": "https://your-api.com/webhooks"
    }

    # Sending the request to Arbitrium to create a deployment
    response = requests.post("https://api.edgegap.com/v1/deploy", json=deployment_request, headers=headers)
    
    # Getting the Request ID from our request
    request_id = response.json().get("request_id")

    # Sending the Request ID to our client. Not mandatory, but it can be helpful to retrieve the deployment.
    return jsonify({"request_id": request_id}), 200


@app.route('/webhooks', methods=['POST'])
def webhook_route():
    # Accessing the data we need to forward to our client.
    data = request.get_json()
    request_id = data.get("request_id")
    
    # This is the URL that our client needs to connect to access the TODO application online.
    # Port 3000 is the one we set in our Dockerfile and our version
    connection_url = data.get("ports", {}).get("3000", {}).get("link")

    connection_data = {
        "request_id": request_id,
        "connection_url": connection_url
    }

    # Notifying our client of the connection informations
    requests.post("https://your-route-to-notify-user-client.com", json=connection_data)

    return 204


if __name__ == '__main__':
    app.run()