@echo off
where python >nul 2>&1 && (
  deactivate >nul 2>&1
  echo [INSTALL] Found Python3
  pip >nul 2>&1 && (
    echo [INSTALL] Found pip
    python -m pip install --no-cache-dir --upgrade pip
  ) || (
    echo [ERROR] pip is not available in PATH
    pause
    exit /b
  )

  if exist "C:\\Program Files\\OpenSSL-Win64\\bin\\openssl.exe" (
    echo [INSTALL] Found OpenSSL executable
  ) else (
   echo [ERROR] OpenSSL executable not found in [C:\\Program Files\\OpenSSL-Win64\\bin\\openssl.exe]
   echo [INFO] Install OpenSSL non-light version - https://slproweb.com/products/Win32OpenSSL.html
   pause
   exit /b
  )

  if exist "C:\\Program Files (x86)\\Microsoft Visual Studio" (
    echo [INSTALL] Found Visual Studio Build Tools
  ) else (
    echo [ERROR] Microsoft Visual C++ 14.0 not found in [C:\\Program Files (x86^)\\Microsoft Visual Studio]
    echo [INFO] Install Microsoft Visual Studio Build Tools - https://visualstudio.microsoft.com/thank-you-downloading-visual-studio/?sku=BuildTools^&rel=16
    pause
    exit /b
  )

  echo [INSTALL] Using venv
  rmdir "venv" /q /s >nul 2>&1
  python -m venv ./venv
  .\venv\Scripts\activate
  python -m pip install --upgrade pip wheel

  set LIB=C:\Program Files\OpenSSL-Win64\lib;%LIB%
  set INCLUDE=C:\Program Files\OpenSSL-Win64\include;%INCLUDE%

  echo [INSTALL] Installing dex enabled yara-python
  pip install --no-index --find-links=scripts/wheels yara-python && (
    rem
  ) || (
    echo [INSTALL] Building dex enabled yara-python
    rmdir /q /s yara-python >nul 2>&1
    pip wheel --wheel-dir=yara-python --build-option="build" --build-option="--enable-dex" "git+https://github.com/VirusTotal/yara-python.git@v3.11.0" && (
      pip install --no-index --find-links=yara-python yara-python
    ) || (
      echo [ERROR] APKiD installation failed. Have you installed Visual Studio Build Tools and other requirements?
      echo Please install all the requirements and run setup.bat again.
      echo Follow the official documentation: https://mobsf.github.io/docs/
      pause
    )
    rmdir /q /s yara-python >nul 2>&1
  )

  echo [INSTALL] Installing Requirements
  pip install --no-cache-dir -r requirements.txt
  
  echo [INSTALL] Clean Up
  CALL scripts/clean.bat y

  echo [INSTALL] Migrating Database
  python manage.py makemigrations
  python manage.py makemigrations StaticAnalyzer
  python manage.py migrate
  echo Download and Install wkhtmltopdf for PDF Report Generation - https://wkhtmltopdf.org/downloads.html
  echo [INSTALL] Installation Complete
  python scripts/check_install.py
) || (
  echo [ERROR] python3 is not installed
)