cs-update.ps1

#First run only. Creating necessary files and folders if needed.
$baseDir = "$env:USERPROFILE\cloudslang"

If (!(Test-Path $baseDir)) {
   New-Item -Path $baseDir -ItemType Directory
}
else {
   Write-Host "Your base directory $baseDir already exists! - ignore message"
}

If (!(Test-Path $baseDir\version)) {
   New-Item -Path $baseDir\version -ItemType File
   "Available versions:" | Out-File -Encoding Ascii $baseDir\version
}
else {
   Write-Host "Your version file already exists! - ignore message"
}

sleep 2

$releaseParams = @{
    Uri = "https://api.github.com/repos/CloudSlang/cloud-slang/releases";
    Method = 'GET';
    Headers = @{Accept = 'application/json'};
    ContentType = 'application/json';
                      }

$result = Invoke-RestMethod @releaseParams

$tag = $result.tag_name[0]
$destDir = "$env:USERPROFILE\cloudslang\$tag"

If (!(Test-Path $destDir)) {
   New-Item -Path $destDir -ItemType Directory
}
else {
   Write-Host "Directory already exists!"
}
Push-Location $destDir

#CheckCSrelease function. Checks whether you have the latest Release agains a version file.
function checkRelease {
$version = Get-Content "$baseDir\version"
foreach ($ver in $version){
If ($version -Match $tag) {
	Write-Host "Release $tag already exists"}
else {
	Write-Host "New release $tag Found. Correct any shortcuts you might have."
    "$tag" | Out-File -Encoding Ascii -Append $baseDir\version
	 }
                          }
 sleep 5
 mainmenu 
                      }

#CslangUpdate function. Updates to the CS version given by the $tag var, if you choose to.
function cslangUpdate {
foreach ($num in 0,4){
$asset = $result.assets[$num].url
$ZipFile = $result.assets[$num].name

$releaseAssetParams = @{
    Uri = $asset;
    Method = 'GET';
    Headers = @{Accept = 'application/octet-stream'};
    ContentType = 'application/octet-stream';
					   }

Invoke-RestMethod @releaseAssetParams -OutFile $ZipFile

add-type -type  @'
using System;
using System.Runtime.InteropServices;
using System.ComponentModel;
using System.IO;

namespace Win32Functions
{
  public class ExtendedFileInfo
  {    
    public static long GetFileSizeOnDisk(string file)
    {
        FileInfo info = new FileInfo(file);
        uint dummy, sectorsPerCluster, bytesPerSector;
        int result = GetDiskFreeSpaceW(info.Directory.Root.FullName, out sectorsPerCluster, out bytesPerSector, out dummy, out dummy);
        if (result == 0) throw new Win32Exception();
        uint clusterSize = sectorsPerCluster * bytesPerSector;
        uint hosize;
        uint losize = GetCompressedFileSizeW(file, out hosize);
        long size;
        size = (long)hosize << 32 | losize;
        return ((size + clusterSize - 1) / clusterSize) * clusterSize;
    }

    [DllImport("kernel32.dll")]
    static extern uint GetCompressedFileSizeW([In, MarshalAs(UnmanagedType.LPWStr)] string lpFileName,
       [Out, MarshalAs(UnmanagedType.U4)] out uint lpFileSizeHigh);

    [DllImport("kernel32.dll", SetLastError = true, PreserveSig = true)]
    static extern int GetDiskFreeSpaceW([In, MarshalAs(UnmanagedType.LPWStr)] string lpRootPathName,
       out uint lpSectorsPerCluster, out uint lpBytesPerSector, out uint lpNumberOfFreeClusters,
       out uint lpTotalNumberOfClusters);  
  }
}
'@

$size = $result.assets[$num].size
$downFile = [Win32Functions.ExtendedFileInfo]::GetFileSizeOnDisk( '$destDir\$ZipFile' )

If (!($downFile -Match $size)) {
   Write-Host "$ZipFile - file integrity check PASSED!"
}
else {
  Write-Host "$ZipFile - file size mismatch, retrying download"
  sleep 2
  Invoke-RestMethod @releaseAssetParams -OutFile $ZipFile

     }

					}
function Expand-ZIPFile($file, $destination)
{
$shell = new-object -com shell.application
$zip = $shell.NameSpace($file)
foreach($item in $zip.items())
{
$shell.Namespace($destination).copyhere($item)
}
}
Write-Host "Unpacking files"  
Expand-ZIPFile –File "$destDir\cslang-builder.zip" –Destination "$destDir\cslang-builder"
Expand-ZIPFile –File "$destDir\cslang-cli.zip" –Destination "$destDir\cslang-cli"
 sleep 5
 mainmenu 
						}
				
 function updateShortcuts{
 $WshShell = New-Object -ComObject WScript.Shell
 $Shortcut = $WshShell.CreateShortcut("$env:USERPROFILE\Desktop\Cslang CLI.lnk")
 $Shortcut.TargetPath = "$destDir\cslang-cli\cslang\bin\cslang.bat"
 $Shortcut.WorkingDirectory ="$destDir\cslang-cli\cslang\bin\"
 $Shortcut.Save()
 Write-Host "Shortcuts have been updated"
  sleep 5
  mainmenu		
                         } 
						 
#Areyousure function. Alows user to select y or n when asked to exit. Y exits and N returns to main menu.  
 function areyousure {$areyousure = read-host "Are you sure you want to exit? (y/n)"  
           if ($areyousure -eq "y"){exit}  
           if ($areyousure -eq "n"){mainmenu}  
           else {write-host -foregroundcolor red "Invalid Selection"   
                 areyousure  
                }  
                     }  					
					 
 #Mainmenu function. Contains the screen output for the menu and waits for and handles user input.  
 function mainmenu{  
 cls  
 echo "-----------------Cloudslang Updater----------------------"  
 echo ""  
 echo ""  
 echo "    1. Check for Update"  
 echo "    2. Backup existing CS version - will implement later?"  
 echo "    3. Update CS version" 
 echo "    4  Update Shortcuts" 
 echo "    5. Exit"  
 echo ""  
 echo ""
 echo "     WARNING: If you choose to update and have any"
 echo "              shortcuts created, please set the right"
 echo "              paths if needed. POC - testing"
 echo "---------------------------------------------------------"  
 $answer = read-host "Please Make a Selection"  
 if ($answer -eq 1){checkRelease}  
 if ($answer -eq 2){calc}  
 if ($answer -eq 3){cslangUpdate}  
 if ($answer -eq 4){updateShortcuts}  
 if ($answer -eq 5){areyousure}
 else {write-host -ForegroundColor red "Invalid Selection"  
       sleep 5  
       mainmenu  
      }  
                }  
 mainmenu