#Requires -Version 5.1

<#
    .NAME
    Deployment-All-DataCollections

    .SYNOPSIS
    This script will create sample Data Collection Rules and Data Collection Endpoint to collect:
        (1)   VMinsight - Linux & Windows
        (2)   Event-logs
        (3)   IIS-logs
        (4)   Syslog
  
    .AUTHOR
    Morten Knudsen, Microsoft MVP - https://mortenknudsen.net

    .LICENSE
    Licensed under the MIT license.

    .PROJECTURI
    https://github.com/KnudsenMorten/AzLogLibrary

    .WARRANTY
    Use at your own risk, no warranty given!
#>


#######################################################################
# Variables
#######################################################################

    $TenantId                  = ""
    $SubscriptionId            = ""

    $WorkspaceResourceId       = ""
    $WorkspaceLocation         = "westeurope"

    $ResourceGroupDeployment   = "rg-azmon-datacollectionrules-p"
    $ResourceGroupDCR          = "rg-azmon-datacollectionrules-p"

    $ResourceGroupDCE          = "rg-azmon-datacollectionendpoints-p"


#######################################################################
# FUNCTIONS
#######################################################################

# Az-module
    $ModuleCheck = Get-Module -Name Az.* -ListAvailable -ErrorAction SilentlyContinue
    If (!($ModuleCheck))
        {
            Write-Output "Powershell module Az was not found !"
            Write-Output "Installing latest version from PsGallery in scope AllUsers (takes up to 30 min) .... Please Wait !"

            Install-module -Name Az -Repository PSGallery -Force -Scope AllUsers
        }


# Az-module
    $ModuleCheck = Get-Module -Name Az.ResourceGraph -ListAvailable -ErrorAction SilentlyContinue
    If (!($ModuleCheck))
        {
            Write-Output "Powershell module Az.ResourceGraph was not found !"
            Write-Output "Installing latest version from PsGallery in scope AllUsers .... Please Wait !"

            Install-module -Name Az.ResourceGraph -Repository PSGallery -Force -Scope AllUsers
        }

#######################################################################
# Connectivitity to Azure
#######################################################################
    Disconnect-AzAccount -WarningAction SilentlyContinue
    Connect-AzAccount -tenant $TenantId -WarningAction SilentlyContinue

    $Context = Set-AzContext -Subscription $SubscriptionId -tenant $TenantId -WarningAction SilentlyContinue

#######################################################################
# Resource Group for DCRs
#######################################################################

    # DCR
        Write-Output ""
        Write-Output "Validating Azure resource group exist [ $($ResourceGroupDCR) ]"
        try {
            Get-AzResourceGroup -Name $ResourceGroupDCR -ErrorAction Stop
        } catch {
            Write-Output ""
            Write-Output "Creating Azure resource group [ $($ResourceGroupDCR) ]"
            New-AzResourceGroup -Name $ResourceGroupDCR -Location $WorkspaceLocation
        }

    # Deployment
        Write-Output ""
        Write-Output "Validating Azure resource group exist [ $($ResourceGroupDeployment) ]"
        try {
            Get-AzResourceGroup -Name $ResourceGroupDeployment -ErrorAction Stop
        } catch {
            Write-Output ""
            Write-Output "Creating Azure resource group [ $($ResourceGroupDeployment) ]"
            New-AzResourceGroup -Name $ResourceGroupDeployment -Location $WorkspaceLocation
        }

    # DCE
        Write-Output ""
        Write-Output "Validating Azure resource group exist [ $($ResourceGroupDCE) ]"
        try {
            Get-AzResourceGroup -Name $ResourceGroupDCE -ErrorAction Stop
        } catch {
            Write-Output ""
            Write-Output "Creating Azure resource group [ $($ResourceGroupDCE) ]"
            New-AzResourceGroup -Name $ResourceGroupDCE -Location $WorkspaceLocation
        }


#######################################################################
# dcr-windows-vmhealth-performance_servicemap
#######################################################################

    $DcrName                    = "dcr-windows-vmhealth-performance_servicemap"

    $parameters = @{
        'Name'                  = $DcrName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDeployment
        'TemplateFile'          = ".\DataCollection\vminsight\" + $DcrName + ".json"
        'DcrName'               = $DcrName
        'DcrResourceGroup'      = $ResourceGroupDCR
        'WorkspaceLocation'     = $WorkspaceLocation
        'WorkspaceResourceId'   = $WorkspaceResourceId
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters


#######################################################################
# dcr-linux-vmhealth-performance_servicemap
#######################################################################

    $DcrName                    = "dcr-linux-vmhealth-performance_servicemap"

    $parameters = @{
        'Name'                  = $DcrName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDeployment
        'TemplateFile'          = ".\datacollection\vminsight\" + $DcrName + ".json"
        'DcrName'               = $DcrName
        'DcrResourceGroup'      = $ResourceGroupDCR
        'WorkspaceLocation'     = $WorkspaceLocation
        'WorkspaceResourceId'   = $WorkspaceResourceId
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters


#######################################################################
# dcr-linux-vmhealth-performance_basic
#######################################################################

    $DcrName                    = "dcr-linux-vmhealth-performance_basic"

    $parameters = @{
        'Name'                  = $DcrName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDeployment
        'TemplateFile'          = ".\datacollection\vminsight\" + $DcrName + ".json"
        'DcrName'               = $DcrName
        'DcrResourceGroup'      = $ResourceGroupDCR
        'WorkspaceLocation'     = $WorkspaceLocation
        'WorkspaceResourceId'   = $WorkspaceResourceId
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters

#######################################################################
# dcr-windows-vmhealth-events-system_application
#######################################################################

    $DcrName                    = "dcr-windows-vmhealth-events-system_application"

    $parameters = @{
        'Name'                  = $DcrName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDeployment
        'TemplateFile'          = ".\datacollection\event-logs\" + $DcrName + ".json"
        'DcrName'               = $DcrName
        'DcrResourceGroup'      = $ResourceGroupDCR
        'WorkspaceLocation'     = $WorkspaceLocation
        'WorkspaceResourceId'   = $WorkspaceResourceId
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters


#######################################################################
# dcr-windows-vmhealth-events-security
#######################################################################

    $DcrName                    = "dcr-windows-vmhealth-events-security"

    $parameters = @{
        'Name'                  = $DcrName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDeployment
        'TemplateFile'          = ".\datacollection\event-logs\" + $DcrName + ".json"
        'DcrName'               = $DcrName
        'DcrResourceGroup'      = $ResourceGroupDCR
        'WorkspaceLocation'     = $WorkspaceLocation
        'WorkspaceResourceId'   = $WorkspaceResourceId
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters

#######################################################################
# dce-iis-logs
#######################################################################

    $DceName                    = "dce-iis-logs"

    $parameters = @{
        'Name'                  = $DceName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDCE
        'TemplateFile'          = ".\datacollection\iis-logs\" + $DceName + ".json"
        'DceName'               = $DceName
        'DceResourceGroup'      = $ResourceGroupDCE
        'DceLocation'           = $WorkspaceLocation
        'PublicNetworkAccess'   = "Enabled"
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters


#######################################################################
# dcr-windows-vmhealth-iis-logs-W3SVCx (1-50)
#######################################################################
    
    # Wait 30 sec to let Azure sync up - we are going to retrieve from Azure Resource Graph
    Start-Sleep -s 30
    
    #-------------------------------
    # Getting DCE Resource Id
    #-------------------------------

        # Get DCEs from Azure Resource Graph
            $DCE_All = @()
            $pageSize = 1000
            $iteration = 0
            $searchParams = @{
                                Query = "Resources `
                                        | where type =~ 'microsoft.insights/datacollectionendpoints' "
                                First = $pageSize
                                }

            $results = do {
                $iteration += 1
                $pageResults = Search-AzGraph -UseTenantScope @searchParams
                $searchParams.Skip += $pageResults.Count
                $DCE_All += $pageResults
            } while ($pageResults.Count -eq $pageSize)

            # Error handling
                $DceInfo = $DCE_All | Where-Object { $_.name -eq $DceName }
                    If (!($DceInfo))
                        {
                            Write-Error "Could not find DCE with name [ $($DceName) ]"
                        }

        # DCE ResourceId (target for DCR ingestion)
        $DceResourceId  = $DceInfo.id
        $DceResourceId

    #-------------------------------
    # Deployment loop (1-15)
    #-------------------------------

    # initial
    $Number                             = 0

    Do
        {
            $Number                    += 1
            $DcrName                    = "dcr-windows-vmhealth-iis-logs-W3SVC" + $Number

            # deployment parameters
            $parameters = @{
                'Name'                  = $DcrName + (Get-Random -Maximum 100000)
                'ResourceGroup'         = $ResourceGroupDeployment
                'TemplateFile'          = ".\datacollection\iis-logs\" + "dcr-windows-vmhealth-iis-logs-W3SVCx" + ".json"
                'DcrName'               = $DcrName
                'DcrResourceGroup'      = $ResourceGroupDCR
                'WorkspaceLocation'     = $WorkspaceLocation
                'WorkspaceResourceId'   = $WorkspaceResourceId
                'DceEndpointId'         = $DceResourceId
                'iisLogDirectory'       = "C:\inetpub\logs\LogFiles\W3SVC" + $Number
                'Verbose'               = $true
            }

            New-AzResourceGroupDeployment  @parameters -AsJob
        }
    Until ($Number -ge 5)


#######################################################################
# dcr-linux-sentinel-forwarder-syslog
#######################################################################

    $DcrName                    = "dcr-linux-sentinel-forwarder-syslog"

    $parameters = @{
        'Name'                  = $DcrName + (Get-Random -Maximum 100000)
        'ResourceGroup'         = $ResourceGroupDeployment
        'TemplateFile'          = ".\datacollection\syslog\" + $DcrName + ".json"
        'DcrName'               = $DcrName
        'DcrResourceGroup'      = $ResourceGroupDCR
        'WorkspaceLocation'     = $WorkspaceLocation
        'WorkspaceResourceId'   = $WorkspaceResourceId
        'Verbose'               = $true
    }

    New-AzResourceGroupDeployment  @parameters

#--------------------------------------------------------------------------------------------------------------

# CLEANUP EXISTING DEPLOYMENTS
<#
    $ActiveDeployments = Get-AzResourceGroupDeployment -ResourceGroup $ResourceGroupDeployment
    $ActiveDeployments

    $CleanupDeployments  = $ActiveDeployments | Where-Object { $_.DeploymentName -like "*performance_serv*" }
    $CleanupDeployments | Remove-AzResourceGroupDeployment

    $CleanupDeployments  = $ActiveDeployments | Where-Object { $_.DeploymentName -like "*PerfAndMapDcr*" }
    $CleanupDeployments | Remove-AzResourceGroupDeployment

    $CleanupDeployments  = $ActiveDeployments | Where-Object { $_.DeploymentName -like "*NoMarketplace*" }
    $CleanupDeployments | Remove-AzResourceGroupDeployment

    $CleanupDeployments  = $ActiveDeployments | Where-Object { $_.DeploymentName -like "*VMI-DCR-Deployment*" }
    $CleanupDeployments | Remove-AzResourceGroupDeployment
     
    $CleanupDeployments  = $ActiveDeployments | Where-Object { $_.DeploymentName -like "*Microsoft.DataCollectionRules*" }
    $CleanupDeployments | Remove-AzResourceGroupDeployment

#>