♻️ Update completion script for new complete structure

Author: gluons

src/completion.ps1

@@ -1,28 +1,48 @@
-@('yarn', 'yarn.cmd') | ForEach-Object {
-	Register-ArgumentCompleter -Native -CommandName $_ -ScriptBlock {
-		param($wordToComplete, $commandAst, $cursorPosition)
+Register-ArgumentCompleter -Native -CommandName @('yarn', 'yarn.cmd') -ScriptBlock {
+	param($wordToComplete, $commandAst, $cursorPosition)
 
-		$subCommand = $commandAst.CommandElements[1].Value
+	. $PSScriptRoot\commands.ps1
+	. $PSScriptRoot\utils.ps1
 
-		switch ($subCommand) {
-			'global' {
-				. $PSScriptRoot\sub-commands\global.ps1
-			}
-			Default {
-				. $PSScriptRoot\commands.ps1
+	$searchBlock = { $_ -like "$wordToComplete*" }
+	$completions = @()
+	$secondPart = if ($commandAst.CommandElements[1]) { $commandAst.CommandElements[1].Value } else { $null } # Main command
+	$thirdPart = if ($commandAst.CommandElements[2]) { $commandAst.CommandElements[2].Value } else { $null } # Sub-command
+	$fourthPart = if ($commandAst.CommandElements[3]) { $commandAst.CommandElements[3].Value } else { $null } # Sub-command's option
 
-				# Don't complete any word after sub-command when it isn't one of above commands.
-				if ($subCommand -in $cmds) {
-					$cmds = @()
-				}
-			}
+	# If word to complete is equal to main command, suggest all commands & options of `install` command
+	if (AreEqual $secondPart $wordToComplete) {
+		$completions += $commands | Where-Object $searchBlock | ForEach-Object {
+			[System.Management.Automation.CompletionResult]::new($_, $_, 'Command', $_)
 		}
+		$completions += $options['install'] | Where-Object $searchBlock | ForEach-Object {
+			[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterName', $_)
+		}
+	}
+	# If word to complete is equal to sub-command, suggest sub-commmands
+	if ((-not $(AreEqual $secondPart $wordToComplete)) -and $(AreEqual $thirdPart $wordToComplete)) {
+		$completions += $subCommands[$secondPart].Keys | Where-Object $searchBlock | ForEach-Object {
+			[System.Management.Automation.CompletionResult]::new($_, $_, 'Command', $_)
+		}
+	}
+	# If word to complete is equal to sub-command's options, suggest sub-command's options
+	if (
+		(-not $(AreEqual $secondPart $wordToComplete)) -and
+		(-not $(AreEqual $thirdPart $wordToComplete)) -and
+		$(AreEqual $fourthPart $wordToComplete)
+	) {
+		$subCommand = $subCommands[$secondPart][$thirdPart]
+		$subCommandOptions = if ($subCommand.options) { $subCommand.options } else { $null }
 
-		$cmds |
-			Where-Object { $_ -like "$wordToComplete*" } |
-			Sort-Object |
-			ForEach-Object {
-				[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
-			}
+		$completions += $subCommandOptions | Where-Object $searchBlock | ForEach-Object {
+			[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterName', $_)
+		}
+	}
+
+	# Always suggest global options
+	$completions += $globalOptions | Where-Object $searchBlock | ForEach-Object {
+		[System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterName', $_)
 	}
+
+	return $completions
 }

src/utils.ps1

@@ -0,0 +1,11 @@
+function AreEqual($a, $b) {
+	try {
+		return (
+			([string]::IsNullOrEmpty($a) -and [string]::IsNullOrEmpty($b)) -or
+			($a -eq $b)
+		)
+	}
+	catch {
+		return $a -eq $b
+	}
+}