#!/usr/bin/env python3

# Script-style deployment testing: any error should fail the test and have non-zero exit code.

import os
import sys
import requests
import urllib.parse
from specklepy.api.client import SpeckleClient
from specklepy.api.models import ServerInfo


# Setting the SPECKLE_SERVER to test on
SPECKLE_SERVER = ""

if len(sys.argv) > 1:
    SPECKLE_SERVER = sys.argv[1]
if not SPECKLE_SERVER:
    SPECKLE_SERVER = os.getenv("SPECKLE_SERVER", "")
if not SPECKLE_SERVER:
    print(
        "Error: No Speckle server specified. Use SPECKLE_SERVER environment variable or pass it as the first command-line argument"
    )
    exit(1)

if not SPECKLE_SERVER.startswith("http://") and not SPECKLE_SERVER.startswith(
    "https://"
):
    SPECKLE_SERVER = "http://" + SPECKLE_SERVER

print(f"Using Speckle server '{SPECKLE_SERVER}'")

# Test if frontend is accessible
frontend_response = requests.get(urllib.parse.urljoin(SPECKLE_SERVER, "logo.svg"))
# don't check for status code, the frontend app will server the 404 page with a status code 200
# even if the rote doesn't exist
assert frontend_response.headers.get("Content-Type", "").startswith(
    "image/"
), "Frontend logo Content-Type is not an image"
print("Frontend accessible")

# Test basic unauthenticated operation using specklepy
client = SpeckleClient(SPECKLE_SERVER, use_ssl=SPECKLE_SERVER.startswith("https://"))
server_info = client.server.get()
assert isinstance(server_info, ServerInfo), "GraphQL ServerInfo query error"
print(f"GraphQL operation succeeded. Server name: {server_info.name}")

# Test that the deployed server version matches the expected version
SERVER_VERSION = ""
if len(sys.argv) > 2:
    SERVER_VERSION = sys.argv[2]
if not SERVER_VERSION:
    SERVER_VERSION = os.getenv("SERVER_VERSION")
if SERVER_VERSION:
    if not SERVER_VERSION == "latest":
        assert server_info.version.startswith(
            SERVER_VERSION
        ), f"The deployed version {server_info.version} should match, or be prefixed by, the expected {SERVER_VERSION}"
        print(f"Server version {SERVER_VERSION} is deployed and available")
    else:
        print("Not testing server version, as it was set to 'latest'")
else:
    print(
        "Not testing server version, as an expected value was not provided via environment variables or command-line argument"
    )

print("Deployment tests PASS")