#!/usr/bin/env python

import json

import os

import fire

from aws_session_assume import get_boto_session

AWS_REGION = os.getenv("AWS_REGION", "eu-west-2")

def main(secret_env_name: str = "dev", bucket_env_name: str = "dev", env: str = "dev"):

boto_session = get_boto_session(env)

print("Getting smoke test parameters from AWS....") # noqa

smoke_test_params_name = f"nhsd-nrlf--{secret_env_name}--smoke-test-parameters"

secretsmanager = boto_session.client("secretsmanager", region_name=AWS_REGION)

smoke_test_params_value = secretsmanager.get_secret_value(

SecretId=smoke_test_params_name

)

print("Parsing parameters....") # noqa

smoke_test_params = json.loads(smoke_test_params_value["SecretString"])

nrlf_app_id = smoke_test_params["nrlf_app_id"]

ods_code = smoke_test_params["ods_code"]

print(f"Adding {ods_code} permissions to {nrlf_app_id} app in S3....") # noqa

bucket = f"nhsd-nrlf--{bucket_env_name}-authorization-store"

s3 = boto_session.client("s3")

s3.put_object(

Bucket=bucket,

Key=f"{nrlf_app_id}/{ods_code}.json",

Body=open(f"./tests/smoke/permissions/{ods_code}.json", "rb"),

ExpectedBucketOwner=boto_session.client("sts")

.get_caller_identity()

.get("Account"),

)

if __name__ == "__main__":

fire.Fire(main)