SCOM MP Class Visualizer 2.0

param(
    [string[]]$ClassNames,
    [switch]$Help
)

if (-not $ClassNames -and -not $help){
    Write-Error "Необходимо указать параметр $ClassNames"
    $Help = $true
}

if ($help){
    Write-Host "SCOM MP Class Visualizer script, version 0.2"
    Write-Host "http://gexeg.blogspot.com"
    Write-Host "Параметры: "
    Write-Host "   -Help                    : вывод справочной информации "
    Write-Host "   -ClassNames <[string[]]> : необходимо указать массив классов,"
    Write-Host "                              по которым будет строиться дерево классов`n"
    Write-Host "Примеры использования:"
    Write-Host "1) .\SCOMDrawClassTree.ps1 -ClassNames ""Microsoft.Windows.Server.2008.AD.DomainControllerRole"", ""Microsoft.Windows.Server.2008.AD.DC.KDC"""
    Write-Host "`n2) .\SCOMDrawClassTree.ps1 -ClassNames ""Microsoft.Windows.Server.2008.AD.DomainControllerRole"""
    Write-Host "`n3) `$classNames = Get-SCOMClass -ManagementPack (Get-ScomManagementPack -Name ""Microsoft.Windows.Server.AD.Library"") | foreach {`$_.Name}"
    Write-Host "   .\SCOMDrawClassTree.ps1 -ClassNames `$classNames"
    exit
}

if (-not (Get-Module OperationsManager)){
    Import-Module OperationsManager
}

[hashtable]$classes = @{}
[hashtable]$shapes = @{}

#Const VisSectionIndices
$visSectionObject = 1
$visSectionParagraph = 4
$visSectionProp = 243

#Const VisRowIndices
$visRowXFormOut = 1
$visRowTextXForm = 12
$visRowPageLayout = 24
$visRowLast = -2
$visRowFill = 3

#Const
$visXFormPinX = 0
$visXFormPinY = 1
$visXFormWidth = 2
$visPLOPlaceStyle = 8
$visPLORouteStyle = 9
$visPLOLineRouteExt = 29
$visCustPropsLabel = 2
$visCustPropsType = 5
$visCustPropsValue = 0
$visFillForegnd = 0
$visFillBkgnd = 1

$visTagDefault = 0
$visLayoutIncrSpace = 2
$visLayoutHorzAlignNone = 0
$visLayoutVertAlignNone = 0
$visMillimeters = 70


function VisioResizeShape($shape){
    $shape.CellsSRC($visSectionObject, $visRowXFormOut, 2).FormulaU = "10 mm"
    $shape.CellsSRC($visSectionObject, $visRowXFormOut, 3).FormulaU = "10 mm"
}

function VisioSetConStyle($con){
    $con.CellsSRC($visSectionObject,2,0).FormulaU = "0.25 pt"
    $con.CellsSRC($visSectionObject,23,19).FormulaU = "0"
}

function VisioSetText($shape, $text){
    $chars = $shape.Characters
    $chars.Begin = 0
    $chars.Begin = 0
    $chars.Text = $text


    $shape.CellsSRC($visSectionObject,11,4).FormulaU = "0"
    $shape.CellsSRC($visSectionParagraph,0,6).FormulaU = "0"

    $shape.CellsSRC($visSectionObject,$visRowTextXForm,$visXFormWidth).FormulaU = "5"
    $shape.CellsSRC($visSectionObject,$visRowTextXForm,$visXFormPinX).FormulaU = "3"
    #$shape.CellsSRC($visSectionObject,$visRowTextXForm,$visXFormPinY).FormulaU = "Height*0.5"

    #$shape.CellsSRC($visSectionObject,$visRowTextXForm, 4).FormulaU = "-1"
    #$shape.CellsSRC($visSectionObject,$visRowTextXForm, 5).FormulaU = "TxtHeight*2"

}

function VisioAddDataRow($shape, $Name, $Value){
    $row = $shape.AddRow($visSectionProp, $visRowLast, $visTagDefault)
    $shape.CellsSRC($visSectionProp, $row, $visCustPropsLabel).FormulaU = """$Name"""
    $shape.CellsSRC($visSectionProp, $row, $visCustPropsType).FormulaU = "0"
    $shape.CellsSRC($visSectionProp, $row, $visCustPropsValue).FormulaU = """$Value"""
}



function VisioRearIcons(){
    $page.PageSheet.CellsSRC($visSectionObject, $visRowPageLayout, $visPLOPlaceStyle).FormulaForceU = "7"
    $page.PageSheet.CellsSRC($visSectionObject, $visRowPageLayout, $visPLORouteStyle).FormulaForceU = "3"
    $page.Layout()
    $visio.ActiveWindow.SelectAll()
    $visio.ActiveWindow.Selection.LayoutIncremental($visLayoutIncrSpace,
        $visLayoutHorzAlignNone, $visLayoutVertAlignNone, 70, 8, $visMillimeters)
    $page.Shapes.ItemFromID(0).CellsSRC($visSectionObject, $visRowPageLayout, $visPLOLineRouteExt).FormulaU = "2"
}

function PopulateClassTree($ClassID){
    while ($ClassID){
        $obj = Get-SCOMClass -Id $ClassID
        if ($obj){
            $ClassID = $obj.Base.Id
            $obj | add-member -Name ParentID -MemberType NoteProperty -Value $ClassID
            $classes[$obj.Id] = $obj
            if (-not $ClassID){
                break
            }
        }
    }
}

function PopulateAttr($Shape, $ClassObj){
    VisioAddDataRow -shape $Shape -Name "::::::::Class Properties::::" -Value ":::::::::::::::"
    VisioAddDataRow -shape $Shape -Name "Abstract" -Value $classObj.Abstract
    VisioAddDataRow -shape $Shape -Name "Accessibility" -Value $classObj.Accessibility
    VisioAddDataRow -shape $Shape -Name "Extension" -Value $classObj.Extension
    VisioAddDataRow -shape $Shape -Name "Hosted" -Value $classObj.Hosted
    VisioAddDataRow -shape $Shape -Name "Singleton" -Value $classObj.Singleton
    VisioAddDataRow -shape $Shape -Name "Description" -Value $classObj.Description

    if ($ClassObj.PropertyCollection.Count -gt 0){
        VisioAddDataRow -shape $Shape -Name "::::::::Attributes::::" -Value ":::::::::::::::"
        $ClassObj.PropertyCollection | sort Name | foreach {
            if ($_.Key -eq $true) { $isKey = "Key" } else { $isKey=$null }
            if ($_.Required -eq $true) { $Required = "Mandatory" } else { $Required = "Optional"}
            $propVal = [string]::Join(",", @($_.SystemType, "Min: $($_.MinLength)", "Max: $($_.MaxLength)", $Required, $isKey))
            VisioAddDataRow -shape $Shape -Name $_.Name -Value $propVal
        }
    }
}

function PopulateBaseAttr($shape, $classObj){
    if ($classObj.ParentID){
        $attribs = @()

        while($classObj.ParentID){
            $classObj = $classes[$classObj.ParentID]
            if ($ClassObj.PropertyCollection.Count -gt 0){
                VisioAddDataRow -shape $Shape -Name ":::$($ClassObj.Name)::" -Value ":Attributes::::"
                $ClassObj.PropertyCollection | sort Name | foreach {
                if ($_.Key -eq $true) { $isKey = "Key" } else { $isKey=$null }
                if ($_.Required -eq $true) { $Required = "Mandatory" } else { $Required = "Optional"}
                $propVal = [string]::Join(",", @($_.SystemType, "Min: $($_.MinLength)", "Max: $($_.MaxLength)", $Required, $isKey))
                VisioAddDataRow -shape $Shape -Name $_.Name -Value $propVal
                }
            }
        }
    }
}

$initVisio = {
    Write-Host "Создание приложения Visio"
    $visio = New-Object -ComObject Visio.Application
    $visio.Visible = $false
    $visio.Documents.Add("BASICD_M.VST") | Out-Null
    $doc = $visio.Documents.Item(1)
    $page = $doc.Pages.Item(1)
    $Stensils = $visio.Documents.Item("BASIC_M.VSS")
    $pentagon = $Stensils.Masters.ItemU("Pentagon")
    $dyncon = $Stensils.Masters.ItemU("Dynamic connector")
}

$unInitVisio = {
    $visio.Visible = $true
    Write-Host "Переключитесь в Visio и сохраните сформированный документ."
    $visio.quit()
    $page = $null
    $doc = $null
    $visio = $null
}

. $initVisio

foreach ($className in $ClassNames){
    Write-Host "Обрабатываем класс $className"
    $classObj = Get-SCOMClass -Name $className
    Write-Host "Заполняем дерево классов"
    PopulateClassTree -ClassID $classObj.ID
}

Write-Host "Отрисовка объектов"
$classes.Keys | foreach {
    $shape = $page.Drop($pentagon, 1,1)
    $shape.Name = $classes[$_].Name

    if ($classes[$_].Singleton -eq $true){
        $shape.CellsSRC($visSectionObject, $visRowFill, $visFillBkgnd).FormulaU = "THEMEGUARD(MSOTINT(THEME(""LineColor""),80))"
    }
    elseif ($classes[$_].Abstract -eq $false){
        $shape.CellsSRC($visSectionObject, $visRowFill, $visFillBkgnd).FormulaU = "THEMEGUARD(MSOTINT(THEME(""AccentColor4""),40))"
    }

    VisioResizeShape -shape $shape
    VisioSetText -shape $shape -text ("$($classes[$_].Name)`n$($classes[$_].DisplayName)`n$($classes[$_].ManagementPackName)")
    PopulateAttr -shape $shape -ClassObj $classes[$_]
    PopulateBaseAttr -shape $shape -ClassObj $classes[$_]
    $shapes[$_] = $shape
}

foreach ($class in $classes.Keys){
    if (-not $classes[$class].ParentID){
        continue
    }

    $con = $page.Drop($dyncon, 1,1)
    $shapeTo = $shapes[$class]
    $shapeFrom = $shapes[$classes[$class].ParentID]
    $con.CellsU("BeginX").GlueTo($shapeFrom.CellsSRC(7,0,0))
    $con.CellsU("EndX").GlueTo($shapeTo.CellsSRC(7,5,0))
    VisioSetConStyle -con $con
}

VisioRearIcons
Write-Host "Отрисовка объектов закончена"
. $unInitVisio