VSCodeBackup.psm1
|
Write-Verbose 'Importing from [C:\MyProjects\VSCodeBackup\VSCodeBackup\private]' # .\VSCodeBackup\private\Close-Application.ps1 function Close-Application { [CmdletBinding()] param ( [Parameter(Mandatory)] [string] $ApplicationName, [Parameter()] [string] $TimeOut = "60" ) begin { } process { $ApplicationName = $ApplicationName -replace "\*", "" $Timeout = New-TimeSpan -Seconds $TimeOut $StopWatch = [diagnostics.stopwatch]::StartNew() $ApplicationRunning = Get-Process -Name "*$($ApplicationName)*" -ErrorAction SilentlyContinue if ($ApplicationRunning) { if ( (Test-AdminElevation) -ne $true ) { throw "This module requires elevation if VS Code is running." } else { do { $ApplicationRunning = Get-Process -Name "*$($ApplicationName)*" -ErrorAction SilentlyContinue foreach ($App in $ApplicationRunning) { $App.CloseMainWindow() | Out-Null } } while (($ApplicationRunning.HasExited -contains $false) -and ($StopWatch.elapsed -lt $timeout)) Start-Sleep -Milliseconds 500 if ($ApplicationRunning.HasExited -contains $false) { Write-Error "Could not close all instances of $($ApplicationName)" } } } else { Write-Verbose "$($ApplicationName) was not open" } } end { } } # .\VSCodeBackup\private\Test-AdminElevation.ps1 function Test-AdminElevation { [CmdletBinding()] param ( ) begin { } process { $user = [Security.Principal.WindowsIdentity]::GetCurrent(); (New-Object Security.Principal.WindowsPrincipal $user).IsInRole([Security.Principal.WindowsBuiltinRole]::Administrator) } end { } } Write-Verbose 'Importing from [C:\MyProjects\VSCodeBackup\VSCodeBackup\public]' # .\VSCodeBackup\public\Backup-VSCode.ps1 function Backup-VSCode { <# .SYNOPSIS Backup VS Code settings and extensions .DESCRIPTION Backup VS Code settings and extensions .PARAMETER Path Location to store zip file .PARAMETER Settings Switch to backup settings .PARAMETER Extensions Switch to backup extensions .EXAMPLE Backup-VSCode -Path c:\Users\bobby\Desktop -Settings -Extensions .NOTES General notes #> [CmdletBinding()] param ( # Parameter help description [Parameter(Mandatory)] [ValidateScript( {Test-Path -Path $_})] [string] $Path, # Parameter help description [Parameter()] [switch] $Settings, # Parameter help description [Parameter()] [switch] $Extensions ) begin { $TimeStamp = Get-Date -Format o | ForEach-Object {$_ -replace ":", "."} $Name = "VSCode-$($TimeStamp).zip" $Path = Resolve-Path -Path $Path } process { #Can't read some files while Code is running try { Close-Application -ApplicationName "code" } catch { $_ } $StartTime = Get-Date -Format o $ExtensionsDirectory = "$env:USERPROFILE\.vscode" | Resolve-Path $SettingsDirectory = "$env:APPDATA\Code\User\settings.json" | Resolve-Path if ($Extensions.IsPresent) { try { Compress-Archive -Path $ExtensionsDirectory -DestinationPath $Path\$Name -Update -CompressionLevel NoCompression } catch { throw $_ } } if ($Settings.IsPresent) { try { Compress-Archive -LiteralPath $SettingsDirectory -DestinationPath $Path\$Name -Update } catch { throw $_ } } $EndTime = Get-Date -Format o $ElapsedTime = New-TimeSpan -Start $StartTime -End $EndTime $ZippedSize = [string]([math]::Round((Get-ChildItem $Path\$Name).Length / 1mb)) + "MB" [PSCustomObject]@{ FileName = [string]$Name FilePath = [string]$Path StartTime = [datetime]$StartTime Duration = $ElapsedTime -replace '\.\d+$' EndTime = [datetime]$EndTime Size = $ZippedSize } } end { } } # .\VSCodeBackup\public\Restore-VSCode.ps1 function Restore-VSCode { <# .SYNOPSIS Restore VS Code from a backup .DESCRIPTION Restore VS Code from a backup .PARAMETER Path Path to backup file .PARAMETER Settings Switch to restore settings .PARAMETER Extensions Switch to restore extensions .EXAMPLE Restore-VSCode -Path .\VSCode-2019-01-31T23.33.58.3351871+01.00.zip -Settings -Extensions .NOTES General notes #> [CmdletBinding()] param ( # Parameter help description [Parameter(Mandatory)] [ValidateScript( {Test-Path -Path $_})] [string] $Path, # Parameter help description [Parameter()] [switch] $Settings, # Parameter help description [Parameter()] [switch] $Extensions ) begin { $Path = Resolve-Path -Path $Path } process { #Can't write some files while Code is running try { Close-Application -ApplicationName "code" } catch { $_ } $ExtensionsDirectory = "$env:USERPROFILE\.vscode" | Resolve-Path $SettingsDirectory = "$env:APPDATA\Code\User\settings.json" | Resolve-Path try { Expand-Archive -Path $Path -DestinationPath $env:TEMP -force } catch { throw $_ } if ($Extensions.IsPresent) { Copy-Item -Path "$env:TEMP\.vscode\extensions" -Destination $ExtensionsDirectory -Force -Recurse } if ($Settings.IsPresent) { Copy-Item -LiteralPath "$env:TEMP\settings.json" -Destination $SettingsDirectory -Force } } end { } } Write-Verbose 'Importing from [C:\MyProjects\VSCodeBackup\VSCodeBackup\classes]' |