param(
    [switch]$debug,
    [switch]$clear
)

if ($debug) {
    $script:debug = $debug
}



$teamsName = "IT Support & Delivery" 
$scheduleGroupName = "IT SUPPORT"

# App Registration details & Tenant ID
$ClientId = "6cb76bbf-f253-4551-87a8-5e1f49d8ace3"
$ClientSecret = "IQg8Q~JTBTDbrHyjKMc.KMSETtQoKWvUhnCr1aY1" 
$tenantid = "e81b4a31-e8ad-4df6-aa30-3bdcc1da2cdd" 
$userIdAdmin = "747bd624-5cd7-4ae0-952f-5dc374e07a5e"
$logPath = (".\log_{0}.log" -f (Get-Date).toString("yyyy_MM_dd"))

function Write-Debug {
    param(
        [string]$message
    )
    if ($script:debug) {
        Write-Host $message
    }
    else {
        $message | Add-Content $logPath
    }
}

#Naplánování konkrétní směny
function Set-Shift {
    param(
        [string]$userId,
        [string]$groupID,
        [string]$shiftName,
        [System.DateTime]$StartDate,
        [System.DateTime]$EndDate,
        [string]$color,
        [string]$teamID,
        [string]$mail
    )

    #$params = @{
    #    UserId            = $userId
    #    schedulingGroupId = $groupID
     #   sharedShift       = @{
    #        notes         = ($shiftName)
     #       startDateTime = [System.DateTime]::Parse($StartDate.ToString("yyyy-MM-dd'T'HH:mm:ssZ"))
    #        endDateTime   = [System.DateTime]::Parse($EndDate.ToString("yyyy-MM-dd'T'HH:mm:ssZ"))
    #        theme         = $color
     #   }
    #}

    #$headers = @{
    #    "MS-APP-ACTS-AS" = $userIdAdmin
    #    "Prefer"          = "no-notification"
    #}

    #$newShift = New-MgTeamScheduleShift -TeamId $teamID -DraftShift -BodyParameter $params -Headers @{ "MS-APP-ACTS-AS" = $userIdAdmin }
    #Write-Debug ("SCHEDULING: {0} {1} - {2}" -f $newShift.SharedShift.StartDateTime.ToString("yyyy-MM-dd HH:mm"), $newShift.SharedShift.EndDateTime.ToString("yyyy-MM-dd HH:mm"), $newShift.SharedShift.Notes) 
    #("SCHEDULING: {0} {1} - {2}" -f $newShift.SharedShift.StartDateTime.ToString("yyyy-MM-dd HH:mm"), $newShift.SharedShift.EndDateTime.ToString("yyyy-MM-dd HH:mm"), $newShift.SharedShift.Notes)  | Add-Content -Path $logPath
    Write-Debug("{0} {1}: {2} {3}" -f $dayDate, $dayDate.DayOfWeek, $mail, $shiftName)
    $newShift = 0
    return $newShift
}

function Invoke-HasTimeOff {
    param(
        [string]$UID,
        [PSCustomObject]$timeoff,
        [string]$mail
    )

    foreach ($toff in $timeoff) {
        if($toff.UserId -eq $UID) {
            Write-Debug("{0} {1}: {2} has time off at this time" -f $dayDate, $dayDate.DayOfWeek, $mail)
            return $true
            break
        }
    }
    return $false
}
function Invoke-HasShift {
    param(
        [string]$UID,
        [PSCustomObject]$shifts,
        [string]$mail
    )

    foreach ($shift in $shifts) {
        if($shift.UserId -eq $UID) {
            Write-Debug("{0} {1}: {2} has a shift at this time" -f $dayDate, $dayDate.DayOfWeek, $mail)
            return $true
            break
        }
    }
    return $false
}

function Invoke-HasShiftorTimeOff {
    param(
        [string]$UID,
        [PSCustomObject]$shifts,
        [PSCustomObject]$timeoff,
        [string]$mail
    )

    foreach ($shift in $shifts) {
        if($shift.UserId -eq $UID) {
            Write-Debug("{0} {1}: {2} has a shift or time off at this time" -f $dayDate, $dayDate.DayOfWeek, $mail)
            return $true
            break
        }
    }

    foreach ($toff in $timeoff) {
        if($toff.UserId -eq $UID) {
            Write-Debug("{0} {1}: {2} has a shift or time off at this time" -f $dayDate, $dayDate.DayOfWeek, $mail)
            return $true
            break
        }
    }
    return $false
}

#Microsoft auth

$body = @{
    grant_type    = "client_credentials"
    resource      = "https://graph.microsoft.com"
    client_id     = $ClientId
    client_secret = $ClientSecret
}
$authUri = "https://login.microsoftonline.com/$tenantid/oauth2/token"
$response = Invoke-RestMethod $authUri -Method 'POST' -Body $body

$token = $response.access_token

Import-Module Microsoft.Graph.Teams
Connect-MgGraph -AccessToken $($token | ConvertTo-SecureString -AsPlainText -Force) >> $null

# Dnešní datum a plán směn

#$today = (Get-Date).Date
#$today = $today_old.AddYears(2)
$today = Get-Date -Day 11 -Month 1 -Year 2027 -Hour 9 -Minute 00 -Second 00 -Millisecond 00

$schedule = Get-Content -Path "./config.json" -Raw | ConvertFrom-Json

$allemails = $schedule.PSObject.Properties | ForEach-Object {
    if ($_.Name -in @("monday", "tuesday", "wednesday", "thursday", "friday")) {
        $_.Value.PSObject.Properties | ForEach-Object {
            if ($_.Name -in @("ds", "ho", "os", "ns")) {
                $_.Value | ForEach-Object {
                    $_
                }
            }
        }
    }
} | Select-Object -Unique

#Časové rozmezí plánování

#$Monday = $today.AddDays(1 - $today.DayOfWeek.value__)
#$Friday = $Monday.AddDays(4)
$daysahead = 14

$startSpanDate = Get-Date -Day $today.Day -Month $today.Month -Year $today.Year -Hour 9 -Minute 00 -Second 00 -Millisecond 00
$endSpanDate = Get-Date -Day $startSpanDate.AddDays($daysahead).Day -Month $startSpanDate.AddDays($daysahead).Month -Year $startSpanDate.AddDays($daysahead).Year -Hour 17 -Minute 00 -Second 00 -Millisecond 00

#RLCZ maintenance
$rlcz_mtnc_email = "jiri.kotlan@itego.cz"
$rlcz_shift_reduction = -3


# SKRIPT
try {
    $team = Get-MgTeam -Property "id,displayName" | Where-Object -property displayName -value $teamsName -eq
    $group = Get-MgTeamScheduleSchedulingGroup -TeamId  $team.id -Headers @{ "MS-APP-ACTS-AS" = $userIdAdmin } |  Where-Object -property displayName -value $scheduleGroupName -eq
    
    $allshifts = [object[]] (Get-MgTeamScheduleShift -TeamId $team.id -Filter "SharedShift/StartDateTime ge $($startSpanDate.toString('yyyy-MM-ddT00:00:00Z'))" -All  -Headers @{ "MS-APP-ACTS-AS" = $userIdAdmin })
    $timeoff_all = [object[]] (Get-MgTeamScheduleTimeOff -TeamId $team.id -Filter "SharedTimeOff/StartDateTime ge $($startSpanDate.toString('yyyy-MM-ddT00:00:00Z'))" -All  -Headers @{ "MS-APP-ACTS-AS" = $userIdAdmin } | Select-Object -Property *,@{name="Date";expression={$_.SharedTimeOff.StartDateTime.AddHours(1).Date}})

    #return

    #$timeoff_today = $timeoff_all | Where-Object -Property Date -eq -Value $(Get-Date -Day 20 -Month 1 -Year 2027).Date


    #if ($clear) {
    #    foreach ($shift in $allshifts) {
    #        Remove-MgTeamSchedu#leShift -TeamId $team.id -ShiftId $shift.id
    #    }
#
#        return
#    }


    #Procházíme dny v časovém rozmezí
    $timeSpan = New-TimeSpan -Start $startSpanDate -End $endSpanDate
    Write-Debug("Planning from {0} to {1}" -f $startSpanDate, $endSpanDate)
    for ($dayNumber = 0; $dayNumber -lt $timespan.Days; $dayNumber++) {
        $dayDate = $today.AddDays($dayNumber);

        #Pokud jde o pracovní den, jinak se jde ihned na další den
        if ($dayDate.DayOfWeek -notin @("Saturday", "Sunday")) {
            #Inicializace dne
            $ds_shift_inplace = $false
        
            #Množina emailů pro tento den v týdnu co mají mít home office
            $ho_emails = $schedule.($dayDate.DayOfWeek.ToString().ToLower()).'ho'

            #Email a ID toho kdo má mít tento den denní směnu
            $ds_email = $schedule.($dayDate.DayOfWeek.ToString().ToLower()).'ds'

            #Emaily těch co jsou on-site u zákazníka
            $os_emails = $schedule.($dayDate.DayOfWeek.ToString().ToLower()).'os'

            #Emaily těch co nemají mít dnes žádnou směnu
            $ns_emails = $schedule.($dayDate.DayOfWeek.ToString().ToLower()).'ns'

            #Upravení času
            $dateStart = $(get-date -Day $dayDate.Day -Month $dayDate.Month -Year $dayDate.Year -Hour 9 -Minute 00 -Second 00).AddHours(-1)
            $dateEnd = $(get-date -Day $dayDate.Day -Month $dayDate.Month -Year $dayDate.Year -Hour 17 -Minute 00 -Second 00).AddHours(-1)

            if (($dayDate.IsDaylightSavingTime()) -eq $false) {
                $dateStart = $dateStart.AddHours(-1)
                $dateEnd = $dateStart.AddHours(-1)
            }
            else {
            }

            $shifts_today = [Object[]] $allshifts | Where-Object -Filter { 
                $_.schedulingGroupId -eq $group.Id -and $_.SharedShift.StartDateTime.ToString("yyyy-MM-dd HH:mm") -ge $dateStart.ToString("yyyy-MM-dd HH:mm") -and $_.SharedShift.EndDateTime.ToString("yyyy-MM-dd HH:mm") -le $dateEnd.ToString("yyyy-MM-dd HH:mm") 
            }

            $timeoff_today = [Object[]] $timeoff_all | Where-Object -Filter { 
                ($_.SharedTimeOff.StartDateTime.Date -eq $dateStart.Date -or $_.SharedTimeOff.EndDateTime.Date -eq $dateEnd.Date)
            }
            
            # Tohle jsou všechny způsoby co jsem zkoušel, nikdy to nematchne, nikdy to nevidí dovolenou
            
            #$timeoff_today = [Object[]] $timeoff_all | Where-Object -Filter { $_.schedulingGroupId -eq $group.Id -and $_.SharedTimeOff.StartDateTime.ToString("yyyy-MM-dd HH:mm") -le $dateEnd.AddDays(1).ToString("yyyy-MM-dd HH:mm") -and $_.SharedTimeOff.EndDateTime.ToString("yyyy-MM-dd HH:mm") -ge $dateStart.AddDays(-1).ToString("yyyy-MM-dd HH:mm") }
            #$timeoff_today = [Object[]] $timeoff_all | Where-Object { $_.schedulingGroupId -eq $group.Id -and $_.SharedTimeOff.StartDateTime -lt $dateEnd -and $_.SharedTimeOff.EndDateTime   -gt $dateStart }
            #$timeoff_today = $timeoff_all | Where-Object { $_.schedulingGroupId -eq $group.Id -and ([datetime]$_.SharedTimeOff.StartDateTime) -lt $dayEnd -and ([datetime]$_.SharedTimeOff.EndDateTime) -gt $dayStart }
            #$timeoff_today = [Object[]] $timeoff_all | Where-Object -Filter { $_.schedulingGroupId -eq $group.Id -and $_.SharedTimeOff.StartDateTime.ToString("yyyy-MM-dd HH:mm") -ge $dateStart.ToString("yyyy-MM-ddT00:00:00Z") -and $_.SharedTimeOff.EndDateTime.ToString("yyyy-MM-dd HH:mm") -le $dateEnd.AddDays(1).ToString("yyyy-MM-ddT00:00:00Z") }
            #$timeoff_today = [Object[]] $timeoff_all | Where-Object -Filter { $_.schedulingGroupId -eq $group.Id -and $_.SharedTimeOff.StartDateTime.ToString("yyyy-MM-dd HH:mm") -ge $dateStart.ToString("yyyy-MM-ddT00:00:00Z") -and $_.SharedTimeOff.EndDateTime.ToString("yyyy-MM-dd HH:mm") -le $dateEnd.ToString("yyyy-MM-ddT00:00:00Z") }
            #Pokus o naplnění množiny manuálně daných denních směn pro tento den
            $manual_dayshifts = [Object[]] $shifts_today | Where-Object -Filter { ($_.SharedShift.Notes -eq "Home Office - Denní směna") -or ($_.SharedShift.Notes -eq "Office - Denní směna") }
            
            if ($null -eq $manual_dayshifts) {
                $ds_shift_inplace = $false
                Write-Debug("{0} {1}: day shift not set" -f $dayDate, $dayDate.DayOfWeek)
                foreach($shift in $manual_dayshifts) {
                    Write-Debug("{0}: {1}-{2}"-f $shift.SharedShift.Notes, $shift.SharedShift.StartDateTime, $shift.SharedShift.EndDateTime)
                }
            }
            else {
                $ds_shift_inplace = $true
                Write-Debug("{0} {1}: day shift set already" -f $dayDate, $dayDate.DayOfWeek)
                foreach($shift in $manual_dayshifts) {
                    Write-Debug("{0}: {1}-{2}"-f $shift.SharedShift.Notes, $shift.SharedShift.StartDateTime, $shift.SharedShift.EndDateTime)
                }
            }

            foreach($email in $allemails) {
                $userId = $(Get-MgUser -Filter "UserPrincipalName eq '$email' or proxyAddresses/any(c:c eq 'smtp:$email')").Id
                
                if($ds_shift_inplace -eq $false) {
                    if($email -eq $ds_email) {
                        Write-Debug("{0} {1}: checking {2} for day shift availability..." -f $dayDate, $dayDate.DayOfWeek, $email)
                        foreach($toff in $timeoff_today) {
                            if($toff.UserId -eq $userId) {
                                Write-Debug("{0}: {1} {2}" -f $email, $toff.SharedTimeOff.StartDateTime, $toff.SharedTimeOff.EndDateTime)
                            }
                        }
                        if((Invoke-HasShiftorTimeOff -UID $userId -shifts $shifts_today -timeoff $timeoff_today -mail $email) -eq $false) {
                            if ($email -in $ho_emails) {
                                $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "Home Office - Denní směna" -StartDate $dateStart -EndDate $dateEnd -color "purple" -teamID $team.id -mail $email
                                $allshifts += [Object[]] $newshift
                                $ds_shift_inplace = $true
                                continue
                            }
                            else {
                                $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "Office - Denní směna" -StartDate $dateStart -EndDate $dateEnd -color "purple" -teamID $team.id -mail $email
                                $allshifts += [Object[]] $newShift
                                $ds_shift_inplace = $true
                                continue
                            }
                        }
                        else { 
                            #Budeme hledat ve zbývajících dnech zvoleného rozsahu
                            Write-Debug("{0} {1}: looking ahead for a day shift replacement" -f $dayDate, $dayDate.DayOfWeek)
                            $remainingdays = $timespan.Days - $dayNumber
                            for ($daysahead = 1; $daysahead -lt $remainingdays; $daysahead++) {
                                $futureday = $dayDate.AddDays($daysahead)
                                #Pokud jsme narazili na pracovní den...
                                    if ($futureday.DayOfWeek -notin @("Saturday", "Sunday")) {
                                        #...jdeme se podívat kdo má mít denní směnu
                                        $futureds_email = $schedule.($futureday.DayOfWeek.ToString().ToLower()).'ds'
                                        Write-Debug(">>> {0} {1}: {2} has it on that day" -f $futureday, $futureday.DayOfWeek, $futureds_email)
                                        $futureds_id = $(Get-MgUser -Filter "UserPrincipalName eq '$futureds_email' or proxyAddresses/any(c:c eq 'smtp:$futureds_email')").Id
                                    #Má další na řadě nastavenou na dnešek směnu? Pokud ne tak má denní a hledání ukončujeme, jinak se jde na další den
                                    if((Invoke-HasTimeOff -UID $futureds_id -timeoff $timeoff_today -mail $futureds_email) -eq $false -and ($futureds_email -notin $os_emails) -and ($futureds_email -notin $ns_emails)) {
                                        Write-Debug(">>> {0} {1}: {2} doesn't have time off, on site or no-shift on {3} {4}" -f $futureday, $futureday.DayOfWeek, $futureds_email, $dayDate, $dayDate.DayOfWeek)
                                        if((Invoke-HasShift -UID $futureds_id -shifts $shifts_today -mail $futureds_email) -eq $false) {
                                            Write-Debug(">>> {0} {1}: {2} ID: {3} doesn't have a shift on {4} {5}" -f $futureday, $futureday.DayOfWeek, $futureds_email, $futureds_id, $dayDate, $dayDate.DayOfWeek)
                                            if($futureds_email -in $ho_emails) {
                                                $newshift = Set-Shift -userId $futureds_id -groupID $group.id -shiftName "Home Office - Denní směna" -StartDate $dateStart -EndDate $dateEnd -color "purple" -teamID $team.id -mail $futureds_email
                                                $allshifts += [Object[]] $newshift
                                                $ds_shift_inplace = $true
                                                break
                                            }
                                            else {
                                                $newshift = Set-Shift -userId $futureds_id -groupID $group.id -shiftName "Office - Denní směna" -StartDate $dateStart -EndDate $dateEnd -color "purple" -teamID $team.id -mail $futureds_email
                                                $allshifts += [Object[]] $newShift
                                                $ds_shift_inplace = $true
                                                break
                                            }
                                        }
                                        else {
                                            Write-Debug(">>> {0} {1}: {2} has a shift on {3} {4}" -f $futureday, $futureday.DayOfWeek, $futureds_email, $dayDate, $dayDate.DayOfWeek)
                                            foreach ($shift in $shifts_today) {
                                                if($shift.UserId -eq $futureds_id) {
                                                    Write-Debug(">>> {0} {1}: removing shift {2} {3} {4} for {5}" -f $futureday, $futureday.DayOfWeek, $shift.Notes, $shift.StartDateTime, $shift.EndDateTime, $futureds_email)
                                                    #Remove-MgTeamScheduleShift -ShiftId $shift.id -TeamId $team.Id
                                                    if($futureds_email -in $ho_emails) {
                                                        $newshift = Set-Shift -userId $futureds_id -groupID $group.id -shiftName "Home Office - Denní směna" -StartDate $dateStart -EndDate $dateEnd -color "purple" -teamID $team.id -mail $futureds_email
                                                        $allshifts += [Object[]] $newshift
                                                        $ds_shift_inplace = $true
                                                        break
                                                    }
                                                    else {
                                                        $newshift = Set-Shift -userId $futureds_id -groupID $group.id -shiftName "Office - Denní směna" -StartDate $dateStart -EndDate $dateEnd -color "purple" -teamID $team.id -mail $futureds_email
                                                        $allshifts += [Object[]] $newShift
                                                        $ds_shift_inplace = $true
                                                        break
                                                    }
                                                }
                                            }
                                        }
                                    }
                                    else {
                                        foreach($timeoff in $timeoff_today) {
                                            if($timeoff.UserId -eq $futureds_id) {
                                                Write-Debug(">>> {0} {1}: {2} has a shift on {3} {4} - {5} {6} {7}" -f $futureday, $futureday.DayOfWeek, $futureds_email, $dayDate, $dayDate.DayOfWeek, $timeoff.SharedTimeOff.Notes, $timeoff.SharedTimeOff.StartDateTime, $timeoff.SharedTimeOff.EndDateTime)
                                            }
                                        }

                                        if($futureds_email -in $os_emails) {
                                            Write-Debug(">>> {0} {1}: {2} is on the on-site list {3} {4}" -f $futureday, $futureday.DayOfWeek, $futureds_email, $dayDate, $dayDate.DayOfWeek)
                                        }

                                        if($futureds_email -in $ns_emails) {
                                            Write-Debug(">>> {0} {1}: {2} is on the no-shift list {3} {4}" -f $futureday, $futureday.DayOfWeek, $futureds_email, $dayDate, $dayDate.DayOfWeek)
                                        }
                                    }
                                }
                            }
                        }
                    }
                }

                if($email -in $os_emails) {
                    if((Invoke-HasShiftorTimeOff -UID $userId -shifts $shifts_today -timeoff $timeoff_today -mail $email) -eq $false) {
                        $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "On-site" -StartDate $dateStart -EndDate $dateEnd -color "gray" -teamID $team.id -mail $email
                        $allshifts += [Object[]] $newshift
                        continue
                    }
                }

                if($email -in $ns_emails) {
                    continue
                }

                if($email -eq $rlcz_mtnc_email) {
                    if ($dayDate.DayOfWeek -eq "Thursday") {
                        $yesterday = $dayDate.AddDays(-1)
                        if (($yesterday.IsDaylightSavingTime()) -eq $false) {
                            $rlcz_shift_start = $(get-date -Day $yesterday.Day -Month $yesterday.Month -Year $yesterday.Year -Hour 18 -Minute 00 -Second 00).AddHours(-1)
                            $rlcz_shift_end = $(get-date -Day $yesterday.Day -Month $yesterday.Month -Year $yesterday.Year -Hour 23 -Minute 00 -Second 00).AddHours(-1)
                        }
                        else {
                            $rlcz_shift_start = $(get-date -Day $yesterday.Day -Month $yesterday.Month -Year $yesterday.Year -Hour 18 -Minute 00 -Second 00).AddHours(-2)
                            $rlcz_shift_end = $(get-date -Day $yesterday.Day -Month $yesterday.Month -Year $yesterday.Year -Hour 23 -Minute 00 -Second 00).AddHours(-2)
                        }
                        $rlcz_shifts = [Object[]] $allshifts | Where-Object -Filter { $_.schedulingGroupId -eq $group.Id -and $_.SharedShift.Notes -eq "RLCZ" -and $_.SharedShift.StartDateTime.ToString("yyyy-MM-dd HH:mm") -ge $rlcz_shift_start.ToString("yyyy-MM-dd HH:mm") -and $_.SharedShift.EndDateTime.ToString("yyyy-MM-dd HH:mm") -le $rlcz_shift_end.ToString("yyyy-MM-dd HH:mm") }
                        if (Invoke-HasShift -UID $userId -shifts $rlcz_shifts -eq $true -mail $email) {
                            if((Invoke-HasShiftorTimeOff -UID $userId -shifts $shifts_today -timeoff $timeoff_toda -mail $email) -eq $false) {
                                if($email -in $ho_emails) {
                                    $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "Home office" -StartDate $dateStart -EndDate $dateEnd.AddHours($rlcz_shift_reduction) -color "pink" -teamID $team.id -mail $email
                                    $allshifts += [Object[]] $newshift
                                    continue
                                }
                                else {
                                    $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "Office" -StartDate $dateStart -EndDate $dateEnd.AddHours($rlcz_shift_reduction) -color "blue" -teamID $team.id -mail $email
                                    $allshifts += [Object[]] $newshift
                                    continue
                                }
                            }
                        }
                    }
                }

                if($email -in $ho_emails) {
                    if((Invoke-HasShiftorTimeOff -UID $userId -shifts $shifts_today -timeoff $timeoff_today -mail $email) -eq $false) {
                        $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "Home office" -StartDate $dateStart -EndDate $dateEnd -color "pink" -teamID $team.id -mail $email
                        $allshifts += [Object[]] $newshift
                        continue
                    }
                }

                if((Invoke-HasShiftorTimeOff -UID $userId -shifts $shifts_today -timeoff $timeoff_today -mail $email) -eq $false) {
                    $newshift = Set-Shift -userId $userId -groupID $group.id -shiftName "Office" -StartDate $dateStart -EndDate $dateEnd -color "blue" -teamID $team.id -mail $email
                    $allshifts += [Object[]] $newshift
                    continue
                }
            }
        }
    }
}
catch {
    $_
    break;
}