|
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
|
|
import shutil, sys, subprocess, argparse, pathlib
|
|
|
|
|
import platform
|
|
|
|
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
|
|
|
|
|
|
parser.add_argument('--private-dir', required=True)
|
|
|
|
|
parser.add_argument('-o', required=True)
|
|
|
|
|
parser.add_argument('cmparr', nargs='+')
|
|
|
|
|
|
|
|
|
|
contents = '''#include<stdio.h>
|
|
|
|
|
|
|
|
|
|
void flob(void) {
|
|
|
|
|
printf("Now flobbing.\\n");
|
|
|
|
|
}
|
|
|
|
|
'''
|
|
|
|
|
|
|
|
|
|
def get_pic_args():
|
|
|
|
|
platname = platform.system().lower()
|
|
|
|
|
if platname in ['windows', 'darwin'] or sys.platform == 'cygwin':
|
|
|
|
|
return []
|
|
|
|
|
return ['-fPIC']
|
|
|
|
|
|
|
|
|
|
def generate_lib_gnulike(outfile, c_file, private_dir, compiler_array):
|
|
|
|
|
if shutil.which('ar'):
|
|
|
|
|
static_linker = 'ar'
|
|
|
|
|
elif shutil.which('llvm-ar'):
|
|
|
|
|
static_linker = 'llvm-ar'
|
|
|
|
|
elif shutil.which('gcc-ar'):
|
|
|
|
|
static_linker = 'gcc-ar'
|
|
|
|
|
else:
|
|
|
|
|
sys.exit('Could not detect a static linker.')
|
|
|
|
|
o_file = c_file.with_suffix('.o')
|
|
|
|
|
compile_cmd = compiler_array + ['-c', '-g', '-O2', '-o', str(o_file), str(c_file)]
|
|
|
|
|
compile_cmd += get_pic_args()
|
|
|
|
|
subprocess.check_call(compile_cmd)
|
|
|
|
|
out_file = pathlib.Path(outfile)
|
|
|
|
|
if out_file.exists():
|
|
|
|
|
out_file.unlink()
|
|
|
|
|
link_cmd = [static_linker, 'csr', outfile, str(o_file)]
|
|
|
|
|
subprocess.check_call(link_cmd)
|
|
|
|
|
return 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def generate_lib_msvc(outfile, c_file, private_dir, compiler_array):
|
|
|
|
|
static_linker = 'lib'
|
|
|
|
|
o_file = c_file.with_suffix('.obj')
|
|
|
|
|
compile_cmd = compiler_array + ['/MDd',
|
|
|
|
|
'/nologo',
|
|
|
|
|
'/ZI',
|
|
|
|
|
'/Ob0',
|
|
|
|
|
'/Od',
|
|
|
|
|
'/c',
|
|
|
|
|
'/Fo' + str(o_file),
|
|
|
|
|
str(c_file)]
|
|
|
|
|
subprocess.check_call(compile_cmd)
|
|
|
|
|
out_file = pathlib.Path(outfile)
|
|
|
|
|
if out_file.exists():
|
|
|
|
|
out_file.unlink()
|
|
|
|
|
link_cmd = [static_linker,
|
|
|
|
|
'/nologo',
|
|
|
|
|
'/OUT:' + str(outfile),
|
|
|
|
|
str(o_file)]
|
|
|
|
|
subprocess.check_call(link_cmd)
|
|
|
|
|
return 0
|
|
|
|
|
|
|
|
|
|
def generate_lib(outfile, private_dir, compiler_array):
|
|
|
|
|
private_dir = pathlib.Path(private_dir)
|
|
|
|
|
if not private_dir.exists():
|
|
|
|
|
private_dir.mkdir()
|
|
|
|
|
c_file = private_dir / 'flob.c'
|
|
|
|
|
c_file.write_text(contents)
|
|
|
|
|
for i in compiler_array:
|
|
|
|
|
if (i.endswith('cl') or i.endswith('cl.exe')) and 'clang-cl' not in i:
|
|
|
|
|
return generate_lib_msvc(outfile, c_file, private_dir, compiler_array)
|
|
|
|
|
return generate_lib_gnulike(outfile, c_file, private_dir, compiler_array)
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
options = parser.parse_args()
|
|
|
|
|
sys.exit(generate_lib(options.o, options.private_dir, options.cmparr))
|
