(import-macros {: map} :lib.macro)

(local str (require :lib.string))
(local array (require :lib.array))
(local {: must} (require :lib.utils))

(local synonyms
  [["шэн" "шен"]
   ["билочунь" "би ло чунь"]
   ["бутылка" "термос"]
   ["доска" "чабань"]
   ["даньцун" "дяньцун" "дянь цун" "дань цун"]
   ["гуанбай" "гуан бай"]
   ["буланшань" "булан шань"]
   ["биндао" "бин дао"]
   ["лунцзин" "лун цзин"]
   ["термос" "бутылка"]
   ["чахай" "сливник"]
   ["те гуань инь" "те гуаньинь" "тегуань инь" "тегуаньинь"]
   ["чжень" "чжэнь"]
   ["точа" "то ча"]
   ["гушу" "гу шу"]])

(fn find-synonyms [word]
  (var input-word (must (luna.utf8.lower word)))
  (var result nil)
  (each [_ synonyms-line (ipairs synonyms) &until result]
    (each [_ synonym-word (ipairs synonyms-line) &until result]
      (when (= input-word synonym-word)
        (set result synonyms-line))))
  (print result)
  result)

(fn replace [line]
  (var results [])
  (let [words (str.split line " ")
        words-count (# words)]
    (for [i 1 words-count]
      (var word nil)
      (for [j i words-count]
        (set word
          (if (not word)
            (. words j)
            (.. word " " (. words j))))
        (table.insert results word))))
  (var final-result line)
  (each [_ result (ipairs results)]
    (local synonyms (find-synonyms result))
    (when synonyms
      (var query (.. (array.join synonyms " OR ") "*"))
      (set final-result
        (: final-result :gsub result (.. "(" query ")")))))
  final-result)

{: replace}