438 lines
10 KiB
TypeScript
438 lines
10 KiB
TypeScript
import * as EventEmitter from 'events';
|
||
import * as bcrypt from 'bcryptjs';
|
||
|
||
import User, { ILocalAccount, IUser, init as initUser } from '../models/user';
|
||
|
||
import getPostSummary from '../../common/get-post-summary';
|
||
import getUserSummary from '../../common/user/get-summary';
|
||
import parseAcct from '../../common/user/parse-acct';
|
||
import getNotificationSummary from '../../common/get-notification-summary';
|
||
|
||
const hmm = [
|
||
'？',
|
||
'ふぅ～む...？',
|
||
'ちょっと何言ってるかわからないです',
|
||
'「ヘルプ」と言うと利用可能な操作が確認できますよ'
|
||
];
|
||
|
||
/**
|
||
* Botの頭脳
|
||
*/
|
||
export default class BotCore extends EventEmitter {
|
||
public user: IUser = null;
|
||
|
||
private context: Context = null;
|
||
|
||
constructor(user?: IUser) {
|
||
super();
|
||
|
||
this.user = user;
|
||
}
|
||
|
||
public clearContext() {
|
||
this.setContext(null);
|
||
}
|
||
|
||
public setContext(context: Context) {
|
||
this.context = context;
|
||
this.emit('updated');
|
||
|
||
if (context) {
|
||
context.on('updated', () => {
|
||
this.emit('updated');
|
||
});
|
||
}
|
||
}
|
||
|
||
public export() {
|
||
return {
|
||
user: this.user,
|
||
context: this.context ? this.context.export() : null
|
||
};
|
||
}
|
||
|
||
protected _import(data) {
|
||
this.user = data.user ? initUser(data.user) : null;
|
||
this.setContext(data.context ? Context.import(this, data.context) : null);
|
||
}
|
||
|
||
public static import(data) {
|
||
const bot = new BotCore();
|
||
bot._import(data);
|
||
return bot;
|
||
}
|
||
|
||
public async q(query: string): Promise<string> {
|
||
if (this.context != null) {
|
||
return await this.context.q(query);
|
||
}
|
||
|
||
if (/^@[a-zA-Z0-9_]+$/.test(query)) {
|
||
return await this.showUserCommand(query);
|
||
}
|
||
|
||
switch (query) {
|
||
case 'ping':
|
||
return 'PONG';
|
||
|
||
case 'help':
|
||
case 'ヘルプ':
|
||
return '利用可能なコマンド一覧です:\n' +
|
||
'help: これです\n' +
|
||
'me: アカウント情報を見ます\n' +
|
||
'login, signin: サインインします\n' +
|
||
'logout, signout: サインアウトします\n' +
|
||
'post: 投稿します\n' +
|
||
'tl: タイムラインを見ます\n' +
|
||
'no: 通知を見ます\n' +
|
||
'@<ユーザー名>: ユーザーを表示します\n' +
|
||
'\n' +
|
||
'タイムラインや通知を見た後、「次」というとさらに遡ることができます。';
|
||
|
||
case 'me':
|
||
return this.user ? `${this.user.name}としてサインインしています。\n\n${getUserSummary(this.user)}` : 'サインインしていません';
|
||
|
||
case 'login':
|
||
case 'signin':
|
||
case 'ログイン':
|
||
case 'サインイン':
|
||
if (this.user != null) return '既にサインインしていますよ！';
|
||
this.setContext(new SigninContext(this));
|
||
return await this.context.greet();
|
||
|
||
case 'logout':
|
||
case 'signout':
|
||
case 'ログアウト':
|
||
case 'サインアウト':
|
||
if (this.user == null) return '今はサインインしてないですよ！';
|
||
this.signout();
|
||
return 'ご利用ありがとうございました <3';
|
||
|
||
case 'post':
|
||
case '投稿':
|
||
if (this.user == null) return 'まずサインインしてください。';
|
||
this.setContext(new PostContext(this));
|
||
return await this.context.greet();
|
||
|
||
case 'tl':
|
||
case 'タイムライン':
|
||
if (this.user == null) return 'まずサインインしてください。';
|
||
this.setContext(new TlContext(this));
|
||
return await this.context.greet();
|
||
|
||
case 'no':
|
||
case 'notifications':
|
||
case '通知':
|
||
if (this.user == null) return 'まずサインインしてください。';
|
||
this.setContext(new NotificationsContext(this));
|
||
return await this.context.greet();
|
||
|
||
case 'guessing-game':
|
||
case '数当てゲーム':
|
||
this.setContext(new GuessingGameContext(this));
|
||
return await this.context.greet();
|
||
|
||
default:
|
||
return hmm[Math.floor(Math.random() * hmm.length)];
|
||
}
|
||
}
|
||
|
||
public signin(user: IUser) {
|
||
this.user = user;
|
||
this.emit('signin', user);
|
||
this.emit('updated');
|
||
}
|
||
|
||
public signout() {
|
||
const user = this.user;
|
||
this.user = null;
|
||
this.emit('signout', user);
|
||
this.emit('updated');
|
||
}
|
||
|
||
public async refreshUser() {
|
||
this.user = await User.findOne({
|
||
_id: this.user._id
|
||
}, {
|
||
fields: {
|
||
data: false
|
||
}
|
||
});
|
||
|
||
this.emit('updated');
|
||
}
|
||
|
||
public async showUserCommand(q: string): Promise<string> {
|
||
try {
|
||
const user = await require('../endpoints/users/show')(parseAcct(q.substr(1)), this.user);
|
||
|
||
const text = getUserSummary(user);
|
||
|
||
return text;
|
||
} catch (e) {
|
||
return `問題が発生したようです...: ${e}`;
|
||
}
|
||
}
|
||
}
|
||
|
||
abstract class Context extends EventEmitter {
|
||
protected bot: BotCore;
|
||
|
||
public abstract async greet(): Promise<string>;
|
||
public abstract async q(query: string): Promise<string>;
|
||
public abstract export(): any;
|
||
|
||
constructor(bot: BotCore) {
|
||
super();
|
||
this.bot = bot;
|
||
}
|
||
|
||
public static import(bot: BotCore, data: any) {
|
||
if (data.type == 'guessing-game') return GuessingGameContext.import(bot, data.content);
|
||
if (data.type == 'post') return PostContext.import(bot, data.content);
|
||
if (data.type == 'tl') return TlContext.import(bot, data.content);
|
||
if (data.type == 'notifications') return NotificationsContext.import(bot, data.content);
|
||
if (data.type == 'signin') return SigninContext.import(bot, data.content);
|
||
return null;
|
||
}
|
||
}
|
||
|
||
class SigninContext extends Context {
|
||
private temporaryUser: IUser = null;
|
||
|
||
public async greet(): Promise<string> {
|
||
return 'まずユーザー名を教えてください:';
|
||
}
|
||
|
||
public async q(query: string): Promise<string> {
|
||
if (this.temporaryUser == null) {
|
||
// Fetch user
|
||
const user: IUser = await User.findOne({
|
||
usernameLower: query.toLowerCase(),
|
||
host: null
|
||
}, {
|
||
fields: {
|
||
data: false
|
||
}
|
||
});
|
||
|
||
if (user === null) {
|
||
return `${query}というユーザーは存在しませんでした... もう一度教えてください:`;
|
||
} else {
|
||
this.temporaryUser = user;
|
||
this.emit('updated');
|
||
return `パスワードを教えてください:`;
|
||
}
|
||
} else {
|
||
// Compare password
|
||
const same = await bcrypt.compare(query, (this.temporaryUser.account as ILocalAccount).password);
|
||
|
||
if (same) {
|
||
this.bot.signin(this.temporaryUser);
|
||
this.bot.clearContext();
|
||
return `${this.temporaryUser.name}さん、おかえりなさい！`;
|
||
} else {
|
||
return `パスワードが違います... もう一度教えてください:`;
|
||
}
|
||
}
|
||
}
|
||
|
||
public export() {
|
||
return {
|
||
type: 'signin',
|
||
content: {
|
||
temporaryUser: this.temporaryUser
|
||
}
|
||
};
|
||
}
|
||
|
||
public static import(bot: BotCore, data: any) {
|
||
const context = new SigninContext(bot);
|
||
context.temporaryUser = data.temporaryUser;
|
||
return context;
|
||
}
|
||
}
|
||
|
||
class PostContext extends Context {
|
||
public async greet(): Promise<string> {
|
||
return '内容:';
|
||
}
|
||
|
||
public async q(query: string): Promise<string> {
|
||
await require('../endpoints/posts/create')({
|
||
text: query
|
||
}, this.bot.user);
|
||
this.bot.clearContext();
|
||
return '投稿しましたよ！';
|
||
}
|
||
|
||
public export() {
|
||
return {
|
||
type: 'post'
|
||
};
|
||
}
|
||
|
||
public static import(bot: BotCore, data: any) {
|
||
const context = new PostContext(bot);
|
||
return context;
|
||
}
|
||
}
|
||
|
||
class TlContext extends Context {
|
||
private next: string = null;
|
||
|
||
public async greet(): Promise<string> {
|
||
return await this.getTl();
|
||
}
|
||
|
||
public async q(query: string): Promise<string> {
|
||
if (query == '次') {
|
||
return await this.getTl();
|
||
} else {
|
||
this.bot.clearContext();
|
||
return await this.bot.q(query);
|
||
}
|
||
}
|
||
|
||
private async getTl() {
|
||
const tl = await require('../endpoints/posts/timeline')({
|
||
limit: 5,
|
||
untilId: this.next ? this.next : undefined
|
||
}, this.bot.user);
|
||
|
||
if (tl.length > 0) {
|
||
this.next = tl[tl.length - 1].id;
|
||
this.emit('updated');
|
||
|
||
const text = tl
|
||
.map(post => `${post.user.name}\n「${getPostSummary(post)}」`)
|
||
.join('\n-----\n');
|
||
|
||
return text;
|
||
} else {
|
||
return 'タイムラインに表示するものがありません...';
|
||
}
|
||
}
|
||
|
||
public export() {
|
||
return {
|
||
type: 'tl',
|
||
content: {
|
||
next: this.next,
|
||
}
|
||
};
|
||
}
|
||
|
||
public static import(bot: BotCore, data: any) {
|
||
const context = new TlContext(bot);
|
||
context.next = data.next;
|
||
return context;
|
||
}
|
||
}
|
||
|
||
class NotificationsContext extends Context {
|
||
private next: string = null;
|
||
|
||
public async greet(): Promise<string> {
|
||
return await this.getNotifications();
|
||
}
|
||
|
||
public async q(query: string): Promise<string> {
|
||
if (query == '次') {
|
||
return await this.getNotifications();
|
||
} else {
|
||
this.bot.clearContext();
|
||
return await this.bot.q(query);
|
||
}
|
||
}
|
||
|
||
private async getNotifications() {
|
||
const notifications = await require('../endpoints/i/notifications')({
|
||
limit: 5,
|
||
untilId: this.next ? this.next : undefined
|
||
}, this.bot.user);
|
||
|
||
if (notifications.length > 0) {
|
||
this.next = notifications[notifications.length - 1].id;
|
||
this.emit('updated');
|
||
|
||
const text = notifications
|
||
.map(notification => getNotificationSummary(notification))
|
||
.join('\n-----\n');
|
||
|
||
return text;
|
||
} else {
|
||
return '通知はありません';
|
||
}
|
||
}
|
||
|
||
public export() {
|
||
return {
|
||
type: 'notifications',
|
||
content: {
|
||
next: this.next,
|
||
}
|
||
};
|
||
}
|
||
|
||
public static import(bot: BotCore, data: any) {
|
||
const context = new NotificationsContext(bot);
|
||
context.next = data.next;
|
||
return context;
|
||
}
|
||
}
|
||
|
||
class GuessingGameContext extends Context {
|
||
private secret: number;
|
||
private history: number[] = [];
|
||
|
||
public async greet(): Promise<string> {
|
||
this.secret = Math.floor(Math.random() * 100);
|
||
this.emit('updated');
|
||
return '0~100の秘密の数を当ててみてください:';
|
||
}
|
||
|
||
public async q(query: string): Promise<string> {
|
||
if (query == 'やめる') {
|
||
this.bot.clearContext();
|
||
return 'やめました。';
|
||
}
|
||
|
||
const guess = parseInt(query, 10);
|
||
|
||
if (isNaN(guess)) {
|
||
return '整数で推測してください。「やめる」と言うとゲームをやめます。';
|
||
}
|
||
|
||
const firsttime = this.history.indexOf(guess) === -1;
|
||
|
||
this.history.push(guess);
|
||
this.emit('updated');
|
||
|
||
if (this.secret < guess) {
|
||
return firsttime ? `${guess}よりも小さいですね` : `もう一度言いますが${guess}より小さいですよ`;
|
||
} else if (this.secret > guess) {
|
||
return firsttime ? `${guess}よりも大きいですね` : `もう一度言いますが${guess}より大きいですよ`;
|
||
} else {
|
||
this.bot.clearContext();
|
||
return `正解です🎉 (${this.history.length}回目で当てました)`;
|
||
}
|
||
}
|
||
|
||
public export() {
|
||
return {
|
||
type: 'guessing-game',
|
||
content: {
|
||
secret: this.secret,
|
||
history: this.history
|
||
}
|
||
};
|
||
}
|
||
|
||
public static import(bot: BotCore, data: any) {
|
||
const context = new GuessingGameContext(bot);
|
||
context.secret = data.secret;
|
||
context.history = data.history;
|
||
return context;
|
||
}
|
||
}
|