thinkphp5敏感词过滤类

<?php /** * 敏感词类库. * User: Lustre * Date: 17/3/9 * Time: 上午9:11 */ namespace DfaFilter; use DfaFilter\Exceptions\PdsBusinessException; class SensitiveHelper { /** * 待检测语句长度 * * @var int */ protected $contentLength = 0; /** * 敏感词单例 * * @var object|null */ private static $_instance = null; /** * 铭感词库树 * * @var HashMap|null */ protected $wordTree = null; /** * 存放待检测语句铭感词 * * @var array|null */ protected static $badWordList = null; /** * 获取单例 * * @return self */ public static function init() { if (! self::$_instance instanceof self) { self::$_instance = new self(); } return self::$_instance; } /** * 构建铭感词树【文件模式】 * * @param string $filepath * * @return $this * @throws \DfaFilter\Exceptions\PdsBusinessException */ public function setTreeByFile($filepath = '') { if (!file_exists($filepath)) { throw new PdsBusinessException('词库文件不存在', PdsBusinessException::CANNOT_FIND_FILE); } // 词库树初始化 $this->wordTree = $this->wordTree ?: new HashMap(); foreach ($this->yieldToReadFile($filepath) as $word) { $this->buildWordToTree(trim($word)); } return $this; } /** * 构建铭感词树【数组模式】 * * @param null $sensitiveWords * * @return $this * @throws \DfaFilter\Exceptions\PdsBusinessException */ public function setTree($sensitiveWords = null) { if (empty($sensitiveWords)) { throw new PdsBusinessException('词库不能为空', PdsBusinessException::EMPTY_WORD_POOL); } $this->wordTree = new HashMap(); foreach ($sensitiveWords as $word) { $this->buildWordToTree($word); } return $this; } /** * 检测文字中的敏感词 * * @param string $content 待检测内容 * @param int $matchType 匹配类型 [默认为最小匹配规则] * @param int $wordNum 需要获取的敏感词数量 [默认获取全部] * @return array * @throws \DfaFilter\Exceptions\PdsSystemException */ public function getBadWord($content, $matchType = 1, $wordNum = 0) { $this->contentLength = mb_strlen($content, 'utf-8'); $badWordList = array(); for ($length = 0; $length < $this->contentLength; $length++) { $matchFlag = 0; $flag = false; $tempMap = $this->wordTree; for ($i = $length; $i < $this->contentLength; $i++) { $keyChar = mb_substr($content, $i, 1, 'utf-8'); // 获取指定节点树 $nowMap = $tempMap->get($keyChar); // 不存在节点树，直接返回 if (empty($nowMap)) { break; } // 存在，则判断是否为最后一个 $tempMap = $nowMap; // 找到相应key，偏移量+1 $matchFlag++; // 如果为最后一个匹配规则,结束循环，返回匹配标识数 if (false === $nowMap->get('ending')) { continue; } $flag = true; // 最小规则，直接退出 if (1 === $matchType) { break; } } if (! $flag) { $matchFlag = 0; } // 找到相应key if ($matchFlag <= 0) { continue; } $badWordList[] = mb_substr($content, $length, $matchFlag, 'utf-8'); // 有返回数量限制 if ($wordNum > 0 && count($badWordList) == $wordNum) { return $badWordList; } // 需匹配内容标志位往后移 $length = $length + $matchFlag - 1; } return $badWordList; } /** * 替换敏感字字符 * * @param $content 文本内容 * @param string $replaceChar 替换字符 * @param bool $repeat true=>重复替换为敏感词相同长度的字符 * @param int $matchType * * @return mixed * @throws \DfaFilter\Exceptions\PdsBusinessException * @throws \DfaFilter\Exceptions\PdsSystemException */ public function replace($content, $replaceChar = '', $repeat = false, $matchType = 1) { if (empty($content)) { throw new PdsBusinessException('请填写检测的内容', PdsBusinessException::EMPTY_CONTENT); } $badWordList = self::$badWordList ? self::$badWordList : $this->getBadWord($content, $matchType); // 未检测到敏感词，直接返回 if (empty($badWordList)) { return $content; } foreach ($badWordList as $badWord) { $hasReplacedChar = $replaceChar; if ($repeat) { $hasReplacedChar = $this->dfaBadWordConversChars($badWord, $replaceChar); } $content = str_replace($badWord, $hasReplacedChar, $content); } return $content; } /** * 标记敏感词 * * @param $content 文本内容 * @param string $sTag 标签开头，如<mark> * @param string $eTag 标签结束，如</mark> * @param int $matchType * * @return mixed * @throws \DfaFilter\Exceptions\PdsBusinessException * @throws \DfaFilter\Exceptions\PdsSystemException */ public function mark($content, $sTag, $eTag, $matchType = 1) { if (empty($content)) { throw new PdsBusinessException('请填写检测的内容', PdsBusinessException::EMPTY_CONTENT); } $badWordList = self::$badWordList ? self::$badWordList : $this->getBadWord($content, $matchType); // 未检测到敏感词，直接返回 if (empty($badWordList)) { return $content; } $badWordList = array_unique($badWordList); foreach ($badWordList as $badWord) { $replaceChar = $sTag . $badWord . $eTag; $content = str_replace($badWord, $replaceChar, $content); } return $content; } /** * 被检测内容是否合法 * * @param $content * * @return bool * @throws \DfaFilter\Exceptions\PdsSystemException */ public function islegal($content) { $this->contentLength = mb_strlen($content, 'utf-8'); for ($length = 0; $length < $this->contentLength; $length++) { $matchFlag = 0; $tempMap = $this->wordTree; for ($i = $length; $i < $this->contentLength; $i++) { $keyChar = mb_substr($content, $i, 1, 'utf-8'); // 获取指定节点树 $nowMap = $tempMap->get($keyChar); // 不存在节点树，直接返回 if (empty($nowMap)) { break; } // 找到相应key，偏移量+1 $tempMap = $nowMap; $matchFlag++; // 如果为最后一个匹配规则,结束循环，返回匹配标识数 if (false === $nowMap->get('ending')) { continue; } return true; } // 找到相应key if ($matchFlag <= 0) { continue; } // 需匹配内容标志位往后移 $length = $length + $matchFlag - 1; } return false; } protected function yieldToReadFile($filepath) { // echo