slackbot.ps1

param(
    [string]$year = $ENV:AOC_YEAR,
    [string]$session = $ENV:AOC_SESSION_COOKIE,
    [string]$leaderboard = $ENV:AOC_LEADERBOARD_ID,
    [string]$live_refresh = $ENV:AOC_REFRESH_DATA, # default false
    [string]$timeout_seconds = $ENV:AOC_REFRESH_RATE_SECONDS,
    [string]$send_leaderboard_update = $ENV:AOC_SEND_LEADERBOARD_STATE, # default true
    [string]$webhook = $ENV:SLACK_WEBHOOK,
    [string]$email = $ENV:EMAIL,
    [string]$send_slack_message = $ENV:SLACK_SEND_MESSAGE, # default false
    [string]$debug = $ENV:SCRIPT_DEBUG # default false
)
$ErrorActionPreference = "Stop"

Import-Module (Join-Path $PSScriptRoot "Logging.psm1") -Force

#region Parameter validation
if ([string]::IsNullOrEmpty($live_refresh)) {
    $live_refresh = "false"
}

if ([string]::IsNullOrEmpty($timeout_seconds)) {
    $timeout_seconds = "900" # 15 minutes, the minimum refresh rate
}

if ([string]::IsNullOrEmpty($debug)) {
    [bool]$debug = $true
}
else {
    [bool]$debug = [bool]::Parse($debug)
}

if ([string]::IsNullOrEmpty($send_slack_message)) {
    [bool]$send_slack_message = $false
}
else {
    [bool]$send_slack_message = [bool]::Parse($send_slack_message)
}

if ([string]::IsNullOrEmpty($send_leaderboard_update)) {
    [bool]$send_leaderboard_update = $true
}
else {
    [bool]$send_leaderboard_update = [bool]::Parse($send_leaderboard_update)
}

if ([string]::IsNullOrEmpty($year)) {
    $year = (Get-Date).Year
}

Write-Info ""
Write-Info "leaderboard: $leaderboard"
Write-Info "year: $year"
Write-Info "live_refresh: $live_refresh"
Write-Info "timeout_seconds: $timeout_seconds"
Write-Info "send_slack_message: $send_slack_message"
if ($debug) {
    Write-Info "session: $session"
    Write-Info "webhook: $webhook"
}
Write-Info ""

#endregion

#region Functions
function Format-TimeSpan($Duration){
    # http://stackoverflow.com/questions/61431951/ddg#61432373
    $Day = switch ($Duration.Days) {
        0 { $null; break }
        1 { "{0} Day," -f $Duration.Days; break }
        Default {"{0} Days," -f $Duration.Days}
    }
    
    $Hour = switch ($Duration.Hours) {
        #0 { $null; break }
        1 { "{0} Hour," -f $Duration.Hours; break }
        Default { "{0} Hours," -f $Duration.Hours }
    }
    
    $Minute = switch ($Duration.Minutes) {
        #0 { $null; break }
        1 { "{0} Minute," -f $Duration.Minutes; break }
        Default { "{0} Minutes," -f $Duration.Minutes }
    }
    
    $Second = switch ($Duration.Seconds) {
        #0 { $null; break }
        1 { "{0} Second" -f $Duration.Seconds; break }
        Default { "{0} Seconds" -f $Duration.Seconds }
    }
    
    return "$Day $Hour $Minute $Second"
}
function Add-Property($Object, $Name, $Value) {
    $Object | Add-Member -MemberType NoteProperty -Name $Name -Value $Value -Force
}
function Get-ObjectKeys($Object) {
    return $Object.PSObject.Properties.name
}
function Get-ObjectValues($Object) {
    return $Object.PSObject.Properties.value
}
function Get-PaddedLength($Values) {
    return $Values | ForEach-Object { "$_".Length } | Sort-Object -Descending | Select-Object -First 1
}
function Get-Participant($Participants, $Id) {
    return $Participants | Where-Object { $_.id -eq $Id } | Select-Object -First 1
}

function Get-Leaderboard($CallApi, $Year, $Session, $LeaderboardId) {
    if ($CallApi) {
        Write-Info "Refreshing data..."
        $url = "https://adventofcode.com/$Year/leaderboard/private/view/$LeaderboardId.json"
        $response = Invoke-WebRequest -Method Get -Uri $url -UserAgent "PowerShell Slack-Integration ($email) (https://github.com/maartengo/adventofcode_slack_bot)" -Headers @{ "Cookie" = "session=$Session" }
        Set-Content -Value $response.Content -Path .\result.json -Encoding utf8BOM
    }
    else {
        Write-Info "Using cached data..."
    }
    return Get-Content .\result.json -Encoding utf8BOM | ConvertFrom-Json
}
function Get-ScoreBoard() {
    $scoreboardMessage = "Current Scores:`n```````n"
    foreach ($participant in $participants) {      
        $index = $participant.position  
        $paddedIndex = "$index".PadRight($indexPadding)
        $paddedName = $participant.name.PadRight($namePadding)
        $paddedScore = "$($participant.local_score)".PadRight($scorePadding)

        if ($null -ne $specialScores."$index") {
            $paddedIndex = $specialScores."$index".PadRight($indexPadding)
        }
    
        $scoreboardMessage += "$paddedIndex) $paddedName $paddedScore ⭐ x $($participant.stars)`n"
    }

    $scoreboardMessage += "``````"
    return $scoreboardMessage
}
function Send-SlackMessage($message) {
    Write-Info "Sending slack message:"
    Write-Info $message
    if ($send_slack_message) {
        $body = @{ "text" = $message } | ConvertTo-Json
        $response = Invoke-WebRequest -Method Post -Uri $webhook -ContentType "application/json; charset=UTF-8" -Body $body
        if ($response.StatusCode -ne 200) {
            Write-Error "Failed to send slack message:"
            $response
        }
    }
}
#endregion


$codeblock = "``````"
$live = [bool]::Parse($live_refresh)
$timeout = [int]::Parse($timeout_seconds)
$timeout = [math]::max(900, $timeout) # 15 minutes, the minimum refresh rate
$leaderboardTimes = @('5:00', '11:01') # post at 5:45 and 12:00 CET
# $leaderboardTimes = @('11:01')
$timeoutInMinutes = [math]::round($timeout / 60)

# Wait until AOC starts
$utcNow = (Get-Date).ToUniversalTime()
if($year -eq $utcNow.Year -and $utcNow.Month -ne 12) {
    $startOfAOC = (Get-Date -Date "$year-12-01 05:00:00Z").ToUniversalTime()
    $secondsUntilDecember = ($startOfAOC - $utcNow).TotalSeconds
    Write-Info "Sleeping $(Format-TimeSpan ($startOfAOC - $utcNow)) until 1 December $year..."
    while($secondsUntilDecember -gt 0) {
        Start-Sleep -Seconds ([math]::min($secondsUntilDecember, 2147483))
        $secondsUntilDecember = $secondsUntilDecember - 2147483
    }
}

$minutesToWait = $timeoutInMinutes - ((Get-Date).minute % $timeoutInMinutes)
if ($timeoutInMinutes -gt $minutesToWait) {
    Write-Info "Waiting $minutesToWait minutes so we start at a rounded number of minutes"
    Start-Sleep ($minutesToWait * 60 - (Get-Date).second)
    Write-Info "Continue..."
}

[datetime]$epoch = '1970-01-01 00:00:00'
$specialScores = @{
    "1"     = "🥇"
    "2"     = "🥈"
    "3"     = "🥉"
}

$init = $true
while ($true) {
    $previous = @{
        content      = $content
        participants = $participants
        members      = $members
        date         = $content.date
    }

    $content = Get-Leaderboard -CallApi $live -Year $year -Session $session -LeaderboardId $leaderboard

    $members = Get-ObjectValues $content.members | Sort-Object -Descending -Property local_score,last_star_ts
    $participants = $members  | Where-Object { $_.local_score -gt 0 } 
    
    $indexPadding = [math]::max(2, "$($participants.Length)".Length)
    $namePadding = Get-PaddedLength $participants.name
    $scorePadding = Get-PaddedLength $participants.local_score
    
    $i = 1
    foreach ($member in $members) {
        Add-Property -Object $member -Name "position" -Value $i
        $i++
    }

    Add-Property -Object $content -Name "date" -Value (Get-Date)
    Set-Content -Value ($content | ConvertTo-Json -Depth 20) -Path .\result.json  -Encoding utf8BOM

    if ($init) {
        $init = $false
        $previous = @{
            content      = $content
            participants = $participants
            members      = $members
            date         = $content.date
        }
    }

    if ($send_leaderboard_update) {
        $now = Get-Date

        $lastStarGained = $participants | Sort-Object -Property last_star_ts -Descending | Select-Object -First 1 -ExpandProperty last_star_ts
        if($null -eq $lastStarGained) {
            $lastStarGained = 0
        }
        $lastStarGainedTime = $epoch.AddSeconds($lastStarGained).ToLocalTime()

        $times = $leaderboardTimes | Foreach-Object { [datetime]::parse($_).ToLocalTime() } | Sort-Object

        for($i = 0; $i -lt $times.Length; $i++) {
            $leaderBoardRefresh = $times[$i]
            if ($now -le $leaderBoardRefresh -and ($now.AddSeconds($timeout) -ge $leaderBoardRefresh)) {
                $previousTime = $times[($i - 1) % $times.Length]
                if($previousTime -gt $now) { $previousTime.AddDays(-1) }

                if($lastStarGainedTime -gt $previousTime) {
                    Send-SlackMessage (Get-ScoreBoard)
                }
                break
            }
        }
    }

    # $gainedStars = @(@{ name = "", day = 1, part = 1, time = 10:32 })
    $gainedStars = @()

    if ($debug) {
        # $previous.participants = $null
    }

    foreach ($participant in $participants) {
        $previousParticipant = Get-Participant -Participants ($previous.participants) -Id $participant.id

        if ($null -ne $previousParticipant -and ($participant.stars - $previousParticipant.stars) -eq 0) {
            continue;
        }
    
        foreach ($day in (Get-ObjectKeys $participant.completion_day_level)) {
            $previousDay = $null
            if ($null -ne $previousParticipant) {
                $previousDay = $previousParticipant.completion_day_level.$day
            }
    
            foreach ($part in (Get-ObjectKeys $participant.completion_day_level.$day)) {
                if ($null -eq $previousDay -or $null -eq $previousDay.$part) {
                    $gainedStars += @{
                        name = $participant.name
                        day  = $day
                        part = $part
                        time = $participant.completion_day_level.$day.$part.get_star_ts
                    }
                }
            }
        }
    }

    if ($gainedStars.Length -gt 0) {
        $message = [System.Collections.ArrayList]::new()
        $gainedStars | Sort-Object -Property time | ForEach-Object {
            $gained = $_
            $paddedName = $gained.name.PadRight($namePadding)
            $paddedDay = "$($gained.day)".PadRight(2)
            $paddedPart = $gained.part -eq "1" ? "1️⃣" : "2️⃣"
            $paddedTime = "{0:HH:mm:ss}" -f $epoch.AddSeconds($gained.time).ToLocalTime()
            $message.Add("$paddedName won a ⭐ for day $paddedDay part $paddedPart at $paddedTime 🎉")
        }
        # Messages can only be 55 lines long
        for ($i = 0; $i * 50 -lt $message.Count; $i++) {
            $text = $message | Select-Object -Skip ($i * 50) -First 50
            $text = $text -join "`n"
            $text = "$codeblock`n$text`n$codeblock"
            Send-SlackMessage $text
            Start-Sleep -Seconds 0.5
        }
    }

    [System.gc]::Collect()
    $interval = ($timeout - (Get-Date).Second % $timeout)
    Write-Info "Sleeping for $timeout seconds... next check at $("{0:yyyy/MM/dd} {0:HH:mm:ss}" -f (Get-Date).AddSeconds($interval).ToLocalTime())"
    Start-Sleep -Seconds $interval
}