pikmin2/tools/transform-dep.py
2023-01-31 01:25:12 -05:00

78 lines
2.1 KiB
Python
Executable File

#!/usr/bin/env python3
# borrowed from prime-decomp
import argparse
import os
from platform import uname
from typing import List
if os.name != 'nt':
wineprefix = os.environ.get('WINEPREFIX', os.path.join(os.environ['HOME'], '.wine'))
winedevices = os.path.join(wineprefix, 'dosdevices')
def in_wsl() -> bool:
# wsl1 has Microsoft, wsl2 has microsoft-standard
release = uname().release
return 'microsoft-standard' in release or 'Microsoft' in release
def convert_path(path: str) -> str:
# lowercase drive letter
path = path[0].lower() + path[1:]
if os.name == 'nt':
return path.replace('\\', '/')
elif path[0] == 'z':
# shortcut for z:
return path[2:].replace('\\', '/')
elif in_wsl():
if path.startswith(r'\\wsl'):
# first part could be wsl$ or wsl.localhost
pos = path.find('\\', 2)
pos = path.find('\\', pos + 1)
path = path[pos:]
return path.replace('\\', '/')
else:
path = path[0:1] + path[2:]
return os.path.join('/mnt', path.replace('\\', '/'))
else:
# use $WINEPREFIX/dosdevices to resolve path
return os.path.realpath(os.path.join(winedevices, path.replace('\\', '/')))
def import_d_file(in_file: str) -> str:
out_lines: List[str] = []
with open(in_file, 'r') as file:
it = iter(file)
line = next(it)
if line.endswith(' \\\n'):
out_lines.append(line[:-3].replace('\\', '/') + " \\\n")
else:
out_lines.append(line.replace('\\', '/'))
return ''.join(out_lines)
def main():
parser = argparse.ArgumentParser(
description="""Transform a .d file from Wine paths to normal paths"""
)
parser.add_argument(
"d_file",
help="""Dependency file in""",
)
parser.add_argument(
"d_file_out",
help="""Dependency file out""",
)
args = parser.parse_args()
output = import_d_file(args.d_file)
with open(args.d_file_out, "w", encoding="UTF-8") as f:
f.write(output)
if __name__ == "__main__":
main()