0

src/api/yjbdc/aiController.js

@@ -228,6 +228,247 @@ export async function generateArticle(content, provider = 'volces1-5') {
     }
 }
 
+/**
+ * 计算两个字符串之间的Levenshtein距离（编辑距离）
+ * @param {string} a - 第一个字符串
+ * @param {string} b - 第二个字符串
+ * @returns {number} - 编辑距离
+ */
+function levenshteinDistance(a, b) {
+    const matrix = [];
+    
+    // 初始化矩阵
+    for (let i = 0; i <= b.length; i++) {
+        matrix[i] = [i];
+    }
+    
+    for (let j = 0; j <= a.length; j++) {
+        matrix[0][j] = j;
+    }
+    
+    // 填充矩阵
+    for (let i = 1; i <= b.length; i++) {
+        for (let j = 1; j <= a.length; j++) {
+            if (b.charAt(i - 1) === a.charAt(j - 1)) {
+                matrix[i][j] = matrix[i - 1][j - 1];
+            } else {
+                matrix[i][j] = Math.min(
+                    matrix[i - 1][j - 1] + 1, // 替换
+                    matrix[i][j - 1] + 1,     // 插入
+                    matrix[i - 1][j] + 1      // 删除
+                );
+            }
+        }
+    }
+    
+    return matrix[b.length][a.length];
+}
+
+/**
+ * 增强FormsOfWords，检测文章中单词的变形形式和拼写错误
+ * @param {Object} jsonObj - 解析后的JSON对象
+ * @param {string} userWords - 用户提供的单词列表，逗号分隔
+ * @returns {Object} - 增强后的JSON对象
+ */
+export function enhanceFormsOfWords(jsonObj, userWords) {
+    if (!jsonObj || !userWords) return jsonObj;
+    
+    // 将用户提供的单词转换为数组并去除空格
+    const userWordsList = userWords.split(',').map(word => word.trim().toLowerCase());
+    
+    // 如果没有ArticleEnglish或FormsOfWords，直接返回
+    if (!jsonObj.ArticleEnglish || !Array.isArray(jsonObj.ArticleEnglish)) {
+        return jsonObj;
+    }
+    
+    // 确保FormsOfWords存在
+    if (!jsonObj.FormsOfWords) {
+        jsonObj.FormsOfWords = [];
+    }
+    
+    // 从文章中提取所有单词
+    const allWordsInArticle = [];
+    jsonObj.ArticleEnglish.forEach(sentence => {
+        // 移除标点符号，分割成单词
+        const words = sentence.toLowerCase()
+            .replace(/[.,\/#!$%\^&\*;:{}=\-_`~()]/g, " ")
+            .replace(/\s+/g, " ")
+            .split(" ");
+        
+        words.forEach(word => {
+            if (word) allWordsInArticle.push(word.trim());
+        });
+    });
+    
+    // 常见的后缀列表
+    const commonSuffixes = [
+        'ed', 'ing', 's', 'es', 'er', 'est', 'ful', 'ly', 'ment', 'ness', 'ity', 
+        'tion', 'sion', 'ation', 'able', 'ible', 'al', 'ial', 'ic', 'ical', 'ious', 
+        'ous', 'ive', 'less', 'y'
+    ];
+    
+    // 不规则动词变化表（部分常见的）
+    const irregularVerbs = {
+        'go': ['went', 'gone', 'goes', 'going'],
+        'be': ['am', 'is', 'are', 'was', 'were', 'been', 'being'],
+        'do': ['did', 'done', 'does', 'doing'],
+        'have': ['has', 'had', 'having'],
+        'say': ['said', 'says', 'saying'],
+        'make': ['made', 'makes', 'making'],
+        'take': ['took', 'taken', 'takes', 'taking'],
+        'come': ['came', 'comes', 'coming'],
+        'see': ['saw', 'seen', 'sees', 'seeing'],
+        'know': ['knew', 'known', 'knows', 'knowing'],
+        'get': ['got', 'gotten', 'gets', 'getting'],
+        'give': ['gave', 'given', 'gives', 'giving'],
+        'find': ['found', 'finds', 'finding'],
+        'think': ['thought', 'thinks', 'thinking'],
+        'tell': ['told', 'tells', 'telling'],
+        'become': ['became', 'becomes', 'becoming'],
+        'show': ['showed', 'shown', 'shows', 'showing'],
+        'leave': ['left', 'leaves', 'leaving'],
+        'feel': ['felt', 'feels', 'feeling'],
+        'put': ['puts', 'putting'],
+        'bring': ['brought', 'brings', 'bringing'],
+        'begin': ['began', 'begun', 'begins', 'beginning'],
+        'keep': ['kept', 'keeps', 'keeping'],
+        'hold': ['held', 'holds', 'holding'],
+        'write': ['wrote', 'written', 'writes', 'writing'],
+        'stand': ['stood', 'stands', 'standing'],
+        'hear': ['heard', 'hears', 'hearing'],
+        'let': ['lets', 'letting'],
+        'mean': ['meant', 'means', 'meaning'],
+        'set': ['sets', 'setting'],
+        'meet': ['met', 'meets', 'meeting'],
+        'run': ['ran', 'runs', 'running'],
+        'pay': ['paid', 'pays', 'paying'],
+        'sit': ['sat', 'sits', 'sitting'],
+        'speak': ['spoke', 'spoken', 'speaks', 'speaking'],
+        'lie': ['lay', 'lain', 'lies', 'lying'],
+        'lead': ['led', 'leads', 'leading'],
+        'read': ['read', 'reads', 'reading'],
+        'sleep': ['slept', 'sleeps', 'sleeping'],
+        'win': ['won', 'wins', 'winning'],
+        'understand': ['understood', 'understands', 'understanding'],
+        'draw': ['drew', 'drawn', 'draws', 'drawing'],
+        'sing': ['sang', 'sung', 'sings', 'singing'],
+        'fall': ['fell', 'fallen', 'falls', 'falling'],
+        'fly': ['flew', 'flown', 'flies', 'flying'],
+        'grow': ['grew', 'grown', 'grows', 'growing'],
+        'lose': ['lost', 'loses', 'losing'],
+        'teach': ['taught', 'teaches', 'teaching'],
+        'eat': ['ate', 'eaten', 'eats', 'eating'],
+        'drink': ['drank', 'drunk', 'drinks', 'drinking']
+    };
+    
+    // 不规则形容词比较级和最高级
+    const irregularAdjectives = {
+        'good': ['better', 'best'],
+        'bad': ['worse', 'worst'],
+        'far': ['further', 'furthest', 'farther', 'farthest'],
+        'little': ['less', 'least'],
+        'many': ['more', 'most'],
+        'much': ['more', 'most']
+    };
+    
+    // 收集所有单词形式
+    const allForms = new Set();
+    
+    userWordsList.forEach(originalWord => {
+        // 添加原始单词
+        allForms.add(originalWord);
+        
+        // 检查不规则动词
+        if (irregularVerbs[originalWord]) {
+            irregularVerbs[originalWord].forEach(form => allForms.add(form));
+        }
+        
+        // 检查不规则形容词
+        if (irregularAdjectives[originalWord]) {
+            irregularAdjectives[originalWord].forEach(form => allForms.add(form));
+        }
+        
+        // 检查文章中的所有单词，寻找可能的变形和拼写错误
+        allWordsInArticle.forEach(articleWord => {
+            // 检查是否是原始单词
+            if (articleWord === originalWord) {
+                allForms.add(articleWord);
+                return;
+            }
+            
+            // 检查拼写错误，使用更严格的条件
+            // 1. 对于短单词（长度<=4），只接受编辑距离为1的情况
+            // 2. 对于中等长度单词（4<长度<=8），只接受编辑距离为1的情况
+            // 3. 对于长单词（长度>8），允许编辑距离为2，但有额外限制
+            if (originalWord.length <= 8) {
+                // 短单词和中等长度单词使用相同的严格条件
+                if (articleWord[0] === originalWord[0] && // 首字母必须相同
+                    Math.abs(articleWord.length - originalWord.length) <= 1 && // 长度差不超过1
+                    levenshteinDistance(originalWord, articleWord) === 1) { // 编辑距离恰好为1
+                    allForms.add(articleWord);
+                }
+            } else {
+                // 长单词（长度>8）的条件
+                if (articleWord[0] === originalWord[0]) { // 首字母必须相同
+                    const editDistance = levenshteinDistance(originalWord, articleWord);
+                    
+                    if (editDistance === 1) {
+                        // 编辑距离为1的情况，长度差不超过1
+                        if (Math.abs(articleWord.length - originalWord.length) <= 1) {
+                            allForms.add(articleWord);
+                        }
+                    } else if (editDistance === 2) {
+                        // 编辑距离为2的情况，需要更严格的条件
+                        // 长度差不超过1且单词长度大于8
+                        if (Math.abs(articleWord.length - originalWord.length) <= 1) {
+                            allForms.add(articleWord);
+                        }
+                    }
+                }
+            }
+            
+            // 检查是否是通过添加后缀形成的变形
+            for (const suffix of commonSuffixes) {
+                if (articleWord.endsWith(suffix)) {
+                    const stem = articleWord.slice(0, -suffix.length);
+                    
+                    // 处理双写字母的情况（如：running -> run）
+                    if (stem.length > 0 && stem[stem.length-1] === stem[stem.length-2]) {
+                        const possibleStem = stem.slice(0, -1);
+                        if (possibleStem === originalWord) {
+                            allForms.add(articleWord);
+                            continue;
+                        }
+                    }
+                    
+                    // 处理去e加ing的情况（如：writing -> write）
+                    if (suffix === 'ing' && originalWord.endsWith('e') && stem + 'e' === originalWord) {
+                        allForms.add(articleWord);
+                        continue;
+                    }
+                    
+                    // 处理y变i的情况（如：studies -> study）
+                    if ((suffix === 'es' || suffix === 'ed') && originalWord.endsWith('y') && 
+                        stem + 'y' === originalWord) {
+                        allForms.add(articleWord);
+                        continue;
+                    }
+                    
+                    // 直接比较
+                    if (stem === originalWord) {
+                        allForms.add(articleWord);
+                    }
+                }
+            }
+        });
+    });
+    
+    // 更新FormsOfWords
+    jsonObj.FormsOfWords = Array.from(new Set([...jsonObj.FormsOfWords, ...allForms]));
+    
+    return jsonObj;
+}
+
 /**
  * 校验和修复JSON结构
  * @param {string} jsonString - JSON字符串
@@ -365,5 +606,6 @@ export function validateAndFixJSON(jsonString) {
 
 export default {
     generateArticle,
+    enhanceFormsOfWords,
     validateAndFixJSON
 };

src/model/yjbdc.js

@@ -102,8 +102,14 @@ class YJBDC {
     static async GetYJBDCArticleList(obj) {
         try {
             if (obj.ID>0){
-                let sql = "SELECT * FROM YJBDC_Articles WHERE UserID=? and ID=? and Flag=0;";
-                return await query(sql, [obj.UserID,obj.ID]);
+                if (obj.UserID<4){
+                    let sql = "SELECT * FROM YJBDC_Articles WHERE ID=? and Flag=0;";
+                    return await query(sql, [obj.ID]);
+                }
+                else{
+                    let sql = "SELECT * FROM YJBDC_Articles WHERE UserID=? and ID=? and Flag=0;";
+                    return await query(sql, [obj.UserID,obj.ID]);
+                }
             }
             else{
                 let sql = "SELECT `ID`,`UserID`,`CreateTime`,`Words`,`Level`,`ArticleStyle`,`ArticleStart` FROM YJBDC_Articles WHERE UserID=? and Flag=0 order by ID desc;";

src/test/data.txt

@@ -0,0 +1,27 @@
+Today is a special anniversary for our family. My parents
+and I decided to do something different to celebrate it.
+First, we went to the court near our home to play
+badminton. We had a great time there and I could feel the
+love and warmth in our family. After that, we went to a
+restaurant. I had a big appetite and ate a lot. My parents
+said they appreciate the time we spend together. On the
+way home, we talked about our ancestors and how they
+lived in the past. My father advocated that we should
+learn from them and work hard. We also discussed some
+alternatives for our future family activities. Suddenly, there
+was an announcement on the radio about a new park in
+our city. We were very interested in it and decided to visit
+it next time. As we walked, I absorbed the beautiful
+scenery around me. When we got home, my parents
+applauded my efforts in today's activities.
+
+absorb
+abvocate
+alternative
+ancestor
+anniversary
+announcement
+appetite
+applaud
+appreciate
+court