optionally follow redirects in `web-request!`
This commit is contained in:
parent
1dec02fd3d
commit
06aab927bb
|
@ -6,6 +6,7 @@
|
||||||
web-respond/status!
|
web-respond/status!
|
||||||
web-request!)
|
web-request!)
|
||||||
|
|
||||||
|
(require racket/dict)
|
||||||
(require (only-in file/sha1 bytes->hex-string))
|
(require (only-in file/sha1 bytes->hex-string))
|
||||||
(require (only-in racket/random crypto-random-bytes))
|
(require (only-in racket/random crypto-random-bytes))
|
||||||
(require net/url)
|
(require net/url)
|
||||||
|
@ -35,20 +36,33 @@
|
||||||
#:headers '())
|
#:headers '())
|
||||||
body))
|
body))
|
||||||
|
|
||||||
(define (web-request! verb urlstr #:body [body #""] #:headers [headers '()])
|
(define (web-request! verb urlstr
|
||||||
|
#:body [request-body #""]
|
||||||
|
#:headers [headers '()]
|
||||||
|
#:redirect-budget [redirect-budget 0])
|
||||||
(define req-id (gensym 'req))
|
(define req-id (gensym 'req))
|
||||||
(define u (string->url urlstr))
|
(define u (string->url urlstr))
|
||||||
(define res (url->resource u))
|
(define res (url->resource u))
|
||||||
(react/suspend (k)
|
(define-values (resp response-body)
|
||||||
(on-start
|
(react/suspend (k)
|
||||||
(printf "~a --> ~a ~a ~v\n" req-id verb urlstr body)
|
(on-start
|
||||||
(send! (web-request req-id
|
(printf "~a --> ~a ~a ~v\n" req-id verb urlstr request-body)
|
||||||
'outbound
|
(send! (web-request req-id
|
||||||
(web-request-header verb res headers (url-query u))
|
'outbound
|
||||||
body)))
|
(web-request-header verb res headers (url-query u))
|
||||||
(stop-when (message (web-response-complete req-id $resp $body))
|
request-body)))
|
||||||
(printf "~a <-- ~v ~v\n" req-id resp body)
|
(stop-when (message (web-response-complete req-id $resp $response-body))
|
||||||
(k resp body))))
|
(printf "~a <-- ~v ~v\n" req-id resp response-body)
|
||||||
|
(k resp response-body))))
|
||||||
|
(define location (dict-ref (web-response-header-headers resp) 'location #f))
|
||||||
|
(if (and (eq? (web-response-header-code-type resp) 'redirection)
|
||||||
|
(positive? redirect-budget)
|
||||||
|
location)
|
||||||
|
(web-request! verb location
|
||||||
|
#:body request-body
|
||||||
|
#:headers headers
|
||||||
|
#:redirect-budget (- redirect-budget 1))
|
||||||
|
(values resp response-body)))
|
||||||
|
|
||||||
(module+ test
|
(module+ test
|
||||||
(check-equal? (extend-url-string-query "http://localhost/" '((a . "hi")))
|
(check-equal? (extend-url-string-query "http://localhost/" '((a . "hi")))
|
||||||
|
|
Loading…
Reference in New Issue