# This file is Copyright 2019 Volatility Foundation and licensed under the Volatility Software License 1.0
# which is available at https://www.volatilityfoundation.org/license/vsl-v1.0
#

import os
import sys

from PyInstaller.building.api import PYZ, EXE
from PyInstaller.building.build_main import Analysis
from PyInstaller.utils.hooks import collect_submodules, collect_data_files, collect_dynamic_libs

block_cipher = None

# NOTE: Issues with default pyinstaller build:
# jsonschema:
#   - https://github.com/pyinstaller/pyinstaller/issues/4100
#   - https://github.com/pyinstaller/pyinstaller/pull/4168

binaries = []
try:
    import capstone

    binaries = collect_dynamic_libs('capstone')
except ImportError:
    pass

# Volatility must be findable in sys.path in order for collect_submodules to work
# This adds the current working directory, which should usually do the trick
sys.path.append(os.getcwd())

vol_analysis = Analysis(['volshell.py'],
                        pathex = [],
                        binaries = binaries,
                        datas = collect_data_files('volatility3.framework') + \
                                collect_data_files('volatility3.framework.automagic', include_py_files = True) + \
                                collect_data_files('volatility3.framework.plugins', include_py_files = True) + \
                                collect_data_files('volatility3.framework.layers', include_py_files = True) + \
                                collect_data_files('volatility3.cli', include_py_files = True) + \
                                collect_data_files('volatility3.schemas') + \
                                collect_data_files('volatility3.plugins', include_py_files = True),
                        hiddenimports = collect_submodules('volatility3.framework.automagic') + \
                                        collect_submodules('volatility3.framework.plugins') + \
                                        collect_submodules('volatility3.framework.symbols'),
                        hookspath = [],
                        runtime_hooks = [],
                        excludes = [],
                        win_no_prefer_redirects = False,
                        win_private_assemblies = False,
                        cipher = block_cipher,
                        noarchive = False)
vol_pyz = PYZ(vol_analysis.pure, vol_analysis.zipped_data,
              cipher = block_cipher)
vol_exe = EXE(vol_pyz,
              vol_analysis.scripts,
              vol_analysis.binaries,
              vol_analysis.zipfiles,
              vol_analysis.datas,
              [('u', None, 'OPTION')],
              name = 'volshell',
              icon = os.path.join('doc', 'source', '_static', 'favicon.ico'),
              debug = False,
              bootloader_ignore_signals = False,
              strip = False,
              upx = True,
              runtime_tmpdir = None,
              console = True)