mirror of
https://github.com/zeldaret/ph
synced 2026-05-24 07:10:52 -04:00
Aetias
68 lines
2.2 KiB
Python
Executable File
68 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
import pyperclip
|
|
import subprocess
|
|
import os
|
|
from pathlib import Path
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description="Generates a context for decomp.me")
|
|
parser.add_argument('file', help="Input file to preprocess")
|
|
parser.add_argument('-f', type=str, dest='out_file', required=False, help='Output context file')
|
|
parser.add_argument('-c', action=argparse.BooleanOptionalAction, dest='clipboard', required=False, help='Copy output to clipboard')
|
|
parser.add_argument('-e', type=str, dest='encoding', required=False, default="utf-8", help='Input file encoding')
|
|
parser.add_argument('-v', action=argparse.BooleanOptionalAction, dest='verbose', required=False, help='Verbose error output')
|
|
args = parser.parse_args()
|
|
|
|
CXX_FLAGS = [
|
|
'-nostdinc',
|
|
'-Iinclude',
|
|
'-Ilibs/c/include',
|
|
'-Ilibs/cpp/include'
|
|
]
|
|
|
|
script_dir = Path(os.path.dirname(os.path.realpath(__file__)))
|
|
root_dir = script_dir / ".."
|
|
|
|
def eprint(*args, **kwargs):
|
|
print(*args, file=sys.stderr, **kwargs)
|
|
|
|
try:
|
|
ctx: str = subprocess.check_output([
|
|
'gcc',
|
|
'-E', '-P', '-fworking-directory', '-undef', '-dD',
|
|
*CXX_FLAGS,
|
|
args.file
|
|
], cwd=root_dir, encoding=args.encoding)
|
|
except subprocess.CalledProcessError as e:
|
|
eprint(f"Failed to preprocess '{args.file}'")
|
|
if args.verbose: eprint(e)
|
|
else: eprint("Use -v for more verbose error output")
|
|
exit(1)
|
|
|
|
|
|
lines = ctx.splitlines(True)
|
|
for i in reversed(range(len(lines))):
|
|
if lines[i].startswith('#define __cplusplus'): lines.pop(i)
|
|
elif lines[i].startswith('#define __STDC_HOSTED__'): lines.pop(i)
|
|
elif lines[i].startswith('#define __STDC__'): lines.pop(i)
|
|
elif lines[i].startswith('#define __STDC_VERSION__'): lines.pop(i)
|
|
|
|
ctx = ''.join(lines)
|
|
|
|
if args.out_file:
|
|
try:
|
|
with open(args.out_file, "w") as file:
|
|
file.write(ctx)
|
|
except OSError as e:
|
|
eprint(f"Failed to write file '{args.out_file}'")
|
|
if args.verbose: eprint(e)
|
|
else: eprint("Use -v for more verbose error output")
|
|
exit(1)
|
|
if args.clipboard:
|
|
pyperclip.copy(ctx)
|
|
eprint("Copied context to clipboard")
|
|
if args.out_file is None and not args.clipboard:
|
|
print(ctx)
|