// 定義一個(gè)Node結(jié)構(gòu)體，代表DFA的一個(gè)節(jié)點(diǎn)。
type Node struct {
	End  bool   // End字段表示是否為一個(gè)單詞的結(jié)束。
	Next map[rune]*Node   // Next字段是一個(gè)映射，用于存儲(chǔ)此節(jié)點(diǎn)的所有子節(jié)點(diǎn)。
}

// 定義一個(gè)DFAMatcher結(jié)構(gòu)體，代表一個(gè)完整的DFA。
type DFAMatcher struct {
	replaceChar rune   // replaceChar字段是替換敏感詞的字符。
	root        *Node   // root字段是DFA的根節(jié)點(diǎn)。
}

//創(chuàng)建出一個(gè)DFA樹(shù)的根節(jié)點(diǎn)實(shí)例
func NewDFAMather() *DFAMatcher {
	return &DFAMatcher{
		root: &Node{
			End: false,
		},
	}
}

// Build方法用于構(gòu)建DFA，它會(huì)將提供的所有單詞添加到DFA中。
func (d *DFAMatcher) Build(words []string) {
	for _, item := range words { // 遍歷提供的所有單詞。
		d.root.AddWord(item) // 將每一個(gè)單詞添加到DFA的根節(jié)點(diǎn)。
	}
}
// AddWord方法用于向當(dāng)前節(jié)點(diǎn)添加一個(gè)單詞。
// 這個(gè)方法會(huì)遍歷單詞的每一個(gè)字符，并為每一個(gè)字符添加一個(gè)子節(jié)點(diǎn)。
func (n *Node) AddWord(word string) {
	node := n                     // 從當(dāng)前節(jié)點(diǎn)開(kāi)始。
	chars := []rune(word)         // 將字符串轉(zhuǎn)化為rune類(lèi)型的切片，以便處理Unicode字符。
	for index, _ := range chars { // 遍歷單詞的每一個(gè)字符。
		node = node.AddChild(chars[index]) // 遞歸地為每一個(gè)字符添加子節(jié)點(diǎn)。
	}
	node.End = true // 設(shè)置最后一個(gè)節(jié)點(diǎn)為單詞的結(jié)束。
}
// AddChild方法向當(dāng)前節(jié)點(diǎn)添加一個(gè)子節(jié)點(diǎn)。
// 如果子節(jié)點(diǎn)已經(jīng)存在，它將不會(huì)被重復(fù)添加。
func (n *Node) AddChild(c rune) *Node {
	if n.Next == nil { // 如果Next字段為nil，則初始化一個(gè)映射。
		n.Next = make(map[rune]*Node)
	}
	//檢查字符c是否已經(jīng)是當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)。
	if next, ok := n.Next[c]; ok { // 如果ok為true，則字符c已經(jīng)是當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)，直接返回該子節(jié)點(diǎn)。
		return next
	} else { // 否則，創(chuàng)建一個(gè)新的節(jié)點(diǎn)，并將其設(shè)置為當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)。
		n.Next[c] = &Node{
			End:  false,
			Next: nil,
		}
		return n.Next[c] // 返回新創(chuàng)建的子節(jié)點(diǎn)。
	}
}

// Match方法用于在文本中查找并替換敏感詞。
// 它返回找到的敏感詞列表和替換后的文本。
func (d *DFAMatcher) Match(text string) (sensitiveWords []string, replaceText string) {
	if d.root == nil { // 如果DFA是空的，直接返回原始文本。
		return nil, text
	}
	textChars := []rune(text)                     // 將文本轉(zhuǎn)化為rune類(lèi)型的切片，以便處理Unicode字符。
	textCharsCopy := make([]rune, len(textChars)) // 創(chuàng)建一個(gè)文本字符的副本，用于替換敏感詞。
	copy(textCharsCopy, textChars)                // 復(fù)制原始文本字符到副本。
	length := len(textChars)                      // 獲取文本的長(zhǎng)度。
	for i := 0; i < length; i++ {                 // 遍歷文本的每一個(gè)字符。
	// 在DFA樹(shù)中查找當(dāng)前字符對(duì)應(yīng)的子節(jié)點(diǎn)
		temp := d.root.FindChild(textChars[i])
		if temp == nil {
			continue // 如果不存在匹配，繼續(xù)檢查下一個(gè)字符
		}
		j := i + 1
		// 遍歷文本中的字符，查找匹配的敏感詞，第一個(gè)匹配上了，就進(jìn)行后面的向下匹配
		for ; j < length && temp != nil; j++ {
			if temp.End {
				// 如果找到一個(gè)敏感詞，將其添加到結(jié)果列表中，并在副本中替換為指定字符
				sensitiveWords = append(sensitiveWords, string(textChars[i:j]))
				replaceRune(textCharsCopy, '*', i, j) //替換敏感詞
			}
			temp = temp.FindChild(textChars[j])
		}
		// 處理文本末尾的情況，如果末尾是一個(gè)完整的敏感詞，添加到結(jié)果列表中，并在副本中替換為指定字符
		if j == length && temp != nil && temp.End {
			sensitiveWords = append(sensitiveWords, string(textChars[i:length]))
			replaceRune(textCharsCopy, '*', i, length)
		}
	}
	return sensitiveWords, string(textCharsCopy) // 返回匹配到的敏感詞列表和替換后的文本
}
// FindChild方法用于在當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)中查找一個(gè)特定的子節(jié)點(diǎn)。
func (n *Node) FindChild(c rune) *Node {
	if n.Next == nil { // 如果Next字段為nil，則直接返回nil。
		return nil
	}
	//檢查字符c是否是當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)。
	if _, ok := n.Next[c]; ok { // 如果ok為true，則字符c是當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)，返回該子節(jié)點(diǎn)。
		return n.Next[c]
	}
	return nil // 否則，返回nil。
}
//替換掉文章中出現(xiàn)的關(guān)鍵詞
func replaceRune(chars []rune, replaceChar rune, begin int, end int) {
	for i := begin; i < end; i++ {
		chars[i] = replaceChar
	}
}