#!/usr/bin/python3
#
# Copyright (C) 2020 The Android Open Source Project
#
# 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.
#

import subprocess
import datetime
import sys


def print_e(*args, **kwargs):
print(*args, file=sys.stderr, **kwargs)


def getJetpadReleaseInfo(date):
try:
rawJetpadReleaseOutput = subprocess.check_output('span sql /span/global/androidx-jetpad:prod_instance \"SELECT GroupId, ArtifactId, ReleaseVersion, PreviousReleaseSHA, ReleaseSHA, Path, RequireSameVersionGroupBuild, ReleaseBuildId, ReleaseBranch FROM LibraryReleases WHERE ReleaseDate = %s\"' % date, shell=True)
except subprocess.CalledProcessError:
print_e("FAIL: Failed to get jetpad release info for %s. "
"This likely means you need to run gcert." % date)
return None
rawJetpadReleaseOutputLines = rawJetpadReleaseOutput.splitlines()
if len(rawJetpadReleaseOutputLines) <= 2:
print_e("Error: Date %s returned zero results from Jetpad. Please check your date" % args.date)
return None
jetpadReleaseOutput = iter(rawJetpadReleaseOutputLines)
return jetpadReleaseOutput

def getReleaseInfoObject(date, includeAllCommits, jetpadReleaseInfo):
releaseDateTime = datetime.datetime.fromtimestamp(float(date)/1000.0)
releaseJsonObject = {}
releaseJsonObject["releaseDate"] = "%02d-%02d-%s" % (releaseDateTime.month, releaseDateTime.day, releaseDateTime.year)
releaseJsonObject["includeAllCommits"] = includeAllCommits
releaseJsonObject["modules"] = {}
for line in jetpadReleaseInfo:
if "androidx" not in line.decode(): continue
# Remove all white space and split line based on '|'
artifactIdReleaseLine = line.decode().replace(" ", "").split('|')
groupId = artifactIdReleaseLine[1]
artifactId = artifactIdReleaseLine[2]
version = artifactIdReleaseLine[3]
fromSHA = artifactIdReleaseLine[4]
untilSHA = artifactIdReleaseLine[5]
path = artifactIdReleaseLine[6]
if path and path[0] == '/': path = path[1:]
requiresSameVersion = False
if artifactIdReleaseLine[7] == "true":
requiresSameVersion = True
buildId = artifactIdReleaseLine[8]
branch = artifactIdReleaseLine[9]
if groupId in releaseJsonObject["modules"]:
releaseJsonObject["modules"][groupId].append({
"groupId": groupId,
"artifactId": artifactId,
"version": version,
"fromSHA": fromSHA,
"untilSHA": untilSHA,
"requiresSameVersion": requiresSameVersion,
"path": path,
"buildId": buildId,
"branch": branch,
})
else:
releaseJsonObject["modules"][groupId] = [{
"groupId": groupId,
"artifactId": artifactId,
"version": version,
"fromSHA": fromSHA,
"untilSHA": untilSHA,
"requiresSameVersion": requiresSameVersion,
"path": path,
"buildId": buildId,
"branch": branch,
}]
return releaseJsonObject

def getJetpadRelease(date, includeAllCommits):
print("Getting the release info from Jetpad...")
jetpadReleaseInfo = getJetpadReleaseInfo(date)
if not jetpadReleaseInfo:
exit(1)
print("Successful")
return getReleaseInfoObject(date, includeAllCommits, jetpadReleaseInfo)