functions/DockerCompose.psm1
|
Function Get-DockerComposeCommands { param( [string]$Command ) $help = $(if ($Command) { docker-compose $Command --help } else { docker-compose --help }) | Select-String -Pattern "^\s{2}\w+" | Where-Object { $_ -notlike "*docker-compose*" } $cmds = @() for ($i = 0; $i -lt $help.Count; $i++) { $cmdline = $help[$i].Line.Trim() $cmds += $cmdline.Substring(0, $cmdline.IndexOf(" ")) } return $cmds } Function Get-DockerComposeServices { param ( [Parameter(Mandatory = $true)] [string]$Path ) if (-not (Test-Path $Path)) { throw } $file = Get-Content (Convert-Path $Path) $categories = $file | Select-String -Pattern "^\w+:" $service = $categories | Where-Object {$_.Line -like "services:*"} $serviceIdx = $categories.IndexOf($service) $startLine = ($categories[$serviceIdx] | Select-Object -ExpandProperty LineNumber) $endLine = $(if ($serviceIdx -lt $categories.Length - 1) {$categories[$serviceIdx+1] | Select-Object -ExpandProperty LineNumber} else {$file.Count}) return $file | Select-Object -Skip $startLine -First ($endLine - $startLine) | Select-String -Pattern "^\s{2}\w+:" | Select-Object @{Name="Service";Expression={$_.Line.Trim().Replace(":","")}} | Sort-Object -Property Service } Function Invoke-DockerCompose { [Alias("dc")] Param ( [Parameter(Mandatory = $true, Position = 0)] [string]$Command, [Parameter(Mandatory = $false, ValueFromRemainingArguments = $true)] [string[]]$Parameters ) $params = @() $params += $Parameters switch ($Command) { "b" { docker-compose build @params } "c" { docker-compose create @params } "d" { docker-compose down @params } "l" { docker-compose logs @params } "s" { docker-compose start @params } "u" { docker-compose up @params } "ud" { docker-compose up --detach @params } "x" { docker-compose stop @params } "rmi" { docker-compose images } default { docker-compose $Command @params } } } Register-ArgumentCompleter -CommandName Invoke-DockerCompose -ParameterName Command -ScriptBlock { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $cmds = Get-DockerComposeCommands return $(if ($wordToComplete) { $cmds | Where-Object { $_ -like "$wordToComplete*" } } else { $cmds }) } Register-ArgumentCompleter -CommandName Invoke-DockerCompose -ParameterName Parameters -ScriptBlock { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameters) $fIdx = $commandAst.CommandElements | Where-Object {$_.Value -eq "-f" -or $_.Value -eq "--file"} $file = $commandAst[$fIdx+1].Value $__dirname = Convert-Path .\ if ($null -eq $file) { $file = [System.IO.Path]::Combine($__dirname, "docker-compose.yml") if (-not (Test-Path $file)) { $file = [System.IO.Path]::Combine($__dirname, "docker-compose.yaml") } } $services = Get-DockerComposeServices -Path $file | Select-Object -ExpandProperty Service if ($wordToComplete) { return $services | Where-Object { $_ -like "$wordToComplete*" } } else { return $services | Where-Object {-not ($commandAst.CommandElements | Select-Object -ExpandProperty Value).Contains($_)} } } |