#------------------------------------------------------------------------------
# Script:  idt-win-installer
#------------------------------------------------------------------------------
# IBM Cloud Developer Tools - CLI installer script for Windows 10 systems
#------------------------------------------------------------------------------
# Copyright (c) 2019, International Business Machines. All Rights Reserved.
#------------------------------------------------------------------------------
$Global:VERSION="1.2.2"
$Global:PROG="IBM Cloud Developer Tools - Installer for Windows"

$Global:INSTALLER_URL="https://ibm.biz/idt-win-installer"
$Global:GIT_URL="https://github.com/IBM-Cloud/ibm-cloud-developer-tools"
$Global:SLACK_URL="https://slack-invite-ibm-cloud-tech.mybluemix.net/"
$Global:IDT_INSTALL_BMX_URL="https://clis.cloud.ibm.com/install"
$Global:IDT_INSTALL_BMX_REPO_NAME="Bluemix"
$Global:IDT_INSTALL_BMX_REPO_URL="https://plugins.cloud.ibm.com"

$Global:FORCE = $false
$Global:NEEDS_REBOOT = $false
$Global:SECS = 0

#------------------------------------------------------------------------------
function help {
  Write-Output @"

  $Global:PROG
  Usage: idt-win-installer [<args>]

  Where <args> is:
    install | update   [Default] Perform install (or update) of all needed CLIs and Plugins
    help               Show this help
    --force | -f       Force updates of dependencies and other settings during update
    --trace            Eanble verbose tracing of all activity

  If "install", "update", or no action, a full CLI installation (or update) will occur:
  1. Pre-req check for 'git', 'docker', and 'kubectl'
  2. Install latest IBM Cloud 'ibmcloud' CLI
  3. Install all required plugins 

  Chat with us on Slack: $Global:SLACK_URL, channel #developer-tools
  Submit any issues to : $Global:GIT_URL/issues

"@
}


#------------------------------------------------------------------------------
function log() {
  Write-Host "[$((Get-PSCallStack)[1].Command)] " -foreground cyan  -nonewline
  Write-Host $args
}

function warn() {
  Write-Host "[$((Get-PSCallStack)[1].Command)] " -foreground cyan  -nonewline
  Write-Host "WARN" -foreground yellow  -nonewline
  Write-Host ": $args"
}

function error() {
  Write-Host "[$((Get-PSCallStack)[1].Command)] " -foreground cyan  -nonewline
  Write-Host "ERROR" -foreground red  -nonewline
  Write-Host ": $args"
  quit
}

#------------------------------------------------------------------------------
function quit() {
  $Global:SECS = (Get-Date)-$Global:SECS
  log "--==[ Finished. Total time: $($Global:SECS.ToString("hh\:mm\:ss")) seconds ]==--"
  Write-Host ""

  #-- Request Restart to save changes to PATH.
  if ( $Global:NEEDS_REBOOT ) {
    $restart = Read-Host -Prompt "A system restart is required. Would you like to restart now (y/N)?"
    if($restart -match "[Yy]" ) {
      Restart-Computer
    } else {
      Write-Host "Note: Reboot still needed to load env variables."
    }
  } else {
    # If running in the console, wait for input before closing.
    if ($Host.Name -eq "ConsoleHost") { 
      Write-Host "Press any key to continue..."
      $Host.UI.RawUI.FlushInputBuffer()   # Make sure buffered input doesn't "press a key" and skip the ReadKey().
      $Host.UI.RawUI.ReadKey("NoEcho,IncludeKeyUp") > $null
      
      #-- turn opff trace
      Set-PSDebug -Trace 0
    }
  }
}

#------------------------------------------------------------------------------
function install() {
  log "Starting Installation/Update..."

  #-- Check if internal IBM setup
  $ibmcloud_command = get-command ibmcloud -erroraction 'silentlycontinue'
  if( $ibmcloud_command )  {
     # The command is set, use it
  } else {
    $ibmcloud_command = 'C:\"Program Files"\IBM\Cloud\bin\ibmcloud.exe'
  }
  $pluginlist = iex "$ibmcloud_command plugin list"
  if($pluginlist -match "\bstage\b") {
    Write-Output
    $reply = Read-Host -Prompt "Use IBM internal repos for install/updates (Y/n)?"
    Write-Output
    if($reply -match "[Yy]*") {
      $Global:IDT_INSTALL_BMX_URL="https://clis.stage1.ng.bluemix.net/install"
      $Global:IDT_INSTALL_BMX_REPO_NAME="stage1"
      $Global:IDT_INSTALL_BMX_REPO_URL="https://plugins.stage1.ng.bluemix.net"
    }
  }

  install_deps
  install_ibmcloud
  install_plugins
  env_setup add

  log "Install finished."

}


#------------------------------------------------------------------------------
#-- Install dependencies - git, docker, kubectl.
function install_deps() {

  [Net.ServicePointManager]::SecurityProtocol = "Tls12, Tls11, Tls, Ssl3"

   #-- git
  log "Checking for external dependency: git"
  if( -not (get-command git -erroraction 'silentlycontinue') -or $Global:FORCE) {
    
    log "Installing/updating external dependency: git"
    Invoke-WebRequest "https://github.com/git-for-windows/git/releases/download/v2.32.0.windows.2/Git-2.32.0.2-64-bit.exe" -UseBasicParsing -outfile "git-installer.exe"
    .\git-installer.exe /SILENT /PathOption="Cmd" | Out-Null
    Remove-Item "git-installer.exe"
    $Global:NEEDS_REBOOT = $true
    log "Install/update completed for: git"
  }

  #-- docker
  log "Checking for external dependency: docker"
  if( -not(get-command docker -erroraction 'silentlycontinue') -or $Global:FORCE) {
    log "Installing/updating external dependency: docker"
    Invoke-WebRequest "https://desktop.docker.com/win/stable/amd64/Docker%20Desktop%20Installer.exe" -UseBasicParsing -outfile "InstallDocker.exe"
    .\InstallDocker.exe | Out-Null
    $Global:NEEDS_REBOOT = $false
    log "Install/update completed for: docker"
  }

  #-- kubectl
  log "Checking for external dependency: kubectl"
  if( -not( get-command kubectl -erroraction 'silentlycontinue') -or $Global:FORCE) {
    log "Installing/updating external dependency: kubectl"

    $response = Invoke-RestMethod "https://containers.cloud.ibm.com/v1/kube-versions" 
    for ($i = 0; $i -lt $response.Length; $i++){
        if ($response[$i].default){
            $kube_version = "v" + $response[$i].major + "." + $response[$i].minor + "."  +$response[$i].patch
            break
        }
    }
    Invoke-WebRequest "https://storage.googleapis.com/kubernetes-release/release/$kube_version/bin/windows/amd64/kubectl.exe" -UseBasicParsing -outfile "kubectl.exe"
    mkdir "C:\Program Files\kubectl" -erroraction 'silentlycontinue'
    Move-Item -Path "kubectl.exe" -Destination "C:\Program Files\kubectl" -force
    add_to_path("C:\Program Files\kubectl")
    $Global:NEEDS_REBOOT = $true
    log "Install/update completed for: kubectl"
  }

  # #-- helm
  # log "Checking for external dependency: helm"
  # if( -not (get-command helm -erroraction 'silentlycontinue') -or $Global:FORCE) {
  #   log "Installing/updating external dependency: helm"
  # [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
  #   $helm_url = ((Invoke-WebRequest https://github.com/helm/helm/releases -UseBasicParsing).Links.OuterHTML | Where-Object{$_ -match '.+?windows-amd64.zip'} | Select-Object -first 1).Split('"')[1]
  #   log "Helm URL : $helm_url"
  #   $helm_file = $helm_url.Split("/")[-1]
  #   log "Helm File: $helm_file"
  #   Invoke-WebRequest $helm_url -UseBasicParsing -outfile "$helm_file"
  #   mkdir "C:\Program Files\helm" -ErrorAction SilentlyContinue
  #   if (-not (Get-Command Expand-7Zip -ErrorAction Ignore)) {
  #       Install-Package -Scope CurrentUser -Force 7Zip4PowerShell > $null
  #   }
  #   Expand-7Zip $helm_file .
  #   $tar_file = $helm_file.Replace('.gz','')
  #   Expand-7Zip $tar_file "C:\Program Files\helm"
  #   Remove-Item $helm_file -erroraction 'silentlycontinue'
  #   Remove-Item $tar_file  -erroraction 'silentlycontinue'
  #   add_to_path("C:\Program Files\helm\windows-amd64")
  #   $Global:NEEDS_REBOOT = $true
  #   log "Install/update completed for: helm"
  # }
}

#------------------------------------------------------------------------------
#-- Add a dir to the system path
function add_to_path {
  Param($path)
  # Directly edit the registery to add kubectl to PATH. Will require a restart to stick.
  $regPath = "HKLM:\SYSTEM\CurrentControlSet\Control\Session Manager\Environment"
  $value = (Get-ItemProperty $regPath -Name Path).Path
  if ( -not ($value -match [Regex]::Escape("$path") )) {
    log "Adding $path to PATH"
    $newValue = "$value;$path"
    Set-ItemProperty -Path $regPath -Name Path -Value $newValue | Out-Null
  }  
}

#------------------------------------------------------------------------------
#-- Install IBM Cloud CLI.
function install_ibmcloud() {
  if( get-command ibmcloud -erroraction 'silentlycontinue') {
      Write-Output "ibmcloud already installed"
      if( $Global:FORCE ){
        # User wants forced update
        ibmcloud update -f
      } else {
        # User will be prompted if they want to update
        ibmcloud update
      }
      
  } else {
    log "Installing 'ibmcloud' CLI for Windows..."
    $url = $Global:IDT_INSTALL_BMX_URL + "/powershell"
    log "Downloading and installing 'ibmcloud' CLI from: $url" 
    Invoke-Expression(New-Object Net.WebClient).DownloadString( $url )
    $Global:NEEDS_REBOOT = $true
    $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path","User") 
  }
  log "IBM Cloud CLI version:"
  $ibmcloud_command = get-command ibmcloud -erroraction 'silentlycontinue'
  if( $ibmcloud_command )  {
     # The command is set, use it
  } else {
    $ibmcloud_command = 'C:\"Program Files"\IBM\Cloud\bin\ibmcloud.exe'
  }
  iex "$ibmcloud_command --version"
}

#------------------------------------------------------------------------------
#-- Install IBM Cloud CLI Plugins.
function install_plugins {
  log "Installing/updating IBM Cloud CLI plugins..."
  $plugins = "Cloud-Functions",
             "container-registry",
             "container-service"
  
  $ibmcloud_command = get-command ibmcloud -erroraction 'silentlycontinue'
  if( $ibmcloud_command )  {
     # The command is set, use it
  } else {
    $ibmcloud_command = 'C:\"Program Files"\IBM\Cloud\bin\ibmcloud.exe'
  }
  $pluginlist = iex "$ibmcloud_command plugin list"

  Foreach ($plugin in $plugins) {
    log "Checking status of plugin: $plugin"
    if($pluginlist -match "\b$plugin\b") {
        log "Updating plugin '$plugin'"
        iex "$ibmcloud_command plugin update -r $Global:IDT_INSTALL_BMX_REPO_NAME $plugin"
    } else {
        log "Installing plugin '$plugin'"
        iex "$ibmcloud_command plugin install -f -r $Global:IDT_INSTALL_BMX_REPO_NAME $plugin"
    }
  }
  log "Running 'ibmcloud plugin list'..."
  iex "$ibmcloud_command plugin list"
  log "Finished installing/updating plugins"
}

#------------------------------------------------------------------------------
#-- Create "ic" script to act as shortcut to "ibmcloud"
function env_setup() {
  $ic_command = get-command ic -erroraction 'silentlycontinue'
  if ( $ic_command ) {
    # the command is set, use its existing location
  } else {
    Write-Output "Creating 'ic' script to act as shortcut to 'ibmcloud' command..."
    $ic_batch = @"
@ECHO OFF
REM #-----------------------------------------------------------
REM # IBM Cloud Shortcut
REM # Shortcut for 'ibmcloud' command
REM #-----------------------------------------------------------
ibmcloud %*
"@
    $ic_command = 'C:\Program Files\IBM\Cloud\bin\ic.bat'
    Write-Output $ic_batch | Out-File -Encoding ascii $ic_command
  }  
}

#------------------------------------------------------------------------------
# MAIN
#------------------------------------------------------------------------------
function main {
  log "--==[ $Global:PROG, v$Global:VERSION ]==--"
  $Global:SECS = (Get-Date)

  #-- Check for Windows 10
  if ([System.Environment]::OSVersion.Version.Major -lt 10) {
    error "This installer requires Windows 10 or higher."
  }

  #-- Check for 64-bit Platform - Dev do not have 32-bit versions.
  if ([Environment]::Is64BitProcess -ne [Environment]::Is64BitOperatingSystem) {
    error "This installer requires 64-bit Windows."
  }

  If(-NOT ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole(`
    [Security.Principal.WindowsBuiltInRole] "Administrator")) {
      error "This script must be run as Administrator. Re-run this script as an Administrator!"
  }

  #-- Running as admin defaults to system32 change to home directory.
  Set-Location ~

  $ACTION="install"

  #-- Parse args
  foreach ($arg in $args) {
    switch -exact ($arg) {
      "--trace" {
        warn "Enabling verbose tracing of all activity"
        Set-PSDebug -Trace 1
      }
      "--force" {
        $Global:FORCE=$true
        warn "Forcing updates for all dependencies and other settings"
      }
      "update"    { $ACTION = "install" }
      "install"   { $ACTION = "install" }
      "help"      { $ACTION = "help" }
      default     { warn "Undefined Arg: $arg" }
    }
  }
  switch -exact ($ACTION) {
    "install"   { install }
    default     { help }
  }

  quit
}

#------------------------------------------------------------------------------
#-- Kick things off
#------------------------------------------------------------------------------
main $args

#------------------------------------------------------------------------------
# EOF
#------------------------------------------------------------------------------