Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 1 | #!/usr/bin/python3 |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 2 | |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 3 | # This script is used to update platform SDK prebuilts, Support Library, and a variety of other |
| 4 | # prebuilt libraries used by Android's Makefile builds. For details on how to use this script, |
| 5 | # visit go/update-prebuilts. |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 6 | import os, sys, getopt, zipfile, re |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 7 | import argparse |
Jiyong Park | 1d1c963 | 2018-05-29 17:45:27 +0900 | [diff] [blame] | 8 | import glob |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 9 | import subprocess |
shepshapard | e4127cf | 2019-09-03 16:08:44 -0700 | [diff] [blame] | 10 | from shutil import copyfile, rmtree, which, move |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 11 | from distutils.version import LooseVersion |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 12 | from functools import reduce |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 13 | import six |
| 14 | import urllib.request, urllib.parse, urllib.error |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 15 | |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 16 | current_path = 'current' |
Andy Wickham | c3c99af | 2020-10-26 20:47:58 +0000 | [diff] [blame] | 17 | framework_sdk_target = 'sdk_phone_armv7-win_sdk' |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 18 | androidx_dir = os.path.join(current_path, 'androidx') |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 19 | gmaven_dir = os.path.join(current_path, 'gmaven') |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 20 | extras_dir = os.path.join(current_path, 'extras') |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 21 | buildtools_dir = 'tools' |
Jeff Gaston | 553a437 | 2018-03-29 18:34:22 -0400 | [diff] [blame] | 22 | jetifier_dir = os.path.join(buildtools_dir, 'jetifier', 'jetifier-standalone') |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 23 | |
Jeff Gaston | a68a8d4 | 2018-03-23 14:00:13 -0400 | [diff] [blame] | 24 | temp_dir = os.path.join(os.getcwd(), "support_tmp") |
| 25 | os.chdir(os.path.dirname(os.path.dirname(os.path.realpath(sys.argv[0])))) |
| 26 | git_dir = os.getcwd() |
| 27 | |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 28 | # See go/fetch_artifact for details on this script. |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 29 | FETCH_ARTIFACT = '/google/data/ro/projects/android/fetch_artifact' |
Jeff Sharkey | abe16df | 2021-05-25 10:09:03 -0600 | [diff] [blame] | 30 | FETCH_ARTIFACT_BEYOND_CORP = '/usr/bin/fetch_artifact' |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 31 | |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 32 | # See (https://developer.android.com/studio/build/dependencies#gmaven-access) |
| 33 | GMAVEN_BASE_URL = 'https://maven.google.com' |
| 34 | |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 35 | # Leave map blank to automatically populate name and path: |
| 36 | # - Name format is MAVEN.replaceAll(':','_') |
| 37 | # - Path format is MAVEN.replaceAll(':','/').replaceAll('.','/') |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 38 | maven_to_make = { |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 39 | # AndroidX |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 40 | 'androidx.benchmark:benchmark-macro': { }, |
| 41 | 'androidx.benchmark:benchmark-macro-junit4': { }, |
| 42 | 'androidx.benchmark:benchmark-common': { }, |
| 43 | 'androidx.benchmark:benchmark-junit4': { }, |
| 44 | 'androidx.tracing:tracing': { }, |
| 45 | 'androidx.tracing:tracing-ktx': { }, |
| 46 | 'androidx.slice:slice-builders': { }, |
| 47 | 'androidx.slice:slice-core': { }, |
| 48 | 'androidx.slice:slice-view': { }, |
| 49 | 'androidx.remotecallback:remotecallback': { }, |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 50 | 'androidx.remotecallback:remotecallback-processor': {'host':True}, |
| 51 | 'androidx.versionedparcelable:versionedparcelable': { }, |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 52 | 'androidx.vectordrawable:vectordrawable-animated': { }, |
| 53 | 'androidx.activity:activity': { }, |
| 54 | 'androidx.activity:activity-ktx': { }, |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 55 | 'androidx.annotation:annotation': {'host_and_device':True}, |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 56 | 'androidx.annotation:annotation-experimental': { }, |
| 57 | 'androidx.asynclayoutinflater:asynclayoutinflater': { }, |
| 58 | 'androidx.collection:collection': { }, |
| 59 | 'androidx.collection:collection-ktx': { }, |
| 60 | 'androidx.concurrent:concurrent-futures': { }, |
| 61 | 'androidx.concurrent:concurrent-listenablefuture-callback': { }, |
| 62 | 'androidx.concurrent:concurrent-listenablefuture': { }, |
| 63 | 'androidx.core:core': { }, |
| 64 | 'androidx.core:core-animation': { }, |
| 65 | 'androidx.core:core-ktx': { }, |
| 66 | 'androidx.contentpaging:contentpaging': { }, |
| 67 | 'androidx.coordinatorlayout:coordinatorlayout': { }, |
| 68 | 'androidx.legacy:legacy-support-core-ui': { }, |
| 69 | 'androidx.legacy:legacy-support-core-utils': { }, |
| 70 | 'androidx.cursoradapter:cursoradapter': { }, |
| 71 | 'androidx.browser:browser': { }, |
| 72 | 'androidx.customview:customview': { }, |
| 73 | 'androidx.documentfile:documentfile': { }, |
| 74 | 'androidx.drawerlayout:drawerlayout': { }, |
| 75 | 'androidx.dynamicanimation:dynamicanimation': { }, |
| 76 | 'androidx.emoji:emoji': { }, |
| 77 | 'androidx.emoji:emoji-appcompat': { }, |
| 78 | 'androidx.emoji:emoji-bundled': { }, |
| 79 | 'androidx.emoji2:emoji2': { }, |
| 80 | 'androidx.emoji2:emoji2-views-helper': { }, |
| 81 | 'androidx.exifinterface:exifinterface': { }, |
| 82 | 'androidx.fragment:fragment': { }, |
| 83 | 'androidx.fragment:fragment-ktx': { }, |
| 84 | 'androidx.heifwriter:heifwriter': { }, |
| 85 | 'androidx.interpolator:interpolator': { }, |
| 86 | 'androidx.loader:loader': { }, |
| 87 | 'androidx.localbroadcastmanager:localbroadcastmanager': { }, |
| 88 | 'androidx.media:media': { }, |
| 89 | 'androidx.media2:media2-player': { }, |
| 90 | 'androidx.media2:media2-session': { }, |
| 91 | 'androidx.media2:media2-common': { }, |
| 92 | 'androidx.media2:media2-exoplayer': { }, |
| 93 | 'androidx.media2:media2-widget': { }, |
| 94 | 'androidx.navigation:navigation-common': { }, |
| 95 | 'androidx.navigation:navigation-common-ktx': { }, |
| 96 | 'androidx.navigation:navigation-fragment': { }, |
| 97 | 'androidx.navigation:navigation-fragment-ktx': { }, |
| 98 | 'androidx.navigation:navigation-runtime': { }, |
| 99 | 'androidx.navigation:navigation-runtime-ktx': { }, |
| 100 | 'androidx.navigation:navigation-ui': { }, |
| 101 | 'androidx.navigation:navigation-ui-ktx': { }, |
| 102 | 'androidx.percentlayout:percentlayout': { }, |
| 103 | 'androidx.print:print': { }, |
| 104 | 'androidx.recommendation:recommendation': { }, |
| 105 | 'androidx.recyclerview:recyclerview-selection': { }, |
| 106 | 'androidx.savedstate:savedstate': { }, |
| 107 | 'androidx.savedstate:savedstate-ktx': { }, |
| 108 | 'androidx.slidingpanelayout:slidingpanelayout': { }, |
| 109 | 'androidx.swiperefreshlayout:swiperefreshlayout': { }, |
| 110 | 'androidx.textclassifier:textclassifier': { }, |
| 111 | 'androidx.transition:transition': { }, |
| 112 | 'androidx.tvprovider:tvprovider': { }, |
| 113 | 'androidx.legacy:legacy-support-v13': { }, |
| 114 | 'androidx.legacy:legacy-preference-v14': { }, |
| 115 | 'androidx.leanback:leanback': { }, |
| 116 | 'androidx.leanback:leanback-preference': { }, |
| 117 | 'androidx.legacy:legacy-support-v4': { }, |
| 118 | 'androidx.appcompat:appcompat': { }, |
| 119 | 'androidx.appcompat:appcompat-resources': { }, |
| 120 | 'androidx.cardview:cardview': { }, |
| 121 | 'androidx.gridlayout:gridlayout': { }, |
| 122 | 'androidx.mediarouter:mediarouter': { }, |
| 123 | 'androidx.palette:palette': { }, |
| 124 | 'androidx.preference:preference': { }, |
| 125 | 'androidx.recyclerview:recyclerview': { }, |
| 126 | 'androidx.vectordrawable:vectordrawable': { }, |
| 127 | 'androidx.viewpager:viewpager': { }, |
| 128 | 'androidx.viewpager2:viewpager2': { }, |
| 129 | 'androidx.wear:wear': { }, |
| 130 | 'androidx.wear:wear-ongoing': { }, |
| 131 | 'androidx.webkit:webkit': { }, |
| 132 | 'androidx.biometric:biometric': { }, |
| 133 | 'androidx.autofill:autofill': { }, |
| 134 | 'androidx.appsearch:appsearch': { }, |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 135 | 'androidx.appsearch:appsearch-local-storage': {'name':'androidx.appsearch_appsearch_local_storage'}, |
| 136 | 'androidx.appsearch:appsearch-platform-storage': { }, |
| 137 | 'androidx.appsearch:appsearch-compiler': {'name':'androidx.appsearch_appsearch-compiler', 'host':True}, |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 138 | 'androidx.car.app:app': { }, |
| 139 | 'androidx.car.app:app-automotive': { }, |
| 140 | 'androidx.car.app:app-testing': { }, |
| 141 | 'androidx.startup:startup-runtime': { }, |
Alan Viverette | cdfc989 | 2021-08-31 19:35:58 +0000 | [diff] [blame] | 142 | 'androidx.window:window': {'optional-uses-libs':{'androidx.window.extensions', 'androidx.window.sidecar'}}, |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 143 | 'androidx.resourceinspection:resourceinspection-annotation': { }, |
| 144 | 'androidx.profileinstaller:profileinstaller': { }, |
Aurimas Liutikas | 6e87bbf | 2021-07-30 10:04:05 -0700 | [diff] [blame] | 145 | |
Alan Viverette | f44e2f9 | 2021-08-31 17:26:26 +0000 | [diff] [blame] | 146 | # AndroidX for Compose |
Colin Cross | 58866d4 | 2021-09-17 14:40:35 -0700 | [diff] [blame] | 147 | 'androidx.compose.compiler:compiler-hosted': { 'host':True }, |
Alan Viverette | f44e2f9 | 2021-08-31 17:26:26 +0000 | [diff] [blame] | 148 | 'androidx.compose.runtime:runtime': { }, |
| 149 | 'androidx.compose.runtime:runtime-saveable': { }, |
| 150 | 'androidx.compose.foundation:foundation': { }, |
| 151 | 'androidx.compose.foundation:foundation-layout': { }, |
| 152 | 'androidx.compose.foundation:foundation-text': { }, |
| 153 | 'androidx.compose.ui:ui': { }, |
| 154 | 'androidx.compose.ui:ui-geometry': { }, |
| 155 | 'androidx.compose.ui:ui-graphics': { }, |
| 156 | 'androidx.compose.ui:ui-text': { }, |
| 157 | 'androidx.compose.ui:ui-unit': { }, |
| 158 | 'androidx.compose.ui:ui-util': { }, |
| 159 | 'androidx.compose.animation:animation-core': { }, |
| 160 | 'androidx.compose.animation:animation': { }, |
| 161 | 'androidx.compose.material:material-icons-core': { }, |
| 162 | 'androidx.compose.material:material-ripple': { }, |
| 163 | 'androidx.compose.material:material': { }, |
| 164 | 'androidx.activity:activity-compose': { }, |
| 165 | |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 166 | # AndroidX for Multidex |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 167 | 'androidx.multidex:multidex': { }, |
| 168 | 'androidx.multidex:multidex-instrumentation': { }, |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 169 | |
| 170 | # AndroidX for Constraint Layout |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 171 | 'androidx.constraintlayout:constraintlayout': {'name':'androidx-constraintlayout_constraintlayout'}, |
| 172 | 'androidx.constraintlayout:constraintlayout-solver': {'name':'androidx-constraintlayout_constraintlayout-solver'}, |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 173 | |
| 174 | # AndroidX for Architecture Components |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 175 | 'androidx.arch.core:core-common': { }, |
| 176 | 'androidx.arch.core:core-runtime': { }, |
| 177 | 'androidx.lifecycle:lifecycle-common': { }, |
| 178 | 'androidx.lifecycle:lifecycle-common-java8': { }, |
| 179 | 'androidx.lifecycle:lifecycle-extensions': { }, |
| 180 | 'androidx.lifecycle:lifecycle-livedata': { }, |
| 181 | 'androidx.lifecycle:lifecycle-livedata-ktx': { }, |
| 182 | 'androidx.lifecycle:lifecycle-livedata-core': { }, |
| 183 | 'androidx.lifecycle:lifecycle-livedata-core-ktx': { }, |
| 184 | 'androidx.lifecycle:lifecycle-process': { }, |
| 185 | 'androidx.lifecycle:lifecycle-runtime': { }, |
| 186 | 'androidx.lifecycle:lifecycle-runtime-ktx': { }, |
| 187 | 'androidx.lifecycle:lifecycle-service': { }, |
| 188 | 'androidx.lifecycle:lifecycle-viewmodel': { }, |
| 189 | 'androidx.lifecycle:lifecycle-viewmodel-ktx': { }, |
| 190 | 'androidx.lifecycle:lifecycle-viewmodel-savedstate': { }, |
| 191 | 'androidx.paging:paging-common': { }, |
| 192 | 'androidx.paging:paging-common-ktx': { }, |
| 193 | 'androidx.paging:paging-runtime': { }, |
| 194 | 'androidx.sqlite:sqlite': { }, |
| 195 | 'androidx.sqlite:sqlite-framework': { }, |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 196 | 'androidx.room:room-common': {'host_and_device':True}, |
Alan Viverette | cdfc989 | 2021-08-31 19:35:58 +0000 | [diff] [blame] | 197 | 'androidx.room:room-compiler': {'host':True, 'extra-static-libs':{'guava-21.0'}}, |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 198 | 'androidx.room:room-migration': {'host_and_device':True}, |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 199 | 'androidx.room:room-runtime': { }, |
| 200 | 'androidx.room:room-testing': { }, |
Alan Viverette | 0723aff | 2021-08-31 17:07:47 +0000 | [diff] [blame] | 201 | 'androidx.room:room-compiler-processing': {'host':True}, |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 202 | 'androidx.work:work-runtime': { }, |
| 203 | 'androidx.work:work-runtime-ktx': { }, |
| 204 | 'androidx.work:work-testing': { }, |
Allen | ab72f01 | 2018-01-29 18:10:23 -0800 | [diff] [blame] | 205 | |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 206 | # Third-party dependencies |
Jeff Gaston | a0bce60 | 2018-08-24 14:05:03 -0400 | [diff] [blame] | 207 | 'com.google.android:flexbox': {'name':'flexbox', 'path':'flexbox'}, |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 208 | |
Jeff Gaston | fa7c7e8 | 2018-04-06 13:35:48 -0400 | [diff] [blame] | 209 | # Androidx Material Design Components |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 210 | 'com.google.android.material:material': { }, |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 211 | } |
| 212 | |
Alan Viverette | cdfc989 | 2021-08-31 19:35:58 +0000 | [diff] [blame] | 213 | # Mapping of POM dependencies to Soong build targets |
| 214 | deps_rewrite = { |
| 215 | 'auto-common':'auto_common', |
| 216 | 'auto-value-annotations':'auto_value_annotations', |
| 217 | 'com.google.auto.value:auto-value':'auto_value_plugin', |
| 218 | 'monitor':'androidx.test.monitor', |
| 219 | 'rules':'androidx.test.rules', |
| 220 | 'runner':'androidx.test.runner', |
| 221 | 'androidx.test:core':'androidx.test.core', |
| 222 | 'com.squareup:javapoet':'javapoet', |
| 223 | 'com.google.guava:listenablefuture':'guava-listenablefuture-prebuilt-jar', |
| 224 | 'sqlite-jdbc':'xerial-sqlite-jdbc', |
| 225 | 'gson':'gson-prebuilt-jar', |
| 226 | 'com.intellij:annotations':'jetbrains-annotations', |
| 227 | 'javax.annotation:javax.annotation-api':'javax-annotation-api-prebuilt-host-jar', |
| 228 | 'org.robolectric:robolectric':'Robolectric_all-target', |
| 229 | 'org.jetbrains.kotlin:kotlin-stdlib-common':'kotlin-stdlib', |
| 230 | 'org.jetbrains.kotlinx:kotlinx-coroutines-core':'kotlinx_coroutines', |
| 231 | 'org.jetbrains.kotlinx:kotlinx-coroutines-android':'kotlinx_coroutines_android', |
| 232 | 'org.jetbrains.kotlinx:kotlinx-metadata-jvm':'kotlinx_metadata_jvm', |
| 233 | } |
| 234 | |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 235 | # List of artifacts that will be updated from GMaven |
| 236 | # Use pattern: `group:library:version:extension` |
| 237 | # e.g.: |
| 238 | # androidx.appcompat:appcompat:1.2.0:aar |
| 239 | # Use `latest` to always fetch the latest version. |
| 240 | # e.g.: |
| 241 | # androidx.appcompat:appcompat:latest:aar |
| 242 | # Also make sure you add `group:library`:{} to maven_to_make as well. |
| 243 | gmaven_artifacts = {} |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 244 | |
| 245 | def name_for_artifact(group_artifact): |
| 246 | return group_artifact.replace(':','_') |
| 247 | |
| 248 | |
| 249 | def path_for_artifact(group_artifact): |
| 250 | return group_artifact.replace('.','/').replace(':','/') |
| 251 | |
| 252 | |
| 253 | # Add automatic entries to maven_to_make. |
| 254 | for key in maven_to_make: |
| 255 | if ('name' not in maven_to_make[key]): |
| 256 | maven_to_make[key]['name'] = name_for_artifact(key) |
| 257 | if ('path' not in maven_to_make[key]): |
| 258 | maven_to_make[key]['path'] = path_for_artifact(key) |
| 259 | |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 260 | # Always remove these files. |
| 261 | blacklist_files = [ |
| 262 | 'annotations.zip', |
| 263 | 'public.txt', |
| 264 | 'R.txt', |
Alan Viverette | 44ad4e4 | 2017-08-09 17:45:00 -0400 | [diff] [blame] | 265 | 'AndroidManifest.xml', |
Alan Viverette | f48d9b4 | 2017-08-17 14:45:46 -0400 | [diff] [blame] | 266 | os.path.join('libs','noto-emoji-compat-java.jar') |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 267 | ] |
| 268 | |
Alan Viverette | 44ad4e4 | 2017-08-09 17:45:00 -0400 | [diff] [blame] | 269 | artifact_pattern = re.compile(r"^(.+?)-(\d+\.\d+\.\d+(?:-\w+\d+)?(?:-[\d.]+)*)\.(jar|aar)$") |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 270 | |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 271 | |
| 272 | class MavenLibraryInfo: |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 273 | def __init__(self, key, group_id, artifact_id, version, dir, repo_dir, file): |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 274 | self.key = key |
| 275 | self.group_id = group_id |
| 276 | self.artifact_id = artifact_id |
| 277 | self.version = version |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 278 | self.dir = dir |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 279 | self.repo_dir = repo_dir |
| 280 | self.file = file |
| 281 | |
| 282 | |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 283 | def print_e(*args, **kwargs): |
| 284 | print(*args, file=sys.stderr, **kwargs) |
| 285 | |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 286 | |
| 287 | def touch(fname, times=None): |
| 288 | with open(fname, 'a'): |
| 289 | os.utime(fname, times) |
| 290 | |
| 291 | |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 292 | def path(*path_parts): |
| 293 | return reduce((lambda x, y: os.path.join(x, y)), path_parts) |
| 294 | |
| 295 | |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 296 | def flatten(list): |
| 297 | return reduce((lambda x, y: "%s %s" % (x, y)), list) |
| 298 | |
| 299 | |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 300 | def rm(path): |
| 301 | if os.path.isdir(path): |
| 302 | rmtree(path) |
| 303 | elif os.path.exists(path): |
| 304 | os.remove(path) |
| 305 | |
| 306 | |
| 307 | def mv(src_path, dst_path): |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 308 | if os.path.exists(dst_path): |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 309 | rm(dst_path) |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 310 | if not os.path.exists(os.path.dirname(dst_path)): |
| 311 | os.makedirs(os.path.dirname(dst_path)) |
Jiyong Park | 1d1c963 | 2018-05-29 17:45:27 +0900 | [diff] [blame] | 312 | for f in (glob.glob(src_path)): |
| 313 | if '*' in dst_path: |
| 314 | dst = os.path.join(os.path.dirname(dst_path), os.path.basename(f)) |
| 315 | else: |
| 316 | dst = dst_path |
shepshapard | e4127cf | 2019-09-03 16:08:44 -0700 | [diff] [blame] | 317 | move(f, dst) |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 318 | |
| 319 | |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 320 | def detect_artifacts(maven_repo_dirs): |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 321 | maven_lib_info = {} |
| 322 | |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 323 | # Find the latest revision for each artifact, remove others |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 324 | for repo_dir in maven_repo_dirs: |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 325 | for root, dirs, files in os.walk(repo_dir): |
| 326 | for file in files: |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 327 | if file[-4:] == ".pom": |
| 328 | # Read the POM (hack hack hack). |
| 329 | group_id = '' |
| 330 | artifact_id = '' |
| 331 | version = '' |
| 332 | file = os.path.join(root, file) |
| 333 | with open(file) as pom_file: |
| 334 | for line in pom_file: |
| 335 | if line[:11] == ' <groupId>': |
| 336 | group_id = line[11:-11] |
| 337 | elif line[:14] == ' <artifactId>': |
| 338 | artifact_id = line[14:-14] |
| 339 | elif line[:11] == ' <version>': |
| 340 | version = line[11:-11] |
| 341 | if group_id == '' or artifact_id == '' or version == '': |
| 342 | print_e('Failed to find Maven artifact data in ' + file) |
| 343 | continue |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 344 | |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 345 | # Locate the artifact. |
| 346 | artifact_file = file[:-4] |
| 347 | if os.path.exists(artifact_file + '.jar'): |
| 348 | artifact_file = artifact_file + '.jar' |
| 349 | elif os.path.exists(artifact_file + '.aar'): |
| 350 | artifact_file = artifact_file + '.aar' |
| 351 | else: |
| 352 | print_e('Failed to find artifact for ' + artifact_file) |
| 353 | continue |
| 354 | |
| 355 | # Make relative to root. |
| 356 | artifact_file = artifact_file[len(root) + 1:] |
| 357 | |
| 358 | # Find the mapping. |
| 359 | group_artifact = group_id + ':' + artifact_id |
Jeff Gaston | 3e622ba | 2018-03-26 23:24:46 -0400 | [diff] [blame] | 360 | if group_artifact in maven_to_make: |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 361 | key = group_artifact |
Jeff Gaston | 3e622ba | 2018-03-26 23:24:46 -0400 | [diff] [blame] | 362 | elif artifact_id in maven_to_make: |
| 363 | key = artifact_id |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 364 | else: |
Alan Viverette | cdfc989 | 2021-08-31 19:35:58 +0000 | [diff] [blame] | 365 | # No mapping entry, skip this library. |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 366 | continue |
| 367 | |
| 368 | # Store the latest version. |
| 369 | version = LooseVersion(version) |
| 370 | if key not in maven_lib_info \ |
| 371 | or version > maven_lib_info[key].version: |
| 372 | maven_lib_info[key] = MavenLibraryInfo(key, group_id, artifact_id, version, |
| 373 | root, repo_dir, artifact_file) |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 374 | |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 375 | return maven_lib_info |
| 376 | |
| 377 | |
Colin Cross | 59b59db | 2018-04-24 13:58:05 -0700 | [diff] [blame] | 378 | def transform_maven_repos(maven_repo_dirs, transformed_dir, extract_res=True, include_static_deps=True): |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 379 | cwd = os.getcwd() |
| 380 | |
| 381 | # Use a temporary working directory. |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 382 | maven_lib_info = detect_artifacts(maven_repo_dirs) |
Jeff Gaston | a68a8d4 | 2018-03-23 14:00:13 -0400 | [diff] [blame] | 383 | working_dir = temp_dir |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 384 | |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 385 | if not maven_lib_info: |
| 386 | print_e('Failed to detect artifacts') |
| 387 | return False |
| 388 | |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 389 | # extract some files (for example, AndroidManifest.xml) from any relevant artifacts |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 390 | for info in maven_lib_info.values(): |
Alan Viverette | d8ce722 | 2017-12-11 17:24:43 -0500 | [diff] [blame] | 391 | transform_maven_lib(working_dir, info, extract_res) |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 392 | |
Colin Cross | 74683bc | 2018-04-11 17:36:18 -0700 | [diff] [blame] | 393 | # generate a single Android.bp that specifies to use all of the above artifacts |
| 394 | makefile = os.path.join(working_dir, 'Android.bp') |
Alan Viverette | 4ec9a17 | 2018-02-20 16:19:31 -0500 | [diff] [blame] | 395 | with open(makefile, 'w') as f: |
Anton Hansson | bd2021c | 2021-04-06 19:21:34 +0000 | [diff] [blame] | 396 | args = ["pom2bp"] |
Aurimas Liutikas | 6e87bbf | 2021-07-30 10:04:05 -0700 | [diff] [blame] | 397 | args.extend(["-sdk-version", "31"]) |
Anton Hansson | bd2021c | 2021-04-06 19:21:34 +0000 | [diff] [blame] | 398 | args.extend(["-default-min-sdk-version", "24"]) |
Jeff Gaston | 1cc0609 | 2018-03-28 15:51:02 -0400 | [diff] [blame] | 399 | if include_static_deps: |
| 400 | args.append("-static-deps") |
Jeff Gaston | 54868ee | 2018-04-25 19:43:35 -0400 | [diff] [blame] | 401 | rewriteNames = sorted([name for name in maven_to_make if ":" in name] + [name for name in maven_to_make if ":" not in name]) |
Jeff Gaston | a0bce60 | 2018-08-24 14:05:03 -0400 | [diff] [blame] | 402 | args.extend(["-rewrite=^" + name + "$=" + maven_to_make[name]['name'] for name in rewriteNames]) |
Alan Viverette | cdfc989 | 2021-08-31 19:35:58 +0000 | [diff] [blame] | 403 | args.extend(["-rewrite=^" + key + "$=" + value for key, value in deps_rewrite.items()]) |
| 404 | args.extend(["-extra-static-libs=" + maven_to_make[name]['name'] + "=" + ",".join(sorted(maven_to_make[name]['extra-static-libs'])) for name in maven_to_make if 'extra-static-libs' in maven_to_make[name]]) |
| 405 | args.extend(["-optional-uses-libs=" + maven_to_make[name]['name'] + "=" + ",".join(sorted(maven_to_make[name]['optional-uses-libs'])) for name in maven_to_make if 'optional-uses-libs' in maven_to_make[name]]) |
Jeff Gaston | 07a3cf7 | 2018-08-24 14:16:34 -0400 | [diff] [blame] | 406 | args.extend(["-host=" + name for name in maven_to_make if maven_to_make[name].get('host')]) |
Tony Mak | 0f65875 | 2019-07-19 11:11:05 +0100 | [diff] [blame] | 407 | args.extend(["-host-and-device=" + name for name in maven_to_make if maven_to_make[name].get('host_and_device')]) |
Jeff Gaston | 7ec1d8f | 2018-03-27 00:25:12 -0400 | [diff] [blame] | 408 | args.extend(["."]) |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 409 | subprocess.check_call(args, stdout=f, cwd=working_dir) |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 410 | |
| 411 | # Replace the old directory. |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 412 | output_dir = os.path.join(cwd, transformed_dir) |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 413 | mv(working_dir, output_dir) |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 414 | return True |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 415 | |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 416 | # moves <artifact_info> (of type MavenLibraryInfo) into the appropriate part of <working_dir> , and possibly unpacks any necessary included files |
Alan Viverette | d8ce722 | 2017-12-11 17:24:43 -0500 | [diff] [blame] | 417 | def transform_maven_lib(working_dir, artifact_info, extract_res): |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 418 | # Move library into working dir |
Jeff Gaston | 9edf86c | 2018-04-18 13:45:23 -0400 | [diff] [blame] | 419 | new_dir = os.path.normpath(os.path.join(working_dir, os.path.relpath(artifact_info.dir, artifact_info.repo_dir))) |
Jeff Gaston | c302bb9 | 2018-03-23 13:53:48 -0400 | [diff] [blame] | 420 | mv(artifact_info.dir, new_dir) |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 421 | |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 422 | matcher = artifact_pattern.match(artifact_info.file) |
| 423 | maven_lib_name = artifact_info.key |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 424 | maven_lib_vers = matcher.group(2) |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 425 | maven_lib_type = artifact_info.file[-3:] |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 426 | |
Alan Viverette | f3c1272 | 2021-08-30 19:50:33 +0000 | [diff] [blame] | 427 | group_artifact = artifact_info.key |
| 428 | make_lib_name = maven_to_make[group_artifact]['name'] |
| 429 | make_dir_name = maven_to_make[group_artifact]['path'] |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 430 | |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 431 | artifact_file = os.path.join(new_dir, artifact_info.file) |
| 432 | |
Colin Cross | 59b59db | 2018-04-24 13:58:05 -0700 | [diff] [blame] | 433 | if maven_lib_type == "aar": |
| 434 | if extract_res: |
| 435 | target_dir = os.path.join(working_dir, make_dir_name) |
| 436 | if not os.path.exists(target_dir): |
| 437 | os.makedirs(target_dir) |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 438 | |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 439 | process_aar(artifact_file, target_dir) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 440 | |
Colin Cross | 59b59db | 2018-04-24 13:58:05 -0700 | [diff] [blame] | 441 | with zipfile.ZipFile(artifact_file) as zip: |
| 442 | manifests_dir = os.path.join(working_dir, "manifests") |
| 443 | zip.extract("AndroidManifest.xml", os.path.join(manifests_dir, make_lib_name)) |
Dan Willemsen | d677b60 | 2017-11-08 22:13:17 -0800 | [diff] [blame] | 444 | |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 445 | print(maven_lib_vers, ":", maven_lib_name, "->", make_lib_name) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 446 | |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 447 | |
Alan Viverette | c960cfb | 2017-12-04 13:09:22 -0500 | [diff] [blame] | 448 | def process_aar(artifact_file, target_dir): |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 449 | # Extract AAR file to target_dir. |
| 450 | with zipfile.ZipFile(artifact_file) as zip: |
| 451 | zip.extractall(target_dir) |
| 452 | |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 453 | # Remove classes.jar |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 454 | classes_jar = os.path.join(target_dir, "classes.jar") |
| 455 | if os.path.exists(classes_jar): |
Dan Willemsen | 814152e | 2017-11-06 13:22:11 -0800 | [diff] [blame] | 456 | os.remove(classes_jar) |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 457 | |
| 458 | # Remove or preserve empty dirs. |
| 459 | for root, dirs, files in os.walk(target_dir): |
| 460 | for dir in dirs: |
| 461 | dir_path = os.path.join(root, dir) |
| 462 | if not os.listdir(dir_path): |
| 463 | os.rmdir(dir_path) |
| 464 | |
| 465 | # Remove top-level cruft. |
| 466 | for file in blacklist_files: |
| 467 | file_path = os.path.join(target_dir, file) |
| 468 | if os.path.exists(file_path): |
| 469 | os.remove(file_path) |
| 470 | |
Alan Viverette | f17c940 | 2017-07-19 12:57:40 -0400 | [diff] [blame] | 471 | |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 472 | class GMavenArtifact(object): |
| 473 | # A map from group:library to the latest available version |
| 474 | key_versions_map = {} |
| 475 | def __init__(self, artifact_glob): |
| 476 | try: |
| 477 | (group, library, version, ext) = artifact_glob.split(':') |
| 478 | except ValueError: |
| 479 | raise ValueError(f'Error in {artifact_glob} expected: group:library:version:ext') |
| 480 | |
| 481 | if not group or not library or not version or not ext: |
| 482 | raise ValueError(f'Error in {artifact_glob} expected: group:library:version:ext') |
| 483 | |
| 484 | self.group = group |
| 485 | self.group_path = group.replace('.', '/') |
| 486 | self.library = library |
| 487 | self.key = f'{group}:{library}' |
| 488 | self.version = version |
| 489 | self.ext = ext |
| 490 | |
| 491 | def get_pom_file_url(self): |
| 492 | return f'{GMAVEN_BASE_URL}/{self.group_path}/{self.library}/{self.version}/{self.library}-{self.version}.pom' |
| 493 | |
| 494 | def get_artifact_url(self): |
| 495 | return f'{GMAVEN_BASE_URL}/{self.group_path}/{self.library}/{self.version}/{self.library}-{self.version}.{self.ext}' |
| 496 | |
| 497 | def get_latest_version(self): |
| 498 | latest_version = GMavenArtifact.key_versions_map[self.key] \ |
| 499 | if self.key in GMavenArtifact.key_versions_map else None |
| 500 | |
| 501 | if not latest_version: |
| 502 | print(f'Fetching latest version for {self.key}') |
| 503 | group_index_url = f'{GMAVEN_BASE_URL}/{self.group_path}/group-index.xml' |
| 504 | import xml.etree.ElementTree as ET |
| 505 | tree = ET.parse(urllib.request.urlopen(group_index_url)) |
| 506 | root = tree.getroot() |
| 507 | libraries = root.findall('./*[@versions]') |
| 508 | for library in libraries: |
| 509 | key = f'{root.tag}:{library.tag}' |
| 510 | GMavenArtifact.key_versions_map[key] = library.get('versions').split(',')[-1] |
| 511 | latest_version = GMavenArtifact.key_versions_map[self.key] |
| 512 | return latest_version |
| 513 | |
| 514 | |
| 515 | def fetch_gmaven_artifact(artifact): |
| 516 | """Fetch a GMaven artifact. |
| 517 | |
| 518 | Downloads a GMaven artifact |
| 519 | (https://developer.android.com/studio/build/dependencies#gmaven-access) |
| 520 | |
| 521 | Args: |
| 522 | artifact_glob: an instance of GMavenArtifact. |
| 523 | """ |
| 524 | download_to = os.path.join('gmaven', artifact.group, artifact.library, artifact.version) |
| 525 | |
| 526 | _DownloadFileToDisk(artifact.get_pom_file_url(), os.path.join(download_to, f'{artifact.library}-{artifact.version}.pom')) |
| 527 | _DownloadFileToDisk(artifact.get_artifact_url(), os.path.join(download_to, f'{artifact.library}-{artifact.version}.{artifact.ext}')) |
| 528 | |
| 529 | return download_to |
| 530 | |
| 531 | |
| 532 | def _DownloadFileToDisk(url, filepath): |
| 533 | """Download the file at URL to the location dictated by the path. |
| 534 | |
| 535 | Args: |
| 536 | url: Remote URL to download file from. |
| 537 | filepath: Filesystem path to write the file to. |
| 538 | """ |
| 539 | print(f'Downloading URL: {url}') |
| 540 | file_data = urllib.request.urlopen(url) |
| 541 | |
| 542 | try: |
| 543 | os.makedirs(os.path.dirname(filepath)) |
| 544 | except os.error: |
| 545 | # This is a common situation - os.makedirs fails if dir already exists. |
| 546 | pass |
| 547 | try: |
| 548 | with open(filepath, 'wb') as f: |
| 549 | f.write(six.ensure_binary(file_data.read())) |
| 550 | except: |
| 551 | os.remove(os.path.dirname(filepath)) |
| 552 | raise |
| 553 | |
| 554 | |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 555 | def fetch_artifact(target, build_id, artifact_path): |
Jeff Sharkey | abe16df | 2021-05-25 10:09:03 -0600 | [diff] [blame] | 556 | global args |
Jiyong Park | 1d1c963 | 2018-05-29 17:45:27 +0900 | [diff] [blame] | 557 | download_to = os.path.join('.', os.path.dirname(artifact_path)) |
| 558 | print('Fetching %s from %s ...' % (artifact_path, target)) |
| 559 | if not os.path.exists(download_to): |
| 560 | os.makedirs(download_to) |
Jeff Sharkey | abe16df | 2021-05-25 10:09:03 -0600 | [diff] [blame] | 561 | if args.beyond_corp: |
| 562 | fetch_cmd = [FETCH_ARTIFACT_BEYOND_CORP, '--use_oauth2', |
| 563 | '--bid', str(build_id), '--target', target, artifact_path, download_to] |
| 564 | else: |
| 565 | fetch_cmd = [FETCH_ARTIFACT, |
| 566 | '--bid', str(build_id), '--target', target, artifact_path, download_to] |
Jeff Gaston | 97a9b76 | 2018-10-02 15:15:12 -0400 | [diff] [blame] | 567 | print("Running: " + ' '.join(fetch_cmd)) |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 568 | try: |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 569 | subprocess.check_output(fetch_cmd, stderr=subprocess.STDOUT) |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 570 | except subprocess.CalledProcessError: |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 571 | print_e('FAIL: Unable to retrieve %s artifact for build ID %s' % (artifact_path, build_id)) |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 572 | print_e('Please make sure you are authenticated for build server access!') |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 573 | return None |
| 574 | return artifact_path |
| 575 | |
| 576 | |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 577 | def fetch_artifacts(target, build_id, artifact_dict): |
| 578 | for artifact, target_path in artifact_dict.items(): |
| 579 | artifact_path = fetch_artifact(target, build_id.url_id, artifact) |
| 580 | if not artifact_path: |
| 581 | return False |
| 582 | mv(artifact_path, target_path) |
| 583 | return True |
| 584 | |
| 585 | |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 586 | def extract_artifact(artifact_path): |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 587 | # Unzip the repo archive into a separate directory. |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 588 | repo_dir = os.path.basename(artifact_path)[:-4] |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 589 | with zipfile.ZipFile(artifact_path) as zipFile: |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 590 | zipFile.extractall(repo_dir) |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 591 | return repo_dir |
| 592 | |
| 593 | |
Alan Viverette | 7e897e2 | 2018-03-09 15:24:10 -0500 | [diff] [blame] | 594 | def fetch_and_extract(target, build_id, file, artifact_path=None): |
| 595 | if not artifact_path: |
| 596 | artifact_path = fetch_artifact(target, build_id, file) |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 597 | if not artifact_path: |
| 598 | return None |
| 599 | return extract_artifact(artifact_path) |
| 600 | |
| 601 | |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 602 | def update_gmaven(gmaven_artifacts): |
| 603 | artifacts = [GMavenArtifact(artifact) for artifact in gmaven_artifacts] |
| 604 | for artifact in artifacts: |
| 605 | if artifact.version == 'latest': |
| 606 | artifact.version = artifact.get_latest_version() |
| 607 | |
| 608 | artifact_dirs = [fetch_gmaven_artifact(artifact) for artifact in artifacts] |
| 609 | if not transform_maven_repos(['gmaven'], gmaven_dir, extract_res=False): |
| 610 | return [] |
| 611 | return [artifact.key for artifact in artifacts] |
| 612 | |
| 613 | |
Jeff Gaston | 08a2ee4 | 2018-05-07 18:08:41 -0400 | [diff] [blame] | 614 | def update_androidx(target, build_id, local_file): |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 615 | if build_id: |
Anthony Chen | acbb487 | 2018-07-02 11:22:48 -0700 | [diff] [blame] | 616 | repo_file = 'top-of-tree-m2repository-all-%s.zip' % build_id.fs_id |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 617 | repo_dir = fetch_and_extract(target, build_id.url_id, repo_file, None) |
| 618 | else: |
| 619 | repo_dir = fetch_and_extract(target, None, None, local_file) |
| 620 | if not repo_dir: |
| 621 | print_e('Failed to extract AndroidX repository') |
| 622 | return False |
| 623 | |
Tony Mak | 0f65875 | 2019-07-19 11:11:05 +0100 | [diff] [blame] | 624 | # Keep JavaPlugins.bp file untounched. |
| 625 | java_plugins_bp_path = os.path.join(androidx_dir, 'JavaPlugins.bp') |
| 626 | tmp_java_plugins_bp_path = os.path.join('/tmp', 'JavaPlugins.bp') |
| 627 | mv(java_plugins_bp_path, tmp_java_plugins_bp_path) |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 628 | |
Tony Mak | 0f65875 | 2019-07-19 11:11:05 +0100 | [diff] [blame] | 629 | # Transform the repo archive into a Makefile-compatible format. |
| 630 | if not transform_maven_repos([repo_dir], androidx_dir, extract_res=False): |
| 631 | return False |
| 632 | |
| 633 | # Import JavaPlugins.bp in Android.bp. |
| 634 | makefile = os.path.join(androidx_dir, 'Android.bp') |
| 635 | with open(makefile, "a+") as f: |
| 636 | f.write('\nbuild = ["JavaPlugins.bp"]\n') |
| 637 | mv(tmp_java_plugins_bp_path, java_plugins_bp_path) |
| 638 | |
| 639 | return True |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 640 | |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 641 | |
Jeff Gaston | 782c3e3 | 2018-02-06 14:36:17 -0500 | [diff] [blame] | 642 | def update_jetifier(target, build_id): |
| 643 | repo_file = 'jetifier-standalone.zip' |
| 644 | repo_dir = fetch_and_extract(target, build_id.url_id, repo_file) |
| 645 | if not repo_dir: |
| 646 | print_e('Failed to extract Jetifier') |
| 647 | return False |
| 648 | |
| 649 | rm(jetifier_dir) |
Jeff Gaston | 553a437 | 2018-03-29 18:34:22 -0400 | [diff] [blame] | 650 | mv(os.path.join(repo_dir, 'jetifier-standalone'), jetifier_dir) |
| 651 | os.chmod(os.path.join(jetifier_dir, 'bin', 'jetifier-standalone'), 0o755) |
Jeff Gaston | 782c3e3 | 2018-02-06 14:36:17 -0500 | [diff] [blame] | 652 | return True |
| 653 | |
Alan Viverette | 7e897e2 | 2018-03-09 15:24:10 -0500 | [diff] [blame] | 654 | |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 655 | def update_constraint(local_file): |
Jeff Gaston | 9edf86c | 2018-04-18 13:45:23 -0400 | [diff] [blame] | 656 | repo_dir = extract_artifact(local_file) |
| 657 | if not repo_dir: |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 658 | print_e('Failed to extract Constraint Layout') |
Jeff Gaston | 9edf86c | 2018-04-18 13:45:23 -0400 | [diff] [blame] | 659 | return False |
| 660 | return transform_maven_repos([repo_dir], os.path.join(extras_dir, 'constraint-layout-x'), extract_res=False) |
| 661 | |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 662 | |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 663 | def update_material(file): |
| 664 | design_dir = extract_artifact(file) |
| 665 | if not design_dir: |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 666 | print_e('Failed to extract Material Design Components') |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 667 | return False |
| 668 | |
| 669 | # Don't bother extracting resources -- this should only be used with AAPT2. |
| 670 | return transform_maven_repos([design_dir], |
| 671 | os.path.join(extras_dir, 'material-design-x'), extract_res=False) |
| 672 | |
| 673 | |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 674 | def update_framework(build_id, sdk_dir): |
Anton Hansson | 83f9217 | 2020-03-26 11:12:03 +0000 | [diff] [blame] | 675 | api_scope_list = ['public', 'system', 'test', 'module-lib', 'system-server'] |
Sundong Ahn | 66091f2 | 2018-08-29 18:54:28 +0900 | [diff] [blame] | 676 | if sdk_dir == 'current': |
| 677 | api_scope_list.append('core') |
| 678 | |
| 679 | for api_scope in api_scope_list: |
Jiyong Park | 1d1c963 | 2018-05-29 17:45:27 +0900 | [diff] [blame] | 680 | target_dir = path(sdk_dir, api_scope) |
Sundong Ahn | 66091f2 | 2018-08-29 18:54:28 +0900 | [diff] [blame] | 681 | if api_scope == 'core': |
| 682 | artifact_to_path = {'core.current.stubs.jar': path(target_dir, 'android.jar')} |
| 683 | else: |
| 684 | artifact_to_path = {'apistubs/android/' + api_scope + '/*.jar': path(target_dir, '*')} |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 685 | |
| 686 | if not fetch_artifacts(framework_sdk_target, build_id, artifact_to_path): |
| 687 | return False |
| 688 | |
Jiyong Park | 1d1c963 | 2018-05-29 17:45:27 +0900 | [diff] [blame] | 689 | if api_scope == 'public': |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 690 | # Fetch a few artifacts from the public sdk. |
Sundong Ahn | 66091f2 | 2018-08-29 18:54:28 +0900 | [diff] [blame] | 691 | artifact = 'sdk-repo-linux-platforms-%s.zip' % build_id.fs_id |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 692 | artifact_path = fetch_artifact(framework_sdk_target, build_id.url_id, artifact) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 693 | if not artifact_path: |
| 694 | return False |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 695 | |
| 696 | with zipfile.ZipFile(artifact_path) as zipFile: |
Colin Cross | d338d70 | 2020-07-10 19:17:54 -0700 | [diff] [blame] | 697 | extra_files = [ |
| 698 | 'android.jar', |
| 699 | 'framework.aidl', |
| 700 | 'uiautomator.jar', |
| 701 | 'core-for-system-modules.jar', |
| 702 | 'data/annotations.zip', |
| 703 | 'data/api-versions.xml'] |
| 704 | for filename in extra_files: |
Anton Hansson | a26e481 | 2020-03-25 12:54:39 +0000 | [diff] [blame] | 705 | matches = list(filter(lambda path: filename in path, zipFile.namelist())) |
| 706 | if len(matches) != 1: |
| 707 | print_e('Expected 1 file named \'%s\' in zip %s, found %d' % |
| 708 | (filename, zipFile.filename, len(matches))) |
| 709 | return False |
| 710 | zip_path = matches[0] |
| 711 | src_path = zipFile.extract(zip_path) |
| 712 | dst_path = path(target_dir, filename) |
| 713 | mv(src_path, dst_path) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 714 | |
Anton Hansson | 73be151 | 2021-06-24 10:56:05 +0100 | [diff] [blame] | 715 | # Filtered API DB is currently only available for "public" |
| 716 | fetch_artifacts(framework_sdk_target, build_id, {'api-versions-public-filtered.xml': path(target_dir, 'data/api-versions-filtered.xml')}) |
| 717 | |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 718 | return True |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 719 | |
Sundong Ahn | cc34cc3 | 2018-07-11 15:18:47 +0900 | [diff] [blame] | 720 | def update_makefile(build_id): |
| 721 | template = '"%s",\n\ |
| 722 | "current"' |
| 723 | makefile = os.path.join(git_dir, 'Android.bp') |
| 724 | |
| 725 | with open(makefile, 'r+') as f: |
| 726 | contents = f.read().replace('"current"', template % build_id) |
| 727 | f.seek(0) |
| 728 | f.write(contents) |
| 729 | |
| 730 | return True |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 731 | |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 732 | def finalize_sdk(build_id, sdk_version): |
| 733 | target_finalize_dir = '%d' % sdk_version |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 734 | |
Anton Hansson | aa554c0 | 2020-04-30 14:06:20 +0100 | [diff] [blame] | 735 | for api_scope in ['public', 'system', 'test', 'module-lib', 'system-server']: |
Sundong Ahn | 66091f2 | 2018-08-29 18:54:28 +0900 | [diff] [blame] | 736 | artifact_to_path = {'apistubs/android/' + api_scope + '/api/*.txt': |
| 737 | path(target_finalize_dir, api_scope, 'api', '*')} |
| 738 | |
| 739 | if not fetch_artifacts(framework_sdk_target, build_id, artifact_to_path): |
| 740 | return False |
| 741 | |
| 742 | return update_framework(build_id, target_finalize_dir) \ |
Sundong Ahn | cc34cc3 | 2018-07-11 15:18:47 +0900 | [diff] [blame] | 743 | and update_makefile(target_finalize_dir) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 744 | |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 745 | |
Anton Hansson | 1d01a03 | 2018-04-09 10:29:37 +0100 | [diff] [blame] | 746 | def update_framework_current(build_id): |
| 747 | return update_framework(build_id, current_path) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 748 | |
| 749 | |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 750 | def update_buildtools(target, arch, build_id): |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 751 | artifact_path = fetch_and_extract(target, build_id.url_id, |
| 752 | "sdk-repo-%s-build-tools-%s.zip" % (arch, build_id.fs_id)) |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 753 | if not artifact_path: |
| 754 | return False |
| 755 | |
| 756 | top_level_dir = os.listdir(artifact_path)[0] |
| 757 | src_path = os.path.join(artifact_path, top_level_dir) |
| 758 | dst_path = path(buildtools_dir, arch) |
Colin Cross | 20648d3 | 2021-02-12 13:30:51 -0800 | [diff] [blame] | 759 | |
| 760 | # There are a few libraries that have been manually added to the |
| 761 | # build tools, copy them from the destination back to the source |
| 762 | # before the destination is overwritten. |
| 763 | files_to_save = ( |
| 764 | 'lib64/libconscrypt_openjdk_jni.dylib', |
| 765 | 'lib64/libconscrypt_openjdk_jni.so', |
| 766 | 'bin/lib64/libwinpthread-1.dll', |
| 767 | ) |
| 768 | for file in files_to_save: |
| 769 | src_file = os.path.join(dst_path, file) |
| 770 | dst_file = os.path.join(src_path, file) |
| 771 | if os.path.exists(dst_path): |
| 772 | mv(src_file, dst_file) |
| 773 | |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 774 | mv(src_path, dst_path) |
| 775 | |
| 776 | # Move all top-level files to /bin and make them executable |
| 777 | bin_path = path(dst_path, 'bin') |
| 778 | top_level_files = filter(lambda e: os.path.isfile(path(dst_path, e)), os.listdir(dst_path)) |
| 779 | for file in top_level_files: |
| 780 | src_file = path(dst_path, file) |
| 781 | dst_file = path(bin_path, file) |
| 782 | mv(src_file, dst_file) |
| 783 | os.chmod(dst_file, 0o755) |
| 784 | |
Colin Cross | 6673ceb | 2021-02-12 13:14:14 -0800 | [diff] [blame] | 785 | # Make the files under lld-bin executable |
| 786 | lld_bin_files = os.listdir(os.path.join(dst_path, 'lld-bin')) |
| 787 | for file in lld_bin_files: |
| 788 | os.chmod(os.path.join(dst_path, 'lld-bin', file), 0o755) |
| 789 | |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 790 | # Remove renderscript |
| 791 | rm(path(dst_path, 'renderscript')) |
| 792 | |
| 793 | return True |
| 794 | |
| 795 | |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 796 | def append(text, more_text): |
| 797 | if text: |
| 798 | return "%s, %s" % (text, more_text) |
| 799 | return more_text |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 800 | |
Alan Viverette | 95f6d36 | 2017-04-06 09:40:50 -0400 | [diff] [blame] | 801 | |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 802 | class buildId(object): |
| 803 | def __init__(self, url_id, fs_id): |
| 804 | # id when used in build server urls |
| 805 | self.url_id = url_id |
| 806 | # id when used in build commands |
| 807 | self.fs_id = fs_id |
| 808 | |
Jeff Gaston | b9b0905 | 2018-02-05 14:16:05 -0500 | [diff] [blame] | 809 | def getBuildId(args): |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 810 | # must be in the format 12345 or P12345 |
Jeff Gaston | b9b0905 | 2018-02-05 14:16:05 -0500 | [diff] [blame] | 811 | source = args.source |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 812 | number_text = source[:] |
| 813 | presubmit = False |
| 814 | if number_text.startswith("P"): |
| 815 | presubmit = True |
| 816 | number_text = number_text[1:] |
| 817 | if not number_text.isnumeric(): |
Alan Viverette | 7e897e2 | 2018-03-09 15:24:10 -0500 | [diff] [blame] | 818 | return None |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 819 | url_id = source |
| 820 | fs_id = url_id |
| 821 | if presubmit: |
| 822 | fs_id = "0" |
| 823 | args.file = False |
| 824 | return buildId(url_id, fs_id) |
Jeff Gaston | b9b0905 | 2018-02-05 14:16:05 -0500 | [diff] [blame] | 825 | |
Colin Cross | 573e121 | 2018-02-12 14:59:52 -0800 | [diff] [blame] | 826 | def getFile(args): |
| 827 | source = args.source |
| 828 | if not source.isnumeric(): |
| 829 | return args.source |
Alan Viverette | 7e897e2 | 2018-03-09 15:24:10 -0500 | [diff] [blame] | 830 | return None |
Colin Cross | 573e121 | 2018-02-12 14:59:52 -0800 | [diff] [blame] | 831 | |
Alan Viverette | 4ec9a17 | 2018-02-20 16:19:31 -0500 | [diff] [blame] | 832 | |
| 833 | def script_relative(rel_path): |
Jeff Gaston | a68a8d4 | 2018-03-23 14:00:13 -0400 | [diff] [blame] | 834 | return os.path.join(script_dir, rel_path) |
Alan Viverette | 4ec9a17 | 2018-02-20 16:19:31 -0500 | [diff] [blame] | 835 | |
| 836 | |
Jeff Gaston | cc296a8 | 2018-03-23 14:33:24 -0400 | [diff] [blame] | 837 | def uncommittedChangesExist(): |
| 838 | try: |
| 839 | # Make sure we don't overwrite any pending changes. |
| 840 | diffCommand = "cd " + git_dir + " && git diff --quiet" |
| 841 | subprocess.check_call(diffCommand, shell=True) |
| 842 | subprocess.check_call(diffCommand + " --cached", shell=True) |
| 843 | return False |
| 844 | except subprocess.CalledProcessError: |
| 845 | return True |
| 846 | |
| 847 | |
Jeff Gaston | a7fba9b | 2018-03-26 23:48:51 -0400 | [diff] [blame] | 848 | rm(temp_dir) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 849 | parser = argparse.ArgumentParser( |
Alan Viverette | 4583709 | 2017-05-12 14:50:53 -0400 | [diff] [blame] | 850 | description=('Update current prebuilts')) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 851 | parser.add_argument( |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 852 | 'source', nargs='?', |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 853 | help='Build server build ID or local Maven ZIP file') |
| 854 | parser.add_argument( |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 855 | '-m', '--material', action="store_true", |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 856 | help='If specified, updates only Material Design Components') |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 857 | parser.add_argument( |
Alan Viverette | 3e57a4a | 2017-08-11 15:49:47 -0400 | [diff] [blame] | 858 | '-c', '--constraint', action="store_true", |
| 859 | help='If specified, updates only Constraint Layout') |
| 860 | parser.add_argument( |
Jeff Gaston | 782c3e3 | 2018-02-06 14:36:17 -0500 | [diff] [blame] | 861 | '-j', '--jetifier', action="store_true", |
| 862 | help='If specified, updates only Jetifier') |
| 863 | parser.add_argument( |
Alan Viverette | 5bfe05b | 2017-06-06 14:21:29 -0400 | [diff] [blame] | 864 | '-p', '--platform', action="store_true", |
| 865 | help='If specified, updates only the Android Platform') |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 866 | parser.add_argument( |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 867 | '-f', '--finalize_sdk', type=int, |
| 868 | help='If specified, imports the source build as the specified finalized SDK version') |
| 869 | parser.add_argument( |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 870 | '-b', '--buildtools', action="store_true", |
| 871 | help='If specified, updates only the Build Tools') |
Jeff Gaston | cc296a8 | 2018-03-23 14:33:24 -0400 | [diff] [blame] | 872 | parser.add_argument( |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 873 | '-x', '--androidx', action="store_true", |
| 874 | help='If specified, updates only the Jetpack (androidx) libraries excluding those covered by other arguments') |
Jeff Gaston | 41fd5e6 | 2018-04-25 17:02:02 -0400 | [diff] [blame] | 875 | parser.add_argument( |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 876 | '-g', '--gmaven', action="store_true", |
| 877 | help='If specified, updates only the artifact from GMaven libraries excluding those covered by other arguments') |
| 878 | parser.add_argument( |
Jeff Gaston | cc296a8 | 2018-03-23 14:33:24 -0400 | [diff] [blame] | 879 | '--commit-first', action="store_true", |
| 880 | help='If specified, then if uncommited changes exist, commit before continuing') |
Jeff Sharkey | abe16df | 2021-05-25 10:09:03 -0600 | [diff] [blame] | 881 | parser.add_argument( |
| 882 | '--beyond-corp', action="store_true", |
| 883 | help='If specified, then fetch artifacts with tooling that works on BeyondCorp devices') |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 884 | args = parser.parse_args() |
Jeff Gaston | 56fe276 | 2018-02-06 14:54:37 -0500 | [diff] [blame] | 885 | args.file = True |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 886 | if not args.source and (args.platform or args.buildtools \ |
| 887 | or args.jetifier or args.androidx or args.material \ |
| 888 | or args.finalize_sdk or args.constraint): |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 889 | parser.error("You must specify a build ID or local Maven ZIP file") |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 890 | sys.exit(1) |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 891 | if not (args.gmaven or args.platform or args.buildtools \ |
Alan Viverette | 6dc4575 | 2020-04-16 14:56:20 +0000 | [diff] [blame] | 892 | or args.jetifier or args.androidx or args.material \ |
| 893 | or args.finalize_sdk or args.constraint): |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 894 | parser.error("You must specify at least one target to update") |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 895 | sys.exit(1) |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 896 | if (args.constraint or args.material or args.androidx or args.gmaven) \ |
Colin Cross | 74683bc | 2018-04-11 17:36:18 -0700 | [diff] [blame] | 897 | and which('pom2bp') is None: |
Louis Pullen-Freilich | 61bb70e | 2019-04-09 16:15:32 +0100 | [diff] [blame] | 898 | parser.error("Cannot find pom2bp in path; please run lunch to set up build environment. You may also need to run 'm pom2bp' if it hasn't been built already.") |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 899 | sys.exit(1) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 900 | |
Jeff Gaston | cc296a8 | 2018-03-23 14:33:24 -0400 | [diff] [blame] | 901 | if uncommittedChangesExist(): |
| 902 | if args.commit_first: |
| 903 | subprocess.check_call("cd " + git_dir + " && git add -u", shell=True) |
| 904 | subprocess.check_call("cd " + git_dir + " && git commit -m 'save working state'", shell=True) |
| 905 | |
| 906 | if uncommittedChangesExist(): |
Anton Hansson | 10861a0 | 2018-04-06 13:57:08 +0100 | [diff] [blame] | 907 | print_e('FAIL: There are uncommitted changes here. Please commit or stash before continuing, because %s will run "git reset --hard" if execution fails' % os.path.basename(__file__)) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 908 | sys.exit(1) |
| 909 | |
| 910 | try: |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 911 | components = None |
| 912 | if args.constraint: |
Jeff Gaston | 9edf86c | 2018-04-18 13:45:23 -0400 | [diff] [blame] | 913 | if update_constraint_x(getFile(args)): |
| 914 | components = append(components, 'Constraint Layout X') |
| 915 | else: |
| 916 | print_e('Failed to update Constraint Layout X, aborting...') |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 917 | sys.exit(1) |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 918 | if args.gmaven: |
| 919 | updated_artifacts = update_gmaven(gmaven_artifacts) |
| 920 | if updated_artifacts: |
| 921 | components = append(components, '\n'.join(updated_artifacts)) |
| 922 | else: |
| 923 | print_e('Failed to update GMaven, aborting...') |
| 924 | sys.exit(1) |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 925 | if args.androidx: |
Jeff Gaston | 17ccf8f | 2018-08-29 11:33:19 -0400 | [diff] [blame] | 926 | if update_androidx('androidx', \ |
Jeff Gaston | bbd6bfc | 2018-03-27 17:35:49 -0400 | [diff] [blame] | 927 | getBuildId(args), getFile(args)): |
| 928 | components = append(components, 'AndroidX') |
| 929 | else: |
| 930 | print_e('Failed to update AndroidX, aborting...') |
| 931 | sys.exit(1) |
Jeff Gaston | 782c3e3 | 2018-02-06 14:36:17 -0500 | [diff] [blame] | 932 | if args.jetifier: |
Jeff Gaston | 17ccf8f | 2018-08-29 11:33:19 -0400 | [diff] [blame] | 933 | if update_jetifier('androidx', getBuildId(args)): |
Jeff Gaston | 782c3e3 | 2018-02-06 14:36:17 -0500 | [diff] [blame] | 934 | components = append(components, 'Jetifier') |
| 935 | else: |
| 936 | print_e('Failed to update Jetifier, aborting...') |
| 937 | sys.exit(1) |
Anton Hansson | be15e8c | 2018-04-26 17:35:45 +0100 | [diff] [blame] | 938 | if args.platform or args.finalize_sdk: |
Anton Hansson | 87654c4 | 2018-04-26 15:19:42 +0100 | [diff] [blame] | 939 | if update_framework_current(getBuildId(args)): |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 940 | components = append(components, 'platform SDK') |
| 941 | else: |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 942 | print_e('Failed to update platform SDK, aborting...') |
Alan Viverette | cd3de26 | 2017-08-14 09:51:30 -0400 | [diff] [blame] | 943 | sys.exit(1) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 944 | if args.finalize_sdk: |
Anton Hansson | be15e8c | 2018-04-26 17:35:45 +0100 | [diff] [blame] | 945 | n = args.finalize_sdk |
| 946 | if finalize_sdk(getBuildId(args), n): |
| 947 | # We commit the finalized dir separately from the current sdk update. |
| 948 | msg = "Import final sdk version %d from build %s" % (n, getBuildId(args).url_id) |
| 949 | subprocess.check_call(['git', 'add', '%d' % n]) |
Sundong Ahn | cc34cc3 | 2018-07-11 15:18:47 +0900 | [diff] [blame] | 950 | subprocess.check_call(['git', 'add', 'Android.bp']) |
Anton Hansson | be15e8c | 2018-04-26 17:35:45 +0100 | [diff] [blame] | 951 | subprocess.check_call(['git', 'commit', '-m', msg]) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 952 | else: |
Anton Hansson | be15e8c | 2018-04-26 17:35:45 +0100 | [diff] [blame] | 953 | print_e('Failed to finalize SDK %d, aborting...' % n) |
Anton Hansson | 9709fd4 | 2018-04-03 16:52:13 +0100 | [diff] [blame] | 954 | sys.exit(1) |
Jeff Gaston | 6e7e750 | 2018-04-04 00:22:45 -0400 | [diff] [blame] | 955 | if args.material: |
| 956 | if update_material(getFile(args)): |
| 957 | components = append(components, 'intermediate-AndroidX Design Library') |
| 958 | else: |
| 959 | print_e('Failed to update intermediate-AndroidX Design Library, aborting...') |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 960 | sys.exit(1) |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 961 | if args.buildtools: |
Jeff Gaston | b9b0905 | 2018-02-05 14:16:05 -0500 | [diff] [blame] | 962 | if update_buildtools('sdk_phone_armv7-sdk_mac', 'darwin', getBuildId(args)) \ |
| 963 | and update_buildtools('sdk_phone_x86_64-sdk', 'linux', getBuildId(args)) \ |
| 964 | and update_buildtools('sdk_phone_armv7-win_sdk', 'windows', getBuildId(args)): |
Alan Viverette | c1c32b6 | 2017-12-20 09:40:36 -0500 | [diff] [blame] | 965 | components = append(components, 'build tools') |
| 966 | else: |
| 967 | print_e('Failed to update build tools, aborting...') |
| 968 | sys.exit(1) |
Jeff Gaston | b70daa9 | 2018-03-29 20:10:22 -0400 | [diff] [blame] | 969 | |
Jeff Gaston | f12f3ce | 2018-03-23 16:41:24 -0400 | [diff] [blame] | 970 | |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 971 | |
Colin Cross | 941d980 | 2018-05-03 21:37:57 -0700 | [diff] [blame] | 972 | subprocess.check_call(['git', 'add', current_path, buildtools_dir]) |
Saeid Farivar Asanjan | 54cd34e | 2021-10-01 16:47:54 +0000 | [diff] [blame^] | 973 | if not args.source and args.gmaven: |
| 974 | src_msg = "GMaven" |
| 975 | elif not args.source.isnumeric(): |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 976 | src_msg = "local Maven ZIP" |
| 977 | else: |
Jeff Gaston | 13e3841 | 2018-02-06 14:45:36 -0500 | [diff] [blame] | 978 | src_msg = "build %s" % (getBuildId(args).url_id) |
Alan Viverette | 129555e | 2018-01-30 09:57:57 -0500 | [diff] [blame] | 979 | msg = "Import %s from %s\n\n%s" % (components, src_msg, flatten(sys.argv)) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 980 | subprocess.check_call(['git', 'commit', '-m', msg]) |
Anton Hansson | be15e8c | 2018-04-26 17:35:45 +0100 | [diff] [blame] | 981 | if args.finalize_sdk: |
| 982 | print('NOTE: Created two commits:') |
| 983 | subprocess.check_call(['git', 'log', '-2', '--oneline']) |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 984 | print('Remember to test this change before uploading it to Gerrit!') |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 985 | |
| 986 | finally: |
| 987 | # Revert all stray files, including the downloaded zip. |
| 988 | try: |
| 989 | with open(os.devnull, 'w') as bitbucket: |
| 990 | subprocess.check_call(['git', 'add', '-Af', '.'], stdout=bitbucket) |
| 991 | subprocess.check_call( |
Jeff Gaston | 9edf86c | 2018-04-18 13:45:23 -0400 | [diff] [blame] | 992 | ['git', 'commit', '-m', 'COMMIT TO REVERT - RESET ME!!!', '--allow-empty'], stdout=bitbucket) |
Alan Viverette | d4527e6 | 2017-05-11 15:03:25 -0400 | [diff] [blame] | 993 | subprocess.check_call(['git', 'reset', '--hard', 'HEAD~1'], stdout=bitbucket) |
| 994 | except subprocess.CalledProcessError: |
Alan Viverette | f5cb0e6 | 2017-11-17 15:15:23 -0500 | [diff] [blame] | 995 | print_e('ERROR: Failed cleaning up, manual cleanup required!!!') |