mirror of
https://github.com/syuilo/ai.git
synced 2024-11-13 01:17:58 +00:00
39 lines
1.6 KiB
TypeScript
39 lines
1.6 KiB
TypeScript
import { hankakuToZenkaku, katakanaToHiragana } from './japanese';
|
||
|
||
export default function(text: string, words: (string | RegExp)[]): boolean {
|
||
if (text == null) return false;
|
||
|
||
text = katakanaToHiragana(hankakuToZenkaku(text));
|
||
words = words.map(word => typeof word == 'string' ? katakanaToHiragana(word) : word);
|
||
|
||
return words.some(word => {
|
||
/**
|
||
* テキストの余分な部分を取り除く
|
||
* 例えば「藍ちゃん好き！」のようなテキストを「好き」にする
|
||
*/
|
||
function cleanup(text: string): string {
|
||
return text.trim()
|
||
.replace(/[！!]+$/, '')
|
||
// 末尾の ー を除去
|
||
// 例えば「おはよー」を「おはよ」にする
|
||
// ただそのままだと「セーラー」などの本来「ー」が含まれているワードも「ー」が除去され
|
||
// 「セーラ」になり、「セーラー」を期待している場合はマッチしなくなり期待する動作にならなくなるので、
|
||
// 期待するワードの末尾にもともと「ー」が含まれている場合は(対象のテキストの「ー」をすべて除去した後に)「ー」を付けてあげる
|
||
.replace(/ー+$/, '') + ((typeof word == 'string' && word[word.length - 1] == 'ー') ? 'ー' : '')
|
||
.replace(/っ+$/, '')
|
||
.replace(/。$/, '')
|
||
.replace(/です$/, '')
|
||
.replace(/^藍/, '')
|
||
.replace(/^ちゃん/, '')
|
||
.replace(/、+$/, '');
|
||
}
|
||
|
||
if (typeof word == 'string') {
|
||
return (text == word) || (cleanup(text) == word);
|
||
} else {
|
||
return (word.test(text)) || (word.test(cleanup(text)));
|
||
}
|
||
});
|
||
}
|
||
|