#!/usr/bin/env python3
import os
import boto3
import botocore

SRC_BUCKET = 'status-im'
DST_BUCKET = 'status-im-prs'

session = boto3.session.Session()
s3 = session.client('s3',
                    region_name='ams3',
                    endpoint_url='https://ams3.digitaloceanspaces.com',
                    aws_access_key_id=os.environ['DO_ID'],
                    aws_secret_access_key=os.environ['DO_SECRET'])

for f in s3.list_objects_v2(Bucket=SRC_BUCKET)['Contents']:
    name = f['Key']
    # ignore is build is not a pr
    if 'pr' != name.split('-')[-1].split('.')[0]:
        continue
    print('{:<25} {} -> {}'.format(name, SRC_BUCKET, DST_BUCKET))
    path = '/tmp/{}'.format(name)
    s3.download_file(SRC_BUCKET, name, path)
    s3.upload_file(path, DST_BUCKET, name)
    s3.delete_object(Bucket=SRC_BUCKET, Key=name)
    os.remove(path)
    break