From fc6d491bee09db4d7bf42de2088ff310407beadc Mon Sep 17 00:00:00 2001 From: syuilo Date: Fri, 31 Aug 2018 17:24:07 +0900 Subject: [PATCH] Improve AI --- src/modules/core/index.ts | 21 ++++---- src/modules/emoji/index.ts | 3 +- src/modules/fortune/index.ts | 5 +- src/modules/guessing-game/index.ts | 3 +- src/modules/reversi/index.ts | 3 +- src/serifs.ts | 2 + src/utils/includes.ts | 10 ++++ src/utils/japanese.ts | 79 ++++++++++++++++++++++++++++++ 8 files changed, 111 insertions(+), 15 deletions(-) create mode 100644 src/utils/includes.ts create mode 100644 src/utils/japanese.ts diff --git a/src/modules/core/index.ts b/src/modules/core/index.ts index 4e3c823..d5f9d2a 100644 --- a/src/modules/core/index.ts +++ b/src/modules/core/index.ts @@ -4,6 +4,7 @@ import MessageLike from '../../message-like'; import serifs from '../../serifs'; import Friend from '../../friend'; import getDate from '../../utils/get-date'; +import includes from '../../utils/includes'; function zeroPadding(num: number, length: number): string { return ('0000000000' + num).slice(-length); @@ -107,8 +108,6 @@ export default class CoreModule implements IModule { } private greet = (msg: MessageLike): boolean => { - if (!msg.text) return false; - const incLove = () => { //#region 1日に1回だけ親愛度を上げる const today = getDate(); @@ -124,19 +123,25 @@ export default class CoreModule implements IModule { //#endregion }; - if (msg.text.includes('こんにちは')) { + if (includes(msg.text, ['こんにちは'])) { msg.reply(serifs.core.hello(msg.friend.name)); incLove(); return true; } - if (msg.text.includes('おはよ')) { + if (includes(msg.text, ['こんばんは'])) { + msg.reply(serifs.core.helloNight(msg.friend.name)); + incLove(); + return true; + } + + if (includes(msg.text, ['おはよ', 'お早う'])) { msg.reply(serifs.core.goodMorning(msg.friend.name)); incLove(); return true; } - if (msg.text.includes('おやすみ')) { + if (includes(msg.text, ['おやすみ', 'お休み'])) { msg.reply(serifs.core.goodNight(msg.friend.name)); incLove(); return true; @@ -146,8 +151,7 @@ export default class CoreModule implements IModule { } private nadenade = (msg: MessageLike): boolean => { - if (!msg.text) return false; - if (!msg.text.includes('なでなで')) return false; + if (!includes(msg.text, ['なでなで'])) return false; // メッセージのみ if (!msg.isMessage) return true; @@ -175,8 +179,7 @@ export default class CoreModule implements IModule { } private kawaii = (msg: MessageLike): boolean => { - if (!msg.text) return false; - if (!msg.text.includes('かわいい') && !msg.text.includes('可愛い')) return false; + if (!includes(msg.text, ['かわいい', '可愛い'])) return false; msg.reply(msg.friend.love >= 5 ? serifs.core.kawaii2 : serifs.core.kawaii1); diff --git a/src/modules/emoji/index.ts b/src/modules/emoji/index.ts index fca4f5e..ffb207c 100644 --- a/src/modules/emoji/index.ts +++ b/src/modules/emoji/index.ts @@ -2,6 +2,7 @@ import 藍 from '../../ai'; import IModule from '../../module'; import MessageLike from '../../message-like'; import serifs from '../../serifs'; +import includes from '../../utils/includes'; const hands = [ '👏', @@ -124,7 +125,7 @@ export default class EmojiModule implements IModule { public install = (ai: 藍) => { } public onMention = (msg: MessageLike) => { - if (msg.text && (msg.text.includes('絵文字') || msg.text.includes('emoji'))) { + if (includes(msg.text, ['絵文字', 'emoji'])) { const hand = hands[Math.floor(Math.random() * hands.length)]; const face = faces[Math.floor(Math.random() * faces.length)]; const emoji = Array.isArray(hand) ? hand[0] + face + hand[1] : hand + face + hand; diff --git a/src/modules/fortune/index.ts b/src/modules/fortune/index.ts index 4db3a23..b455f75 100644 --- a/src/modules/fortune/index.ts +++ b/src/modules/fortune/index.ts @@ -3,6 +3,7 @@ import IModule from '../../module'; import MessageLike from '../../message-like'; import serifs from '../../serifs'; import * as seedrandom from 'seedrandom'; +import includes from '../../utils/includes'; const omikujis = [ '大大吉', @@ -30,9 +31,7 @@ export default class FortuneModule implements IModule { public install = (ai: 藍) => { } public onMention = (msg: MessageLike) => { - if (msg.text == null) return false; - - if (msg.text.includes('占') || msg.text.includes('うらな') || msg.text.includes('運勢') || msg.text.includes('おみくじ')) { + if (includes(msg.text, ['占', 'うらな', '運勢', 'おみくじ'])) { const date = new Date(); const seed = `${date.getFullYear()}/${date.getMonth()}/${date.getDay()}@${msg.userId}`; const rng = seedrandom(seed); diff --git a/src/modules/guessing-game/index.ts b/src/modules/guessing-game/index.ts index cc429ff..62d8ce1 100644 --- a/src/modules/guessing-game/index.ts +++ b/src/modules/guessing-game/index.ts @@ -4,6 +4,7 @@ import IModule from '../../module'; import MessageLike from '../../message-like'; import serifs from '../../serifs'; import getCollection from '../../utils/get-collection'; +import includes from '../../utils/includes'; export default class GuessingGameModule implements IModule { public readonly name = 'guessingGame'; @@ -28,7 +29,7 @@ export default class GuessingGameModule implements IModule { } public onMention = (msg: MessageLike) => { - if (msg.text && (msg.text.includes('数当て') || msg.text.includes('数あて'))) { + if (includes(msg.text, ['数当て', '数あて'])) { const exist = this.guesses.findOne({ userId: msg.userId, isEnded: false diff --git a/src/modules/reversi/index.ts b/src/modules/reversi/index.ts index 2a40a17..f96511a 100644 --- a/src/modules/reversi/index.ts +++ b/src/modules/reversi/index.ts @@ -9,6 +9,7 @@ import * as WebSocket from 'ws'; import Friend from '../../friend'; import getDate from '../../utils/get-date'; import { User } from '../../misskey/user'; +import includes from '../../utils/includes'; export default class ReversiModule implements IModule { public readonly name = 'reversi'; @@ -45,7 +46,7 @@ export default class ReversiModule implements IModule { } public onMention = (msg: MessageLike) => { - if (msg.text && (msg.text.includes('リバーシ') || msg.text.includes('りばーし') || msg.text.includes('オセロ') || msg.text.includes('おせろ') || msg.text.toLowerCase().includes('reversi'))) { + if (includes(msg.text, ['リバーシ', 'オセロ', 'reversi', 'othello'])) { if (config.reversiEnabled) { msg.reply(serifs.reversi.ok); diff --git a/src/serifs.ts b/src/serifs.ts index 36eaf4b..8d43412 100644 --- a/src/serifs.ts +++ b/src/serifs.ts @@ -8,6 +8,8 @@ export default { hello: name => name ? `こんにちは、${name}♪` : `こんにちは♪`, + helloNight: name => name ? `こんばんは、${name}♪` : `こんばんは♪`, + goodMorning: name => name ? `おはようございます、${name}！` : 'おはようございます！', goodNight: name => name ? `おやすみなさい、${name}！` : 'おやすみなさい！', diff --git a/src/utils/includes.ts b/src/utils/includes.ts new file mode 100644 index 0000000..e92d81f --- /dev/null +++ b/src/utils/includes.ts @@ -0,0 +1,10 @@ +import { hiraganaToKatagana, zenkakuToHankaku, hankakuToZenkaku } from './japanese'; + +export default function(text: string, words: string[]): boolean { + if (text == null) return false; + + text = hankakuToZenkaku(hiraganaToKatagana(text)); + words = words.map(word => hiraganaToKatagana(word)); + + return words.some(word => text.includes(word)); +} diff --git a/src/utils/japanese.ts b/src/utils/japanese.ts new file mode 100644 index 0000000..43ab538 --- /dev/null +++ b/src/utils/japanese.ts @@ -0,0 +1,79 @@ +// Utilities for Japanese + +const kanaMap: string[][] = [ + ['ア', 'ｱ'], ['イ', 'ｲ'], ['ウ', 'ｳ'], ['エ', 'ｴ'], ['オ', 'ｵ'], + ['カ', 'ｶ'], ['キ', 'ｷ'], ['ク', 'ｸ'], ['ケ', 'ｹ'], ['コ', 'ｺ'], + ['サ', 'ｻ'], ['シ', 'ｼ'], ['ス', 'ｽ'], ['セ', 'ｾ'], ['ソ', 'ｿ'], + ['タ', 'ﾀ'], ['チ', 'ﾁ'], ['ツ', 'ﾂ'], ['テ', 'ﾃ'], ['ト', 'ﾄ'], + ['ナ', 'ﾅ'], ['ニ', 'ﾆ'], ['ヌ', 'ﾇ'], ['ネ', 'ﾈ'], ['ノ', 'ﾉ'], + ['ハ', 'ﾊ'], ['ヒ', 'ﾋ'], ['フ', 'ﾌ'], ['ヘ', 'ﾍ'], ['ホ', 'ﾎ'], + ['マ', 'ﾏ'], ['ミ', 'ﾐ'], ['ム', 'ﾑ'], ['メ', 'ﾒ'], ['モ', 'ﾓ'], + ['ヤ', 'ﾔ'], ['ユ', 'ﾕ'], ['ヨ', 'ﾖ'], + ['ラ', 'ﾗ'], ['リ', 'ﾘ'], ['ル', 'ﾙ'], ['レ', 'ﾚ'], ['ロ', 'ﾛ'], + ['ワ', 'ﾜ'], ['ヲ', 'ｦ'], ['ン', 'ﾝ'], + ['ガ', 'ｶﾞ'], ['ギ', 'ｷﾞ'], ['グ', 'ｸﾞ'], ['ゲ', 'ｹﾞ'], ['ゴ', 'ｺﾞ'], + ['ザ', 'ｻﾞ'], ['ジ', 'ｼﾞ'], ['ズ', 'ｽﾞ'], ['ゼ', 'ｾﾞ'], ['ゾ', 'ｿﾞ'], + ['ダ', 'ﾀﾞ'], ['ヂ', 'ﾁﾞ'], ['ヅ', 'ﾂﾞ'], ['デ', 'ﾃﾞ'], ['ド', 'ﾄﾞ'], + ['バ', 'ﾊﾞ'], ['ビ', 'ﾋﾞ'], ['ブ', 'ﾌﾞ'], ['ベ', 'ﾍﾞ'], ['ボ', 'ﾎﾞ'], + ['パ', 'ﾊﾟ'], ['ピ', 'ﾋﾟ'], ['プ', 'ﾌﾟ'], ['ペ', 'ﾍﾟ'], ['ポ', 'ﾎﾟ'], + ['ヴ', 'ｳﾞ'], ['ヷ', 'ﾜﾞ'], ['ヺ', 'ｦﾞ'], + ['ァ', 'ｧ'], ['ィ', 'ｨ'], ['ゥ', 'ｩ'], ['ェ', 'ｪ'], ['ォ', 'ｫ'], + ['ッ', 'ｯ'], ['ャ', 'ｬ'], ['ュ', 'ｭ'], ['ョ', 'ｮ'], + ['ー', 'ｰ'] +]; + +/** + * カタカナをひらがなに変換します + * @param str カタカナ + * @returns ひらがな + */ +export function katakanaToHiragana(str: string): string { + return str.replace(/[\u30a1-\u30f6]/g, match => { + const char = match.charCodeAt(0) - 0x60; + return String.fromCharCode(char); + }); +} + +/** + * ひらがなをカタカナに変換します + * @param str ひらがな + * @returns カタカナ + */ +export function hiraganaToKatagana(str: string): string { + return str.replace(/[\u3041-\u3096]/g, match => { + const char = match.charCodeAt(0) + 0x60; + return String.fromCharCode(char); + }); +} + +/** + * 全角カタカナを半角カタカナに変換します + * @param str 全角カタカナ + * @returns 半角カタカナ + */ +export function zenkakuToHankaku(str: string): string { + const reg = new RegExp('(' + kanaMap.map(x => x[0]).join('|') + ')', 'g'); + + return str + .replace(reg, match => + kanaMap.find(x => x[0] == match)[1] + ) + .replace(/゛/g, 'ﾞ') + .replace(/゜/g, 'ﾟ'); +}; + +/** + * 半角カタカナを全角カタカナに変換します + * @param str 半角カタカナ + * @returns 全角カタカナ + */ +export function hankakuToZenkaku(str: string): string { + const reg = new RegExp('(' + kanaMap.map(x => x[1]).join('|') + ')', 'g'); + + return str + .replace(reg, match => + kanaMap.find(x => x[1] == match)[0] + ) + .replace(/ﾞ/g, '゛') + .replace(/ﾟ/g, '゜'); +};