<#
    .SYNOPSIS
    Initialize-Instance.ps1

    .DESCRIPTION
    This script downloads and installs the required PowerShell modules to create and configure Active Directory Certificate Authorities. 
    It also creates a self signed certificate to be used with PowerShell DSC.
    
    .EXAMPLE
    .\Initialize-Instance
#>

[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12

#==================================================
# Variables
#==================================================

$Modules = @(
    @{
        Name = 'NetworkingDsc'
        Version = '8.2.0'
    },
    @{
        Name = 'ComputerManagementDsc'
        Version = '8.4.0'
    }
)

#==================================================
# Main
#==================================================

Write-Output 'Installing NuGet Package Provider'
Try {
    $Null = Install-PackageProvider -Name 'NuGet' -MinimumVersion '2.8.5' -Force -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed to install NuGet Package Provider $_"
    Exit 1
}

Write-Output 'Setting PSGallery Respository to trusted'
Try {
    Set-PSRepository -Name 'PSGallery' -InstallationPolicy 'Trusted' -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed to set PSGallery Respository to trusted $_"
    Exit 1
}

Write-Output 'Installing the needed Powershell DSC modules for this Quick Start'
Foreach ($Module in $Modules) {
    Try {
        Install-Module -Name $Module.Name -RequiredVersion $Module.Version -ErrorAction Stop
    } Catch [System.Exception] {
        Write-Output "Failed to Import Modules $_"
        Exit 1
    }
}

Write-Output 'Temporarily disabling Windows Firewall'
Try {
    Get-NetFirewallProfile -ErrorAction Stop | Set-NetFirewallProfile -Enabled False -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed to disable Windows Firewall $_"
    Exit 1
}

Write-Output 'Creating Directory for DSC Public Cert'
Try {
    $Null = New-Item -Path 'C:\AWSQuickstart\publickeys' -ItemType 'Directory' -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed to create publickeys directory $_"
    Exit 1
}

Write-Output 'Creating DSC Certificate to Encrypt Credentials in MOF File'
Try {
    $cert = New-SelfSignedCertificate -Type 'DocumentEncryptionCertLegacyCsp' -DnsName 'AWSQSDscEncryptCert' -HashAlgorithm 'SHA256' -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed to create self signed cert $_"
    Exit 1
}

Write-Output 'Exporting the public key certificate'
Try {
    $Null = $cert | Export-Certificate -FilePath 'C:\AWSQuickstart\publickeys\AWSQSDscPublicKey.cer' -Force -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed to copy self signed cert to publickeys directory $_"
    Exit 1
}

Write-Output 'Finding RAW Disk'
$Counter = 0
Do {
    Try {
        $BlankDisk = Get-Disk -ErrorAction Stop | Where-Object { $_.partitionstyle -eq 'raw' }
    } Catch [System.Exception] {
        Write-Output "Failed to get disk $_"
        $BlankDisk = $Null
    }
    If (-not $BlankDisk) {
        $Counter ++
        Write-Output 'RAW Disk not found sleeping 10 seconds and will try again.'
        Start-Sleep -Seconds 10
    }
} Until ($BlankDisk -or $Counter -eq 12)

If ($Counter -ge 12) {
    Write-Output 'RAW Disk not found sleeping exitiing'
    Exit 1
}

Write-Output 'Data Volume not initialized attempting to bring online'
Try{
    Initialize-Disk -Number $BlankDisk.Number -PartitionStyle 'GPT' -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed attempting to bring online Data Volume $_"
    Exit 1
}

Start-Sleep -Seconds 5

Write-Output 'Data Volume creating new partition'
Try {
    $Null = New-Partition -DiskNumber $BlankDisk.Number -DriveLetter 'D' -UseMaximumSize -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed creating new partition $_"
    Exit 1
}

Start-Sleep -Seconds 5

Write-Output 'Data Volume formatting partition'
Try {
    $Null = Format-Volume -DriveLetter 'D' -FileSystem 'NTFS' -NewFileSystemLabel 'Data' -Confirm:$false -Force -ErrorAction Stop
} Catch [System.Exception] {
    Write-Output "Failed formatting partition $_"
    Exit 1
}