blob: 074c14d347042880a93327b96ed74845ee42df65 [file] [log] [blame]
#!/usr/bin/env python
"""
Copyright 2018-present Open Networking Foundation
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
from subprocess import call
import tempfile
import hashlib
import requests, os
import xml.etree.ElementTree, shutil
import time
SONATYPE_USER=os.environ.get("SONATYPE_USER")
SONATYPE_PASSWORD=os.environ.get("SONATYPE_PASSWORD")
SONATYPE_PROFILE=os.environ.get("SONATYPE_PROFILE")
CREATE_REPO_REQUEST_TEMPLATE = '''\
<promoteRequest>
<data>
<description>%(description)</description>
</data>
</promoteRequest>
'''
CLOSE_REPO_REQUEST_TEMPLATE = '''\
<promoteRequest>
<data>
<description>%(description)</description>
<stagedRepositoryId>%(repo_id)</stagedRepositoryId>
</data>
</promoteRequest>
'''
CLOSE_RETRY_ATTEMPTS = 12 * 2
def hashlib_compute(hash, input_file, output_file):
with open(input_file, 'rb') as f:
for block in iter(lambda: f.read(100000), b''):
hash.update(block)
md5_string = hash.hexdigest()
output = open(output_file, "w")
output.write(md5_string + "\n")
f.close()
output.close()
def generate_metadata_files(input_file, dest):
# create a temporary directory to hold the metadata files
global tempdir
base_metadata_filename = tempdir + "/" + os.path.basename(dest)
files = []
# generate the signature file
signature_filename = base_metadata_filename + ".asc"
call(["gpg", "--armor", "--detach-sig", "--output", signature_filename, input_file])
files.append(signature_filename)
# generate the md5 checksum file
md5_filename = base_metadata_filename + ".md5"
md5 = hashlib.md5()
hashlib_compute(md5, input_file, md5_filename)
files.append(md5_filename)
# generate the SHA checksum file
sha1_filename = base_metadata_filename + ".sha1"
sha1 = hashlib.sha1()
hashlib_compute(sha1, input_file, sha1_filename)
files.append(sha1_filename)
# generate the base artifact
base_artifact_filename = base_metadata_filename
shutil.copyfile(input_file, base_artifact_filename)
files.append(base_artifact_filename)
return files
def create_staging_repo(description):
if destination_repo_url is None:
return None
create_request = CREATE_REPO_REQUEST_TEMPLATE.replace("%(description)", description)
url = "https://" + destination_repo_url + "/service/local/staging/profiles" + "/" + SONATYPE_PROFILE + "/start"
headers = {'Content-Type': 'application/xml'}
r = requests.post(url, create_request, headers=headers, auth=(SONATYPE_USER, SONATYPE_PASSWORD))
root = xml.etree.ElementTree.fromstring(r.text)
repo_id = root.find("data").find("stagedRepositoryId").text
return repo_id
def close_staging_repo(description, repo_id):
if repo_id is None:
return
close_request = CLOSE_REPO_REQUEST_TEMPLATE.replace("%(description)", description).replace("%(repo_id)", repo_id)
url = "https://" + destination_repo_url + "/service/local/staging/profiles" + "/" + SONATYPE_PROFILE + "/finish"
headers = {'Content-Type': 'application/xml'}
r = requests.post(url, close_request, headers=headers, auth=(SONATYPE_USER, SONATYPE_PASSWORD))
def wait_for_staging_repo(description, repo_id):
base_url = "https://" + destination_repo_url + "/service/local/staging/profiles" + "/" + SONATYPE_PROFILE
if repo_id is None:
return
close_request = CLOSE_REPO_REQUEST_TEMPLATE.replace("%(description)", description).replace("%(repo_id)", repo_id)
url = base_url + "/finish"
headers = {'Content-Type': 'application/xml'}
repo_query_url = "https://oss.sonatype.org/service/local/staging/repository/" + repo_id
attempt = 1
print ("waiting for repo to close...")
while True:
r = requests.get(repo_query_url, close_request, headers=headers, auth=(SONATYPE_USER, SONATYPE_PASSWORD))
root = xml.etree.ElementTree.fromstring(r.text)
transitioning = root.find("transitioning").text
if transitioning != "true":
break
if attempt == CLOSE_RETRY_ATTEMPTS:
print ("Unable to close repo")
sys.exit(1)
attempt = attempt + 1
time.sleep(5)
print ("Repo closed successfully")
def stage_file(file, repo_id, dest):
filename_in_repo = os.path.dirname(dest) + "/" + os.path.basename(file)
if destination_repo_url is not None:
# deploy to Nexus repo
upload_base = "https://" + destination_repo_url + "/service/local/staging/deployByRepositoryId"
url = upload_base + "/" + repo_id + "/" + filename_in_repo
headers = {'Content-Type': 'application/xml'}
with open(file, 'rb') as f:
r = requests.post(url, data=f.read(), headers=headers, auth=(SONATYPE_USER, SONATYPE_PASSWORD))
if r.status_code != 201:
print (r.status_code)
print (r.text)
sys.exit(1)
else:
# deploy to local repo
file_in_local_repo = os.path.expanduser(local_maven_repo + "/" + filename_in_repo)
dir_in_local_repo = os.path.dirname(file_in_local_repo)
if not os.path.isdir(dir_in_local_repo):
os.makedirs(dir_in_local_repo)
shutil.copyfile(src, file_in_local_repo)
def stage_files(files, dest):
for file in files:
stage_file(file=file, repo_id=repo_id, dest=dest)
def upload_file(src, dest):
print ("publishing: " + dest.replace("org/onosproject", ""))
files = generate_metadata_files(src, dest)
stage_files(files, dest)
if __name__ == '__main__':
import sys
if len(sys.argv) < 2:
print 'USAGE: upload-maven-artifacts catalog-file-name [nexus root url]'
sys.exit(1)
input_list_file = sys.argv[1]
local_maven_repo = None
destination_repo_url = None
if len(sys.argv) == 3:
destination_repo_url = sys.argv[2]
else:
local_maven_repo = os.environ.get("MAVEN_REPO")
if local_maven_repo is None:
local_maven_repo = "~/.m2/repository"
if destination_repo_url is not None:
if SONATYPE_USER is None:
print "Environment variable SONATYPE_USER must be set"
sys.exit(1)
if SONATYPE_PASSWORD is None:
print "Environment variable SONATYPE_PASSWORD must be set"
sys.exit(1)
if SONATYPE_PROFILE is None:
print "Environment variable SONATYPE_PROFILE must be set"
sys.exit(1)
print ("Uploading to remote repo: " + destination_repo_url)
else:
print ("Installing in local repo: " + local_maven_repo)
list_file = open(input_list_file, "r")
lines = list_file.readlines()
list_file.close()
tempdir = tempfile.mkdtemp(prefix="upload-maven-artifacts-")
description = "test repo"
repo_id = create_staging_repo(description)
for line in lines:
s = line.split()
src = s[0]
dest = s[1]
upload_file(src, dest)
close_staging_repo(repo_id=repo_id, description=description)
wait_for_staging_repo(repo_id=repo_id, description=description)
shutil.rmtree(tempdir)