mirror of
https://github.com/derrod/legendary.git
synced 2026-04-16 09:09:05 +00:00
cdf6645d5c
* update action steps versions * added support for multiple architectures * switch to zipapp building on Linux * drop support for deb package CI
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import os
|
|
import sys
|
|
import zipfile
|
|
import filelock
|
|
import zlib
|
|
|
|
# We assume zipapp is created on Linux only
|
|
|
|
cache_path = os.environ.get('XDG_CACHE_HOME')
|
|
if cache_path:
|
|
cache_path = os.path.join(cache_path, 'legendary')
|
|
else:
|
|
cache_path = os.path.expanduser('~/.cache/legendary')
|
|
|
|
vendored_packages_path = os.path.join(cache_path, 'vendored')
|
|
vendored_packages_lock = os.path.join(cache_path, 'vendored.lock')
|
|
|
|
# At the moment only Cryptodome AES uses a native module
|
|
# Thus we only handle the extraction of that
|
|
|
|
if zipfile.is_zipfile(os.path.dirname(__file__)):
|
|
with filelock.FileLock(vendored_packages_lock) as lock:
|
|
with zipfile.ZipFile(os.path.dirname(__file__)) as zf:
|
|
# First see if we need to do the extraction
|
|
should_extract = True
|
|
init_path = os.path.join(vendored_packages_path, 'Cryptodome/__init__.py')
|
|
if os.path.exists(init_path):
|
|
file = zf.getinfo('Cryptodome/__init__.py')
|
|
with open(init_path, 'rb') as init:
|
|
should_extract = zlib.crc32(init.read()) != file.CRC
|
|
|
|
# We extract only dependencies that require native code
|
|
if should_extract:
|
|
for file in zf.infolist():
|
|
if file.filename.startswith('Cryptodome'):
|
|
extracted = zf.extract(file.filename, vendored_packages_path)
|
|
os.chmod(extracted, file.external_attr >> 16)
|
|
sys.path.insert(0, vendored_packages_path)
|
|
|
|
# Run CLI
|
|
import legendary.cli
|
|
legendary.cli.main() |