base.ps1
|
$script:FountEncoding = [System.Text.Encoding]::UTF8 $script:FountClient = $null $script:FountStream = $null function Set-FountClient($ComputerName = "localhost", $Port = 16698, $timeout = 100) { if ($script:FountClient) { if ($script:FountClient.Connected) { return $script:FountClient } else { Close-FountClient } } # 创建新的 TCP 客户端 $Client = New-Object System.Net.Sockets.TcpClient if ($Client.ConnectAsync($ComputerName, $Port).Wait($timeout)) { if ($Client.Connected) { Write-Verbose "成功连接到 Fount 服务器。" $script:FountClient = $Client return } } Write-Error "无法连接到 Fount 服务器" -ErrorAction Stop } function Close-FountClient { if ($script:FountStream) { $script:FountStream.Dispose() $script:FountStream = $null } if ($script:FountClient) { if ($script:FountClient.Connected) { $script:FountClient.Close() } $script:FountClient.Dispose() $script:FountClient = $null } } function Invoke-FountIPC( [string]$Type, [hashtable]$Data, [string]$ComputerName = "localhost", [int]$Port = 16698 ) { # 构建要发送的完整 JSON 对象 $CommandObject = @{ type = $Type data = $Data } $JsonCommand = $CommandObject | ConvertTo-Json -Compress -Depth 100 $JsonCommand += "`n" # 添加换行符作为消息结束符 try { # 获取或建立连接 if (-not $script:FountClient) { Set-FountClient -ComputerName $ComputerName -Port $Port if (-not $script:FountClient) { Write-Error "无法连接到 Fount 服务器" -ErrorAction Stop } } if (-not $script:FountClient.Connected) { $script:FountClient.Connect($ComputerName, $Port) if ($script:FountStream) { $script:FountStream.Dispose() } $script:FountStream = $null } if (-not $script:FountStream) { $script:FountStream = $script:FountClient.GetStream() } # 获取网络流 $Stream = $script:FountStream $Encoding = $script:FountEncoding # 发送数据 $Bytes = $Encoding.GetBytes($JsonCommand) $Stream.Write($Bytes, 0, $Bytes.Length) $Stream.Flush() # 读取响应（支持大响应和分块读取） $MemoryStream = New-Object System.IO.MemoryStream $Buffer = New-Object byte[] 4096 $FoundTerminator = $false do { $Read = $Stream.Read($Buffer, 0, $Buffer.Length) if ($Read -gt 0) { $MemoryStream.Write($Buffer, 0, $Read) # 检查是否包含换行符 $CurrentData = $Encoding.GetString($MemoryStream.ToArray()) if ($CurrentData -match "`n") { $FoundTerminator = $true break } } } while ($Read -gt 0 -and -not $FoundTerminator) # 提取有效响应 $ResponseString = $Encoding.GetString($MemoryStream.ToArray()) # 解析 JSON 响应 if (-not [string]::IsNullOrEmpty($ResponseString)) { $result = ConvertFrom-Json -InputObject $ResponseString if ($result.status -ne 'ok') { Write-Error "与 Fount 服务器通信失败: $($result.message)" -ErrorAction Stop } else { return $result.data } } else { Write-Error "收到空响应" -ErrorAction Stop } } finally { if ($MemoryStream) { $MemoryStream.Dispose() } } } function Start-FountShell { param( [Parameter(Mandatory)] [string]$UserName, [Parameter(Mandatory)] [string]$ShellName, [Parameter(ValueFromRemainingArguments)] [array]$Arguments ) Invoke-FountIPC -Type "runshell" -Data @{ username = $UserName shellname = $ShellName args = $Arguments } } function Invoke-FountShell { param ( [Parameter(Mandatory)] [string]$UserName, [Parameter(Mandatory)] [string]$ShellName, $Data ) Invoke-FountIPC -Type "invokeshell" -Data @{ username = $UserName shellname = $ShellName data = $Data } } function Test-FountRunning { try { Invoke-FountIPC -Type "ping" -ErrorAction Stop | Out-Null $true } catch { $false } } function Install-Fount { $scriptContent = Invoke-RestMethod https://raw.githubusercontent.com/steve02081504/fount/refs/heads/master/src/runner/main.ps1 Invoke-Expression "function fountInstaller { $scriptContent }" fountInstaller @args } function Start-Fount( [switch] $NoAutoInstall ) { if (Get-Command fount -ErrorAction SilentlyContinue) { fount @args } elseif (!$NoAutoInstall) { Install-Fount @args } else { Write-Error "Fount 未安装" } } function Stop-Fount { Invoke-FountIPC -Type "shutdown" Close-FountClient } |