$ErrorActionPreference = 'Stop'
 
$ipText = @"
10.10.103.31
10.10.103.61
"@
 
[string[]]$ipArry = $ipText.Split([char[]]"`r`n") | % { $_.Trim() } | ? { $_.Length -gt 0 } 
# $ips = $ipTxt.Split([char[]]"`r`n") | % { $_.Trim() } | ? { $_.Length -gt 0 } 
 
Write-Output "共有待处理 ip 数： $($ipArry.Length)"
 
Import-Module powershell-yaml
 
class YamlParser {
 
    $yamlObject = $null
 
    [void] load($filePath) {
 
        $yamlContent = Get-Content -Path $filePath -Raw
        $this.yamlObject = ConvertFrom-Yaml -Yaml $yamlContent
    }
 
    # 根据组名找出该组下的所有节点名称
    [string[]] findProxyNamesByGroupName([string] $groupName) {
 
        [array]$groups = $this.yamlObject['proxy-groups']
 
        for ($idx = 0; $idx -lt $groups.Count; $idx++) {
 
            $group = $groups[$idx]
            # $name = $group['name']
 
            if ($group['name'] -eq $groupName) {
 
                $proxies = $group.proxies
                return $proxies
            }            
        }
 
        return @()        
    }
 
    [string] parseGroupName([string] $rule) {
 
        # SRC-IP-CIDR,10.10.103.36/32,Relay-us-int-home-8
        # 使用 , 进行拆分，然后取最后一个值
 
        if ([string]::IsNullOrWhiteSpace($rule)) {
            return $null
        }
 
        $ruleArray = $rule.Split(',')
        $groupName = $ruleArray[-1]
 
        return $groupName
    }
 
    [string] findRuleByIp([string] $ip) {
 
        # 初始化结果变量
        $result = $null
 
        # 遍历 rules 部分
        foreach ($rule in $this.yamlObject.rules) {
            # 检查规则是否包含目标 IP 地址
            if ($rule -match $ip) {
                $result = $rule
                break
            }
        }
 
        # 输出结果
        if ($result) {
            Write-Output "找到匹配的规则: $result"
        }
        else {
            Write-Output "未找到匹配的规则。"
        }
 
        return $result    
    }
}
 
$yamlFilePath = 'config.yml'
 
$parser = [YamlParser]::new()
$parser.load($yamlFilePath)
 
foreach ($ip in $ipArry) {
 
    $rule = $parser.findRuleByIp($ip)
    $group = $parser.parseGroupName($rule)
    $proxies = $parser.findProxyNamesByGroupName($group)
    $proxies = $proxies | ? { $_ -ne 'Gateway' }
    $proxyNems = $proxies -join ','
 
    # Write-Output "$ip 找到规则 $rule, 组名 $group, 节点名 $proxyNems"
 
    Write-Output "$ip  $group $($proxyNems)"
}