#!/usr/bin/env bash
## Estimate the consumed GitHub Full-Time Runners for the day so far (by UTC time).
## We Accumulate the GitHub Job Duration and Extrapolate the GitHub Runner Hours.

## Job Duration to Runner Hours is inflated slightly
## https://docs.google.com/spreadsheets/d/1ujGKmUyy-cGY-l1pDBfle_Y6LKMsNp7o3rbfT1UkiZE/edit?gid=1163309346#gid=1163309346
duration_hours_to_runner_hours=7.4

## For Testing:
## date=2024-10-21
## hours=24
date=$(date -u +'%Y-%m-%d')
hours=$(bc -e "1+$(date -u +'%H')")

## Set the GitHub Token: export GITHUB_TOKEN=...
## Any Token with Read-Access to NuttX Repo will do:
## "public_repo" (Access public repositories)
. $HOME/github-token.sh

## Accumulate the Job Duration and Extrapolate the Runner Hours
function add_runner_hours {
  local repo=$1
  local run_ids=$(
    gh run list \
      --repo $repo \
      --limit 1000 \
      --created $date \
      --json databaseId,createdAt,displayTitle,name \
      --jq '.[].databaseId'
  )

  ## Call GitHub API to get the Job Duration
  ## https://docs.github.com/en/rest/actions/workflow-runs?apiVersion=2022-11-28#get-workflow-run-usage
  for run_id in $run_ids; do
    local run_duration_ms=$(
      curl -L --silent \
        -H "Accept: application/vnd.github+json" \
        -H "Authorization: Bearer $GITHUB_TOKEN" \
        -H "X-GitHub-Api-Version: 2022-11-28" \
        https://api.github.com/repos/$repo/actions/runs/$run_id/timing \
        | jq '.run_duration_ms'
    )
    # Beware of GitHub API Rate Limits
    sleep 0.1
    if [[ "$run_duration_ms" == "null" ]]; then
      continue
    fi

    ## Extrapolate the Job Duration to Runner Hours
    total_job_hours=$(
      bc -l -e "$total_job_hours+($run_duration_ms/1000/60/60)"
    )
    local runner_hours=$(
      bc -l -e "$duration_hours_to_runner_hours*$run_duration_ms/1000/60/60"
    )
    if [[ "$runner_hours" != "0" ]]; then
      local runner_hours_rounded=$(
        bc -l -e "r($runner_hours,1)"
      )
      echo $run_id,$run_duration_ms,$runner_hours_rounded
    fi
  done
}

## Accumulate the Runner Hours for nuttx and nuttx-apps repos
echo run_id,run_duration_ms,runner_hours
total_job_hours=0
add_runner_hours apache/nuttx
add_runner_hours apache/nuttx-apps

## Compute the Full-Time Runners
total_runner_hours=$(
  bc -l -e "$duration_hours_to_runner_hours*$total_job_hours"
)
fulltime_runners=$(
  bc -e "$total_runner_hours/$hours"
)
total_job_hours_rounded=$(
  bc -l -e "r($total_job_hours,1)"
)
total_runner_hours_rounded=$(
  bc -l -e "r($total_runner_hours,0)"
)
echo date=$date
echo hours=$hours
echo total_job_hours=$total_job_hours_rounded
echo total_runner_hours=$total_runner_hours_rounded
echo fulltime_runners=$fulltime_runners