import os
import sys
import json

from framework_list import frameworks
from functions import log, run_command
import time


def get_cocoapods_trunk_token():
    (exit_code, out, error) = run_command([
        "aws", "secretsmanager", "get-secret-value", "--secret-id", os.environ['COCOAPODS_SECRET_ARN']
    ])
    if exit_code != 0:
        log(f"Failed to get CocoaPods token from AWS. Output: {out}, Error: {error}")
        sys.exit(exit_code)

    return json.loads(out)["SecretString"]

log(f"Retrieving CocoaPods token")
os.environ['COCOAPODS_TRUNK_TOKEN'] = get_cocoapods_trunk_token()
log("Retrieved CocoaPods token")

def publish(framework: str, retryAttempt: int = 0):

    # Most pods take a few minutes to build, and a few seconds to push to trunk. However, the
    # AWSiOSSDK podspec can take a long time to build, since it builds each dependent pod as
    # part of its linting process, so set the timeout accordingly.
    (exit_code, out, err) = run_command(
        ["bundle", "exec", "pod", "trunk", "push", f"{framework}.podspec", "--allow-warnings", "--synchronous"],
        keepalive_interval=300,
        timeout=3600,
    )

    if retryAttempt < 3 and exit_code != 0 and "Unable to accept duplicate entry for" not in str(out):
        retryAttempt += 1
        log(f"Retrying {framework}")
        return publish(framework=framework, retryAttempt=retryAttempt)
    return (exit_code, out, err)

log("Publishing CocoaPods")

for framework in frameworks:
    log(f"Publishing {framework}")

    (exit_code, out, err) = publish(framework=framework)

    if exit_code != 0 and "Unable to accept duplicate entry for" in str(out):
        log(f"Already published {framework}")
    elif exit_code == 0:
        log(f"Published {framework}")
    else:
        log(f"Could not publish {framework}: output: {out}; error: {err}")
        sys.exit(exit_code)

    if framework == "AWSCore":
        log(f"pod repo update after {framework}")
        (exit_code, out, err) = run_command(
            ["bundle", "exec", "pod", "repo", "update"],
            keepalive_interval=300,
            timeout=3600,
        )
        if exit_code != 0:
            log(f"Failed to update CocoaPods repo'; output={out}, error={err}")

sys.exit(0)