core/init.ps1
|
. $PSScriptRoot\function.ps1 $_psc = @{} $_psc.version = '1.0.5' $_psc.root_dir = Split-Path $PSScriptRoot -Parent $_psc.completions = $_psc.root_dir + '\completions' $_psc.core = $_psc.root_dir + '\core' $_psc.list_path = $_psc.core + '\.list' $_psc.old_list_path = $_psc.core + '\.old_list' $_psc.update_path = $_psc.core + '\.update' $_psc.lang = (Get-WinSystemLocale).name $_psc.langs = @('zh-CN', 'en-US') $_psc.alias_path = $_psc.completions + '\PSCompletions\.alias' $_psc.list = '' $_psc.init = $false $_psc.old=(Get-ChildItem ((Get-ChildItem (Split-Path $_psc.root_dir -Parent) | Where-Object { $_.Name -lt $_psc.version } | Sort-Object Name -Descending | Select-Object -First 1).FullName + '\completions')) foreach($_ in $_psc.old){ if($_.BaseName -ne 'PSCompletions'){ Move-Item $_.FullName $_psc.completions -ErrorAction SilentlyContinue } } $_psc.old= $null if (Get-Command Set-PSReadLineKeyHandler -ErrorAction SilentlyContinue) { Set-PSReadLineKeyHandler 'Tab' MenuComplete } if (!(Test-Path($_psc.list_path)) -or ([environment]::GetEnvironmentvariable("abgox_PSCompletions", "User") -split ';')[0] -ne $_psc.version) { $module_version = $_psc.version $root_cmd = 'psc' $github = 'https://github.com/abgox/PSCompletions' $gitee = 'https://gitee.com/abgox/PSCompletions' $language = $_psc.lang $update = 1 [environment]::SetEnvironmentvariable('abgox_PSCompletions', ($module_version + ';' + $root_cmd + ';' + $github + ';' + $gitee + ';' + $language + ';' + $update), 'User') $_psc.init = $true } function psc_init() { $_psc.config = _psc_get_config $p_c = $_psc.config function get_root { ([environment]::GetEnvironmentvariable("abgox_PSCompletions", "User") -split ';')[1] } $_psc.root_cmd = get_root if ($p_c.language) { $_psc.lang = $p_c.language } if (!( $_psc.lang -in $_psc.langs)) { $_psc.lang = 'en-US' } $_psc.github = $p_c.github.Replace('github.com', 'raw.githubusercontent.com') + '/main' $_psc.gitee = $p_c.gitee + '/raw/main' function _do($var, $var2) { return $(if ($p_c.$var) { $_psc.$var } elseif ($p_c.$var2) { $_psc.$var2 }) } if ($_psc.lang -eq 'zh-CN') { $_psc.url = _do 'gitee' 'github' } else { $_psc.url = _do 'github' 'gitee' } $psc_json_path = $_psc.completions + '\PSCompletions\json\' + $_psc.lang + '.json' if (!(Test-Path($psc_json_path))) { _psc_add_completion 'PSCompletions' $false } if (!(Test-Path($_psc.alias_path))) { echo $_psc.config.root_cmd > $_psc.alias_path } $psc_alias = _psc_get_cmd ($_psc.completions + '\PSCompletions') 'psc' if ($psc_alias -ne $_psc.root_cmd) { _psc_set_config 'root_cmd' $psc_alias $_psc.root_cmd = get_root } $_psc.json = (Get-Content -Path $psc_json_path -Raw -Encoding UTF8 | ConvertFrom-Json).PSCompletions_core_info $_psc.installed = Get-ChildItem -Path $_psc.completions -Filter "*.ps1" -Recurse if (!(Test-Path($_psc.update_path))) { New-Item $_psc.update_path > $null } if (!(Test-Path($_psc.list_path))) { New-Item $_psc.list_path > $null _psc_download_list } function _do([scriptblock]$do) { try { & $do }catch {} } _do { if (!(Test-Path($_psc.old_list_path))) { Copy-Item $_psc.list_path $_psc.old_list_path -Force } } _do { $_psc.list = _psc_get_content $_psc.list_path } $_psc.update = (_psc_get_content $_psc.update_path) -split ',' foreach ($_ in $_psc.installed) { . $_.FullName } } psc_init if ($_psc.init -and (Test-Path($_psc.list_path))) { Write-Host (_psc_replace $_psc.json.init_info) -f DarkCyan } if ($_psc.config.update -eq 2) { Write-Host (_psc_replace $_psc.json.module_update @{'confirm' = $_psc.json.confirm }) -f Yellow $choice = $host.UI.RawUI.ReadKey("NoEcho, IncludeKeyDown") if ($choice.Character -eq 13) { Update-Module 'PSCompletions' } else { Write-Host (_psc_replace $_psc.json.module_cancel) -f Green } } if ($_psc.config.update -ne 0) { $add = _psc_get_content ($_psc.core + '\.add') if ($_psc.update -or $add) { Write-Host (_psc_replace $_psc.json.update_has ) -f Cyan if ($add) { Write-Host (_psc_replace $_psc.json.update_has1 @{'add_list' = $add } ) -f Cyan } if ($_psc.update) { Write-Host (_psc_replace $_psc.json.update_has2 @{'update_list' = $_psc.update } ) -f Cyan } Write-Host (_psc_replace $_psc.json.update_tip) -f Cyan } } New-Alias $_psc.config.root_cmd 'PSCompletions' $null = Start-Job -ScriptBlock { param( $_psc, $_psc_get_config ) $get_config = Get-Command $_psc_get_config -CommandType Function $_psc.config = &$get_config function set_config($key, $value) { $config = $_psc.config $config.$key = $value $res = $config.module_version + ';' + $config.root_cmd + ';' + $config.github + ';' + $config.gitee + ';' + $config.language + ';' + $config.update [environment]::SetEnvironmentvariable('abgox_PSCompletions', $res, 'User') } function get_content($path) { try { return (Get-Content $path -Encoding utf8).Trim() } catch { return "" } } function _do($do) { try { & $do }catch {} } _do { $response = Invoke-WebRequest -Uri ($_psc.url + '/core/.version') if ($response.StatusCode -eq 200) { $content = ($response.Content).Trim() if ($content -ne $_psc.version) { set_config 'update' 2 } } } echo (Compare-Object -ReferenceObject (get_content $_psc.list_path) -DifferenceObject (get_content $_psc.old_list_path) -PassThru) > ($_psc.core + '\.add') _do { $response = Invoke-WebRequest -Uri ($_psc.url + '/core/.list') if ($response.StatusCode -eq 200) { $content = ($response.Content).Trim() Move-Item $_psc.list_path $_psc.old_list_path -Force echo $content > $_psc.list_path } } $res = New-Object System.Collections.ArrayList $installed = (Get-ChildItem -Path $_psc.completions -Filter "*.ps1" -Recurse).BaseName foreach ($_ in $installed) { $url = $_psc.url + '/completions/' + $_ + '/.guid' $response = Invoke-WebRequest -Uri $url if ($response.StatusCode -eq 200) { $content = ($response.Content).Trim() $guid = (get_content ($_psc.completions + '\' + $_ + '\.guid')).Trim() if ($guid -ne $content) { $res.Add($_) > $null } } echo $res > $_psc.update_path } } -ArgumentList $_psc, '_psc_get_config' |