FilesExpand file tree

main

/

make

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

114 lines (85 loc) · 3.46 KB

main

/

make

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

114 lines (85 loc) · 3.46 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

#!/usr/bin/env python3

from __future__ import annotations

import argparse

import base64

import hashlib

import io

import os.path

import shutil

import subprocess

import tempfile

import zipapp

import zipfile

HERE = os.path.dirname(os.path.realpath(__file__))

IMG = 'make-pre-commit-zipapp'

def _msg(s: str) -> None:

print(f'\033[7m{s}\033[m')

def _exit_if_retv(*cmd: str) -> None:

if subprocess.call(cmd):

raise SystemExit(1)

def _check_no_shared_objects(wheeldir: str) -> None:

for zip_filename in os.listdir(wheeldir):

with zipfile.ZipFile(os.path.join(wheeldir, zip_filename)) as zipf:

for filename in zipf.namelist():

if filename.endswith('.so') or '.so.' in filename:

raise AssertionError(zip_filename, filename)

def _add_shim(dest: str) -> None:

shim = os.path.join(HERE, 'python')

shutil.copy(shim, dest)

bio = io.BytesIO()

with zipfile.ZipFile(bio, 'w') as zipf:

zipf.write(shim, arcname='__main__.py')

with tempfile.TemporaryDirectory() as tmpdir:

_exit_if_retv(

'podman', 'run', '--rm', '--volume', f'{tmpdir}:/out:rw', IMG,

'cp', '/venv/lib/python3.12/site-packages/distlib/t32.exe', '/out',

)

with open(os.path.join(dest, 'python.exe'), 'wb') as f:

with open(os.path.join(tmpdir, 't32.exe'), 'rb') as t32:

f.write(t32.read())

f.write(b'#!py.exe -3\n')

f.write(bio.getvalue())

def _write_cache_key(version: str, wheeldir: str, dest: str) -> None:

cache_hash = hashlib.sha256(f'{version}\n'.encode())

for filename in sorted(os.listdir(wheeldir)):

cache_hash.update(f'{filename}\n'.encode())

with open(os.path.join(HERE, 'python'), 'rb') as f:

cache_hash.update(f.read())

with open(os.path.join(dest, 'CACHE_KEY'), 'wb') as f:

f.write(base64.urlsafe_b64encode(cache_hash.digest()).rstrip(b'='))

def main() -> int:

parser = argparse.ArgumentParser()

parser.add_argument('version')

args = parser.parse_args()

with tempfile.TemporaryDirectory() as tmpdir:

wheeldir = os.path.join(tmpdir, 'wheels')

os.mkdir(wheeldir)

_msg('building podman image...')

_exit_if_retv('podman', 'build', '-q', '-t', IMG, HERE)

_msg('populating wheels...')

_exit_if_retv(

'podman', 'run', '--rm', '--volume', f'{wheeldir}:/wheels:rw', IMG,

'pip', 'wheel', f'pre_commit=={args.version}', 'setuptools',

'--wheel-dir', '/wheels',

)

_msg('validating wheels...')

_check_no_shared_objects(wheeldir)

_msg('adding __main__.py...')

mainfile = os.path.join(tmpdir, '__main__.py')

shutil.copy(os.path.join(HERE, 'entry'), mainfile)

_msg('adding shim...')

_add_shim(tmpdir)

_msg('copying file_lock.py...')

file_lock_py = os.path.join(HERE, '../../pre_commit/file_lock.py')

file_lock_py_dest = os.path.join(tmpdir, 'pre_commit/file_lock.py')

os.makedirs(os.path.dirname(file_lock_py_dest))

shutil.copy(file_lock_py, file_lock_py_dest)

_msg('writing CACHE_KEY...')

_write_cache_key(args.version, wheeldir, tmpdir)

filename = f'pre-commit-{args.version}.pyz'

_msg(f'writing {filename}...')

shebang = '/usr/bin/env python3'

zipapp.create_archive(tmpdir, filename, interpreter=shebang)

return 0

if __name__ == '__main__':

raise SystemExit(main())