breezewiki/src/config.rkt

116 lines
3.9 KiB
Racket
Raw Normal View History

2022-10-04 09:00:44 +00:00
#lang typed/racket/base
(require racket/function
racket/pretty
2022-08-30 10:13:26 +00:00
racket/runtime-path
racket/string
typed/ini)
(provide
config-parameter
2022-08-24 08:59:11 +00:00
config-true?
config-get)
2022-10-04 09:00:44 +00:00
(module+ test
(require "../lib/typed-rackunit.rkt"))
2022-10-04 09:00:44 +00:00
2022-08-24 08:59:11 +00:00
(define-runtime-path path-config "../config.ini")
2022-10-04 09:00:44 +00:00
(: config-parameter (Symbol -> (Parameterof String)))
(define (config-parameter key)
(hash-ref config key))
2022-10-04 09:00:44 +00:00
(: config-true? (Symbol -> Boolean))
2022-08-24 08:59:11 +00:00
(define (config-true? key)
(not (member ((config-parameter key)) '("" "false"))))
2022-10-04 09:00:44 +00:00
(: config-get (Symbol -> String))
(define (config-get key)
((config-parameter key)))
(define default-config
'((application_name . "BreezeWiki")
(canonical_origin . "")
2022-08-24 08:59:11 +00:00
(debug . "false")
(feature_search_suggestions . "true")
(instance_is_official . "false") ; please don't turn this on, or you will make me very upset
(log_outgoing . "true")
(port . "10416")
(strict_proxy . "false")
(feature_offline::enabled . "false")
(feature_offline::format . "json.gz")
(feature_offline::only . "false")
(access_log::enabled . "false")))
(define loaded-alist
(with-handlers
([exn:fail:filesystem:errno?
(λ (exn)
(displayln "note: config file not detected, using defaults")
'())]
[exn:fail:contract?
(λ (exn)
(displayln "note: config file empty or missing [] section, using defaults")
'())])
(define h (in-hash
(ini->hash
(call-with-input-file path-config
(λ (in)
(read-ini in))))))
(define l
(for*/list : (Listof (Pairof Symbol String))
([(section-key section) h]
[(key value) (in-hash section)])
(if (eq? section-key '||)
(cons key value)
(cons (string->symbol (string-append (symbol->string section-key)
"::"
(symbol->string key)))
value))))
(printf "note: ~a items loaded from config file~n" (length l))
l))
(define env-alist
(for/list : (Listof (Pairof Symbol String))
([name (environment-variables-names (current-environment-variables))]
#:when (string-prefix? (string-downcase (bytes->string/latin-1 name)) "bw_"))
(cons
;; key: convert to string, remove bw_ prefix, convert to symbol
(string->symbol (string-downcase (substring (bytes->string/latin-1 name) 3)))
;; value: convert to string
(bytes->string/latin-1
(cast (environment-variables-ref (current-environment-variables) name) Bytes)))))
(when (> (length env-alist) 0)
(printf "note: ~a items loaded from environment variables~n" (length env-alist)))
(define combined-alist (append default-config loaded-alist env-alist))
(define config
(for/hasheq ([pair combined-alist]) : (Immutable-HashTable Symbol (Parameter String))
(values (car pair) (make-parameter (cdr pair)))))
2022-08-24 08:59:11 +00:00
(when (config-true? 'debug)
2022-08-30 12:45:37 +00:00
; all values here are optimised for maximum prettiness
2022-08-30 10:13:26 +00:00
(parameterize ([pretty-print-columns 80])
(display "config: ")
2022-10-04 09:00:44 +00:00
(pretty-write ((inst sort (Pairof Symbol String))
(hash->list (make-immutable-hasheq combined-alist))
symbol<?
#:key car))))
(when (not (config-true? 'debug))
(when (not (config-true? 'canonical_origin))
(displayln
(string-append "warning: configuring canonical_origin is highly recommended for production!\n"
" see https://docs.breezewiki.com/Configuration.html"))))
2022-10-04 09:00:44 +00:00
(module+ test
; this is just a sanity check
(parameterize ([(config-parameter 'application_name) "JeffWiki"]
[(config-parameter 'strict_proxy) ""])
(check-equal? (config-get 'application_name) "JeffWiki")
(check-false (config-true? 'strict_proxy))
(check-equal? (string? (config-get 'feature_offline::format)) #t)))