improve synonyms

1 files changed, 55 insertions, 0 deletions

diff --git a/lib/synonyms.fnl b/lib/synonyms.fnl
new file mode 100644
index 0000000..d3f314f
--- /dev/null
+++ b/lib/synonyms.fnl
@@ -0,0 +1,55 @@
+(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}