From 620c5cc97a321ef49a21863d3c2341cead30f1e7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ond=C5=99ej=20Samohel?= <ondrej@samohel.cz>
Date: Thu, 10 Sep 2020 12:17:46 +0200
Subject: [PATCH] pimped ps build script

---
 build.ps1 | 80 ++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 70 insertions(+), 10 deletions(-)

diff --git a/build.ps1 b/build.ps1
index bbd3884473..89f19c29b4 100644
--- a/build.ps1
+++ b/build.ps1
@@ -1,7 +1,61 @@
-Write-Host "Building Pype ..."
-Write-Host "Detecting host Python ..."
+<#
+.SYNOPSIS
+  Helper script to build Pype.
+
+.DESCRIPTION
+  This script will detect Python installation, create venv and install
+  all necessary packages from `requirements.txt` needed by Pype to be
+  included during application freeze on Windows.
+
+.EXAMPLE
+
+PS> .\build.ps1
+
+#>
+
+
+function Exit-WithCode($exitcode) {
+   # Only exit this host process if it's a child of another PowerShell parent process...
+   $parentPID = (Get-CimInstance -ClassName Win32_Process -Filter "ProcessId=$PID" | Select-Object -Property ParentProcessId).ParentProcessId
+   $parentProcName = (Get-CimInstance -ClassName Win32_Process -Filter "ProcessId=$parentPID" | Select-Object -Property Name).Name
+   if ('powershell.exe' -eq $parentProcName) { $host.SetShouldExit($exitcode) }
+
+   exit $exitcode
+}
+
+$art = @'
+
+
+     ____________
+   / \      __   \
+   \  \     \/_\  \
+    \  \     _____/ ______
+     \  \    \___// \     \
+      \  \____\   \  \_____\
+       \/_____/    \/______/  PYPE Club .
+
+'@
+
+Write-Host $art -ForegroundColor DarkGreen
+
+$version_file = Get-Content -Path "version.py"
+$pype_version = $version_file -match '(\d+\.\d+.\d+)'
+if (-not $pype_version) {
+  Write-Host "!!! " -ForegroundColor yellow -NoNewline
+  Write-Host "Cannot determine Pype version."
+  Exit-WithCode 1
+}
+
+Write-Host ">>> " -NoNewline -ForegroundColor green
+Write-Host "Building Pype [ " -NoNewline -ForegroundColor white
+Write-host "v$($matches[1])"  -NoNewline -ForegroundColor green
+Write-Host " ]..." -ForegroundColor white
+
+Write-Host ">>> " -NoNewline -ForegroundColor green
+Write-Host "Detecting host Python ... " -NoNewline
 if (-not (Get-Command "python" -ErrorAction SilentlyContinue)) {
-    Write-Host "!!! Python not detected"
+    Write-Host "!!! Python not detected" -ForegroundColor red
+    Exit-WithCode 1
 }
 $version_command = @'
 import sys
@@ -12,26 +66,32 @@ $p = & python -c $version_command
 $env:PYTHON_VERSION = $p
 $m = $p -match '(\d+)\.(\d+)'
 if(-not $m) {
-  Write-Host "!!! Cannot determine version".
-  return 1
+  Write-Host "!!! Cannot determine version" -ForegroundColor red
+  Exit-WithCode 1
 }
 # We are supporting python 3.6 and up
 if(($matches[1] -lt 3) -or ($matches[2] -lt 7)) {
-  Write-Host "FAILED Version [ $p ] is old and unsupported"
-  return 1
+  Write-Host "FAILED Version [ $p ] is old and unsupported" -ForegroundColor red
+  Exit-WithCode 1
 }
-Write-Host "... got [ $p ]"
+Write-Host "OK [ $p ]" -ForegroundColor green
+Write-Host ">>> " -NoNewline -ForegroundColor green
 Write-Host "Creating virtual env ..."
 & python -m venv venv
+Write-Host ">>> " -NoNewline -ForegroundColor green
 Write-Host "Entering venv..."
 try {
   . (".\venv\Scripts\Activate.ps1")
 }
 catch {
-  Write-Host "!!! Failed to activate."
+  Write-Host "!!! Failed to activate" -ForegroundColor red
   Write-Host $_.Exception.Message
-  return 1
+  Exit-WithCode 1
 }
+Write-Host ">>> " -NoNewline -ForegroundColor green
+Write-Host "Installing packages to new venv ..."
 & pip install -r .\requirements.txt
+Write-Host ">>> " -NoNewline -ForegroundColor green
+Write-Host "Building Pype ..."
 & python setup.py build
 deactivate