# get EdgeDX devices
function Get-EdgeDxDevices {
    param (
        [Parameter(Mandatory = $true)]
        [string]$ApiKey
    )


    $baseUrl = "https://api.controlup.com/edge/api" # Base URL
    $endpoint = "devices"
    $method = "GET"


    $headers = @{
        "Authorization" = "Bearer $ApiKey"
        "Content-Type"  = "application/json"
    }


    $url = "$baseUrl/$endpoint"


    try {
        # Invoke the API to get the list of devices
        $response = Invoke-RestMethod -Uri $url -Method $method -Headers $headers
        return $response
    } catch {
        throw "Error invoking API: $_"
    }
}


$devices = (Get-EdgeDxDevices -ApiKey $apikey).rows | Select-Object -Property Name,_id,_platform_name_friendly,last_communication,tags
$devices

$deviceIDs = @($devices | Where-Object { -not $_.tags -and $_._platform_name_friendly -eq "Microsoft Windows"})._id

#Update EdgeDX Devices Tags 
function Update-EdgeDxDeviceTags {
    [CmdletBinding(DefaultParameterSetName = "CreateTags")]
    param (
        [Parameter(Mandatory = $true)]
        [string]$ApiKey,             # API key provided by the user
        [Parameter(Mandatory = $true)]
        [string[]]$DeviceIds,        # Array of device IDs to update
        [Parameter(Mandatory = $true)]
        [string[]]$Tags,             # The tags to update
        [Parameter(Mandatory = $true, ParameterSetName = "CreateTags")]
        [switch]$Create,             # Specify this switch to create tags
        [Parameter(Mandatory = $true, ParameterSetName = "DeleteTags")]
        [switch]$Delete              # Specify this switch to delete tags
    )


    # Determine the action based on the parameter set
    $action = if ($PSCmdlet.ParameterSetName -eq "CreateTags") {
        "create"
    } elseif ($PSCmdlet.ParameterSetName -eq "DeleteTags") {
        "delete"
    } else {
        throw "Unexpected parameter set."
    }


    # Validate tags
    $invalidTags = $Tags | Where-Object { -not ($_ -match '^[a-z0-9\-_ ]{1,150}$') }
    if ($invalidTags) {
        throw "Invalid tag(s) provided: $($invalidTags -join ', '). Tags may contain only: a-z, 0-9, -, _, spaces, and must be 1-150 characters long."
    }


    # Build the request body
    $body = @{
        "action" = $action
        "ids"    = $DeviceIds
        "tags"   = $Tags
    }


    # Base URL and endpoint
    $baseUrl = "https://api.controlup.com/edge/api"
    $endpoint = "devices/tags"
    $url = "$baseUrl/$endpoint"


    # Headers with API key
    $headers = @{
        "Authorization" = "Bearer $ApiKey"
        "Content-Type"  = "application/json"
    }


    try {
        # Make the API call
        $jsonBody = $body | ConvertTo-Json -Compress -Depth 2
        $response = Invoke-RestMethod -Uri $url -Method "POST" -Headers $headers -Body $jsonBody


        # Extract device details from the response and display them
        foreach ($device in $response) {
            Write-Output "Device $($device.name) now has these tags: $($device.tags -join ', ')"
        }
    } catch {
        throw "Error invoking API: $_"
    }
}
 

Update-EdgeDxDeviceTags -ApiKey $apikey -DeviceIds $deviceIDs -Tags "windows" -Create

# set EdgeDX Device Groups
function Set-EdgeDxDeviceGroups {
    param (
        [Parameter(Mandatory = $true)]
        [string]$ApiKey,           # API key provided by the user
        [Parameter(Mandatory = $true)]
        [string]$GroupName,        # The name of the group to assign devices to
        [Parameter(Mandatory = $true)]
        [string[]]$DeviceIds       # Array of device IDs to associate with the group
    )


    # Build the request body
    $body = @{
        "group"      = $GroupName
        "device_ids" = $DeviceIds
    }


    # Base URL and endpoint
    $baseUrl = "https://api.controlup.com/edge/api"
    $endpoint = "devices/set-group"
    $url = "$baseUrl/$endpoint"


    # Headers with API key
    $headers = @{
        "Authorization" = "Bearer $ApiKey"
        "Content-Type"  = "application/json"
    }


    try {
        # Make the API call
        $jsonBody = $body | ConvertTo-Json -Compress -Depth 2
        $response = Invoke-RestMethod -Uri $url -Method "POST" -Headers $headers -Body $jsonBody
        return $response
    } catch {
        throw "Error invoking API: $_"
    }
}

Set-EdgeDxDeviceGroups -ApiKey $apikey -GroupName "Amsterdam" -DeviceIds $deviceIDs

# Define the current date
$currentDate = Get-Date


# Define the threshold for communication 
$thresholdDate = $currentDate.AddDays(-60) #fill in the number of days of no communication


# Filter devices with no communication in the last X days
$filteredDevices = $devices | Where-Object {
    -not $_.last_communication -or ([datetime]::Parse($_.last_communication) -lt $thresholdDate)
}
$filteredDevices
$deviceIDs = @($filteredDevices)._id

function Remove-EdgeDxDevice {
    param (
        [Parameter(Mandatory = $true)]
        [string]$ApiKey,           # API key provided by the user
        [Parameter(Mandatory = $true)]
        [array]$DeviceIds          # Array of device IDs to delete
    )


    # Build the body
    $body = @{
        "device_ids" = $DeviceIds
    }


    # Base URL and endpoint
    $baseUrl = "https://api.controlup.com/edge/api"
    $endpoint = "devices/delete"
    $url = "$baseUrl/$endpoint"


    # Headers with API key
    $headers = @{
        "Authorization" = "Bearer $ApiKey"
        "Content-Type"  = "application/json"
    }


    try {
        # Make the API call
        $jsonBody = $body | ConvertTo-Json -Compress -Depth 2
        $response = Invoke-RestMethod -Uri $url -Method "POST" -Headers $headers -Body $jsonBody


        return $response
    } catch {
        throw "Error invoking API: $_"
    }
}


Remove-EdgeDxDevice -ApiKey $apikey -DeviceIds $deviceIDs

# Define a function to create the message box
function Get-ApiKey {
    Add-Type -AssemblyName System.Windows.Forms


    # Create a form for the input
    $form = New-Object System.Windows.Forms.Form
    $form.Text = "Enter API Key"
    $form.Size = New-Object System.Drawing.Size(400, 150)
    $form.StartPosition = "CenterScreen"


    # Create a label
    $label = New-Object System.Windows.Forms.Label
    $label.Text = "Please enter your API key:"
    $label.AutoSize = $true
    $label.Location = New-Object System.Drawing.Point(10, 10)
    $form.Controls.Add($label)


    # Create a textbox for the input
    $textBox = New-Object System.Windows.Forms.TextBox
    $textBox.Location = New-Object System.Drawing.Point(10, 40)
    $textBox.Width = 360
    $form.Controls.Add($textBox)


    # Create an OK button
    $okButton = New-Object System.Windows.Forms.Button
    $okButton.Text = "OK"
    $okButton.Location = New-Object System.Drawing.Point(300, 70)
    $okButton.Add_Click({ $form.Close() })
    $form.Controls.Add($okButton)


    # Show the form
    $form.ShowDialog() | Out-Null


    # Return the input from the textbox
    return $textBox.Text
}


# Call the function and save the API key
$ApiKey = Get-ApiKey