import autobind from 'autobind-decorator'; import * as loki from 'lokijs'; import Module from '../../module'; import Message from '../../message'; import serifs from '../../serifs'; import getCollection from '../../utils/get-collection'; export default class GuessingGameModule extends Module { public readonly name = 'guessingGame'; private guesses: loki.Collection<{ userId: string; secret: number; tries: number[]; isEnded: boolean; startedAt: number; endedAt: number; }>; @autobind public install() { //#region Init DB this.guesses = getCollection(this.ai.db, 'guessingGame', { indices: ['userId'] }); //#endregion return { mentionHook: this.mentionHook, contextHook: this.contextHook }; } @autobind private mentionHook(msg: Message) { if (!msg.includes(['数当て', '数あて'])) return false; const exist = this.guesses.findOne({ userId: msg.userId, isEnded: false }); if (!msg.isDm) { if (exist != null) { msg.reply(serifs.guessingGame.arleadyStarted); } else { msg.reply(serifs.guessingGame.plzDm); } return true; } const secret = Math.floor(Math.random() * 100); this.guesses.insertOne({ userId: msg.userId, secret: secret, tries: [], isEnded: false, startedAt: Date.now(), endedAt: null }); msg.reply(serifs.guessingGame.started).then(reply => { this.subscribeReply(msg.userId, msg.isDm, msg.isDm ? msg.userId : reply.id); }); return true; } @autobind private contextHook(msg: Message) { if (msg.text == null) return; const exist = this.guesses.findOne({ userId: msg.userId, isEnded: false }); if (msg.text.includes('やめ')) { msg.reply(serifs.guessingGame.cancel); exist.isEnded = true; exist.endedAt = Date.now(); this.guesses.update(exist); this.unsubscribeReply(msg.userId); return; } const guess = msg.text.toLowerCase().replace(this.ai.account.username.toLowerCase(), '').match(/[0-9]+/); if (guess == null) { msg.reply(serifs.guessingGame.nan).then(reply => { this.subscribeReply(msg.userId, msg.isDm, reply.id); }); return; } if (guess.length > 3) return; const g = parseInt(guess[0], 10); const firsttime = exist.tries.indexOf(g) === -1; exist.tries.push(g); let text: string; let end = false; if (exist.secret < g) { text = firsttime ? serifs.guessingGame.less(g.toString()) : serifs.guessingGame.lessAgain(g.toString()); } else if (exist.secret > g) { text = firsttime ? serifs.guessingGame.grater(g.toString()) : serifs.guessingGame.graterAgain(g.toString()); } else { end = true; text = serifs.guessingGame.congrats(exist.tries.length.toString()); } if (end) { exist.isEnded = true; exist.endedAt = Date.now(); this.unsubscribeReply(msg.userId); } this.guesses.update(exist); msg.reply(text).then(reply => { if (!end) { this.subscribeReply(msg.userId, msg.isDm, reply.id); } }); } }