import argparse
import sys
import time

import requests


def parse_args():
    parser = argparse.ArgumentParser()

    parser.add_argument("pr_id", help="Pull Request ID")
    parser.add_argument("merge_sha", help="Merge Commit SHA")
    parser.add_argument("head_sha", help="HEAD Commit SHA")
    parser.add_argument("run_attempt", help="# of Run Attempt")
    parser.add_argument("branch_name", help="PR's Branch Name")
    parser.add_argument("URL", help="URL to submit build task")
    parser.add_argument("client_id", help="CF Access client id")
    parser.add_argument("secret", help="CF Access client secret")

    return parser.parse_args()


if __name__ == "__main__":
    args = parse_args()

    # Submit build job
    headers = {
        "CF-Access-Client-Id": args.client_id,
        "CF-Access-Client-Secret": args.secret,
    }

    payload = {
        "pr_id": args.pr_id,
        "merge_commit_sha": args.merge_sha,
        "head_commit_sha": args.head_sha,
        "run_attempt": args.run_attempt,
        "branch_name": args.branch_name,
    }

    workflow_id = ""
    try:
        resp = requests.post(args.URL, headers=headers, json=payload)
        resp.raise_for_status()

        workflow_id = resp.json()["workflow_id"]
    except Exception as err:
        print(f"Unexpected error {err=}, {type(err)=}, exiting.")
        sys.exit(1)

    print("Internal build submitted.")

    time.sleep(30)

    # Poll build status
    failed_requests = 0
    FAILED_REQUEST_LIMIT = 10

    while failed_requests < FAILED_REQUEST_LIMIT:
        try:
            resp = requests.get(args.URL, headers=headers, params={"id": workflow_id})
            resp.raise_for_status()

            status = resp.json()["status"]
            if status == "errored":
                print("Internal build failed.")
                print(resp.json()["error"]["message"])
                print(
                    "If you are a Cloudflare employee, please check your internal"
                    " branch and refer this doc for further details:"
                    " https://cflare.co/workerd-internal-build"
                )
                print(
                    "If you are an external contributor, please ask the auto-assigned"
                    " reviewers to check the internal build failure."
                )
                sys.exit(1)
            elif status == "complete":
                break

            print("Waiting for build to finish..")

        except Exception as err:
            print(f"Unexpected error {err=}, {type(err)=}")
            failed_requests = failed_requests + 1
        time.sleep(30)

    if failed_requests == FAILED_REQUEST_LIMIT:
        print(f"{failed_requests=} == {FAILED_REQUEST_LIMIT=}, exiting.")
        sys.exit(1)

    print("Internal build succeeded.")