#!/usr/bin/env python3

import itertools
import os
import shutil
import sys

import toml
from PyInstaller.__main__ import run

if sys.platform == "win32":
	from PyInstaller.utils.win32.versioninfo import (FixedFileInfo, StringFileInfo, StringStruct,
													 StringTable, VarFileInfo, VarStruct, VSVersionInfo)
	#from PyInstaller.utils.win32.versioninfo import SetVersion

SCRIPT_PATH = os.path.dirname(os.path.realpath(__file__))

"""Load pyproject.toml information."""
project = toml.load(os.path.join(SCRIPT_PATH, "pyproject.toml"))
poetry = project["tool"]["poetry"]

"""Configuration options that may be changed or referenced often."""
DEBUG = False  # When False, removes un-needed data after build has finished
NAME = poetry["name"]
AUTHOR = "vinetrimmer contributors"
VERSION = poetry["version"]
ICON_FILE = "assets/icon.ico"  # pass None to use default icon
ONE_FILE = False  # Must be False if using setup.iss
CONSOLE = True  # If build is intended for GUI, set to False
ADDITIONAL_DATA = [
	# (local file path, destination in build output)
]
HIDDEN_IMPORTS = []
EXTRA_ARGS = [
	"-y", "--win-private-assemblies", "--win-no-prefer-redirects"
]

"""Prepare environment to ensure output data is fresh."""
shutil.rmtree("build", ignore_errors=True)
shutil.rmtree("dist/vinetrimmer", ignore_errors=True)
# we don't want to use any spec, only the configuration set in this file
try:
	os.unlink(f"{NAME}.spec")
except FileNotFoundError:
	pass

"""Run PyInstaller with the provided configuration."""
run([
	"vinetrimmer/vinetrimmer.py",
	"-n", NAME,
	"-i", ["NONE", ICON_FILE][bool(ICON_FILE)],
	["-D", "-F"][ONE_FILE],
	["-w", "-c"][CONSOLE],
	*itertools.chain(*[["--add-data", os.pathsep.join(x)] for x in ADDITIONAL_DATA]),
	*itertools.chain(*[["--hidden-import", x] for x in HIDDEN_IMPORTS]),
	*EXTRA_ARGS
])

if sys.platform == "win32":
	"""Set Version Info Structure."""
	VERSION_4_TUP = tuple(map(int, f"{VERSION}.0".split(".")))
	VERSION_4_STR = ".".join(map(str, VERSION_4_TUP))
	#SetVersion(
	#	"dist/{0}/{0}.exe".format(NAME),
	#	VSVersionInfo(
	#		ffi=FixedFileInfo(
	#			filevers=VERSION_4_TUP,
	#			prodvers=VERSION_4_TUP
	#		),
	#		kids=[
	#			StringFileInfo([StringTable(
	#				"040904B0",  # ?
	#				[
	#					StringStruct("Comments", NAME),
	#					StringStruct("CompanyName", AUTHOR),
	#					StringStruct("FileDescription", "Widevine DRM downloader and decrypter"),
	#					StringStruct("FileVersion", VERSION_4_STR),
	#					StringStruct("InternalName", NAME),
	#					StringStruct("LegalCopyright", f"Copyright (C) 2019-2021 {AUTHOR}"),
	#					StringStruct("OriginalFilename", ""),
	#					StringStruct("ProductName", NAME),
	#					StringStruct("ProductVersion", VERSION_4_STR)
	#				]
	#			)]),
	#			VarFileInfo([VarStruct("Translation", [0, 1200])])  # ?
	#		]
	#	)
	#)

if not DEBUG:
	shutil.rmtree("build", ignore_errors=True)
	# we don't want to keep the generated spec
	try:
		os.unlink(f"{NAME}.spec")
	except FileNotFoundError:
		pass