|
| 1 | +#!/usr/bin/env python |
| 2 | + |
| 3 | +# This script finds and prints the various versions in this project: wasi-sdk |
| 4 | +# itself, LLVM, and the Git revisions of dependencies. |
| 5 | +# |
| 6 | +# Usage: version [wasi-sdk|llvm|llvm-major|dump]? |
| 7 | + |
| 8 | +import argparse |
| 9 | +import subprocess |
| 10 | + |
| 11 | +# The number of characters to use for the abbreviated Git revision. |
| 12 | +GIT_REF_LEN = 12 |
| 13 | + |
| 14 | + |
| 15 | +def main(action, llvm_dir): |
| 16 | + if action == 'wasi-sdk': |
| 17 | + print(git_version()) |
| 18 | + elif action == 'llvm': |
| 19 | + major, minor, path = llvm_cmake_version(llvm_dir) |
| 20 | + print(f'{major}.{minor}.{path}') |
| 21 | + elif action == 'llvm-major': |
| 22 | + major, _, _ = llvm_cmake_version(llvm_dir) |
| 23 | + print(major) |
| 24 | + elif action == 'dump': |
| 25 | + print(git_version()) |
| 26 | + print(f'wasi-libc: {git_commit("src/wasi-libc")}') |
| 27 | + print(f'llvm: {git_commit(llvm_dir)}') |
| 28 | + major, minor, path = llvm_cmake_version(llvm_dir) |
| 29 | + print(f'llvm-version: {major}.{minor}.{path}') |
| 30 | + print(f'config: {git_commit("src/config")}') |
| 31 | + |
| 32 | + |
| 33 | +def git_version(): |
| 34 | + version = exec(['git', 'describe', '--long', '--candidates=999', |
| 35 | + '--match=wasi-sdk-*', '--dirty=+m', f'--abbrev={GIT_REF_LEN}']) |
| 36 | + # Parse, e.g.: wasi-sdk-21-0-g317548590b40+m |
| 37 | + parts = version.replace('+', '-').split('-') |
| 38 | + assert (parts[0:2] == ['wasi', 'sdk']) |
| 39 | + major, minor, git, dirty = parts[2:] |
| 40 | + version = f'{major}.{minor}' |
| 41 | + if git: |
| 42 | + assert (git.startswith('g')) |
| 43 | + version += git |
| 44 | + if dirty: |
| 45 | + assert (dirty == 'm') |
| 46 | + version += '+m' |
| 47 | + return version |
| 48 | + |
| 49 | + |
| 50 | +def git_commit(dir='.'): |
| 51 | + return exec(['git', 'rev-parse', f'--short={GIT_REF_LEN}', 'HEAD'], dir) |
| 52 | + |
| 53 | + |
| 54 | +def llvm_cmake_version(llvm_dir): |
| 55 | + with open(f'{llvm_dir}/llvm/CMakeLists.txt', 'r') as file: |
| 56 | + for line in file: |
| 57 | + line = line.strip() |
| 58 | + if line.startswith('set(LLVM_VERSION_MAJOR'): |
| 59 | + llvm_version_major = parse_cmake_set(line) |
| 60 | + elif line.startswith('set(LLVM_VERSION_MINOR'): |
| 61 | + llvm_version_minor = parse_cmake_set(line) |
| 62 | + elif line.startswith('set(LLVM_VERSION_PATCH'): |
| 63 | + llvm_version_patch = parse_cmake_set(line) |
| 64 | + return llvm_version_major, llvm_version_minor, llvm_version_patch |
| 65 | + |
| 66 | + |
| 67 | +def parse_cmake_set(line): |
| 68 | + return line.split(' ')[1].split(')')[0] |
| 69 | + |
| 70 | + |
| 71 | +def exec(command, cwd='.'): |
| 72 | + result = subprocess.run(command, capture_output=True, text=True, cwd=cwd) |
| 73 | + return result.stdout.strip() |
| 74 | + |
| 75 | + |
| 76 | +if __name__ == '__main__': |
| 77 | + parser = argparse.ArgumentParser( |
| 78 | + description='Print the various kinds of versions in wasi-sdk') |
| 79 | + parser.add_argument('action', |
| 80 | + choices=['wasi-sdk', 'llvm', 'llvm-major', 'dump'], |
| 81 | + nargs='?', |
| 82 | + default='wasi-sdk', |
| 83 | + help='Which kind of version to print (default: wasi-sdk).') |
| 84 | + parser.add_argument('--llvm-dir', |
| 85 | + nargs='?', |
| 86 | + default='src/llvm-project', |
| 87 | + help='Override the location of the LLVM source directory (default: src/llvm-project).') |
| 88 | + args = parser.parse_args() |
| 89 | + main(args.action, args.llvm_dir) |
0 commit comments