package me.mymilkbottles;

import org.apache.commons.lang.CharUtils;

import java.io.File;
import java.util.HashMap;
import java.util.Map;


public class SensitiveWordFilter {

    private WordTree wordTree = new WordTree();

    private static final String SENSITIVE_WORD_REPLACE = "***";

    private
 
 class WordTree {

        private Map<Character, WordTree> tree = new HashMap<Character, WordTree>();

        private boolean isWordEndFlag;

        public WordTree() {
        }

        public WordTree(boolean isWordEndFlag) {
            this.isWordEndFlag = isWordEndFlag;
        }

        public
 
 boolean getIsWordEndFlag() {
            return isWordEndFlag;
        }

        public void set(Character ch, WordTree wordTree) {
            tree.put(ch, wordTree);
        }

        public WordTree get(Character ch) {
            return tree.get(ch);
        }
    }

    public SensitiveWordFilter addSensitiveWord
 
(String sensitiveWord) {

        WordTree wordTreeNode = wordTree;

        for (int i = 0, len = sensitiveWord.length(); i < len; ++i) {

            Character nowCharacter = sensitiveWord.charAt(i);

            WordTree tempWordTreeNode = new WordTree(i == len - 1);

            wordTreeNode.set(nowCharacter, tempWordTreeNode);

            wordTreeNode = tempWordTreeNode;
        }
        return this;
    }

    public String sensitiveFilter(String filterText) {

        StringBuilder afterFilterText = new StringBuilder();

        int len = filterText.length();

        int nowPosition = 0, testPosition, filterPosition;

        while (nowPosition < len) {
            WordTree wordTreeNode = wordTree;

            Character nowCharacter = filterText.charAt(nowPosition);

            WordTree tempWordTreeNode = wordTreeNode.get(nowCharacter);

            if (tempWordTreeNode == null) {
                afterFilterText.append(nowCharacter);
                ++nowPosition;
            } else {
                testPosition = nowPosition;
                filterPosition = -1;
                while (tempWordTreeNode != null) {
                    if (tempWordTreeNode.getIsWordEndFlag()) {
                        filterPosition = testPosition;
                        break;
                    } else {
                        while ((++testPosition) < len) {
                            nowCharacter = filterText.charAt(testPosition);
                            if (CharUtils.isAsciiAlphanumeric(nowCharacter) || (nowCharacter >= 0x4e00 && nowCharacter <= 0x9fbb)) {
                                break;
                            }
                        }
                        tempWordTreeNode = tempWordTreeNode.get(nowCharacter);
                    }
                }
                if (filterPosition != -1) {
                    afterFilterText.append(SENSITIVE_WORD_REPLACE);
                    nowPosition = filterPosition + 1;
                } else {
                    afterFilterText.append(filterText.charAt(nowPosition));
                    ++nowPosition;
                }
            }
        }

        return afterFilterText.toString();
    }

    public static void main(String[] args) {
        SensitiveWordFilter sensitiveWordFilter = new SensitiveWordFilter();

        String[] sensitiveWords =
                new String[]{"色情", "暴力", "賭博", "上網", "通宵", "sb", "傻逼", "智障", "你媽"};

        for (String sensitiveWord : sensitiveWords) {
            sensitiveWordFilter.addSensitiveWord(sensitiveWord);
        }