forked from cadence/breezewiki
Compare commits
3 commits
51bf087b30
...
19aeeea60f
Author | SHA1 | Date | |
---|---|---|---|
19aeeea60f | |||
ba806d0633 | |||
bc07a37bf7 |
5 changed files with 37 additions and 16 deletions
|
@ -9,6 +9,8 @@
|
|||
"url-utils.rkt")
|
||||
|
||||
(provide
|
||||
; header to not send referers to fandom
|
||||
referrer-policy
|
||||
; timeout durations for http-easy requests
|
||||
timeouts
|
||||
; generates a consistent footer
|
||||
|
@ -22,6 +24,7 @@
|
|||
(require rackunit
|
||||
html-writing))
|
||||
|
||||
(define referrer-policy (header #"Referrer-Policy" #"no-referrer"))
|
||||
(define timeouts (easy:make-timeout-config #:lease 5 #:connect 5))
|
||||
|
||||
(define (application-footer source-url #:license [license-in #f])
|
||||
|
|
|
@ -113,6 +113,7 @@
|
|||
(xexp->html body))
|
||||
(response/output
|
||||
#:code 200
|
||||
#:headers (list referrer-policy)
|
||||
(λ (out)
|
||||
(write-html body out))))))
|
||||
(module+ test
|
||||
|
|
|
@ -33,13 +33,24 @@
|
|||
(easy:response-headers-ref dest-res 'content-type))
|
||||
|
||||
(define (get-media-html url content-type)
|
||||
(define maybe-proxied-url (if (config-true? 'strict_proxy) (u-proxy-url url) url))
|
||||
(cond
|
||||
[(eq? content-type #f) `""]
|
||||
[(regexp-match? #rx"(?i:^image/)" content-type) `(img (@ (src ,url)))]
|
||||
[(regexp-match? #rx"(?i:^image/)" content-type) `(img (@ (src ,maybe-proxied-url)))]
|
||||
[(regexp-match? #rx"(?i:^audio/|^application/ogg(;|$))" content-type)
|
||||
`(audio (@ (src ,url) (controls)))]
|
||||
[(regexp-match? #rx"(?i:^video/)" content-type) `(video (@ (src ,url) (controls)))]
|
||||
`(audio (@ (src ,maybe-proxied-url) (controls)))]
|
||||
[(regexp-match? #rx"(?i:^video/)" content-type) `(video (@ (src ,maybe-proxied-url) (controls)))]
|
||||
[else `""]))
|
||||
(module+ test
|
||||
(require rackunit)
|
||||
(parameterize ([config-parameter 'strict_proxy "true"])
|
||||
(check-equal? (get-media-html "https://static.wikia.nocookie.net/a" "image/jpeg") (img (@ (src "/proxy?dest=https%3A%2F%2Fstatic.wikia.nocookie.net%2Fa"))))
|
||||
(check-equal? (get-media-html "https://static.wikia.nocookie.net/b" "audio/mp3") (audio (@ (src "/proxy?dest=https%3A%2F%2Fstatic.wikia.nocookie.net%2Fb")))))
|
||||
(parameterize ([config-parameter 'strict_proxy "no"])
|
||||
(check-equal? (get-media-html "https://static.wikia.nocookie.net/c" "application/ogg") (audio (@ (src "/proxy?dest=https%3A%2F%2Fstatic.wikia.nocookie.net%2Fc"))))
|
||||
(check-equal? (get-media-html "https://static.wikia.nocookie.net/d" "video/mp4") (video (@ (src "/proxy?dest=https%3A%2F%2Fstatic.wikia.nocookie.net%2Fd")))))
|
||||
(check-equal? (get-media-html "https://example.com" "who knows") "")
|
||||
(check-equal? (get-media-html #f "who knows") ""))
|
||||
|
||||
(define (generate-results-page #:source-url source-url
|
||||
#:wikiname wikiname
|
||||
|
@ -125,6 +136,7 @@
|
|||
; convert to string with error checking, error will be raised if xexp is invalid
|
||||
(xexp->html body))
|
||||
(response/output #:code 200
|
||||
#:headers (list referrer-policy)
|
||||
(λ (out) (write-html body out)))))))
|
||||
;(module+ test
|
||||
; (check-not-false ((query-selector (attribute-selector 'href "/test/wiki/Ankle_Monitor")
|
||||
|
|
|
@ -81,6 +81,7 @@
|
|||
(xexp->html body))
|
||||
(response/output
|
||||
#:code 200
|
||||
#:headers (list referrer-policy)
|
||||
(λ (out)
|
||||
(write-html body out))))))
|
||||
(module+ test
|
||||
|
|
|
@ -152,15 +152,17 @@
|
|||
(λ (v) (dict-update v 'rel (λ (s)
|
||||
(list (string-append (car s) " noreferrer")))
|
||||
'(""))))
|
||||
; proxy images from inline styles
|
||||
(curry attribute-maybe-update 'style
|
||||
(λ (style)
|
||||
(regexp-replace #rx"url\\(['\"]?(.*?)['\"]?\\)" style
|
||||
(λ (whole url)
|
||||
(string-append
|
||||
"url("
|
||||
(u-proxy-url url)
|
||||
")")))))
|
||||
; proxy images from inline styles, if strict_proxy is set
|
||||
(curry u
|
||||
(λ (v) (config-true? 'strict_proxy))
|
||||
(λ (v) (attribute-maybe-update 'style
|
||||
(λ (style)
|
||||
(regexp-replace #rx"url\\(['\"]?(.*?)['\"]?\\)" style
|
||||
(λ (whole url)
|
||||
(string-append
|
||||
"url("
|
||||
(u-proxy-url url)
|
||||
")")))) v)))
|
||||
; and also their links, if strict_proxy is set
|
||||
(curry u
|
||||
(λ (v)
|
||||
|
@ -168,8 +170,10 @@
|
|||
(eq? element-type 'a)
|
||||
(has-class? "image-thumbnail" v)))
|
||||
(λ (v) (attribute-maybe-update 'href u-proxy-url v)))
|
||||
; proxy images from src attributes
|
||||
(curry attribute-maybe-update 'src u-proxy-url)
|
||||
; proxy images from src attributes, if strict_proxy is set
|
||||
(curry u
|
||||
(λ (v) (config-true? 'strict_proxy))
|
||||
(λ (v) (attribute-maybe-update 'src u-proxy-url v)))
|
||||
; don't lazyload images
|
||||
(curry u
|
||||
(λ (v) (dict-has-key? v 'data-src))
|
||||
|
@ -276,8 +280,8 @@
|
|||
(define headers (if redirect-msg
|
||||
(let* ([dest (get-attribute 'href (bits->attributes ((query-selector (λ (t a c) (eq? t 'a)) redirect-msg))))]
|
||||
[value (bytes-append #"0;url=" (string->bytes/utf-8 dest))])
|
||||
(list (header #"Refresh" value)))
|
||||
(list)))
|
||||
(list (header #"Refresh" value) referrer-policy))
|
||||
(list referrer-policy)))
|
||||
(when (config-true? 'debug)
|
||||
; used for its side effects
|
||||
; convert to string with error checking, error will be raised if xexp is invalid
|
||||
|
|
Loading…
Reference in a new issue