# Author:  Lucas Arnström
# Contact: Lucke92@gmail.com

Write-Host
Write-Host "DiskQuota 1.4 - Alpha | Designed for Office IT-Partner @ Gavle, Sweden"
Write-Host

# Lägg in användarna i arrayen users
function Compile-Users {
    Process {
        for($i = 0 ; $i -le $arrComputers.Length ; $i++) {
            [array]$users[$i] = $_.User
        }
    }
}

# Skapa ett objekt och lägg in domän\dator och diskutrymmet för att sedan skicka vidare det i pipelinen
# fortsätt utveckla for-loopen så den sätter ihop redan existerande användare med den nya för att undgå kopior
function Compute {
    Process {
        for($i = 0 ; $i -le $arrComputers.Length ; $i++) {
            if($_.User -eq $users[$i]) {
                # Här ska en ihopkoppling av användare och deras diskanvändning sammankopplas och beräknas
            }
        }

        $obj = New-Object PSObject
        $obj | Add-Member NoteProperty "Domain\Computer" ($_.User -replace ('.+n="(.+)".+e="(.+)"','$1\$2'))
        $obj | Add-Member NoteProperty "DiskspaceUsed(MB)" ([Math]::Round(($_.DiskSpaceUsed / 1048576), 1))

        Write-Output $obj   # Skickar vidare objektet i pipelinen
    }
}

$strComputer = Read-Host @"
Please enter computername.
It could also be a domainname, a NetBIOS name, or IP-adress.
Only place a dot if you want to check localhost.
COMPUTERNAME
"@

Write-Host
Write-Host "Please wait..."

$arrComputers = $strComputer.Split(" ")                # Delar upp inmatningarna till en array ifall flera datorer / servrar har angetts

foreach($computer in $arrComputers) {
    $colItems += get-wmiobject -class "Win32_DiskQuota" -namespace "root\CIMV2" `
    -computername $computer
}

$colItems | Sort-Object -Property User | Compile-Users | Compute | Format-Table * -auto

Read-Host "Press any key to continue...."