From 60a42f934c84e9a5f5c247373bfd10299dd47830 Mon Sep 17 00:00:00 2001
From: ShikiSuen <shikisuen@outlook.com>
Date: Wed, 16 Feb 2022 16:39:06 +0800
Subject: [PATCH] mgrLM // +chkIfUserPhraseExists to writeUserPhrase.

---
 Source/Modules/IMEModules/ctlInputMethod.swift  | 2 +-
 Source/Modules/LangModelRelated/mgrLangModel.h  | 2 +-
 Source/Modules/LangModelRelated/mgrLangModel.mm | 5 ++++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/Source/Modules/IMEModules/ctlInputMethod.swift b/Source/Modules/IMEModules/ctlInputMethod.swift
index ed0a0b0f..ecc3a120 100644
--- a/Source/Modules/IMEModules/ctlInputMethod.swift
+++ b/Source/Modules/IMEModules/ctlInputMethod.swift
@@ -603,7 +603,7 @@ extension ctlInputMethod: KeyHandlerDelegate {
         if !state.validToWrite {
             return false
         }
-        mgrLangModel.writeUserPhrase(state.userPhrase, inputMode: keyHandler.inputMode)
+        mgrLangModel.writeUserPhrase(state.userPhrase, inputMode: keyHandler.inputMode, areWeDuplicating: state.chkIfUserPhraseExists)
         return true
     }
 }
diff --git a/Source/Modules/LangModelRelated/mgrLangModel.h b/Source/Modules/LangModelRelated/mgrLangModel.h
index 5a7ac5e7..81ac9df8 100644
--- a/Source/Modules/LangModelRelated/mgrLangModel.h
+++ b/Source/Modules/LangModelRelated/mgrLangModel.h
@@ -33,7 +33,7 @@ NS_ASSUME_NONNULL_BEGIN
 + (BOOL)checkIfUserDataFolderExists;
 
 + (BOOL)checkIfUserPhraseExist:(NSString *)userPhrase key:(NSString *)key NS_SWIFT_NAME(checkIfExist(userPhrase:key:));
-+ (BOOL)writeUserPhrase:(NSString *)userPhrase inputMode:(InputMode)mode;
++ (BOOL)writeUserPhrase:(NSString *)userPhrase inputMode:(InputMode)mode areWeDuplicating:(BOOL)areWeDuplicating;
 + (void)setPhraseReplacementEnabled:(BOOL)phraseReplacementEnabled;
 + (void)setCNSEnabled:(BOOL)cnsEnabled;
 + (NSString *)userPhrasesDataPath:(InputMode)mode;
diff --git a/Source/Modules/LangModelRelated/mgrLangModel.mm b/Source/Modules/LangModelRelated/mgrLangModel.mm
index e950c1e6..9555b122 100644
--- a/Source/Modules/LangModelRelated/mgrLangModel.mm
+++ b/Source/Modules/LangModelRelated/mgrLangModel.mm
@@ -219,7 +219,7 @@ static void LTLoadLanguageModelFile(NSString *filenameWithoutExtension, vChewing
     return NO;
 }
 
-+ (BOOL)writeUserPhrase:(NSString *)userPhrase inputMode:(InputMode)mode;
++ (BOOL)writeUserPhrase:(NSString *)userPhrase inputMode:(InputMode)mode areWeDuplicating:(BOOL)areWeDuplicating
 {
     if (![self checkIfUserLanguageModelFilesExist]) {
         return NO;
@@ -251,6 +251,9 @@ static void LTLoadLanguageModelFile(NSString *filenameWithoutExtension, vChewing
     //     [currentMarkedPhrase appendString:@"\n"];
     // }
     [currentMarkedPhrase appendString:userPhrase];
+    if (areWeDuplicating) {
+        [currentMarkedPhrase appendString:@"\tFreqOverrideByDuplication"];
+    }
     [currentMarkedPhrase appendString:@"\n"];
 
     NSFileHandle *writeFile = [NSFileHandle fileHandleForUpdatingAtPath:path];