typesetting/pitfall/restructure/string.rkt

#lang restructure/racket
(require "number.rkt" (prefix-in utils- "utils.rkt") "stream.rkt")
(provide (all-defined-out))

#|
approximates
https://github.com/mbutterick/restructure/blob/master/src/String.coffee
|#

(define (byteLength val encoding)
  (define encoder
    (caseq encoding
         [(ascii utf8) string->bytes/utf-8]))
  (bytes-length (encoder (format "~a" val))))

(define-subclass Streamcoder (StringT [length_ #f] [encoding_ 'ascii])
         
  (define/augment (decode stream [parent #f])
    (define length__
      (cond
        [length_ (utils-resolveLength length_ stream parent)]
        [else (send stream count-nonzero-chars)]))
    (define encoding__
      (cond
        [(procedure? encoding_) (or (encoding_ parent) 'ascii)]
        [else encoding_]))
    (define string (send stream readString length__ encoding__))
    (when (and (not length_) (< (send stream pos) (send stream length)))
      (send stream pos (add1 (send stream pos))))
    string)
    

  (define/augment (encode stream val-in [parent #f])
    (define val (format "~a" val-in))
    (define encoding__
         (cond
           [(procedure? encoding_) (or (encoding_ (and parent (· parent val)) 'ascii))]
           [else encoding_]))
    (when (NumberT? length_)
      (send length_ encode stream (byteLength val encoding__)))
    (send stream writeString val encoding__)
    (when (not length_)
      (send stream writeUInt8 #x00)))
    
    

  (define/override (size [val #f] [parent #f])
    ;; Use the defined value if no value was given
    (cond
      [(not val) (utils-resolveLength length_ #f parent)]
      [else
       (define encoding__
         (cond
           [(procedure? encoding_) (or (encoding_ (and parent (· parent val)) 'ascii))]
           [else encoding_]))
       (when (eq? encoding__ 'utf16be)
         (set! encoding__ 'utf16le))
       (define size (byteLength val encoding__))
       (when (NumberT? length_)
         (increment! size (send length_ size)))
       (when (not length_)
         (increment! size))
       size]))
    

  )

(define-values (String? +String) (values StringT? +StringT))

#;(test-module
   (require "stream.rkt")
   (define stream (+DecodeStream #"\2BCDEF"))
   (define S (+String uint8 'utf8))
   (check-equal? (send S decode stream) "BC")
   (define os (+EncodeStream))
   (send S encode os "Mike")
   (check-equal? (send os dump) #"\4Mike")
   (check-equal? (send (+String) size "foobar") 6))