#!/usr/bin/env python3

import json
import os
import pprint
import re
import subprocess
import sys
import urllib.request

event_name = os.environ["GITHUB_EVENT_NAME"]

is_pull_request = is_push = False
if event_name.startswith("pull_request"):
    is_pull_request = True
elif event_name.startswith("push"):
    is_push = True
elif event_name.startswith("schedule"):
    # Don't publish; scheduled workflows run against the latest commit of every
    # implementation, so they are likely to have failed tests for the wrong reasons
    sys.exit(0)
else:
    print("Unexpected event name:", event_name)

with open(os.environ["GITHUB_EVENT_PATH"]) as fd:
    github_event = json.load(fd)

pprint.pprint(github_event)

context_suffix = ""

command = ["netlify", "deploy", "--dir=dashboard/"]
if is_pull_request:
    pr_number = github_event["number"]
    sha = github_event.get("after") or github_event["pull_request"]["head"]["sha"]
    # Aliases can't exceed 37 chars
    command.extend(["--alias", f"pr-{pr_number}-{sha[0:10]}"])
    context_suffix = " (pull_request)"
elif is_push:
    ref = github_event["ref"]
    m = re.match("refs/heads/(.*)", ref)
    if m:
        branch = m.group(1)
        sha = github_event["head_commit"]["id"]

        if branch in ("main", "master"):
            command.extend(["--prod"])
        else:
            command.extend(["--alias", f"br-{branch[0:23]}-{sha[0:10]}"])
            context_suffix = " (push)"
    else:
        # TODO
        pass


print("Running", command)
proc = subprocess.run(command, capture_output=True)

output = proc.stdout.decode()
assert proc.returncode == 0, (output, proc.stderr.decode())
print(output)

m = re.search("https://[^ ]*--[^ ]*netlify.app", output)
assert m
netlify_site_url = m.group(0)
target_url = f"{netlify_site_url}/index.xhtml"

print("Published to", netlify_site_url)


def send_status() -> None:
    statuses_url = github_event["repository"]["statuses_url"].format(sha=sha)

    payload = {
        "state": "success",
        "context": f"Dashboard{context_suffix}",
        "description": "Table of all test results",
        "target_url": target_url,
    }
    request = urllib.request.Request(
        statuses_url,
        data=json.dumps(payload).encode(),
        headers={
            "Authorization": f'token {os.environ["GITHUB_TOKEN"]}',
            "Content-Type": "text/json",
            "Accept": "application/vnd.github+json",
        },
    )

    response = urllib.request.urlopen(request)

    assert response.status == 201, response.read()


send_status()


def send_pr_comment() -> None:
    comments_url = github_event["pull_request"]["_links"]["comments"]["href"]

    payload = {
        "body": f"[Test results]({target_url})",
    }
    request = urllib.request.Request(
        comments_url,
        data=json.dumps(payload).encode(),
        headers={
            "Authorization": f'token {os.environ["GITHUB_TOKEN"]}',
            "Content-Type": "text/json",
            "Accept": "application/vnd.github+json",
        },
    )

    response = urllib.request.urlopen(request)

    assert response.status == 201, response.read()


if is_pull_request:
    send_pr_comment()