blob: 226e16238f5b081d1b2f3fb7c72176fbeaf65050 [file] [log] [blame]
#!/usr/bin/env python3
import argparse
import boto3
import os
import sys
S3_DEFAULT_BUCKET = 'archives.webkit.org'
S3_EWS_BUCKET = 'ews-archives.webkit.org'
S3_MINIFIED_BUCKET = 'minified-archives.webkit.org'
S3_REGION_PREFIX = 'https://s3-us-west-2.amazonaws.com'
def uploadToS3(archive_path, bucket, identifier, revision):
print('Transferring {} to S3...'.format(archive_path))
key = '/'.join([identifier, revision + '.zip'])
print('\tS3 Bucket: {}\n\tS3 Key: {}'.format(bucket, key))
s3 = boto3.client('s3')
s3.upload_file(archive_path, bucket, key)
print('\tS3 URL: {}/{}/{}'.format(S3_REGION_PREFIX, bucket, key))
def archiveExists(archive):
if archive:
if os.path.exists(archive):
return True
else:
print('WARNING: Archive does not exist: {}'.format(archive))
return False
def main():
parser = argparse.ArgumentParser(add_help=True)
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--revision', action="store", help='Revision number or change identifier for the built archive')
group.add_argument('--change-id', dest='change_id', action="store", help='patch id or hash of specified change')
parser.add_argument('--identifier', action="store", required=True, help='S3 destination identifier, in the form of fullPlatform-architecture-configuration. [mac-mojave-x86_64-release]')
parser.add_argument('--archive', action="store", required=True, help='Path to the full size archive. [path/to/123456.zip]')
args = parser.parse_args()
parentdir, filename = os.path.split(str(args.archive))
minifiedArchive = os.path.join(parentdir, 'minified-' + filename)
s3_bucket = S3_DEFAULT_BUCKET
if args.change_id:
s3_bucket = S3_EWS_BUCKET
if archiveExists(args.archive):
uploadToS3(args.archive, s3_bucket, args.identifier, args.revision or args.change_id)
if not args.change_id and archiveExists(minifiedArchive):
uploadToS3(minifiedArchive, S3_MINIFIED_BUCKET, args.identifier, args.revision)
if __name__ == "__main__":
main()