#!/usr/bin/env python3
#
# mmgen = Multi-Mode GENerator, a command-line cryptocurrency wallet
# Copyright (C)2013-2023 The MMGen Project <mmgen@tuta.io>
# Licensed under the GNU General Public License, Version 3:
#   https://www.gnu.org/licenses
# Public project repositories:
#   https://github.com/mmgen/mmgen
#   https://gitlab.com/mmgen/mmgen

import os
from subprocess import run,PIPE
from setuptools import setup,Extension
from setuptools.command.build_ext import build_ext

cache_path = os.path.join(os.environ['HOME'],'.cache','mmgen')
ext_path = os.path.join(cache_path,'secp256k1')

def build_libsecp256k1():

	def fix_broken_libpython_fn():
		from pathlib import Path
		path = Path(Path().resolve().anchor) / 'msys64/mingw64/lib'
		old = path / 'libpython3.10.dll.a'
		new = path / 'libpython310.dll.a'
		if old.exists() and not new.exists():
			import shutil
			print(f'Fixing broken library filename: {old.name!r} -> {new.name!r}')
			shutil.copy2(old,new)

	def fix_broken_aclocal_path():
		os.environ['ACLOCAL_PATH'] = '/ucrt64/share/aclocal:/usr/share/aclocal'

	import platform
	if platform.system() == 'Windows':
		fix_broken_libpython_fn()
		if os.getenv('MSYSTEM') == 'UCRT64':
			fix_broken_aclocal_path()

	if not os.path.exists(cache_path):
		os.makedirs(cache_path)

	if not os.path.exists(ext_path):
		print('\nCloning libsecp256k1')
		run(['git','clone','https://github.com/bitcoin-core/secp256k1.git'],check=True,cwd=cache_path)

	if not os.path.exists(os.path.join(ext_path,'.libs/libsecp256k1.a')):
		print('\nBuilding libsecp256k1')
		cmds = {
			'Windows': (
				['sh','./autogen.sh'],
				['sh','./configure','CFLAGS=-g -O2 -fPIC','--disable-dependency-tracking'],
				['mingw32-make','MAKE=mingw32-make']
			),
			'Linux': (
				['./autogen.sh'],
				['./configure','CFLAGS=-g -O2 -fPIC'],
				['make'] + ([] if have_arm else ['-j4']),
			),
		}[platform.system()]
		for cmd in cmds:
			print('Executing {}'.format(' '.join(cmd)))
			run(cmd,check=True,cwd=ext_path)

uname = {k:run(['uname',f'-{k}'],stdout=PIPE,check=True).stdout.strip().decode() for k in ('m','s')}

have_arm   = uname['m'] in ('aarch64','armv7l') # x86_64, aarch64, armv7l
have_msys2 = uname['s'].startswith('MSYS_NT')   # Linux, MSYS_NT.*

class my_build_ext(build_ext):
	def build_extension(self,ext):
		build_libsecp256k1()
		build_ext.build_extension(self,ext)

setup(
	cmdclass = { 'build_ext': my_build_ext },
	ext_modules = [Extension(
		name          = 'mmgen.proto.secp256k1.secp256k1',
		sources       = ['extmod/secp256k1mod.c'],
		libraries     = ([],['gmp'])[have_msys2],
		extra_objects = [os.path.join(ext_path,'.libs/libsecp256k1.a')],
		include_dirs  = [os.path.join(ext_path,'include')],
	)]
)