aoc2022/scripts/aoc-lib.rkt

118 lines
4.3 KiB
Racket
Raw Normal View History

2022-12-01 05:11:06 +00:00
#lang racket
;; library to interact with the AoC API and track solutions progress
(require net/uri-codec net/http-client json)
(provide aoc-fetch-input aoc-fetch-challenge aoc-submit-answer aoc-fetch-leaderboard
aoc-complete? aoc-set-complete!)
(define *host* "adventofcode.com")
(define *status-file* ".status.rktd")
;; generates API paths
(define/contract (puzzle-path year day endpoint)
(-> string? string? (or/c "input" "answer" false/c) path?)
(define base (build-path "/" year "day" day))
(if endpoint (build-path base endpoint) base))
(define/contract (leaderboard-path year lb)
(-> string? string? path?)
(build-path "/" year "leaderboard" "private" "view" (string-append lb ".json")))
;; sets up necessary headers for API
(define (make-headers session)
(list (string-append "Cookie: session=" session)
"Content-Type: application/x-www-form-urlencoded"))
;; http request helper
(define (aoc-request endpoint session [method 'GET] [data #f])
(define (parse-headers hlist)
(for/list ([h (in-list hlist)])
(match h
[(pregexp #px"^([^:]+): (.*?)$" (list _ k v))
(cons (string->symbol (string-downcase (bytes->string/utf-8 k)))
(bytes->string/utf-8 v))]
[x (cons 'unknown x)])))
(define (do-request path headers method data)
(define-values [status headers-out content]
(http-sendrecv *host* path #:ssl? #t #:headers headers #:method method #:data data))
(define headers-out/parsed (parse-headers headers-out))
(match status
[(pregexp #px"^HTTP/1\\.[10] 200") content]
[(pregexp #px"^HTTP/1\\.[10] 302")
(define location (cdr (or (assoc 'location headers-out/parsed)
(error "got 302 with no location"))))
(printf "got redirect to ~a\n" location)
(close-input-port content)
(do-request location headers 'GET #f)]
[(pregexp #px"^HTTP/1\\.[10] 404")
(error "endpoint returned 404\n response: " (port->bytes content))]
[stat
(error "endpoint returned unexpected data\n status: " stat "\n response: "
(port->bytes content))]))
(do-request (path->string endpoint) (make-headers session) method data))
;; gets the input file for a challenge
(define/contract (aoc-fetch-input year day session)
(-> string? string? string? input-port?)
(aoc-request (puzzle-path year day "input") session))
;; submits an answer to the server
(define/contract (aoc-submit-answer year day session part answer)
(-> string? string? string? (or/c 1 2 "1" "2") string? (or/c symbol? bytes?))
(define data `((level . ,(~a part))
(answer . ,answer)))
(define resp
(port->bytes (aoc-request (puzzle-path year day "answer") session 'POST (alist->form-urlencoded data))))
(match resp
[(pregexp #px"That's the right answer")
(aoc-set-complete! day part)
'answer-correct]
[(pregexp #px"That's not the right answer") 'answer-incorrect]
[(pregexp #px"You gave an answer too recently") 'rate-limited]
[(pregexp #px"fifty stars")
(aoc-set-complete! day part)
'year-complete]
[(pregexp #px"Did you already complete it?")
(aoc-set-complete! day part)
'already-completed]
[x x]))
;; fetches the HTML page for a challenge
(define/contract (aoc-fetch-challenge year day session)
(-> string? string? string? input-port?)
(aoc-request (puzzle-path year day #f) session))
;; gets a private leaderboard as json
(define/contract (aoc-fetch-leaderboard year lb session)
(-> string? string? string? jsexpr?)
(read-json (aoc-request (leaderboard-path year lb) session)))
;; helper to generate entries for the challenge status file
(define (day+part->key day part)
(when (string? day)
(set! day (string->number day)))
(when (string? part)
(set! part (string->number part)))
(cons day part))
;; get the challenge status file
(define (aoc-get-status)
(cond [(file-exists? *status-file*)
(call-with-input-file *status-file* read)]
[else '()]))
;; is a challenge complete already?
(define (aoc-complete? day part)
(set-member? (aoc-get-status) (day+part->key day part)))
;; mark a challenge as completed
(define (aoc-set-complete! day part)
(define status (set-add (aoc-get-status) (day+part->key day part)))
(call-with-output-file
*status-file* (lambda (out) (write status out))
#:mode 'binary #:exists 'replace))