Synchronizing Quotas - Part 2/3: Quotas-Acquire.ps1

Note: Did this in a bit of a rush...

########################

## Quotas-Acquire.ps1 ##

########################

Param($User,$Pass,$Cluster,$SVM)

Import-Module DataONTAP

"";"<<<<< Quotas-Acquire.ps1 >>>>>";""

"The script is designed to store Quota Policy Rule information in special hidden CIFS shares and their comments field.";""

If(!$User) {$User = Read-Host ">>> Enter Admin User "}

If(!$Pass) {$Pass = Read-Host ">>> Enter Password " -AsSecureString}

else {$Pass = $Pass | ConvertTo-SecureString -AsPlainText -Force}

$Cred = New-Object System.Management.Automation.PsCredential($User,$Pass)

If(!$Cluster){$Cluster = Read-Host ">>> Enter Source Cluster "}

If(!$SVM) {$SVM = Read-Host ">>> Enter Source SVM "}

"";"STEP 1: CONNECTING TO SOURCE CLUSTER $Cluster AND FOCUSING ON SVM $SVM <<<"

[Void](Connect-NcController -Name $Cluster -Credential $Cred)

$Global:CurrentNcController.Vserver = $SVM

"";"STEP 2: GET THE APPLIED QUOTA POLICY"

$SVM_Attrs = Get-NcVserver -Template

$SVM_Attrs.QuotaPolicy = ""

$Quota_Pol = (Get-NcVserver -Attributes $SVM_Attrs).QuotaPolicy

"";"STEP 3: GET THE LIST OF VOLUMES WITH QUOTA ENABLED (ON)"

$QUOTA_Status_Attrs = Get-NcQuotaStatus -Template

$QUOTA_Status_Attrs.Volume = ""

$QUOTA_Status_Query = Get-NcQuotaStatus -Template

$QUOTA_Status_Query.Status = "on"

$QUOTA_On_Vols = (Get-NcQuotaStatus -Attributes $QUOTA_Status_Attrs -Query $QUOTA_Status_Query).Volume

$QUOTA_Vols = @{}

$QUOTA_On_Vols | foreach { $QUOTA_Vols.$_ = $true }

"";"STEP 4: GET THE VOLUME SIZES (for Tracking Quotas)"

$VOL_Attrs = Get-NcVol -Template

Initialize-NcObjectProperty -object $VOL_Attrs -name VolumeSpaceAttributes

$VOL_Attrs.VolumeSpaceAttributes.Size = ""

$VOL_Sizes = Get-NcVol -Attributes $VOL_Attrs -Vserver $SVM

$VOL_Size = @{}

$VOL_Sizes | foreach { $VOL_Size.($_.Name) = $_.TotalSize }

"";"STEP 5: GET THE QUOTA POLICY RULES PER VOLUME PER QTREE (Belonging to the Active Policy)"

$QUOTA_Query = Get-NcQuota -Template

$QUOTA_Query.Policy = $Quota_Pol

$QUOTA_Query.Vserver = $SVM

$QUOTA_Rules = Get-NcQuota -Query $Quota_Query

$VOL_Attrs = Get-NcVol -Template

$VOL_List = (Get-NcVol -Attributes $VOL_Attrs -Vserver $SVM).Name

$VOL_Hash = @{} # HASH gives true for .VolName is VolName exists!

$QTREE_Attrs = Get-NcQtree -Template

$QTREE_Attrs.Qtree = ""

$QTREE_V_Q = @{} # HASH gives true for .VolName.Qtree if Qtree exists!

$QUOTA_V_Q = @{} # HASH gives Disk Limit for .VolName.Qtree if it exists or null!

Foreach ($VolName in $VOL_List) {

$VOL_Hash.$VolName = $true

$QTREE_V_Q.$VolName = @{}

$QUOTA_V_Q.$VolName = @{}

$QTREE_List = (Get-NcQtree -Attributes $QTREE_Attrs -Volume $VolName -Vserver $SVM).Qtree

Foreach ($QtreeName in $QTREE_List){

If ($QtreeName){

$QTREE_V_Q.$VolName.$QtreeName = $true

$QUOTA_V_Q.$VolName.$QtreeName = $null

}

Foreach ($QUOTA_Rule in $QUOTA_Rules){

$VolName = $QUOTA_Rule.Volume

$QtreeName = (($QUOTA_Rule.QuotaTarget).Split("/"))[3]

# Get-NcQuota returns the Qtree in the QuotaTarget fields as /vol/VOLNAME/QTREENAME, even if the volume is mounted to /. So the split is always on 3 as in 0/1/2/3. It does not return the Qtree in .Qtree! #

$DiskLimit = $QUOTA_Rule.DiskLimit

If ($QtreeName){ $QUOTA_V_Q.$VolName.$QtreeName = $DiskLimit }

# A tracking quota with no defined DiskLimit gives $DiskLimit -eq "-" #

}

"";"STEP 6: CHECK EXISTING QUOTA.RULE.* SHARES"

$Shares = Get-NcCifsShare -ShareName "QUOTA.RULE.*"

$QR_Shares_V = @{} # HASH records if volumes already have a share

$QR_Shares_V_Q = @{} # HASH records if qtrees already have a share

$VOL_List | foreach { $QR_Shares_V_Q.$_ = @{} }

Foreach ($Share in $Shares){

$ShareName = $Share.ShareName

$ShareNameSplit = $ShareName.Split(".")

$SplitCount = $ShareNameSplit.Count

$VolName = $ShareNameSplit[2]

$QtreeName = $ShareNameSplit[3]

If (!$ShareName.EndsWith(".$")){

Write-Host "Invalid Quota Rule share - $ShareName (does not end with .$) - removing!" -ForegroundColor Red