File: package-github-release-artifacts

package info (click to toggle)
perfetto 54.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 133,812 kB
  • sloc: cpp: 338,350; python: 74,464; sql: 46,895; ansic: 18,340; javascript: 2,557; java: 2,160; sh: 1,444; yacc: 776; xml: 563; makefile: 226
file content (168 lines) | stat: -rwxr-xr-x 5,649 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
#!/usr/bin/env python3
# Copyright (C) 2021 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.
"""Packages prebuilts and SDK sources for GitHub releases.

Usage: ./tools/release/package-github-release-artifacts v20.0

IMPORTANT: This script must be run from the git tag being packaged. The SDK
source files are generated from the current working directory, so running this
script from a different commit will result in mismatched SDK sources.

This will generate:
- One .zip file for every os-arch combo (e.g. android-arm.zip)
- SDK source zips (perfetto-cpp-sdk-src.zip, perfetto-c-sdk-src.zip)
All files will be placed into /tmp/perfetto-v20.0-github-release/ .
"""

import argparse
import subprocess
import os
import sys
import tempfile
import shutil


def exec(*args):
  print(' '.join(args))
  subprocess.check_call(args)


def get_repo_root():
  """Returns the root directory of the Perfetto repository."""
  script_dir = os.path.dirname(os.path.abspath(__file__))
  return os.path.abspath(os.path.join(script_dir, '..', '..'))


def verify_git_state(expected_version):
  """Verifies git is on the correct tag with no uncommitted changes."""
  warnings = []

  # Check for uncommitted changes
  try:
    result = subprocess.run(['git', 'status', '--porcelain'],
                            capture_output=True,
                            text=True)
    if result.returncode == 0 and result.stdout.strip():
      warnings.append(
          f'Working directory has uncommitted changes:\n{result.stdout}')
  except Exception as e:
    warnings.append(f'Could not check git status: {e}')

  # Check current tag
  try:
    result = subprocess.run(['git', 'describe', '--exact-match', '--tags'],
                            capture_output=True,
                            text=True)
    if result.returncode == 0:
      current_tag = result.stdout.strip()
      if current_tag != expected_version:
        warnings.append(
            f'On tag {current_tag}, but packaging {expected_version}')
    else:
      warnings.append(f'Not on a git tag (expected {expected_version})')
  except Exception as e:
    warnings.append(f'Could not check git tag: {e}')

  if warnings:
    print('WARNING: SDK sources may not match the release tag:')
    for warning in warnings:
      print(f'  - {warning}')
    return input('\nContinue anyway? [y/N] ').lower().strip() in ['y', 'yes']

  print(f'✓ On tag {expected_version} with clean working directory')
  return True


def generate_sdk_sources(tmpdir):
  """Generates SDK source zips using gen_amalgamated."""
  repo_root = get_repo_root()
  gen_amalgamated = os.path.join(repo_root, 'tools', 'gen_amalgamated')

  sdk_zips = []

  # Generate C++ SDK (default targets)
  print('\n--- Generating C++ SDK amalgamated sources ---')
  cpp_sdk_staging = tempfile.mkdtemp(prefix='perfetto-cpp-sdk-')
  try:
    exec('python3', gen_amalgamated, '--output',
         os.path.join(cpp_sdk_staging, 'perfetto'))
    os.chdir(tmpdir)
    exec('zip', '-9rj', 'perfetto-cpp-sdk-src.zip', cpp_sdk_staging)
    sdk_zips.append('perfetto-cpp-sdk-src.zip')
    print('C++ SDK source zip created: perfetto-cpp-sdk-src.zip')
  finally:
    shutil.rmtree(cpp_sdk_staging, ignore_errors=True)

  # Generate C SDK
  # TODO(lalitm): determine the correct target for C SDK
  print('\n--- Generating C SDK amalgamated sources ---')
  c_sdk_staging = tempfile.mkdtemp(prefix='perfetto-c-sdk-')
  try:
    # For now, use the same targets as C++ SDK
    # This should be updated with the correct C SDK target
    exec('python3', gen_amalgamated, '--output',
         os.path.join(c_sdk_staging, 'perfetto'))
    os.chdir(tmpdir)
    exec('zip', '-9rj', 'perfetto-c-sdk-src.zip', c_sdk_staging)
    sdk_zips.append('perfetto-c-sdk-src.zip')
    print('C SDK source zip created: perfetto-c-sdk-src.zip')
  finally:
    shutil.rmtree(c_sdk_staging, ignore_errors=True)

  return sdk_zips


def main():
  parser = argparse.ArgumentParser(epilog='Example: %s v19.0' % __file__)
  parser.add_argument('version', help='Version tag (e.g., v20.0)')

  args = parser.parse_args()

  # Verify we're on the correct tag with no uncommitted changes
  if not verify_git_state(args.version):
    print('Aborted.')
    return 1

  tmpdir = '/tmp/perfetto-%s-github-release' % args.version
  src = 'gs://perfetto-luci-artifacts/%s/' % args.version
  os.makedirs(tmpdir, exist_ok=True)

  # Download and package prebuilts
  print('--- Downloading prebuilts from GCS ---')
  os.chdir(tmpdir)
  exec('gsutil', '-m', 'rsync', '-rc', src, tmpdir + '/')

  zips = []
  for arch in os.listdir(tmpdir):
    if not os.path.isdir(arch):
      continue
    exec('zip', '-9r', '%s.zip' % arch, arch)
    zips.append(arch + '.zip')

  # Generate SDK source zips
  sdk_zips = generate_sdk_sources(tmpdir)
  zips.extend(sdk_zips)

  print('')
  print('=' * 70)
  print('%d zip files saved in %s' % (len(zips), tmpdir))
  print('Prebuilt binaries: %d' % (len(zips) - len(sdk_zips)))
  print('SDK sources: %d' % len(sdk_zips))
  print('Files: %s' % ', '.join(sorted(zips)))
  print('=' * 70)


if __name__ == '__main__':
  sys.exit(main())