search for combo

words/main.rkt

@@ -4,6 +4,7 @@
 
 (define (make-words #:letters [letters "etaoinshrdluw"]
                     #:mandatory [mandatory #f]
+                    #:combo [combo #f]
                     #:min [min-length 5]
                     #:max [max-length 10]
                     #:hide-plurals [hide-plurals? #t]
@@ -12,7 +13,9 @@
                     #:all-caps [all-caps? #f]
                     #:initial-caps [initial-caps? #f])
   (define mandatory-cs
-    (if mandatory (remove-duplicates (for/list ([c (in-string mandatory)])
+    (if (or mandatory combo)
+        (remove-duplicates
+         (for/list ([c (in-string (string-append (or mandatory "") (or combo "")))])
                    (char-downcase c)) char=?) null))
   (define letter-cs-charidx
     (word->charidx
@@ -34,17 +37,19 @@
              ([idx (in-list (shuffle (range (vector-length wordrecs))))]
               [rec (in-value (vector-ref wordrecs idx))]
               [word-charidx (in-value (word-rec-charint rec))]
+              [word (in-value (word-rec-word rec))]
               #:break (= count (or max-words +inf.0))
               #:when (and
                       ;; between min and max length
                       ((if (<= min-length max-length) <= >=) min-length (word-rec-length rec) max-length)
                       ;; word contains each mandatory char, case-insensitive
-                      (or (not mandatory)
                       (for/and ([mc (in-list mandatory-cs)])
-                            (word-charidx . contains-char? . mc)))
+                               (word-charidx . contains-char? . mc))
                       ;; word contains only letters + mandatory, case-insensitive
                       (for/and ([wc (in-list (map char-downcase (charidx->chars word-charidx)))])
                                (letter-cs-charidx . contains-char? . wc))
+                      (or (not combo)
+                          (regexp-match combo word))
                       ;; maybe only proper names
                       (if proper-names?
                           (capitalized? word-charidx)
@@ -52,10 +57,10 @@
                       ;; maybe hide plurals
                       (or (not hide-plurals?)
                           (not (word-rec-plural? rec)))))
-    (values (cons (caser (word-rec-word rec)) word-acc) (add1 count))))
+    (values (cons (caser word) word-acc) (add1 count))))
 
 (module+ test
   (require rackunit)
   (time (make-words))
-  (check-equal? (sort (make-words #:mandatory "xyz") string<?)
+  (check-equal? (sort (make-words #:mandatory "xyz" #:combo #false) string<?)
                   '("azoxy" "dysoxidize" "isazoxy" "oxytonize" "rhizotaxy" "zootaxy")))