修改数据结构

src/hooks/useSplitArticle.ts

@@ -1,117 +1,159 @@
-import {Sentence} from "@/types.ts";
+import {DefaultArticleWord, Sentence, Word} from "@/types.ts";
+import {cloneDeep} from "lodash";
 
 interface KeyboardMap {
-  Period: string,
-  Comma: string,
-  Slash: string,
-  Exclamation: string,
-  Quote: string,
+    Period: string,
+    Comma: string,
+    Slash: string,
+    Exclamation: string,
+    Quote: string,
 }
 
 export const CnKeyboardMap: KeyboardMap = {
-  Period: '。',
-  Comma: '，',
-  Slash: '？',
-  Exclamation: '！',
-  Quote: '”',
+    Period: '。',
+    Comma: '，',
+    Slash: '？',
+    Exclamation: '！',
+    Quote: '”',
 }
 export const EnKeyboardMap: KeyboardMap = {
-  Period: '.',
-  Comma: ',',
-  Slash: '?',
-  Exclamation: '!',
-  Quote: '"',
+    Period: '.',
+    Comma: ',',
+    Slash: '?',
+    Exclamation: '!',
+    Quote: '"',
 }
 
+
 export function useSplitArticle(article: string, lang: string = 'en', keyboardMap: KeyboardMap = EnKeyboardMap): Sentence[][] {
-  let sections = []
-  let section = []
-  let sentence = {
-    sentence: '',
-    words: []
-  }
-  let sentences = []
-  let word = '';
-  // console.log(article,)
-  //加\n用于添加最后一段
-  article += '\n'
-  if (lang === 'en') {
-    article = article.replaceAll(`‘`, '"')
-    article = article.replaceAll(`’`, '"')
-    article = article.replaceAll(`“`, '"')
-    article = article.replaceAll(`”`, '"')
-  }
-  // console.log('article', article)
-
-  article.split('').map(v => {
-    switch (v) {
-      case ' ':
-        sentence.words.push(word)
-        word = ''
-        break
-      case keyboardMap.Period:
-      case keyboardMap.Comma:
-      case keyboardMap.Slash:
-      case keyboardMap.Exclamation:
-        word += v
-        sentence.words.push(word)
-        sentence.words = sentence.words.filter(v => v)
-        sentence.sentence = sentence.words.join(' ')
-        sentences.push({
-          target: sentence.sentence,
-          trans: '',
-          location: `${sections.length}-${section.length}`
-        })
-        // sentence.words.push(word)
-        // sentence.words = sentence.words.filter(v => v)
-        // sentence.sentence = sentence.words.join(' ')
-        // sentence.sentence += v
-        // sentence.words.push(v)
-        section.push(sentence)
-        sentence = {
-          sentence: '',
-          words: []
-        }
-        word = ''
-        break
-      case keyboardMap.Quote:
-        let lastSentence = {
-          sentence: '',
-          words: []
-        }
-        if (section.length) {
-          lastSentence = section[section.length - 1]
-        } else {
-          let lastSection = sections[sections.length - 1]
-          lastSentence = lastSection[lastSection.length - 1]
-        }
-        if (lastSentence.sentence.includes(keyboardMap.Quote)) {
-          lastSentence.sentence += keyboardMap.Quote
-          lastSentence.words[lastSentence.words.length - 1] += keyboardMap.Quote
-        } else {
-          word += v
-        }
-        // console.log('lastSentence', lastSentence)
-        break
-      case '\n':
-        if (section.length) {
-          sections.push(section)
-          section = []
-          sentence = {
-            sentence: '',
-            words: []
-          }
-          word = ''
-        }
-        break
-      default:
-        word += v
-        break
+    let sections: Sentence[][] = []
+    let section: Sentence[] = []
+    let sentence: Sentence = {
+        sentence: '',
+        words: []
     }
-  })
-  if (!sections.length && section.length) {
+    section.push(sentence)
     sections.push(section)
-  }
+    let word = cloneDeep({...DefaultArticleWord, name: '', nextSpace: true});
+    //加\n用于添加最后一段
+    article += '\n'
+    if (lang === 'en') {
+        article = article.replaceAll(`‘`, '"')
+        article = article.replaceAll(`’`, '"')
+        article = article.replaceAll(`“`, '"')
+        article = article.replaceAll(`”`, '"')
+    }
 
-  return sections
+    // console.log('article', article)
+
+    article.split('').map((v, i, arr) => {
+        switch (v) {
+            case ' ':
+                if (word.name) {
+                    sentence.words.push(word)
+                    word = cloneDeep(DefaultArticleWord)
+                }
+                break
+            case keyboardMap.Period:
+            case keyboardMap.Comma:
+            case keyboardMap.Slash:
+            case keyboardMap.Exclamation:
+                word.nextSpace = false
+                sentence.words.push(word)
+                sentence.words.push(cloneDeep({...DefaultArticleWord, name: v, nextSpace: true}))
+                // sentence.words = sentence.words.filter(v => v)
+                sentence.sentence = sentence.words.map(v => v.name).join(' ')
+                section.push({
+                    sentence: '',
+                    words: []
+                })
+                sentence = section[section.length - 1]
+                word = cloneDeep(DefaultArticleWord)
+                break
+            case keyboardMap.Quote:
+                let lastSentence = {
+                    sentence: '',
+                    words: []
+                }
+                let startSymbol = null
+                let indexs = {
+                    a: -1,
+                    b: -1,
+                    c: -1
+                }
+                //TODO 可以优化成for+break
+                sections.toReversed().map((sectionItem, a) => {
+                    sectionItem.toReversed().map((sentenceItem, b) => {
+                        sentenceItem.words.toReversed().map((wordItem, c) => {
+                            if (wordItem.symbolPosition !== '' && !startSymbol) {
+                                startSymbol = wordItem.symbolPosition === 'end'
+                                indexs = {a, b, c}
+                            }
+                        })
+                    })
+                })
+
+                debugger
+                if (startSymbol || startSymbol === null) {
+                    sentence.words.push(cloneDeep({
+                        ...DefaultArticleWord,
+                        name: v,
+                        nextSpace: false,
+                        isSymbol: true,
+                        symbolPosition: 'start'
+                    }))
+                    word = cloneDeep(DefaultArticleWord)
+                } else {
+                    let addCurrent = false
+                    sentence.words.toReversed().map((wordItem, c) => {
+                        if (wordItem.symbolPosition === 'start' && !addCurrent) {
+                            addCurrent = true
+                        }
+                    })
+                    if (addCurrent) {
+                        sentence.words.push(cloneDeep({
+                            ...DefaultArticleWord,
+                            name: v,
+                            nextSpace: true,
+                            isSymbol: true,
+                            symbolPosition: 'end'
+                        }))
+                        word = cloneDeep(DefaultArticleWord)
+                    } else {
+                        section[section.length - 2].words.push(cloneDeep({
+                            ...DefaultArticleWord,
+                            name: v,
+                            nextSpace: true,
+                            isSymbol: true,
+                            symbolPosition: 'end'
+                        }))
+                        section[section.length - 2].sentence = section[section.length - 2].words.map(v => v.name).join(' ')
+                    }
+                }
+                sentence.sentence = sentence.words.map(v => v.name).join(' ')
+
+                break
+            case '\n':
+                section.pop()
+                if (i !== arr.length - 1) {
+                    sections.push([])
+                    section = sections[sections.length - 1]
+                    section.push({
+                        sentence: '',
+                        words: []
+                    })
+                    sentence = section[section.length - 1]
+                    word = cloneDeep(DefaultArticleWord)
+                }
+                break
+            default:
+                word.name += v
+                break
+        }
+    })
+    // if (!sections.length && section.length) {
+    //     sections.push(section)
+    // }
+
+    return sections
 }